Add unit tests for http error.

newrelic · Aug 17, 2020 · 7a03190 · 7a03190
1 parent 5b3470c
commit 7a03190
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/tests/agent_unittests/test_agent_protocol.py b/tests/agent_unittests/test_agent_protocol.py
@@ -23,6 +23,7 @@
     DiscardDataForRequest,
     ForceAgentDisconnect,
     ForceAgentRestart,
+    NetworkInterfaceException,
     RetryDataForRequest,
 )
 
@@ -88,6 +89,18 @@ def close_connection(self):
         HttpClientRecorder.STATE -= 1
 
 
+class HttpClientException(DeveloperModeClient):
+    def send_request(
+        self,
+        method="POST",
+        path="/agent_listener/invoke_raw_method",
+        params=None,
+        headers=None,
+        payload=None,
+    ):
+        raise NetworkInterfaceException
+
+
 @pytest.fixture(autouse=True)
 def clear_sent_values():
     yield
@@ -246,6 +259,14 @@ def test_status_code_exceptions(status_code, expected_exc, log_level, caplog):
     assert "123LICENSEKEY" not in message
 
 
+def test_protocol_http_error_causes_retry():
+    protocol = AgentProtocol(
+        finalize_application_settings(), client_cls=HttpClientException
+    )
+    with pytest.raises(RetryDataForRequest):
+        protocol.send("analytic_event_data")
+
+
 def test_protocol_context_manager():
     protocol = AgentProtocol(
         finalize_application_settings(), client_cls=HttpClientRecorder