update http stream tests

Author: pnilan

unit_tests/sources/streams/http/error_handlers/test_http_status_error_handler.py

@@ -34,13 +34,13 @@ def test_given_ok_response_http_status_error_handler_returns_success_action(mock
             403,
             ResponseAction.FAIL,
             FailureType.config_error,
-            "Status Code: 403. Error: Forbidden. You don't have permission to access this resource.",
+            "HTTP Status Code: 403. Error: Forbidden. You don't have permission to access this resource.",
         ),
         (
             404,
             ResponseAction.FAIL,
             FailureType.system_error,
-            "Status Code: 404. Error: Not found. The requested resource was not found on the server.",
+            "HTTP Status Code: 404. Error: Not found. The requested resource was not found on the server.",
         ),
     ],
 )

unit_tests/sources/streams/http/test_http.py

@@ -210,10 +210,10 @@ def get_error_handler(self) -> Optional[ErrorHandler]:
     req.status_code = HTTPStatus.TOO_MANY_REQUESTS
     send_mock = mocker.patch.object(requests.Session, "send", return_value=req)
 
-    with pytest.raises(UserDefinedBackoffException, match="Too many requests") as excinfo:
+    with pytest.raises(
+        AirbyteTracedException, match="Exception: HTTP Status Code: 429. Error: Too many requests."
+    ):
         list(stream.read_records(SyncMode.full_refresh))
-    assert isinstance(excinfo.value.request, requests.PreparedRequest)
-    assert isinstance(excinfo.value.response, requests.Response)
     if retries <= 0:
         assert send_mock.call_count == 1
     else:

unit_tests/sources/streams/http/test_http_client.py

@@ -555,7 +555,7 @@ def backoff_time(self, *args, **kwargs):
     session_send.return_value = mocked_response
 
     with patch.object(requests.Session, "send", return_value=mocked_response) as mocked_send:
-        with pytest.raises(UserDefinedBackoffException):
+        with pytest.raises(AirbyteTracedException) as e:
             http_client.send_request(
                 http_method="get", url="https://test_base_url.com/v1/endpoint", request_kwargs={}
             )
@@ -583,7 +583,7 @@ def backoff_time(self, *args, **kwargs):
     session_send.return_value = mocked_response
 
     with patch.object(requests.Session, "send", return_value=mocked_response) as mocked_send:
-        with pytest.raises(UserDefinedBackoffException):
+        with pytest.raises(AirbyteTracedException) as e:
             http_client.send_request(
                 http_method="get", url="https://test_base_url.com/v1/endpoint", request_kwargs={}
             )
@@ -613,7 +613,7 @@ def backoff_time(self, *args, **kwargs):
     session_send.return_value = mocked_response
 
     with patch.object(requests.Session, "send", return_value=mocked_response) as mocked_send:
-        with pytest.raises(UserDefinedBackoffException):
+        with pytest.raises(AirbyteTracedException) as e:
             http_client.send_request(
                 http_method="get", url="https://test_base_url.com/v1/endpoint", request_kwargs={}
             )
@@ -652,7 +652,7 @@ def backoff_time(self, *args, **kwargs):
     session_send.return_value = mocked_response
 
     with patch.object(requests.Session, "send", return_value=mocked_response) as mocked_send:
-        with pytest.raises(UserDefinedBackoffException):
+        with pytest.raises(AirbyteTracedException) as e:
             http_client.send_request(
                 http_method="get", url="https://test_base_url.com/v1/endpoint", request_kwargs={}
             )
@@ -680,7 +680,7 @@ def backoff_time(self, *args, **kwargs):
     session_send.return_value = mocked_response
 
     with patch.object(requests.Session, "send", return_value=mocked_response) as mocked_send:
-        with pytest.raises(UserDefinedBackoffException):
+        with pytest.raises(AirbyteTracedException) as e:
             http_client.send_request(
                 http_method="get", url="https://test_base_url.com/v1/endpoint", request_kwargs={}
             )
@@ -709,7 +709,7 @@ def test_backoff_strategy_endless(
     session_send.return_value = mocked_response
 
     with patch.object(requests.Session, "send", return_value=mocked_response) as mocked_send:
-        with pytest.raises(expected_error):
+        with pytest.raises(AirbyteTracedException) as e:
             http_client.send_request(
                 http_method="get",
                 url="https://test_base_url.com/v1/endpoint",