import sqlite3
from sfm.database import SeismoDb
from minimateplus.models import Event, Timestamp


def _ins(db, key, serial, pvs, ts):
    ev = Event(index=0)
    ev._waveform_key = bytes.fromhex(key)
    ev.timestamp = ts
    # peak_vector_sum comes from peak_values; simplest: insert then UPDATE pvs directly
    db.insert_events([ev], serial=serial)
    row = [r for r in db.query_events(serial=serial) if r["waveform_key"] == key][0]
    with sqlite3.connect(db.db_path) as c:
        c.execute("UPDATE events SET peak_vector_sum=? WHERE id=?", (pvs, row["id"]))
    return row["id"]


def test_propagate_copies_flags_to_twins(tmp_path):
    db = SeismoDb(tmp_path / "s.db")
    base = Timestamp(raw=b"", flag=0x10, year=2026, unknown_byte=0, month=2, day=25, hour=20, minute=19, second=5)
    twin = Timestamp(raw=b"", flag=0x10, year=2026, unknown_byte=0, month=2, day=25, hour=20, minute=19, second=45)
    other = Timestamp(raw=b"", flag=0x10, year=2026, unknown_byte=0, month=2, day=25, hour=20, minute=19, second=44)

    primary_id = _ins(db, "01110001", "BE1", 0.4763, base)
    twin_id = _ins(db, "01110002", "BE1", 0.4763, twin)        # twin: same serial+pvs, 40s apart
    non_twin_id = _ins(db, "01110003", "BE1", 0.9999, other)   # near time but different pvs

    db.update_event_review(primary_id, {"false_trigger": True})
    moved = db.propagate_review_to_twins(primary_id)

    assert twin_id in moved
    assert db.get_event(twin_id)["false_trigger"] == 1
    assert db.get_event(non_twin_id)["false_trigger"] == 0