Skip to content

Commit fdfa6df

Browse files
Adjust PSK tests to use testing_context()
1 parent 3458f2e commit fdfa6df

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

Lib/test/test_ssl.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4364,14 +4364,14 @@ def test_session_handling(self):
43644364
def test_psk(self):
43654365
psk = bytes.fromhex('deadbeef')
43664366

4367-
client_context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
4367+
client_context, server_context, _ = testing_context()
4368+
43684369
client_context.check_hostname = False
43694370
client_context.verify_mode = ssl.CERT_NONE
43704371
client_context.maximum_version = ssl.TLSVersion.TLSv1_2
43714372
client_context.set_ciphers('PSK')
43724373
client_context.set_psk_client_callback(lambda hint: (None, psk))
43734374

4374-
server_context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
43754375
server_context.maximum_version = ssl.TLSVersion.TLSv1_2
43764376
server_context.set_ciphers('PSK')
43774377
server_context.set_psk_server_callback(lambda identity: psk)
@@ -4443,14 +4443,14 @@ def server_callback(identity):
44434443
self.assertEqual(identity, client_identity)
44444444
return psk
44454445

4446-
client_context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
4446+
client_context, server_context, _ = testing_context()
4447+
44474448
client_context.check_hostname = False
44484449
client_context.verify_mode = ssl.CERT_NONE
44494450
client_context.minimum_version = ssl.TLSVersion.TLSv1_3
44504451
client_context.set_ciphers('PSK')
44514452
client_context.set_psk_client_callback(client_callback)
44524453

4453-
server_context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
44544454
server_context.minimum_version = ssl.TLSVersion.TLSv1_3
44554455
server_context.set_ciphers('PSK')
44564456
server_context.set_psk_server_callback(server_callback, identity_hint)

0 commit comments

Comments
 (0)