def test_cancel_message_during_normal(self): # Setup user_input = UserInput('cm') settings = Settings() window = TxWindow(name='Alice', type=WIN_TYPE_CONTACT, type_print='contact', uid='*****@*****.**') window.window_contacts = [create_contact()] self.queues[MESSAGE_PACKET_QUEUE].put( ('testmessage1', settings, '*****@*****.**', '*****@*****.**', False, False, '*****@*****.**')) self.queues[MESSAGE_PACKET_QUEUE].put( ('testmessage2', settings, '*****@*****.**', '*****@*****.**', False, False, '*****@*****.**')) self.queues[MESSAGE_PACKET_QUEUE].put( ('testmessage3', settings, '*****@*****.**', '*****@*****.**', False, False, '*****@*****.**')) time.sleep(0.1) # Test self.assertIsNone( cancel_packet(user_input, window, settings, self.queues)) time.sleep(0.1) self.assertEqual(self.queues[MESSAGE_PACKET_QUEUE].qsize(), 2)
def test_cancel_message_when_nothing_to_cancel(self): # Setup window = Window(name='Alice', type='contact', uid='*****@*****.**') user_input = UserInput('cm') window.window_contacts = [create_contact('Alice')] settings = Settings() queues = {MESSAGE_PACKET_QUEUE: Queue(), FILE_PACKET_QUEUE: Queue()} # Test self.assertIsNone(cancel_packet(user_input, window, settings, queues)) time.sleep(0.2) self.assertEqual(queues[FILE_PACKET_QUEUE].qsize(), 0)
def test_cancel_message_when_nothing_to_cancel(self): # Setup user_input = UserInput('cm') settings = Settings() window = TxWindow(name='Alice', type=WIN_TYPE_CONTACT, type_print='contact', uid='*****@*****.**') window.window_contacts = [create_contact()] # Test self.assertIsNone( cancel_packet(user_input, window, settings, self.queues)) time.sleep(0.1) self.assertEqual(self.queues[FILE_PACKET_QUEUE].qsize(), 0)
def test_cancel_group_message_during_normal(self): # Setup window = Window(name='testgroup', type='group', uid='testgroup') user_input = UserInput('cm') window.window_contacts = [create_contact('Alice')] settings = Settings() queues = {MESSAGE_PACKET_QUEUE: Queue(), FILE_PACKET_QUEUE: Queue()} # Test queues[MESSAGE_PACKET_QUEUE].put( ('testmessage1', settings, '*****@*****.**', '*****@*****.**', False, 'testgroup')) queues[MESSAGE_PACKET_QUEUE].put( ('testmessage2', settings, '*****@*****.**', '*****@*****.**', False, 'testgroup')) time.sleep(0.2) self.assertIsNone(cancel_packet(user_input, window, settings, queues)) time.sleep(0.2) self.assertEqual(queues[MESSAGE_PACKET_QUEUE].qsize(), 1)
def test_cancel_message_during_trickle(self): # Setup user_input = UserInput('cm') window = Window() window.window_contacts = [create_contact('Alice')] settings = Settings(session_trickle=True) queues = {MESSAGE_PACKET_QUEUE: Queue(), FILE_PACKET_QUEUE: Queue()} # Test queues[MESSAGE_PACKET_QUEUE].put(('testmessage1', { '*****@*****.**': False })) queues[MESSAGE_PACKET_QUEUE].put(('testmessage2', { '*****@*****.**': False })) time.sleep(0.2) self.assertIsNone(cancel_packet(user_input, window, settings, queues)) time.sleep(0.2) self.assertEqual(queues[MESSAGE_PACKET_QUEUE].qsize(), 1)
def test_cancel_file_during_traffic_masking(self): # Setup user_input = UserInput('cf') settings = Settings(session_traffic_masking=True) window = TxWindow() window.window_contacts = [create_contact()] self.queues[FILE_PACKET_QUEUE].put(('testfile1', { '*****@*****.**': False })) self.queues[FILE_PACKET_QUEUE].put(('testfile2', { '*****@*****.**': False })) time.sleep(0.1) # Test self.assertIsNone( cancel_packet(user_input, window, settings, self.queues)) time.sleep(0.1) self.assertEqual(self.queues[FILE_PACKET_QUEUE].qsize(), 1)