Parcourir la source

[2353] Test BoB.init_socket_srv() and BoB.remove_socket_srv()

Mukund Sivaraman il y a 12 ans
Parent
commit
4a9b794019
2 fichiers modifiés avec 26 ajouts et 0 suppressions
  1. 3 0
      src/bin/bind10/bind10_src.py.in
  2. 23 0
      src/bin/bind10/tests/bind10_test.py.in

+ 3 - 0
src/bin/bind10/bind10_src.py.in

@@ -916,6 +916,9 @@ class BoB:
             self._srv_socket.close()
             os.remove(self._socket_path)
             os.rmdir(self._tmpdir)
+            self._srv_socket = None
+            self._tmpdir = None
+            self._socket_path = None
 
     def _srv_accept(self):
         """

+ 23 - 0
src/bin/bind10/tests/bind10_test.py.in

@@ -25,6 +25,7 @@ import bind10_src
 import unittest
 import sys
 import os
+import os.path
 import copy
 import signal
 import socket
@@ -366,6 +367,28 @@ class TestBoB(unittest.TestCase):
         self.assertEqual(creator, bob._socket_cache._creator)
         self.assertRaises(ValueError, bob.set_creator, creator)
 
+    def test_socket_srv(self):
+        """Tests init_socket_srv() and remove_socket_srv() work as expected."""
+        bob = BoB()
+
+        self.assertIsNone(bob._srv_socket)
+        self.assertIsNone(bob._tmpdir)
+        self.assertIsNone(bob._socket_path)
+
+        bob.init_socket_srv()
+
+        self.assertIsNotNone(bob._srv_socket)
+        self.assertIsNotNone(bob._tmpdir)
+        self.assertTrue(os.path.isdir(bob._tmpdir))
+        self.assertIsNotNone(bob._socket_path)
+        self.assertTrue(os.path.exists(bob._socket_path))
+
+        bob.remove_socket_srv()
+
+        self.assertIsNone(bob._srv_socket)
+        self.assertIsNone(bob._tmpdir)
+        self.assertIsNone(bob._socket_path)
+
     def test_init_alternate_socket(self):
         bob = BoB("alt_socket_file")
         self.assertEqual(bob.verbose, False)