From a97326d8718f41c871c24c45c2642dcbacfecf87 Mon Sep 17 00:00:00 2001 From: Shalev Roda <65566801+shalevr@users.noreply.github.com> Date: Fri, 6 Jan 2023 19:04:49 +0200 Subject: [PATCH] Add uninstrument test for starlette (#1558) --- .../tests/test_starlette_instrumentation.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/instrumentation/opentelemetry-instrumentation-starlette/tests/test_starlette_instrumentation.py b/instrumentation/opentelemetry-instrumentation-starlette/tests/test_starlette_instrumentation.py index a367ab0e4..9c658e009 100644 --- a/instrumentation/opentelemetry-instrumentation-starlette/tests/test_starlette_instrumentation.py +++ b/instrumentation/opentelemetry-instrumentation-starlette/tests/test_starlette_instrumentation.py @@ -320,6 +320,18 @@ class TestAutoInstrumentation(TestStarletteManualInstrumentation): self.assertEqual(span.resource.attributes["key1"], "value1") self.assertEqual(span.resource.attributes["key2"], "value2") + def test_uninstrument(self): + self._client.get("/foobar") + spans = self.memory_exporter.get_finished_spans() + self.assertEqual(len(spans), 3) + + self.memory_exporter.clear() + self._instrumentor.uninstrument() + + self._client.get("/foobar") + spans = self.memory_exporter.get_finished_spans() + self.assertEqual(len(spans), 0) + class TestAutoInstrumentationHooks(TestStarletteManualInstrumentationHooks): """