Merge pull request #1 from UiPath/fix/lint_tests

fix: check static types in tests

Author: cristipufu

tests/conftest.py

@@ -5,15 +5,14 @@
 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:
     """Helper to capture and analyze spans."""
 
     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)
 

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))

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