2024-06-07 02:34:17 -07:00
|
|
|
import pytest
|
2024-06-18 03:40:43 -07:00
|
|
|
|
2024-06-07 02:34:17 -07:00
|
|
|
import saveweb_tracker.tracker
|
2024-06-18 03:40:43 -07:00
|
|
|
from saveweb_tracker.item import PreparedItem
|
2024-06-07 02:34:17 -07:00
|
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
|
|
async def test_tracker():
|
2024-06-18 03:40:43 -07:00
|
|
|
async with saveweb_tracker.tracker.Tracker("test", "1.1", "test-python") as tracker:
|
2024-06-07 02:34:17 -07:00
|
|
|
task = await tracker.claim_task_async(with_delay=False)
|
|
|
|
print(task)
|
|
|
|
assert task is not None
|
|
|
|
text = await tracker.update_task_async(task_id=task.id, status="DONE")
|
|
|
|
print(text)
|
|
|
|
text = await tracker.insert_item_async(
|
|
|
|
item_id=task.id, item_status=None, payload={
|
|
|
|
"kjasd": "111111111122222111111111122222111111111122222111111111122222111111111122222111111111122222111111111122222111111111122222",
|
|
|
|
}
|
|
|
|
)
|
2024-06-18 03:40:43 -07:00
|
|
|
print(text)
|
|
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
|
|
async def test_tracker_insert_many():
|
|
|
|
async with saveweb_tracker.tracker.Tracker("test", "1.1", "test-python") as tracker:
|
|
|
|
items = []
|
|
|
|
for i in range(15):
|
|
|
|
item = PreparedItem(item_id=i, item_status=i, payload={"kjaasd": 111})
|
|
|
|
items.append(item)
|
|
|
|
|
|
|
|
text = await tracker.insert_many_async(items)
|
2024-06-07 02:34:17 -07:00
|
|
|
print(text)
|