Browse Source

[1828] Cleanup code by using with (for accept_sock)

Mukund Sivaraman 13 years ago
parent
commit
74b4b2f998
1 changed files with 4 additions and 5 deletions
  1. 4 5
      src/lib/python/isc/util/cio/tests/socketsession_test.py

+ 4 - 5
src/lib/python/isc/util/cio/tests/socketsession_test.py

@@ -237,11 +237,10 @@ class TestForwarder(unittest.TestCase):
         s = socket.socket(socket.AF_UNIX, SOCK_STREAM, 0)
         s.setblocking(False)
         s.connect(TEST_UNIX_FILE)
-        accept_sock = self.accept_forwarder()
-        receiver = SocketSessionReceiver(accept_sock)
-        s.close()
-        self.assertRaises(SocketSessionError, receiver.pop)
-        accept_sock.close()
+        with self.accept_forwarder() as accept_sock:
+            receiver = SocketSessionReceiver(accept_sock)
+            s.close()
+            self.assertRaises(SocketSessionError, receiver.pop)
 
 class TestReceiver(unittest.TestCase):
     # We only check a couple of failure cases on construction.  Valid cases