Merge pull request #6 from real-easypy/callable-concurrent-fix

concurrency: fix arg handling in concurrent calls in recent feature u…

easypy/concurrency.py

@@ -974,8 +974,9 @@ def __repr__(self):
             flags += 'T'
         return "<%s[%s] '%s'>" % (self.__class__.__name__, self.threadname, flags)
 
-    def _logged_func(self, kwargs=None):
-        kwargs = {**self.kwargs, **(kwargs or {})}
+    def _logged_func(self, *args, **kwargs):
+        args = self.args + args
+        kwargs = {**self.kwargs, **kwargs}
         stack = ExitStack()
         self.exc = None
         self.timer = Timer()
@@ -986,7 +987,7 @@ def _logged_func(self, kwargs=None):
                 stack.enter_context(_logger.suppressed())
             _logger.debug("%s - starting", self)
             while True:
-                self._result = self.func(*self.args, **kwargs)
+                self._result = self.func(*args, **kwargs)
                 if not self.loop:
                     return
                 if self.wait(self.sleep):
@@ -1046,10 +1047,10 @@ def paused(self):
 
     @contextmanager
     def _running(self, *args, **kwargs):
-        func = lambda *args, **kwargs: self._logged_func(*args, **kwargs)
+        func = lambda: self._logged_func(*args, **kwargs)
 
         if DISABLE_CONCURRENCY:
-            self._logged_func(*args, **kwargs)
+            func()
             yield self
             return
 

tests/test_concurrency.py

@@ -25,9 +25,18 @@ def test_thread_stacks():
 
 
 def test_call_concurrent():
-    func = lambda a, b: a + b
+    func = lambda a, b: a * 10 + b
     c = concurrent(func, 1, b=2, threadname='add')
-    assert c() == 3
+    assert c() == 12
+
+    c = concurrent(func, 1, 2, threadname='add')
+    assert c() == 12
+
+    c = concurrent(func, 1, threadname='add')
+    assert c(2) == 12
+
+    c = concurrent(func, threadname='add')
+    assert c(1, 2) == 12
 
 
 def test_call_concurrent_timeout():