def test_cancel_file_when_nothing_to_cancel(self): # Setup user_input = UserInput('cf') settings = Settings(traffic_masking=True) window = TxWindow(name='Alice', type=WIN_TYPE_CONTACT, type_print='contact', uid=nick_to_pub_key("Alice")) window.window_contacts = [create_contact('Alice')] # Test self.assertIsNone(cancel_packet(user_input, window, settings, self.queues)) self.assertEqual(self.queues[TM_FILE_PACKET_QUEUE].qsize(), 0)
def test_cancel_file_during_traffic_masking(self): # Setup user_input = UserInput('cf') settings = Settings(traffic_masking=True) window = TxWindow() window.window_contacts = [create_contact('Alice')] self.queues[TM_FILE_PACKET_QUEUE].put(('testfile1', {nick_to_pub_key("Alice"): False})) self.queues[TM_FILE_PACKET_QUEUE].put(('testfile2', {nick_to_pub_key("Alice"): False})) # Test self.assertIsNone(cancel_packet(user_input, window, settings, self.queues)) self.assertEqual(self.queues[TM_FILE_PACKET_QUEUE].qsize(), 1)