import threading

from chatmaild.filedict import FileDict, write_bytes_atomic


def test_basic(tmp_path):
    fdict = FileDict(tmp_path.joinpath("metadata"))
    assert fdict.read() == {}
    with fdict.modify() as d:
        d["devicetoken"] = [1, 2, 3]
        d["456"] = 4.2
    new = fdict.read()
    assert new["devicetoken"] == [1, 2, 3]
    assert new["456"] == 4.2


def test_bad_marshal_file(tmp_path, caplog):
    fdict1 = FileDict(tmp_path.joinpath("metadata"))
    fdict1.path.write_bytes(b"l12k3l12k3l")
    assert fdict1.read() == {}
    assert "corrupt" in caplog.records[0].msg


def test_write_bytes_atomic_concurrent(tmp_path):
    p = tmp_path.joinpath("somefile.ext")
    write_bytes_atomic(p, b"hello")

    threads = []
    for i in range(30):
        content = f"hello{i}".encode("ascii")
        t = threading.Thread(target=lambda: write_bytes_atomic(p, content))
        t.start()
        threads.append(t)

    for t in threads:
        t.join()

    assert p.read_text().strip() != "hello"
    assert len(list(p.parent.iterdir())) == 1