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)