push

Author: hassiebp

tests/test_decorators.py

@@ -23,6 +23,12 @@
 mock_kwargs = {"a": 1, "b": 2, "c": 3}
 
 
+def removeMockResourceManagerInstances():
+    for public_key in LangfuseResourceManager._instances.keys():
+        if public_key != "pk-lf-1234567890":
+            LangfuseResourceManager._instances.pop(public_key)
+
+
 def test_nested_observations():
     mock_name = "test_nested_observations"
     langfuse = get_client()
@@ -1144,6 +1150,9 @@ def level_1_function(*args, **kwargs):
     assert len(trace_data.observations) == 3
     assert trace_data.name == mock_name
 
+    # Reset instances to not leak to other test suites
+    removeMockResourceManagerInstances()
+
 
 def test_multiproject_context_propagation_deep_nesting():
     client1 = Langfuse()  # Reads from environment
@@ -1204,6 +1213,9 @@ def level_1_function(*args, **kwargs):
     ]
     assert set(levels) == {"1", "2", "3", "4"}
 
+    # Reset instances to not leak to other test suites
+    removeMockResourceManagerInstances()
+
 
 def test_multiproject_context_propagation_override():
     # Initialize two separate Langfuse instances
@@ -1255,6 +1267,9 @@ def level_1_function(*args, **kwargs):
     assert len(trace_data.observations) == 2
     assert trace_data.name == mock_name
 
+    # Reset instances to not leak to other test suites
+    removeMockResourceManagerInstances()
+
 
 def test_multiproject_context_propagation_no_public_key():
     # Initialize two separate Langfuse instances
@@ -1307,6 +1322,9 @@ def level_1_function(*args, **kwargs):
         # Trace not found is also expected - tracing was completely disabled
         pass
 
+    # Reset instances to not leak to other test suites
+    removeMockResourceManagerInstances()
+
 
 @pytest.mark.asyncio
 async def test_multiproject_async_context_propagation_basic():
@@ -1374,6 +1392,9 @@ async def async_level_1_function(*args, **kwargs):
     ]
     assert all(async_flags)
 
+    # Reset instances to not leak to other test suites
+    removeMockResourceManagerInstances()
+
 
 @pytest.mark.asyncio
 async def test_multiproject_mixed_sync_async_context_propagation():
@@ -1443,6 +1464,9 @@ async def async_level_1_function(*args, **kwargs):
     assert "sync" in types
     assert "async" in types
 
+    # Reset instances to not leak to other test suites
+    removeMockResourceManagerInstances()
+
 
 @pytest.mark.asyncio
 async def test_multiproject_concurrent_async_context_isolation():
@@ -1531,6 +1555,9 @@ async def async_level_1_function(task_id, *args, **kwargs):
     assert "async_level_2_function" in trace_2_names
     assert "async_level_3_function" in trace_2_names
 
+    # Reset instances to not leak to other test suites
+    removeMockResourceManagerInstances()
+
 
 @pytest.mark.asyncio
 async def test_multiproject_async_generator_context_propagation():
@@ -1593,6 +1620,9 @@ async def async_consumer_function():
     # This confirms that the async generator inherited the public key context
     assert len(trace_data.observations) == 2
 
+    # Reset instances to not leak to other test suites
+    removeMockResourceManagerInstances()
+
 
 @pytest.mark.asyncio
 async def test_multiproject_async_context_exception_handling():
@@ -1658,4 +1688,5 @@ async def async_root_function(*args, **kwargs):
     )
     assert caught_obs is not None
 
-    LangfuseResourceManager.reset()
+    # Reset instances to not leak to other test suites
+    removeMockResourceManagerInstances()