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)