pool free test added

alttch · Aug 20, 2019 · 4f732bc · 4f732bc
1 parent 88bc9ea
commit 4f732bc
Showing 1 changed file with 22 additions and 16 deletions.
diff --git a/tests/test.py b/tests/test.py
@@ -14,20 +14,19 @@
 
 from types import SimpleNamespace
 
-result = SimpleNamespace(
-    g=None,
-    function_collection=0,
-    task_collection=0,
-    background_task_annotated=None,
-    background_task_thread=None,
-    background_task_mp=None,
-    background_worker=0,
-    background_interval_worker=0,
-    background_interval_worker_async_ex=0,
-    background_queue_worker=0,
-    background_event_worker=0,
-    locker_success=False,
-    locker_failed=False)
+result = SimpleNamespace(g=None,
+                         function_collection=0,
+                         task_collection=0,
+                         background_task_annotated=None,
+                         background_task_thread=None,
+                         background_task_mp=None,
+                         background_worker=0,
+                         background_interval_worker=0,
+                         background_interval_worker_async_ex=0,
+                         background_queue_worker=0,
+                         background_event_worker=0,
+                         locker_success=False,
+                         locker_failed=False)
 
 sys.path.insert(0, Path().absolute().parent.as_posix())
 
@@ -197,8 +196,9 @@ def test_background_interval_worker_mp(self):
 
     def test_locker(self):
 
-        with_lock = Locker(
-            mod='test (broken is fine!)', relative=False, timeout=0.5)
+        with_lock = Locker(mod='test (broken is fine!)',
+                           relative=False,
+                           timeout=0.5)
 
         @with_lock
         def test_locker():
@@ -213,6 +213,12 @@ def locker_ok():
         self.assertTrue(result.locker_success)
         self.assertFalse(result.locker_failed)
 
+    def test_supervisor(self):
+        result = task_supervisor.get_stats()
+
+        self.assertEqual(len(result['threads']), 0)
+        self.assertEqual(len(result['mps']), 0)
+
 
 task_supervisor.set_thread_pool(pool_size=20, reserve_normal=5, reserve_high=5)
 task_supervisor.set_mp_pool(pool_size=20, reserve_normal=5, reserve_high=5)