ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter9/browser-use-rpa/test_state_predicates.py

@@ -0,0 +1,45 @@
+import tempfile
+import unittest
+
+from learning_agent import KnowledgeBase
+from learning_agent.workflow import WorkflowStatus
+from workflow_validation_demo import (
+    candidate_workflow,
+    changed_environment,
+    replay,
+    validate_candidate,
+)
+
+
+class StatePredicateTest(unittest.TestCase):
+    def test_candidate_must_pass_reset_replay_before_storage(self):
+        workflow = candidate_workflow()
+        self.assertEqual(WorkflowStatus.CANDIDATE, workflow.validation_status)
+        with tempfile.TemporaryDirectory() as directory:
+            store = KnowledgeBase(directory)
+            store.save_candidate(workflow)
+            with self.assertRaises(ValueError):
+                store.save_workflow(workflow)
+
+            result = validate_candidate(workflow)
+            self.assertTrue(result["success"])
+            store.publish_validated(workflow)
+            self.assertIn(workflow.workflow_id, store.workflows)
+
+    def test_page_change_invalidates_workflow_and_requests_fallback(self):
+        workflow = candidate_workflow()
+        validate_candidate(workflow)
+        result = replay(workflow, changed_environment())
+        self.assertFalse(result["success"])
+        self.assertTrue(result["fallback_required"])
+        self.assertIn("send button is visible", result["error"])
+
+    def test_predicates_survive_serialization(self):
+        original = candidate_workflow()
+        restored = type(original).from_json(original.to_json())
+        self.assertEqual("element_visible", restored.steps[0].preconditions[0].predicate_type.value)
+        self.assertEqual("element_text_contains", restored.final_predicates[0].predicate_type.value)
+
+
+if __name__ == "__main__":
+    unittest.main()