Python Packet.assemble_and_store_fileの例

コード例 #1

ファイル: test_packet.py プロジェクト: gtog/tfc

    def test_short_file(self):
        # Setup
        packets = split_to_assembly_packets(self.short_f_data, FILE)

        # Test
        self.assertFalse(os.path.isfile(f'{DIR_RECV_FILES}Alice/testfile.txt'))
        self.assertFalse(
            os.path.isfile(f'{DIR_RECV_FILES}Alice/testfile.txt.1'))

        packet = Packet(self.onion_pub_key, ORIGIN_CONTACT_HEADER, FILE,
                        self.contact, self.settings)
        packet.long_active = True

        for p in packets:
            packet.add_packet(p)
        self.assertIsNone(
            packet.assemble_and_store_file(self.ts, self.onion_pub_key,
                                           self.window_list))
        self.assertTrue(os.path.isfile(f'{DIR_RECV_FILES}Alice/testfile.txt'))

        for p in packets:
            packet.add_packet(p)
        self.assertIsNone(
            packet.assemble_and_store_file(self.ts, self.onion_pub_key,
                                           self.window_list))
        self.assertTrue(
            os.path.isfile(f'{DIR_RECV_FILES}Alice/testfile.txt.1'))

コード例 #2

ファイル: test_packet.py プロジェクト: tannercollin/tfc

    def test_long_file(self):
        # Setup
        packet             = Packet(self.onion_pub_key, ORIGIN_CONTACT_HEADER, FILE, self.contact, self.settings)
        packet.long_active = True
        packet_list        = assembly_packet_creator(FILE)

        for p in packet_list:
            packet.add_packet(p)

        # Test
        self.assertIsNone(packet.assemble_and_store_file(self.ts, self.onion_pub_key, self.window_list))
        self.assertEqual(os.path.getsize(f'{DIR_RECV_FILES}Alice/test_file.txt'), 10000)