def test_handle_stale_conversation_retry(self): conversation = StubConversation(12344) conversation.retries_made = 0 table = ConversationTable() performer = Performer(self.sock, table, 10, 6) performer._handle_stale_conversation(conversation) self.assertEqual(self.sock.sent_data, "stub_packet_data") self.assertEqual(self.sock.sent_addr, ("stub_host", "stub_port"))
def test_handle_stale_conversation_giveup(self): conversation = StubConversation(12344) conversation.retries_made = 6 table = ConversationTable() table.add_conversation("stub_host", "stub_port", conversation) performer = Performer(self.sock, table, 10, 6) performer._handle_stale_conversation(conversation) self.assertEqual(self.sock.sent_data, '\x00\x05\x00\x00Conversation Timed Out\x00') self.assertEqual(self.sock.sent_addr, ("stub_host", "stub_port")) self.assertIsNone(table.get_conversation("stub_host", "stub_port"), None)