18 lines
570 B
Python
18 lines
570 B
Python
|
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
|