mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-05-23 04:17:34 +00:00
eaefc4f2ec
Patch _now_ts so two answers share the same second; verify both fetch paths return rows ordered by id when timestamps match. Co-authored-by: nic <nicsins@users.noreply.github.com>
96 lines
4.1 KiB
Python
96 lines
4.1 KiB
Python
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from python.surveys.db import SurveyDB
|
|
from python.surveys.parser import parse_survey_page
|
|
from python.surveys.schemas import FieldKind, Persona, SurveyField, UserProfile
|
|
from python.surveys.profile_refiner import _deep_merge
|
|
|
|
|
|
class TestSurveyParser(unittest.TestCase):
|
|
def test_groups_radio_checkbox(self):
|
|
dom = """
|
|
<h1>Demo</h1>
|
|
<div>Preferred mobile platform</div>
|
|
<div><input type="radio" name="mobile" selector="1a" /> Android</div>
|
|
<div><input type="radio" name="mobile" selector="2a" /> iOS</div>
|
|
<div>Topics</div>
|
|
<div><input type="checkbox" name="topics" selector="3a" /> Music</div>
|
|
<div><input type="checkbox" name="topics" selector="4a" /> Sports</div>
|
|
"""
|
|
page = parse_survey_page(dom, url="file://demo")
|
|
radios = [f for f in page.fields if f.kind == FieldKind.RADIO]
|
|
checks = [f for f in page.fields if f.kind == FieldKind.CHECKBOX]
|
|
self.assertEqual(len(radios), 1)
|
|
self.assertEqual(len(checks), 1)
|
|
self.assertIn("android", " ".join(radios[0].options).lower())
|
|
self.assertTrue(radios[0].option_selectors)
|
|
self.assertEqual(set(checks[0].option_selectors.values()), {"3a", "4a"})
|
|
|
|
|
|
class TestSurveyDB(unittest.TestCase):
|
|
def test_db_roundtrip_and_events(self):
|
|
db = SurveyDB(":memory:")
|
|
try:
|
|
persona = Persona(id="p1", name="Test", description="x", constraints={"a": 1})
|
|
db.upsert_persona(persona)
|
|
got = db.get_persona("p1")
|
|
self.assertIsNotNone(got)
|
|
self.assertEqual(got.name, "Test")
|
|
|
|
profile = UserProfile(id="default", persona_id="p1", data={"demographics": {"country": "DE"}})
|
|
db.upsert_profile(profile)
|
|
|
|
db.create_session("s1", url="file://demo", persona_id="p1", profile_id="default")
|
|
field = SurveyField(selector="1a", kind=FieldKind.TEXT, label="Email")
|
|
db.insert_answer("a1", "s1", "Email", field, "test@example.com")
|
|
|
|
evs = db.fetch_unprocessed_answer_events()
|
|
self.assertEqual(len(evs), 1)
|
|
self.assertEqual(evs[0]["profile_id"], "default")
|
|
self.assertEqual(evs[0]["persona_id"], "p1")
|
|
self.assertEqual(evs[0]["answer_text"], "test@example.com")
|
|
|
|
db.mark_answers_processed(["a1"])
|
|
self.assertEqual(db.fetch_unprocessed_answer_events(), [])
|
|
finally:
|
|
db.close()
|
|
|
|
def test_unprocessed_order_stable_when_created_at_ties(self):
|
|
"""Same-second inserts must not rely on undefined SQLite ordering."""
|
|
fixed_ts = 1_700_000_000
|
|
with patch("python.surveys.db._now_ts", return_value=fixed_ts):
|
|
db = SurveyDB(":memory:")
|
|
try:
|
|
persona = Persona(id="p1", name="Test", description="x", constraints={})
|
|
db.upsert_persona(persona)
|
|
profile = UserProfile(id="default", persona_id="p1", data={})
|
|
db.upsert_profile(profile)
|
|
db.create_session("s1", url="file://demo", persona_id="p1", profile_id="default")
|
|
field = SurveyField(selector="1a", kind=FieldKind.TEXT, label="Q")
|
|
# Insert lexicographically later id first; tie-break must still order by id ASC.
|
|
db.insert_answer("b2", "s1", "Q", field, "second")
|
|
db.insert_answer("a1", "s1", "Q", field, "first")
|
|
|
|
ids_plain = [r["id"] for r in db.fetch_unprocessed_answers()]
|
|
ids_events = [r["id"] for r in db.fetch_unprocessed_answer_events()]
|
|
self.assertEqual(ids_plain, ["a1", "b2"])
|
|
self.assertEqual(ids_events, ["a1", "b2"])
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
class TestDeepMerge(unittest.TestCase):
|
|
def test_deep_merge(self):
|
|
dst = {"a": {"b": 1}, "x": 1}
|
|
src = {"a": {"c": 2}, "x": 2, "y": 3}
|
|
out = _deep_merge(dst, src)
|
|
self.assertEqual(out["a"]["b"], 1)
|
|
self.assertEqual(out["a"]["c"], 2)
|
|
self.assertEqual(out["x"], 2)
|
|
self.assertEqual(out["y"], 3)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|
|
|