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()
