add tests

Author: rcholic

tests/integration/test_agent_workflows.py

@@ -189,9 +189,9 @@ def test_agent_workflow_with_retry(self):
             from sentience.models import ActionResult
 
             mock_snapshot.return_value = create_mock_snapshot()
-            # First call fails, second succeeds
+            # First call raises exception (triggers retry), second succeeds
             mock_click.side_effect = [
-                ActionResult(success=False, duration_ms=100, outcome="error"),
+                RuntimeError("Element not found"),
                 ActionResult(success=True, duration_ms=150, outcome="dom_updated"),
             ]
 
@@ -232,7 +232,7 @@ def test_agent_workflow_finish_action(self):
         llm = MockLLMProvider(responses=["FINISH()"])
         agent = SentienceAgent(browser, llm, verbose=False)
 
-        with patch("sentience.snapshot.snapshot") as mock_snapshot:
+        with patch("sentience.agent.snapshot") as mock_snapshot:
             mock_snapshot.return_value = create_mock_snapshot()
 
             result = agent.act("Task is complete", max_retries=0)
@@ -421,4 +421,3 @@ def test_agent_step_count_increments(self):
 
             agent.act("Second action", max_retries=0)
             assert agent._step_count == initial_count + 2
-

tests/test_utils_browser.py

@@ -149,4 +149,3 @@ def test_save_storage_state_prints_success_message(self, capsys):
             assert "✅" in captured.out
             assert "Saved storage state" in captured.out
             assert str(file_path) in captured.out
-