from datetime import datetime, timedelta import pytest from copanier import config from copanier.models import Delivery, Product, Person, Order, ProductOrder now = datetime.now def test_can_create_delivery(): delivery = Delivery( producer="Andines", from_date=now(), to_date=now(), order_before=now(), contact="some@one.to", ) assert delivery.producer == "Andines" assert delivery.where == "Marché de la Briche" assert delivery.from_date.year == now().year assert delivery.id def test_wrong_datetime_raise_valueerror(): with pytest.raises(ValueError): Delivery( producer="Andines", order_before=now(), to_date=now(), from_date="pouet", contact="some@one.to", ) def test_delivery_is_open_when_order_before_is_in_the_future(delivery): delivery.order_before = now() + timedelta(hours=1) assert delivery.is_open delivery.order_before = now() - timedelta(days=1) assert not delivery.is_open # We don't take the hour into account delivery.order_before = now() - timedelta(hours=1) assert delivery.is_open def test_delivery_status(delivery): delivery.order_before = now() + timedelta(hours=1) assert delivery.status == delivery.OPEN delivery.order_before = now() - timedelta(days=1) assert delivery.status == delivery.CLOSED def test_can_create_product(): product = Product(name="Lait 1.5L", ref="123", price=1.5) assert product.ref == "123" assert product.price == 1.5 def test_can_create_delivery_with_products(): delivery = Delivery( producer="Andines", from_date=now(), to_date=now(), order_before=now(), products=[Product(name="Lait", ref="123", price=1.5)], contact="some@one.to", ) assert len(delivery.products) == 1 assert delivery.products[0].ref == "123" def test_can_add_product_to_delivery(delivery): delivery.products = [] assert not delivery.products delivery.products.append(Product(name="Chocolat", ref="choco", price=10)) assert len(delivery.products) == 1 def test_can_create_person(): person = Person(email="foo@bar.fr", first_name="Foo") assert person.email == "foo@bar.fr" assert person.first_name == "Foo" def test_can_create_order_with_products(): order = Order(products={"123": ProductOrder(wanted=2)}) assert len(order.products) == 1 assert order.products["123"].wanted == 2 def test_can_add_product_to_order(): order = Order() assert len(order.products) == 0 order.products["123"] = ProductOrder(wanted=2) assert order.products["123"].wanted == 2 def test_order_has_adjustments(): order = Order() assert not order.has_adjustments order.products["123"] = ProductOrder(wanted=2) assert not order.has_adjustments order.products["123"] = ProductOrder(wanted=2, adjustment=1) assert order.has_adjustments def test_can_persist_delivery(delivery): delivery.persist() def test_can_load_delivery(delivery): delivery.producer = "Corto" delivery.persist() loaded = Delivery.load(delivery.id) assert loaded.producer == "Corto" def test_person_is_staff_if_email_is_in_config(monkeypatch): monkeypatch.setattr(config, "STAFF", ["foo@bar.fr"]) person = Person(email="foo@bar.fr") assert person.is_staff person = Person(email="foo@bar.org") assert not person.is_staff def test_person_is_staff_if_no_staff_in_config(monkeypatch): monkeypatch.setattr(config, "STAFF", []) person = Person(email="foo@bar.fr") assert person.is_staff def test_productorder_quantity(): choice = ProductOrder(wanted=3) assert choice.wanted == 3 assert choice.quantity == 3 choice = ProductOrder(wanted=3, adjustment=2) assert choice.quantity == 5 choice = ProductOrder(wanted=3, adjustment=-1) assert choice.quantity == 2