def test_create_message_id_list(self): """Test message ID list request creation""" msg1 = Message('M1') msg2 = Message('M2') msg3 = Message('M3') tc = b'\x01\x03\x03\x07' tc_str_ok = dandelion.util.encode_bytes(tc).decode() str_ = Protocol.create_message_id_list(tc, [msg1, msg2, msg3])[:-1] tc_str, m1_str, m2_str, m3_str = str_.split(';') self.assertEqual(tc_str, tc_str_ok) self.assertEqual(msg1.id, dandelion.util.decode_bytes(m1_str.encode())) self.assertEqual(msg2.id, dandelion.util.decode_bytes(m2_str.encode())) self.assertEqual(msg3.id, dandelion.util.decode_bytes(m3_str.encode())) str_ = Protocol.create_message_id_list(tc, None)[:-1] self.assertEqual(str_, tc_str) str_ = Protocol.create_message_id_list(tc, [])[:-1] self.assertEqual(str_, tc_str) """Testing bad input""" self.assertRaises(TypeError, Protocol.create_message_id_list, 1337, None) self.assertRaises(TypeError, Protocol.create_message_id_list, tc, msg1) self.assertRaises(TypeError, Protocol.create_message_id_list, [msg1], tc) self.assertRaises(AttributeError, Protocol.create_message_id_list, tc, tc) self.assertRaises(ValueError, Protocol.create_message_id_list, None, []) self.assertRaises(TypeError, Protocol.create_message_id_list, 0, None) self.assertRaises(AttributeError, Protocol.create_message_id_list, tc, ['fo'])
def test_create_message_id_list(self): """Test message ID list request creation""" msg1 = Message('M1') msg2 = Message('M2') msg3 = Message('M3') tc = b'\x01\x03\x03\x07' tc_str_ok = dandelion.util.encode_bytes(tc).decode() str_ = Protocol.create_message_id_list(tc, [msg1, msg2, msg3])[:-1] tc_str, m1_str, m2_str, m3_str = str_.split(';') self.assertEqual(tc_str, tc_str_ok) self.assertEqual(msg1.id, dandelion.util.decode_bytes(m1_str.encode())) self.assertEqual(msg2.id, dandelion.util.decode_bytes(m2_str.encode())) self.assertEqual(msg3.id, dandelion.util.decode_bytes(m3_str.encode())) str_ = Protocol.create_message_id_list(tc, None)[:-1] self.assertEqual(str_, tc_str) str_ = Protocol.create_message_id_list(tc, [])[:-1] self.assertEqual(str_, tc_str) """Testing bad input""" self.assertRaises(TypeError, Protocol.create_message_id_list, 1337, None) self.assertRaises(TypeError, Protocol.create_message_id_list, tc, msg1) self.assertRaises(TypeError, Protocol.create_message_id_list, [msg1], tc) self.assertRaises(AttributeError, Protocol.create_message_id_list, tc, tc) self.assertRaises(ValueError, Protocol.create_message_id_list, None, []) self.assertRaises(TypeError, Protocol.create_message_id_list, 0, None) self.assertRaises(AttributeError, Protocol.create_message_id_list, tc, ['fo'])
def test_basic_server_transaction(self): """Tests the server transaction protocol and logic""" db = ContentDB() tc = db.add_messages([Message('fubar'), Message('foo'), Message('bar')]) with TestServerHelper() as server_helper, TestClientHelper() as client_helper: srv_transaction = ServerTransaction(server_helper.sock, db) test_client = SocketTransaction(client_helper.sock, b'\n') """Run the server transaction in a separate thread to allow client access""" thread = threading.Thread(target=srv_transaction.process) thread.start() """Check greeting from server""" rcv = test_client._read() self.assertEqual(rcv, Protocol.create_greeting_message(db.id).encode()) """Check response to mdgid list req""" test_client._write(Protocol.create_message_id_list_request(tc).encode()) rcv = test_client._read() self.assertEqual(rcv, Protocol.create_message_id_list(tc, None).encode()) """Check response to mdg req""" test_client._write(Protocol.create_message_list_request([msg.id for msg in db.get_messages()]).encode()) rcv = test_client._read() self.assertEqual(rcv, Protocol.create_message_list(db.get_messages()).encode()) """Wait for server (will time out if no requests)""" thread.join(2*TIMEOUT)
def _process_data(self, bdata): """Internal helper function that processes what should be a server request.""" try: data = bdata.decode() if Protocol.is_message_id_list_request(data): tc = Protocol.parse_message_id_list_request(data) tc, msgs = self._db.messages_since(tc) response_str = Protocol.create_message_id_list(tc, msgs) self._write(response_str.encode()) elif Protocol.is_message_list_request(data): msgids = Protocol.parse_message_list_request(data) msgs = self._db.get_messages(msgids) response_str = Protocol.create_message_list(msgs) self._write(response_str.encode()) else: raise ProtocolParseError except (ProtocolParseError, ValueError, TypeError): #print("SERVER TRANSACTION: Error processing data from client") raise ServerTransaction._AbortTransactionException
def test_basic_client_transaction(self): """Tests the client transaction protocol and logic""" client_db = ContentDB() srv_db = ContentDB() tc = srv_db.add_messages( [Message('fubar'), Message('foo'), Message('bar')]) self.assertEqual(client_db.message_count, 0) self.assertEqual(srv_db.message_count, 3) with TestServerHelper() as server_helper, TestClientHelper( ) as client_helper: client_transaction = ClientTransaction(client_helper.sock, client_db) srv_sock = SocketTransaction(server_helper.sock, b'\n') """Run the client transaction in a separate thread""" thread = threading.Thread(target=client_transaction.process) thread.start() """Send a greeting (should be req. by client)""" srv_sock._write( Protocol.create_greeting_message(srv_db.id).encode()) """Reading msg id list request""" rcv = srv_sock._read() self.assertEqual( rcv, Protocol.create_message_id_list_request().encode()) """Sending the msg id list""" srv_sock._write( Protocol.create_message_id_list( tc, srv_db.get_messages()).encode()) """Reading msg list request""" rcv = srv_sock._read() self.assertEqual( rcv, Protocol.create_message_list_request( [msg.id for msg in srv_db.get_messages()]).encode()) """Sending the msg id list""" srv_sock._write( Protocol.create_message_list(srv_db.get_messages()).encode()) """Wait for client to hang up""" thread.join(2 * TIMEOUT) """Make sure the client has updated the db""" self.assertEqual(client_db.message_count, 3) self.assertEqual(srv_db.message_count, 3) self.assertEqual( len([ srvmsg for srvmsg in srv_db.get_messages() if srvmsg not in client_db.get_messages() ]), 0)
def test_roundtrip_message_id_list(self): """Test message ID list response creation / parsing by a round trip""" msg1 = Message('M1') msg2 = Message('M2') msg3 = Message('M3') tc, msgids = Protocol.parse_message_id_list(Protocol.create_message_id_list(b'24', [msg1, msg2, msg3])) self.assertEqual(tc, b'24') self.assertEqual(len(msgids), 3) self.assertTrue(msg1.id in msgids) self.assertTrue(msg2.id in msgids) self.assertTrue(msg3.id in msgids)
def test_roundtrip_message_id_list(self): """Test message ID list response creation / parsing by a round trip""" msg1 = Message('M1') msg2 = Message('M2') msg3 = Message('M3') tc, msgids = Protocol.parse_message_id_list( Protocol.create_message_id_list(b'24', [msg1, msg2, msg3])) self.assertEqual(tc, b'24') self.assertEqual(len(msgids), 3) self.assertTrue(msg1.id in msgids) self.assertTrue(msg2.id in msgids) self.assertTrue(msg3.id in msgids)
def test_basic_client_transaction(self): """Tests the client transaction protocol and logic""" client_db = ContentDB() srv_db = ContentDB() tc = srv_db.add_messages([Message('fubar'), Message('foo'), Message('bar')]) self.assertEqual(client_db.message_count, 0) self.assertEqual(srv_db.message_count, 3) with TestServerHelper() as server_helper, TestClientHelper() as client_helper: client_transaction = ClientTransaction(client_helper.sock, client_db) srv_sock = SocketTransaction(server_helper.sock, b'\n') """Run the client transaction in a separate thread""" thread = threading.Thread(target=client_transaction.process) thread.start() """Send a greeting (should be req. by client)""" srv_sock._write(Protocol.create_greeting_message(srv_db.id).encode()) """Reading msg id list request""" rcv = srv_sock._read() self.assertEqual(rcv, Protocol.create_message_id_list_request().encode()) """Sending the msg id list""" srv_sock._write(Protocol.create_message_id_list(tc, srv_db.get_messages()).encode()) """Reading msg list request""" rcv = srv_sock._read() self.assertEqual(rcv, Protocol.create_message_list_request([msg.id for msg in srv_db.get_messages()]).encode()) """Sending the msg id list""" srv_sock._write(Protocol.create_message_list(srv_db.get_messages()).encode()) """Wait for client to hang up""" thread.join(2*TIMEOUT) """Make sure the client has updated the db""" self.assertEqual(client_db.message_count, 3) self.assertEqual(srv_db.message_count, 3) self.assertEqual(len([srvmsg for srvmsg in srv_db.get_messages() if srvmsg not in client_db.get_messages()]), 0)
def test_basic_server_transaction(self): """Tests the server transaction protocol and logic""" db = ContentDB() tc = db.add_messages( [Message('fubar'), Message('foo'), Message('bar')]) with TestServerHelper() as server_helper, TestClientHelper( ) as client_helper: srv_transaction = ServerTransaction(server_helper.sock, db) test_client = SocketTransaction(client_helper.sock, b'\n') """Run the server transaction in a separate thread to allow client access""" thread = threading.Thread(target=srv_transaction.process) thread.start() """Check greeting from server""" rcv = test_client._read() self.assertEqual(rcv, Protocol.create_greeting_message(db.id).encode()) """Check response to mdgid list req""" test_client._write( Protocol.create_message_id_list_request(tc).encode()) rcv = test_client._read() self.assertEqual( rcv, Protocol.create_message_id_list(tc, None).encode()) """Check response to mdg req""" test_client._write( Protocol.create_message_list_request( [msg.id for msg in db.get_messages()]).encode()) rcv = test_client._read() self.assertEqual( rcv, Protocol.create_message_list(db.get_messages()).encode()) """Wait for server (will time out if no requests)""" thread.join(2 * TIMEOUT)