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)