from sfm.database import SeismoDb
from minimateplus.models import Event


def _ins(db, eid_key="01110000", serial="BE1"):
    ev = Event(index=0); ev._waveform_key = bytes.fromhex(eid_key)
    db.insert_events([ev], serial=serial)
    return db.query_events(serial=serial)[0]["id"]


def test_confirm_real_sets_and_clears_ft(tmp_path):
    db = SeismoDb(tmp_path/"s.db"); eid = _ins(db)
    db.update_event_review(eid, {"false_trigger": True})
    assert db.get_event(eid)["false_trigger"] == 1
    db.update_event_review(eid, {"reviewed_real": True})   # confirming real clears FT
    row = db.get_event(eid)
    assert row["reviewed_real"] == 1 and row["false_trigger"] == 0


def test_flag_ft_clears_reviewed_real(tmp_path):
    db = SeismoDb(tmp_path/"s.db"); eid = _ins(db)
    db.update_event_review(eid, {"reviewed_real": True})
    db.update_event_review(eid, {"false_trigger": True})
    row = db.get_event(eid)
    assert row["false_trigger"] == 1 and row["reviewed_real"] == 0