fix failed tests

open-telemetry · Feb 4, 2022 · fb685b0 · fb685b0
1 parent 3a43261
commit fb685b0
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/instrumentation/opentelemetry-instrumentation-httpx/tests/test_httpx_integration.py b/instrumentation/opentelemetry-instrumentation-httpx/tests/test_httpx_integration.py
@@ -617,6 +617,15 @@ async def _perform_request():
 
         return _async_call(_perform_request())
 
+    def test_basic_multiple(self):
+        # We need to create separate clients because in httpx >= 0.19,
+        # closing the client after "with" means the second http call fails
+        self.perform_request(self.URL,
+                             client=self.create_client(self.transport))
+        self.perform_request(self.URL,
+                             client=self.create_client(self.transport))
+        self.assert_span(num_spans=2)
+
 
 class TestSyncInstrumentationIntegration(BaseTestCases.BaseInstrumentorTest):
     def create_client(
@@ -664,3 +673,10 @@ async def _perform_request():
                 return await _client.request(method, url, headers=headers)
 
         return _async_call(_perform_request())
+
+    def test_basic_multiple(self):
+        # We need to create separate clients because in httpx >= 0.19,
+        # closing the client after "with" means the second http call fails
+        self.perform_request(self.URL, client=self.create_client())
+        self.perform_request(self.URL, client=self.create_client())
+        self.assert_span(num_spans=2)