def test_illegal_acknowledgement_packet_during_uninitialized_state(self): packet = packets.AcknowledgementPacket(3) conversation = TFTPConversation(self.client_host, self.client_port, StubResponseRouterTwo()) response_packet = conversation.handle_packet(packet) self.assertEqual(conversation.state, tftp_conversation.COMPLETED) self.assertEqual(response_packet.__class__, packets.ErrorPacket) self.assertEqual(response_packet.error_code, 5)
def test_no_action_for_writing(self): packet = packets.WriteRequestPacket("example_filename", "netascii") conversation = TFTPConversation(self.client_host, self.client_port, NoActionAvailableResponseRouterStub()) response_packet = conversation.handle_packet(packet) self.assertEqual(conversation.state, tftp_conversation.COMPLETED) self.assertEqual(response_packet.__class__, packets.ErrorPacket) self.assertEqual(response_packet.error_code, 2)
def test_out_of_lock_step_block_num(self): packet = packets.DataPacket(2, "") conversation = TFTPConversation(self.client_host, self.client_port, StubResponseRouterTwo()) conversation.cached_packet = "stub packet" conversation.state = tftp_conversation.WRITING conversation.current_block_num = 3 response_packet = conversation.handle_packet(packet) self.assertEqual(conversation.state, tftp_conversation.WRITING) self.assertEqual(response_packet.__class__, packets.NoOpPacket)
def test_begin_reading(self): packet = packets.ReadRequestPacket("example_filename", "netascii") conversation = TFTPConversation(self.client_host, self.client_port, StubResponseRouter()) response_packet = conversation.handle_packet(packet) self.assertEqual(conversation.state, tftp_conversation.READING) self.assertEqual(conversation.filename, "example_filename") self.assertEqual(conversation.mode, "netascii") self.assertEqual(conversation.current_block_num, 1) self.assertEqual(conversation.read_buffer.__class__, StubReadBuffer) self.assertEqual(response_packet.__class__, packets.DataPacket) self.assertEqual(conversation.cached_packet, response_packet)
def test_illegal_packet_type_during_writing_state(self): packet = packets.AcknowledgementPacket(2) conversation = TFTPConversation(self.client_host, self.client_port, StubResponseRouterTwo()) conversation.cached_packet = "stub packet" conversation.state = tftp_conversation.WRITING conversation.read_buffer = StubReadBufferTwo() response_packet = conversation.handle_packet(packet) self.assertEqual(conversation.state, tftp_conversation.WRITING) self.assertEqual(response_packet.__class__, packets.ErrorPacket) self.assertEqual(response_packet.error_code, 0) self.assertEqual(conversation.cached_packet, "stub packet")
def test_finish_reading(self): packet = packets.AcknowledgementPacket(3) conversation = TFTPConversation(self.client_host, self.client_port, StubResponseRouterTwo()) conversation.filename = "example_filename" conversation.state = tftp_conversation.READING conversation.current_block_num = 3 conversation.read_buffer = StubReadBufferTwo() response_packet = conversation.handle_packet(packet) self.assertEqual(conversation.state, tftp_conversation.COMPLETED) self.assertEqual(response_packet.__class__, packets.NoOpPacket) self.assertEqual(conversation.cached_packet, response_packet)
def test_begin_writing(self): packet = packets.WriteRequestPacket("example_filename", "netascii") conversation = TFTPConversation(self.client_host, self.client_port, StubResponseRouter()) response_packet = conversation.handle_packet(packet) self.assertEqual(conversation.state, tftp_conversation.WRITING) self.assertEqual(conversation.filename, "example_filename") self.assertEqual(conversation.mode, "netascii") self.assertEqual(conversation.current_block_num, 0) self.assertEqual(conversation.write_buffer.__class__, WriteBuffer) self.assertEqual(conversation.cached_packet, response_packet) self.assertEqual(response_packet.__class__, packets.AcknowledgementPacket) self.assertEqual(response_packet.block_num, 0)
def test_continue_reading(self): packet = packets.AcknowledgementPacket(1) conversation = TFTPConversation(self.client_host, self.client_port, StubResponseRouterTwo()) conversation.state = tftp_conversation.READING conversation.read_buffer = StubReadBufferTwo() conversation.current_block_num = 1 response_packet = conversation.handle_packet(packet) self.assertEqual(conversation.state, tftp_conversation.READING) self.assertEqual(conversation.current_block_num, 2) self.assertEqual(response_packet.data, "X" * 512) self.assertEqual(response_packet.__class__, packets.DataPacket) self.assertEqual(conversation.cached_packet, response_packet)
def test_finish_writing(self): packet = packets.DataPacket(3, "O" * 511) conversation = TFTPConversation(self.client_host, self.client_port, StubResponseRouterTwo()) conversation.state = tftp_conversation.WRITING conversation.write_buffer = WriteBuffer() conversation.write_buffer.data = "X" * 512 conversation.filename = "stub_filename" conversation.current_block_num = 2 write_action_wrapper = StubWriteActionWrapper() conversation.write_action = write_action_wrapper.stub_action response_packet = conversation.handle_packet(packet) self.assertEqual(conversation.state, tftp_conversation.COMPLETED) self.assertEqual(conversation.current_block_num, 3) self.assertEqual(conversation.cached_packet, response_packet) self.assertEqual(response_packet.__class__, packets.AcknowledgementPacket) self.assertEqual(response_packet.block_num, 3) # action should get invoked, saving this state in the wrapper class self.assertEqual(write_action_wrapper.received_state, ("10.26.0.3", 12345, "stub_filename", "X" * 512 + "O" * 511))