From 366b0c6f166491da062bcf0e1d930a03d7f0fd0e Mon Sep 17 00:00:00 2001
From: Cristian Pufu <cristian.pufu@uipath.com>
Date: Sun, 2 Nov 2025 11:55:31 +0200
Subject: [PATCH] fix: check static types in tests

---
 tests/conftest.py                   | 3 +--
 tests/tracing/test_serialization.py | 8 ++++++--
 tests/tracing/test_span_nesting.py  | 1 -
 3 files changed, 7 insertions(+), 5 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index 11aa76d..ed5b894 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -5,7 +5,6 @@
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import SimpleSpanProcessor
 from opentelemetry.sdk.trace.export.in_memory_span_exporter import InMemorySpanExporter
-from opentelemetry.sdk.trace.sampling import ALWAYS_ON, ParentBased
 
 
 class SpanCapture:
@@ -13,7 +12,7 @@ class SpanCapture:
 
     def __init__(self):
         self.exporter = InMemorySpanExporter()
-        self.provider = TracerProvider(sampler=ParentBased(root=ALWAYS_ON))
+        self.provider = TracerProvider()
         self.provider.add_span_processor(SimpleSpanProcessor(self.exporter))
         trace.set_tracer_provider(self.provider)
 
diff --git a/tests/tracing/test_serialization.py b/tests/tracing/test_serialization.py
index 5a077f2..ab4ed6a 100644
--- a/tests/tracing/test_serialization.py
+++ b/tests/tracing/test_serialization.py
@@ -118,13 +118,15 @@ def process_timestamp(timestamp: datetime):
 
 def test_traced_with_complex_return_value(span_capture: SpanCapture):
     """Test tracing with complex return value."""
+    from typing import Any, Dict
+
     from pydantic import BaseModel
 
     from uipath.core.tracing.decorators import traced
 
     class Result(BaseModel):
         success: bool
-        data: dict
+        data: Dict[str, Any]
 
     @traced(name="get_result")
     def get_result():
@@ -145,10 +147,12 @@ def get_result():
 
 def test_traced_with_set_and_tuple(span_capture: SpanCapture):
     """Test tracing with set and tuple inputs."""
+    from typing import Set, Tuple
+
     from uipath.core.tracing.decorators import traced
 
     @traced(name="process_collections")
-    def process_collections(items: set, pair: tuple):
+    def process_collections(items: Set[int], pair: Tuple[int, ...]) -> int:
         return len(items) + len(pair)
 
     result = process_collections({1, 2, 3}, (4, 5))
diff --git a/tests/tracing/test_span_nesting.py b/tests/tracing/test_span_nesting.py
index 613aba7..f4b5b96 100644
--- a/tests/tracing/test_span_nesting.py
+++ b/tests/tracing/test_span_nesting.py
@@ -1,7 +1,6 @@
 """Test span nesting behavior for traced decorators."""
 
 import pytest
-from opentelemetry import trace
 
 from tests.conftest import SpanCapture