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_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_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)
