tests: use socket's contextmanager implementation

sullyj3 · Jul 31, 2024 · eeaec04 · eeaec04
1 parent 594e7f4
commit eeaec04
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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