gavindiaz/aiomql
1
0
Fork 0
mirror of https://github.com/Ichinga-Samuel/aiomql.git synced 2026-07-29 05:07:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

v4.0.9

Browse Source
This commit is contained in:
Ichinga Samuel
2025-01-23 18:18:18 +01:00
parent 12f61f3b6e
commit 28811df0ce
20 changed files with 284 additions and 157 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/live/unit/test_task_queue.py
+1 -5
View File
@@ -6,7 +6,7 @@ from aiomql.core.task_queue import TaskQueue, QueueItem
class TestTaskQueue:
@classmethod
def setup_class(cls):
cls.task_queue = TaskQueue(timeout=5, worker_timeout=1)
cls.task_queue = TaskQueue(absolute_timeout=6, worker_timeout=1)
cls.data = {}
async def task_one(self):
@@ -27,14 +27,10 @@ class TestTaskQueue:
async def test_queue(self):
item_one = QueueItem(self.task_one)
self.task_queue.add(item=item_one, must_complete=False)
assert len(self.task_queue.priority_tasks) == 0
assert self.task_queue.queue.qsize() == 1
self.task_queue.add(item=QueueItem(self.task_two), must_complete=True)
assert len(self.task_queue.priority_tasks) == 1
assert self.task_queue.queue.qsize() == 2
self.task_queue.add(item=QueueItem(self.task_three), must_complete=False)
await self.task_queue.run()
assert len(self.data["task_one"]) >= 2
assert len(self.data["task_two"]) == 10
assert len(self.data["task_three"]) == 1
assert len(self.task_queue.priority_tasks) == 0