p1-taskqueue 0.1.7__tar.gz → 0.1.8__tar.gz

{p1_taskqueue-0.1.7 → p1_taskqueue-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: p1-taskqueue
-Version: 0.1.7
+Version: 0.1.8
 Summary: A Task Queue Wrapper for Dekoruma Backend
 Author-email: Chalvin <engineering@dekoruma.com>
 Project-URL: Homepage, https://github.com/Dekoruma/p1-taskqueue

{p1_taskqueue-0.1.7 → p1_taskqueue-0.1.8}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "p1-taskqueue"
 # DO NOT CHANGE THIS VERSION - it gets automatically replaced by CI/CD with the git tag version
-version = "0.1.7"
+version = "0.1.8"
 description = "A Task Queue Wrapper for Dekoruma Backend"
 authors = [
     {name = "Chalvin", email = "engineering@dekoruma.com"}

{p1_taskqueue-0.1.7 → p1_taskqueue-0.1.8}/src/p1_taskqueue.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: p1-taskqueue
-Version: 0.1.7
+Version: 0.1.8
 Summary: A Task Queue Wrapper for Dekoruma Backend
 Author-email: Chalvin <engineering@dekoruma.com>
 Project-URL: Homepage, https://github.com/Dekoruma/p1-taskqueue

{p1_taskqueue-0.1.7 → p1_taskqueue-0.1.8}/src/taskqueue/libs/helper_test.py

@@ -58,10 +58,15 @@ def celery_worker_burst(include_func_names: List[str], channel: str = "default")
                             decoded_body) > 1 else {}
 
                         full_func_name = ""
-                        if task_name.endswith("dynamic_function_executor") and len(task_args) >= 2:
-                            full_func_name = f"{task_args[0]}.{task_args[1]}"
-                        elif task_name.endswith("dynamic_class_method_executor") and len(task_args) >= 3:
-                            full_func_name = f"{task_args[0]}.{task_args[1]}.{task_args[2]}"
+                        if task_name.endswith("dynamic_function_executor"):
+                            module_path = task_kwargs.get('module_path', '')
+                            function_name = task_kwargs.get('function_name', '')
+                            full_func_name = f"{module_path}.{function_name}"
+                        elif task_name.endswith("dynamic_class_method_executor"):
+                            module_path = task_kwargs.get('module_path', '')
+                            class_name = task_kwargs.get('class_name', '')
+                            method_name = task_kwargs.get('method_name', '')
+                            full_func_name = f"{module_path}.{class_name}.{method_name}"
 
                         should_execute = full_func_name in included_set if full_func_name else False
 

{p1_taskqueue-0.1.7 → p1_taskqueue-0.1.8}/tests/test_test_utils.py

@@ -65,7 +65,7 @@ class TestCeleryWorkerBurst:
         mock_message.acknowledged = False
 
         mock_loads.return_value = [
-            ['module.submodule', 'test_function', [], {}], {}]
+            [], {'module_path': 'module.submodule', 'function_name': 'test_function', 'args': [], 'kwargs': {}}]
 
         mock_queue = MagicMock()
         mock_queue.get.side_effect = [mock_message, None]
@@ -87,8 +87,8 @@ class TestCeleryWorkerBurst:
 
         mock_message.ack.assert_called_once()
         mock_task.apply.assert_called_once_with(
-            args=['module.submodule', 'test_function', [], {}],
-            kwargs={}
+            args=[],
+            kwargs={'module_path': 'module.submodule', 'function_name': 'test_function', 'args': [], 'kwargs': {}}
         )
 
     @patch('taskqueue.libs.helper_test.current_app')
@@ -107,7 +107,7 @@ class TestCeleryWorkerBurst:
         mock_message.acknowledged = False
 
         mock_loads.return_value = [
-            ['module.submodule', 'TestClass', 'test_method', [], {}], {}]
+            [], {'module_path': 'module.submodule', 'class_name': 'TestClass', 'method_name': 'test_method', 'args': [], 'kwargs': {}, 'init_args': [], 'init_kwargs': {}}]
 
         mock_queue = MagicMock()
         mock_queue.get.side_effect = [mock_message, None]
@@ -129,8 +129,8 @@ class TestCeleryWorkerBurst:
 
         mock_message.ack.assert_called_once()
         mock_task.apply.assert_called_once_with(
-            args=['module.submodule', 'TestClass', 'test_method', [], {}],
-            kwargs={}
+            args=[],
+            kwargs={'module_path': 'module.submodule', 'class_name': 'TestClass', 'method_name': 'test_method', 'args': [], 'kwargs': {}, 'init_args': [], 'init_kwargs': {}}
         )
 
     @patch('taskqueue.libs.helper_test.current_app')
@@ -149,7 +149,7 @@ class TestCeleryWorkerBurst:
         mock_message.acknowledged = False
 
         mock_loads.return_value = [
-            ['module.submodule', 'other_function', [], {}], {}]
+            [], {'module_path': 'module.submodule', 'function_name': 'other_function', 'args': [], 'kwargs': {}}]
 
         mock_queue = MagicMock()
         mock_queue.get.side_effect = [mock_message, None]
@@ -226,7 +226,7 @@ class TestCeleryWorkerBurst:
         mock_message.acknowledged = False
 
         mock_loads.return_value = [
-            ['module.submodule', 'test_function', [], {}], {}]
+            [], {'module_path': 'module.submodule', 'function_name': 'test_function', 'args': [], 'kwargs': {}}]
 
         mock_queue = MagicMock()
         mock_queue.get.side_effect = [mock_message, None]
@@ -271,7 +271,7 @@ class TestCeleryWorkerBurst:
             lambda self: get_acknowledged())
 
         mock_loads.return_value = [
-            ['module.submodule', 'test_function', [], {}], {}]
+            [], {'module_path': 'module.submodule', 'function_name': 'test_function', 'args': [], 'kwargs': {}}]
 
         mock_queue = MagicMock()
         mock_queue.get.side_effect = [mock_message, None]