fix tests

Author: SentienceDEV

tests/unit/test_extract.py

@@ -24,9 +24,7 @@ def model_name(self) -> str:
         return "stub"
 
 
-class BrowserStub:
-    page = True
-
+class PageStub:
     def __init__(self, content: str):
         self._content = content
 
@@ -40,6 +38,30 @@ def evaluate(self, _script: str, _opts: dict):
         }
 
 
+class AsyncPageStub:
+    def __init__(self, content: str):
+        self._content = content
+
+    async def evaluate(self, _script: str, _opts: dict):
+        return {
+            "status": "success",
+            "url": "https://example.com",
+            "format": "markdown",
+            "content": self._content,
+            "length": len(self._content),
+        }
+
+
+class BrowserStub:
+    def __init__(self, content: str):
+        self.page = PageStub(content)
+
+
+class AsyncBrowserStub:
+    def __init__(self, content: str):
+        self.page = AsyncPageStub(content)
+
+
 class ItemSchema(BaseModel):
     name: str
     price: str
@@ -64,7 +86,7 @@ def test_extract_schema_invalid_json() -> None:
 
 @pytest.mark.asyncio
 async def test_extract_async_schema_success() -> None:
-    browser = BrowserStub("Product: Widget")
+    browser = AsyncBrowserStub("Product: Widget")
     llm = LLMStub('{"name":"Widget","price":"$10"}')
     from sentience.read import extract_async
 

tests/unit/test_tool_registry.py

@@ -3,7 +3,13 @@
 import pytest
 from pydantic import BaseModel, ValidationError
 
-from sentience.tools import ToolContext, ToolRegistry, ToolSpec, register_default_tools
+from sentience.tools import (
+    ToolContext,
+    ToolRegistry,
+    ToolSpec,
+    UnsupportedCapabilityError,
+    register_default_tools,
+)
 
 
 class EchoInput(BaseModel):
@@ -143,7 +149,7 @@ def can(self, _name: str) -> bool:
             return False
 
     ctx = ToolContext(RuntimeStub())
-    with pytest.raises(ValueError, match="unsupported_capability"):
+    with pytest.raises(UnsupportedCapabilityError, match="unsupported_capability"):
         ctx.require("tabs")
 
 
@@ -213,10 +219,10 @@ async def snapshot(self, **_kwargs):
     ctx = ToolContext(RuntimeStub())
     register_default_tools(registry, ctx)
 
-    with pytest.raises(ValueError, match="unsupported_capability"):
+    with pytest.raises(UnsupportedCapabilityError, match="unsupported_capability"):
         await registry.execute("press", {"key": "Enter"}, ctx=ctx)
 
-    with pytest.raises(ValueError, match="unsupported_capability"):
+    with pytest.raises(UnsupportedCapabilityError, match="unsupported_capability"):
         await registry.execute(
             "scroll_to_element",
             {"element_id": 1, "behavior": "instant", "block": "center"},