mirror of
https://github.com/almet/copanier.git
synced 2025-04-28 19:42:37 +02:00
90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
import pytest
|
|
|
|
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()
|
|
)
|
|
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"
|
|
)
|
|
|
|
|
|
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_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)],
|
|
)
|
|
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_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"
|