from tamq.store import Store


def test_disconnect_endpoint(tmp_path):
    store = Store(tmp_path / "queue.sqlite3")
    store.register_endpoint("ep", 1, "tamq", ["a"])
    store.disconnect_endpoint("ep")
    assert store.endpoint("ep") is None


def test_disconnect_all_marks_active_endpoints(tmp_path):
    store = Store(tmp_path / "queue.sqlite3")
    store.register_endpoint("ep-1", 1, "tamq", ["a"])
    store.register_endpoint("ep-2", 2, "tamq", ["b"])
    store.disconnect_all()
    assert store.endpoint("ep-1") is None
    assert store.endpoint("ep-2") is None