diff --git a/tests/test_trustme.py b/tests/test_trustme.py index 68d61e6..754af7e 100644 --- a/tests/test_trustme.py +++ b/tests/test_trustme.py @@ -258,6 +258,7 @@ def fake_ssl_client(ca, raw_client_sock, hostname): # Send and receive some data to prove the connection is good wrapped_client_sock.send(b"x") assert wrapped_client_sock.recv(1) == b"y" + wrapped_client_sock.close() except: # pragma: no cover sys.excepthook(*sys.exc_info()) raise @@ -271,6 +272,7 @@ def fake_ssl_server(server_cert, raw_server_sock): # Prove that we're connected assert wrapped_server_sock.recv(1) == b"x" wrapped_server_sock.send(b"y") + wrapped_server_sock.close() except: # pragma: no cover sys.excepthook(*sys.exc_info()) raise