def test_prf_name_with_sha384(self):
state = ConnectionState()
state.cipher = constants.CipherSuite.TLS_AES_256_GCM_SHA384
self.assertEqual(state.prf_name, "sha384")
def test_prf_name_with_sha256(self):
state = ConnectionState()
state.cipher = constants.CipherSuite.TLS_AES_128_GCM_SHA256
self.assertEqual(state.prf_name, "sha256")
def test_get_last_message_of_type_with_no_messages(self):
state = ConnectionState()
msg = state.get_last_message_of_type(messages.ClientHello)
self.assertIsNone(msg)
def test_get_server_public_key(self):
state = ConnectionState()
with self.assertRaises(StopIteration):
state.get_server_public_key()
def test___init__(self):
state = ConnectionState()
self.assertIsNotNone(state)
def setUp(self):
self.state = ConnectionState()
self.hello_gen = ClientHelloGenerator()
self.vanilla_hello = self.hello_gen.generate(self.state).write()
