copanier/tests/test_models.py
Yohan Boniface 6db17fa56e Allow to archive deliveries
- Delivery.id is not persisted anymore: the file name and relaltive
  directory *is* the id, and we don't want to denormalize this info
- Delivery.load will filter out non fields keys from the storage: this
  allows more resilient upgrade of the data model, and is usefull when
  dealing with multiple branches with different data models
- Delivery.archive/unarchive will move the delivery storage from
  root to archive/
- Delivery.all now accepts "is_archived" to retrieve archived instead
  of active deliveries
- import products is now moved to a modal
- archived deliveries are now listed in a separate page
2019-04-24 17:51:56 +02:00

165 lines
4.6 KiB
Python

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 not 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):
with pytest.raises(AssertionError):
delivery.path
assert not delivery.id
delivery.persist()
assert delivery.id
assert delivery.path.exists()
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
def test_archive_delivery(delivery):
delivery.persist()
old_id = delivery.id
old_path = delivery.path
assert str(old_path).endswith(f"delivery/{delivery.id}.yml")
assert old_path.exists()
delivery.archive()
assert delivery.is_archived
assert delivery.id.startswith("archive/")
new_path = delivery.path
assert str(new_path).endswith(f"delivery/archive/{old_id}.yml")
assert not old_path.exists()
assert new_path.exists()
delivery.unarchive()
assert not delivery.id.startswith("archive/")
assert old_path.exists()
assert not new_path.exists()
assert not delivery.is_archived