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_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)