def test_get_last_message_of_type(self): state = ConnectionState() msg = messages.ServerHello() msg.server_version = (3, 1) state.handshake_messages.append(msg) msg = messages.ServerHello() msg.server_version = (3, 3) state.handshake_messages.append(msg) msg = state.get_last_message_of_type(messages.ServerHello) self.assertEqual(msg.server_version, (3, 3))
def test_get_last_message_of_type_with_no_messages_of_that_type(self): state = ConnectionState() msg = messages.ServerHello() msg.server_version = (3, 1) state.handshake_messages.append(msg) msg = state.get_last_message_of_type(messages.ClientHello) self.assertIsNone(msg)