fix: manual cleanup

Author: andreiancuta-uipath

packages/uipath-platform/tests/common/test_interrupt_models_compatibility.py

@@ -1,11 +1,8 @@
 import importlib
 import subprocess
 import sys
-import textwrap
 import warnings
 
-import pytest
-
 # TODO: Remove these tests with the compatibility bridge in the next breaking-change release.
 INTERRUPT_MODEL_NAMES = (
     "CreateBatchTransform",
@@ -62,8 +59,6 @@ def test_compatibility_exports_are_canonical_objects():
 
     assert set(INTERRUPT_MODEL_NAMES) == set(legacy.__all__)
     assert set(INTERRUPT_MODEL_NAMES) <= set(common.__all__)
-    assert set(legacy.__all__) <= set(dir(legacy))
-    assert set(INTERRUPT_MODEL_NAMES) <= set(dir(common))
 
 
 def test_importing_common_does_not_load_resume_triggers_or_warn():
@@ -83,30 +78,16 @@ def test_importing_common_does_not_load_resume_triggers_or_warn():
     )
 
 
-@pytest.mark.parametrize(
-    "imports",
-    [
-        """
-from uipath.platform.resume_triggers import WaitUntil as canonical
-from uipath.platform.common import WaitUntil as legacy_root
-from uipath.platform.common.interrupt_models import WaitUntil as legacy_direct
-""",
-        """
-from uipath.platform.common.interrupt_models import WaitUntil as legacy_direct
-from uipath.platform.common import WaitUntil as legacy_root
-from uipath.platform.resume_triggers import WaitUntil as canonical
-""",
-    ],
-)
-def test_import_orders_are_cycle_safe(imports: str):
-    indented_imports = textwrap.indent(imports.strip(), "    ")
+def test_legacy_first_import_is_cycle_safe_and_warns():
     _run_python(
-        f"""
+        """
 import warnings
 
 with warnings.catch_warnings(record=True) as caught:
     warnings.simplefilter("always")
-{indented_imports}
+    from uipath.platform.common.interrupt_models import WaitUntil as legacy_direct
+    from uipath.platform.common import WaitUntil as legacy_root
+    from uipath.platform.resume_triggers import WaitUntil as canonical
 
 assert canonical is legacy_root is legacy_direct
 compatibility_warnings = [
@@ -118,66 +99,3 @@ def test_import_orders_are_cycle_safe(imports: str):
 assert compatibility_warnings
 """
     )
-
-
-def test_legacy_imports_warn_at_caller_site():
-    _run_python(
-        """
-import warnings
-
-with warnings.catch_warnings(record=True) as caught:
-    warnings.simplefilter("always")
-    from uipath.platform.common import WaitUntil
-    from uipath.platform.common.interrupt_models import WaitTask
-
-compatibility_warnings = [
-    item
-    for item in caught
-    if issubclass(item.category, DeprecationWarning)
-    and "uipath.platform.resume_triggers" in str(item.message)
-]
-assert len(compatibility_warnings) == 2
-assert all(item.filename == "<string>" for item in compatibility_warnings), [
-    item.filename for item in compatibility_warnings
-]
-"""
-    )
-
-
-def test_cached_resolution_warns_once_per_name():
-    _run_python(
-        """
-import warnings
-
-import uipath.platform.common as common
-
-with warnings.catch_warnings(record=True) as caught:
-    warnings.simplefilter("always")
-    common.WaitUntil
-    common.WaitUntil
-    common.WaitJob
-
-compatibility_warnings = [
-    item
-    for item in caught
-    if issubclass(item.category, DeprecationWarning)
-    and "uipath.platform.resume_triggers" in str(item.message)
-]
-assert len(compatibility_warnings) == 2
-"""
-    )
-
-
-def test_star_import_exports_all_interrupt_models():
-    _run_python(
-        """
-import warnings
-
-warnings.simplefilter("ignore", DeprecationWarning)
-
-from uipath.platform.common.interrupt_models import *  # noqa: F403
-
-for name in ("InvokeProcess", "WaitUntil", "CreateTask", "WaitIntegrationEvent"):
-    assert name in globals(), name
-"""
-    )

packages/uipath-platform/tests/services/test_guardrails_decorators.py

@@ -989,32 +989,6 @@ def test_custom_validator_path_delegates_to_run(self):
         )
         assert result == _PASSED
 
-    def test_built_in_validator_uses_injected_service(self):
-        """BuiltInGuardrailValidator.run() delegates to an injected service."""
-        from uipath.platform.guardrails.decorators.validators import (
-            BuiltInGuardrailValidator,
-        )
-        from uipath.platform.guardrails.guardrails import BuiltInValidatorGuardrail
-
-        mock_built_in = MagicMock(spec=BuiltInValidatorGuardrail)
-
-        class _TestBuiltIn(BuiltInGuardrailValidator):
-            def get_built_in_guardrail(self, name, description, enabled_for_evals):
-                return mock_built_in
-
-        validator = _TestBuiltIn()
-        fake_service = MagicMock()
-        fake_service.evaluate_guardrail.return_value = _PASSED
-        validator._service = fake_service
-        evaluator = _make_evaluator(validator, "G", None, True)
-
-        result = evaluator({"text": "hello"}, GuardrailExecutionStage.PRE, None, None)
-
-        fake_service.evaluate_guardrail.assert_called_once_with(
-            {"text": "hello"}, mock_built_in
-        )
-        assert result == _PASSED
-
     def test_built_in_validator_path_lazy_initializes_default_service(self):
         """BuiltInGuardrailValidator.run() creates the default service once."""
         from uipath.platform.guardrails.decorators.validators import (
@@ -1071,28 +1045,6 @@ def get_built_in_guardrail(self, name, description, enabled_for_evals):
             evaluator({"text": "hello"}, GuardrailExecutionStage.PRE, None, None)
 
 
-# ---------------------------------------------------------------------------
-# 15b. default_guardrails_service
-# ---------------------------------------------------------------------------
-
-
-class TestDefaultGuardrailsService:
-    def test_builds_guardrails_service_from_env(self, monkeypatch):
-        """The default factory builds a GuardrailsService from env config."""
-        from uipath.platform.guardrails._guardrails_service import (
-            GuardrailsService,
-            default_guardrails_service,
-        )
-
-        monkeypatch.setenv("UIPATH_URL", "https://test.uipath.com")
-        monkeypatch.setenv("UIPATH_ACCESS_TOKEN", "token")
-
-        service = default_guardrails_service()
-        assert isinstance(service, GuardrailsService)
-        assert service._config.base_url == "https://test.uipath.com"
-        assert service._config.secret == "token"
-
-
 # ---------------------------------------------------------------------------
 # 16. Joke-agent integration scenarios (plain functions)
 # ---------------------------------------------------------------------------

packages/uipath-platform/tests/services/test_guardrails_service.py

@@ -17,6 +17,7 @@
     GuardrailsService,
     MapEnumParameterValue,
 )
+from uipath.platform.guardrails._guardrails_service import default_guardrails_service
 
 
 @pytest.fixture
@@ -29,6 +30,18 @@ def service(
     return GuardrailsService(config=config, execution_context=execution_context)
 
 
+class TestDefaultGuardrailsService:
+    def test_builds_guardrails_service_from_env(self, monkeypatch):
+        """The default factory builds a GuardrailsService from env config."""
+        monkeypatch.setenv("UIPATH_URL", "https://test.uipath.com")
+        monkeypatch.setenv("UIPATH_ACCESS_TOKEN", "token")
+
+        service = default_guardrails_service()
+        assert isinstance(service, GuardrailsService)
+        assert service._config.base_url == "https://test.uipath.com"
+        assert service._config.secret == "token"
+
+
 class TestGuardrailsService:
     """Test GuardrailsService functionality."""