Add invalid response error tests

HardNorth · HardNorth · commit d061864022d5 · 2023-03-27T19:17:59.000+03:00
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -31,6 +31,14 @@ def response_error(*args, **kwargs):
     return result
 
 
+def invalid_response(*args, **kwargs):
+    result = Response()
+    result._content = \
+        '<html><head><title>405 Not Allowed</title></head></html>'
+    result.status_code = 405
+    return result
+
+
 @pytest.mark.parametrize(
     'requests_method, client_method, client_params, expected_result',
     [
@@ -59,6 +67,29 @@ def test_connection_errors(rp_client, requests_method, client_method,
     assert result == expected_result
 
 
+@pytest.mark.parametrize(
+    'requests_method, client_method, client_params',
+    [
+        ('put', 'finish_launch', [timestamp()]),
+        ('put', 'finish_test_item', ['test_item_id', timestamp()]),
+        ('get', 'get_item_id_by_uuid', ['test_item_uuid']),
+        ('get', 'get_launch_info', []),
+        ('get', 'get_launch_ui_id', []),
+        ('get', 'get_launch_ui_url', []),
+        ('get', 'get_project_settings', []),
+        ('post', 'start_launch', ['Test Launch', timestamp()]),
+        ('post', 'start_test_item', ['Test Item', timestamp(), 'STEP']),
+        ('put', 'update_test_item', ['test_item_id'])
+    ]
+)
+def test_invalid_responses(rp_client, requests_method, client_method,
+                           client_params):
+    rp_client.launch_id = 'test_launch_id'
+    getattr(rp_client.session, requests_method).side_effect = invalid_response
+    result = getattr(rp_client, client_method)(*client_params)
+    assert result is None
+
+
 LAUNCH_ID = 333
 EXPECTED_DEFAULT_URL = 'http://endpoint/ui/#project/launches/all/' + str(
     LAUNCH_ID)
