def test_client_get_network_info():
alice = env.clients[0]
alice.DATABASE_NAME = 'test.db'
alice.get_network_info()
assert alice.pubs_mixes == group_layered_topology(env.pubs_mixes)
assert alice.pubs_providers == env.pubs_providers
assert alice.befriended_clients == env.pubs_clients
def test_mix_get_network_info():
mix = env.mixes[0]
mix.DATABASE_NAME = 'test.db'
mix.get_network_info()
other_mixes = [m for m in env.pubs_mixes if mix.name != m.name]
assert mix.pubs_mixes == group_layered_topology(other_mixes)
assert mix.pubs_providers == env.pubs_providers
provider = env.providers[0]
provider.DATABASE_NAME = 'test.db'
provider.get_network_info()
assert provider.pubs_mixes == group_layered_topology(env.pubs_mixes)
other_providers = [
p for p in env.pubs_providers if provider.name != p.name
]
assert provider.pubs_providers == other_providers
def test_client_startProtocol():
alice = env.clients[0]
alice.startProtocol()
pkt, addr = alice.transport.written[0]
data = ['SUBSCRIBE', alice.name, alice.host, alice.port]
assert pkt == petlib.pack.encode(data)
assert alice.befriended_clients == env.pubs_clients
assert alice.pubs_providers == env.pubs_providers
assert alice.pubs_mixes == group_layered_topology(env.pubs_mixes)
pkt, addr = alice.transport.written[1]
assert pkt == petlib.pack.encode(['PULL', alice.name])
def register_mixes(self, mixes):
self.pubs_mixes = group_layered_topology(mixes)
def test_mix_register_mixes():
[mix.register_mixes(env.pubs_mixes) for mix in env.mixes]
mix = env.mixes[0]
assert mix.pubs_mixes == group_layered_topology(env.pubs_mixes)
