Coverage for polar/billing_entry/tasks.py: 62%

1import uuid 

2 

3from polar.worker import AsyncSessionMaker, TaskPriority, actor 

4 

5from .repository import BillingEntryRepository 

6 

7 

8@actor(actor_name="billing_entry.set_order_item", priority=TaskPriority.LOW) 

9async def set_order_item( 

10 billing_entries: list[uuid.UUID], order_item_id: uuid.UUID 

11) -> None: 

12 async with AsyncSessionMaker() as session: 

13 repository = BillingEntryRepository.from_session(session) 

14 await repository.update_order_item_id(billing_entries, order_item_id)