asynchronous/process: add pool parameter tests

Signed-off-by: Matteo Cafasso <noxdafox@gmail.com>

test/test_asynchronous_process_fork.py

@@ -9,7 +9,7 @@
 import multiprocessing
 from concurrent.futures import CancelledError, TimeoutError
 
-from pebble import asynchronous, ProcessExpired
+from pebble import asynchronous, ProcessExpired, ProcessPool
 
 
 # set start method
@@ -107,6 +107,11 @@ def daemon_keyword_decorated():
     return multiprocessing.current_process().daemon
 
 
+@asynchronous.process(pool=ProcessPool(1, context=mp_context))
+def pool_decorated(_argument, _keyword_argument=0):
+    return multiprocessing.current_process().pid
+
+
 class ProcessAsynchronousObj:
     a = 0
 
@@ -431,3 +436,10 @@ async def test():
             return await callable_object(1)
 
         self.assertEqual(asyncio.run(test()), 1)
+
+    def test_pool_decorated(self):
+        """Process Fork results are produced."""
+        async def test():
+            return await pool_decorated(1, 1)
+
+        self.assertEqual(asyncio.run(test()), asyncio.run(test()))

test/test_asynchronous_process_forkserver.py

@@ -9,7 +9,7 @@
 import multiprocessing
 from concurrent.futures import CancelledError, TimeoutError
 
-from pebble import asynchronous, ProcessExpired
+from pebble import asynchronous, ProcessExpired, ProcessPool
 
 
 # set start method
@@ -107,6 +107,11 @@ def daemon_keyword_decorated():
     return multiprocessing.current_process().daemon
 
 
+@asynchronous.process(pool=ProcessPool(1, context=mp_context))
+def pool_decorated(_argument, _keyword_argument=0):
+    return multiprocessing.current_process().pid
+
+
 class ProcessAsynchronousObj:
     a = 0
 
@@ -404,3 +409,10 @@ async def test():
             return await callable_object(1)
 
         self.assertEqual(asyncio.run(test()), 1)
+
+    def test_pool_decorated(self):
+        """Process Forkserver results are produced."""
+        async def test():
+            return await pool_decorated(1, 1)
+
+        self.assertEqual(asyncio.run(test()), asyncio.run(test()))

test/test_asynchronous_process_spawn.py

@@ -9,7 +9,7 @@
 import multiprocessing
 from concurrent.futures import CancelledError, TimeoutError
 
-from pebble import asynchronous, ProcessExpired
+from pebble import asynchronous, ProcessExpired, ProcessPool
 
 
 # set start method
@@ -107,6 +107,11 @@ def daemon_keyword_decorated():
     return multiprocessing.current_process().daemon
 
 
+@asynchronous.process(pool=ProcessPool(1, context=mp_context))
+def pool_decorated(_argument, _keyword_argument=0):
+    return multiprocessing.current_process().pid
+
+
 class ProcessAsynchronousObj:
     a = 0
 
@@ -404,3 +409,10 @@ async def test():
             return await callable_object(1)
 
         self.assertEqual(asyncio.run(test()), 1)
+
+    def test_pool_decorated(self):
+        """Process Spawn results are produced."""
+        async def test():
+            return await pool_decorated(1, 1)
+
+        self.assertEqual(asyncio.run(test()), asyncio.run(test()))