Fix threading instrumentation contexxt types

Add None check to context handling in thread instrumentation
Add testcases for None context

Author: jomcgi

instrumentation/opentelemetry-instrumentation-threading/src/opentelemetry/instrumentation/threading/__init__.py

@@ -150,7 +150,8 @@ def __wrap_threading_run(
             token = context.attach(instance._otel_context)
             return call_wrapped(*args, **kwargs)
         finally:
-            context.detach(token)
+            if token is not None:
+                context.detach(token)
 
     @staticmethod
     def __wrap_thread_pool_submit(
@@ -169,7 +170,8 @@ def wrapped_func(*func_args: Any, **func_kwargs: Any) -> R:
                 token = context.attach(otel_context)
                 return original_func(*func_args, **func_kwargs)
             finally:
-                context.detach(token)
+                if token is not None:
+                    context.detach(token)
 
         # replace the original function with the wrapped function
         new_args: tuple[Callable[..., Any], ...] = (wrapped_func,) + args[1:]

instrumentation/opentelemetry-instrumentation-threading/tests/test_threading.py

@@ -15,12 +15,14 @@
 import threading
 from concurrent.futures import ThreadPoolExecutor
 from typing import List
+from unittest.mock import MagicMock, patch
 
-from opentelemetry import trace
+from opentelemetry import context, trace
 from opentelemetry.instrumentation.threading import ThreadingInstrumentor
 from opentelemetry.test.test_base import TestBase
 
 
+# pylint: disable=too-many-public-methods
 class TestThreading(TestBase):
     def setUp(self):
         super().setUp()
@@ -224,3 +226,46 @@ def test_uninstrumented(self):
         self.assertEqual(len(spans), 1)
 
         ThreadingInstrumentor().instrument()
+
+    def test_threading_with_none_context_token(self):
+        with self.get_root_span(), patch(
+            "opentelemetry.context.attach", return_value=None
+        ), patch("opentelemetry.context.detach") as mock_detach:
+            thread = threading.Thread(target=self.fake_func)
+            thread.start()
+            thread.join()
+            mock_detach.assert_not_called()
+
+    def test_threading_with_valid_context_token(self):
+        mock_token = MagicMock(spec=context.Token)
+        with patch(
+            "opentelemetry.context.attach", return_value=mock_token
+        ), self.get_root_span(), patch(
+            "opentelemetry.context.detach", autospec=True
+        ) as mock_detach:
+            thread = threading.Thread(target=self.fake_func)
+            thread.start()
+            thread.join()
+            mock_detach.assert_called_once()
+
+    def test_thread_pool_with_none_context_token(self):
+        with self.get_root_span(), patch(
+            "opentelemetry.context.attach", return_value=None
+        ), patch(
+            "opentelemetry.context.detach"
+        ) as mock_detach, ThreadPoolExecutor(max_workers=1) as executor:
+            future = executor.submit(self.get_current_span_context_for_test)
+            future.result()
+
+            mock_detach.assert_not_called()
+
+    def test_threadpool_with_valid_context_token(self):
+        mock_token = MagicMock(spec=context.Token)
+        with self.get_root_span(), patch(
+            "opentelemetry.context.attach", return_value=mock_token
+        ), patch(
+            "opentelemetry.context.detach", autospec=True
+        ) as mock_detach, ThreadPoolExecutor(max_workers=1) as executor:
+            future = executor.submit(self.get_current_span_context_for_test)
+            future.result()
+            mock_detach.assert_called_once()