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)