From efb029d33408b8d9af59f99cc817bd8dd146f969 Mon Sep 17 00:00:00 2001
From: Vadzim Hushchanskou <vadzim_hushchanskou@epam.com>
Date: Tue, 10 Oct 2023 17:09:14 +0300
Subject: [PATCH] Fix tests

---
 tests/aio/test_aio_client.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/tests/aio/test_aio_client.py b/tests/aio/test_aio_client.py
index b63f3ae..73d1255 100644
--- a/tests/aio/test_aio_client.py
+++ b/tests/aio/test_aio_client.py
@@ -22,7 +22,7 @@
 import aiohttp
 # noinspection PyPackageRequirements
 import pytest
-from aiohttp import ServerTimeoutError
+from aiohttp import ServerTimeoutError, ServerConnectionError
 
 from reportportal_client import OutputType
 # noinspection PyProtectedMember
@@ -371,7 +371,7 @@ async def test_launch_uuid_print_default_print(mock_stdout):
 
 
 def connection_error(*args, **kwargs):
-    raise ServerTimeoutError()
+    raise ServerConnectionError()
 
 
 def json_error(*args, **kwargs):
@@ -405,8 +405,10 @@ def response_error(*args, **kwargs):
 async def test_connection_errors(aio_client, requests_method, client_method,
                                  client_params):
     getattr(await aio_client.session(), requests_method).side_effect = connection_error
-    result = await getattr(aio_client, client_method)(*client_params)
-    assert result is None
+    try:
+        await getattr(aio_client, client_method)(*client_params)
+    except Exception as e:
+        assert type(e) == ServerConnectionError
 
     getattr(await aio_client.session(), requests_method).side_effect = response_error
     result = await getattr(aio_client, client_method)(*client_params)