Add stub for openai clients

Author: seratch

tests/unit/test_runtime.py

@@ -29,14 +29,36 @@
 
 @pytest.fixture(autouse=True)
 def stub_openai_module(monkeypatch: pytest.MonkeyPatch) -> Iterator[types.ModuleType]:
-    """Provide a stub ``openai.AsyncOpenAI`` for modules under test."""
+    """Provide a stub ``openai.AsyncOpenAI`` and patch imports in guardrails.*.
+
+    Ensures tests don't require real OPENAI_API_KEY or networked clients.
+    """
     module = types.ModuleType("openai")
 
-    class AsyncOpenAI:
+    class AsyncOpenAI:  # noqa: D401 - simple stub
+        """Stubbed AsyncOpenAI client."""
+
         pass
 
     module.__dict__["AsyncOpenAI"] = AsyncOpenAI
+    # Ensure any downstream import finds our stub module
     monkeypatch.setitem(sys.modules, "openai", module)
+    # Also patch already-imported symbols on guardrails modules
+    try:
+        import guardrails.runtime as gr_runtime  # type: ignore
+
+        monkeypatch.setattr(gr_runtime, "AsyncOpenAI", AsyncOpenAI, raising=False)
+    except Exception:
+        pass
+    try:
+        import guardrails.types as gr_types  # type: ignore
+
+        monkeypatch.setattr(gr_types, "AsyncOpenAI", AsyncOpenAI, raising=False)
+    except Exception:
+        pass
+    # Provide dummy API key to satisfy any code paths that inspect env
+    monkeypatch.setenv("OPENAI_API_KEY", "test-key")
+
     yield module
     monkeypatch.delitem(sys.modules, "openai", raising=False)
 

tests/unit/test_types.py

@@ -10,14 +10,29 @@
 
 @pytest.fixture(autouse=True)
 def stub_openai_module(monkeypatch: pytest.MonkeyPatch) -> Iterator[types.ModuleType]:
-    """Provide a stub ``openai.AsyncOpenAI`` for modules under test."""
+    """Provide a stub ``openai.AsyncOpenAI`` and patch guardrails types symbol.
+
+    Ensures tests don't require real OPENAI_API_KEY or networked clients.
+    """
     module = types.ModuleType("openai")
 
-    class AsyncOpenAI:
+    class AsyncOpenAI:  # noqa: D401 - simple stub
+        """Stubbed AsyncOpenAI client."""
+
         pass
 
     module.__dict__["AsyncOpenAI"] = AsyncOpenAI
     monkeypatch.setitem(sys.modules, "openai", module)
+    # Patch already-imported symbol in guardrails.types if present
+    try:
+        import guardrails.types as gr_types  # type: ignore
+
+        monkeypatch.setattr(gr_types, "AsyncOpenAI", AsyncOpenAI, raising=False)
+    except Exception:
+        pass
+    # Provide dummy API key in case any code inspects env
+    monkeypatch.setenv("OPENAI_API_KEY", "test-key")
+
     yield module
     monkeypatch.delitem(sys.modules, "openai", raising=False)