def test_clear_addrs(): store = PeerStore() store.add_addrs("peer1", ["/foo1", "/bar1"], 10) store.add_addrs("peer2", ["/foo2"], 10) store.clear_addrs("peer1") assert store.addrs("peer1") == [] assert store.addrs("peer2") == ["/foo2"] store.add_addrs("peer1", ["/foo1", "/bar1"], 10) assert store.addrs("peer1") == ["/foo1", "/bar1"]
def test_add_addrs_multiple(): store = PeerStore() store.add_addrs("peer1", ["/foo1", "/bar1"], 10) store.add_addrs("peer2", ["/foo2"], 10) assert store.addrs("peer1") == ["/foo1", "/bar1"] assert store.addrs("peer2") == ["/foo2"]
def test_add_addr_single(): store = PeerStore() store.add_addr("peer1", "/foo", 10) store.add_addr("peer1", "/bar", 10) store.add_addr("peer2", "/baz", 10) assert store.addrs("peer1") == ["/foo", "/bar"] assert store.addrs("peer2") == ["/baz"]
def test_addrs_empty(): with pytest.raises(PeerStoreError): store = PeerStore() val = store.addrs("peer") assert not val