"""Tests for OrderProduct model.""" from ows_product_physical.connector.mysql import delivery_db_session from ows_product_physical.models.order import Order from ows_product_physical.models.order import OrderProduct from ows_product_physical.models.order_product import \ get_order_products_by_order_id from tests.utils import db def test_get_order_products_by_order_id(delivery_db_fixture, delivery_orders, delivery_order_products): """Test that get_order_products_by_order_id returns db entries.""" db.populate_delivery_table(Order, delivery_orders) db.populate_delivery_table(OrderProduct, delivery_order_products) order_id = delivery_orders[0]['order_id'] with delivery_db_session() as session: result = get_order_products_by_order_id(order_id, session) assert result == session.query(OrderProduct) \ .filter(OrderProduct.order_id == order_id) \ .all()