예제 #1
0
    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)
예제 #2
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)