def test_mark_retry(self): original_packet = packets.AcknowledgementPacket(3) conversation = TFTPConversation(self.client_host, self.client_port, StubResponseRouterTwo()) conversation.cached_packet = original_packet retry_packet = conversation.mark_retry() self.assertEqual(conversation.retries_made, 1) self.assertEqual(retry_packet, original_packet)