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)
