Coverage for api/utils/clear.py: 30%

1from typing import Any 

2 

3from sotrans_models.models.orders.order import OrderDBModel, OrderStatus 

4 

5 

6def clear_on_move(order_data: dict[str, Any], order_status: OrderStatus): 

7 if order_status in (OrderStatus.exchange, OrderStatus.appointment): 

8 for f in ( 

9 OrderDBModel.driver, 

10 OrderDBModel.carrier, 

11 OrderDBModel.truck, 

12 OrderDBModel.trailer, 

13 OrderDBModel.best_bid, 

14 OrderDBModel.document_draft, 

15 "documents", 

16 ): 

17 order_data.pop(f, None)