From eeaec0483a145b7d19a334d13e56bc1988e64839 Mon Sep 17 00:00:00 2001 From: James Sully Date: Wed, 31 Jul 2024 19:58:34 +1000 Subject: [PATCH] tests: use socket's contextmanager implementation --- test_daemon.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/test_daemon.py b/test_daemon.py index 667592d..a161741 100755 --- a/test_daemon.py +++ b/test_daemon.py @@ -87,12 +87,9 @@ def main(): @contextmanager def client_socket(): - try: - client_sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) + with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as client_sock: client_sock.connect(SOCKET_PATH) yield client_sock - finally: - client_sock.close() def test_msg_and_response(test_name, msg, expected): global failure