def test_queue_command_normal(self): # Setup packet_list = split_to_assembly_packets(os.urandom(200), COMMAND) # Test self.assertIsNone(queue_assembly_packets(packet_list, COMMAND, *self.args)) self.assertEqual(self.queues[COMMAND_PACKET_QUEUE].qsize(), 1) packet = self.queues[COMMAND_PACKET_QUEUE].get() self.assertIsInstance(packet, bytes)
def test_queue_command_traffic_masking(self): # Setup packet_list = split_to_assembly_packets(os.urandom(200), COMMAND) self.settings.traffic_masking = True # Test self.assertIsNone(queue_assembly_packets(packet_list, COMMAND, *self.args)) self.assertEqual(self.queues[TM_COMMAND_PACKET_QUEUE].qsize(), 1) data = self.queues[TM_COMMAND_PACKET_QUEUE].get() self.assertIsInstance(data, bytes)
def test_queue_file_traffic_masking(self): # Setup packet_list = split_to_assembly_packets(os.urandom(200), FILE) self.settings.traffic_masking = True # Test self.assertIsNone(queue_assembly_packets(packet_list, FILE, *self.args)) self.assertEqual(self.queues[TM_FILE_PACKET_QUEUE].qsize(), 1) packet, log_messages, log_as_ph = self.queues[TM_FILE_PACKET_QUEUE].get() self.assertIsInstance(packet, bytes) self.assertTrue(log_messages) self.assertFalse(log_as_ph)
def test_queue_message_normal(self): # Setup packet_list = split_to_assembly_packets(os.urandom(200), MESSAGE) # Test self.assertIsNone(queue_assembly_packets(packet_list, MESSAGE, *self.args)) self.assertEqual(self.queues[MESSAGE_PACKET_QUEUE].qsize(), 1) packet, pub_key, log_setting, log_as_ph, win_uid = self.queues[MESSAGE_PACKET_QUEUE].get() self.assertIsInstance(packet, bytes) self.assertEqual(pub_key, nick_to_pub_key("Alice")) self.assertEqual(win_uid, nick_to_pub_key("Alice")) self.assertTrue(log_setting) self.assertFalse(log_as_ph)