class TrustedServerTest(unittest.TestCase): def setUp(self): self.trusted = TrustedServer(keys={}, max_connections=123, invoke_workers=False) def tearDown(self): if self.trusted.running: self.trusted.finish() self.trusted.join() def put_message_on_queue(self): self.trusted.input_queue.put('messsage') def start_multiple_connections(self, number): for _ in range(number): self.trusted.connect() def test_is_subclass_of_thread(self): self.assertTrue(isinstance(self.trusted, Thread)) def test_has_field_input_queue_that_is_a_queue(self): self.assertTrue(isinstance(self.trusted.input_queue, Queue)) def test_has_field_output_queue(self): self.assertTrue(isinstance(self.trusted.output_queue, Queue)) def test_creates_a_tuple_with_two_queues_when_connection_is_started(self): self.trusted.start() self.trusted.input_queue.put('message') output = self.trusted.output_queue.get() self.assertTrue(isinstance(output, tuple)) def put_multiple_messages_on_queue(self, number): for _ in range(number): self.put_message_on_queue()
def setUp(self): self.trusted = TrustedServer(keys={}, max_connections=123, invoke_workers=False)
from Utils import generate_random_key from TrustedServer import TrustedServer from Server import Server from Client import Client from threading import active_count, enumerate, current_thread, Thread import sys id_for_alice = 'alice' id_for_bob = 'bob' key_for_alice = generate_random_key() key_for_bob = generate_random_key() trusted_server = TrustedServer(keys={id_for_alice: key_for_alice, id_for_bob: key_for_bob},max_connections=10) trusted_server.start() server = Server(server_id=id_for_bob, server_key=key_for_bob, max_connections=10, trusted_server=trusted_server) server.start() client = Client(client_id=id_for_alice, client_key=key_for_alice, server=server, server_id=id_for_bob) client.start() def clean(): client.join(120) server.finish() trusted_server.finish() server.join(120) trusted_server.join(120) t = Thread(target=clean) t.start() t.join(360)
from Client import Client from Server import Server from TrustedServer import TrustedServer from Utils import generate_random_key id_for_alice = 'alice' id_for_bob = 'bob' key_for_alice = generate_random_key() key_for_bob = generate_random_key() trusted_server = TrustedServer(keys={id_for_alice: key_for_alice, id_for_bob: key_for_bob}, max_connections=10) trusted_server.start() server = Server(server_id=id_for_bob, server_key=key_for_bob, max_connections=10, trusted_server=trusted_server) server.start() client = Client(client_id=id_for_alice, client_key=key_for_alice, server=server, server_id=id_for_bob) client.start() client.join() server.finish() trusted_server.finish()