mirror of
https://github.com/almet/copanier.git
synced 2025-04-28 19:42:37 +02:00
118 lines
3 KiB
Python
118 lines
3 KiB
Python
import os
|
|
from datetime import datetime, timedelta
|
|
|
|
import pytest
|
|
from roll.extensions import traceback
|
|
from roll.testing import Client as BaseClient
|
|
|
|
from copanier import app as copanier_app
|
|
from copanier import config as kconfig
|
|
from copanier.utils import create_token
|
|
from copanier.models import Delivery, Person, Product, Producer, Groups, Group
|
|
|
|
|
|
def pytest_configure(config):
|
|
os.environ["COPANIER_DATA_ROOT"] = "tmp/db"
|
|
os.environ["COPANIER_SEND_EMAILS"] = ""
|
|
os.environ["COPANIER_STAFF"] = ""
|
|
kconfig.init()
|
|
assert str(kconfig.DATA_ROOT) == "tmp/db"
|
|
|
|
|
|
def pytest_runtest_setup(item):
|
|
for path in Delivery.get_root().glob("*.yml"):
|
|
path.unlink()
|
|
|
|
|
|
class Client(BaseClient):
|
|
content_type = "application/x-www-form-urlencoded; charset=utf-8"
|
|
headers = {}
|
|
|
|
async def request(
|
|
self, path, method="GET", body=b"", headers=None, content_type=None
|
|
):
|
|
# TODO move this to Roll upstream?
|
|
headers = headers or {}
|
|
for key, value in self.headers.items():
|
|
headers.setdefault(key, value)
|
|
return await super().request(path, method, body, headers, content_type)
|
|
|
|
def login(self, email="foo@bar.org"):
|
|
token = create_token(email)
|
|
self.headers["Cookie"] = f"token={token.decode()}"
|
|
|
|
def logout(self):
|
|
try:
|
|
del self.headers["Cookie"]
|
|
except KeyError:
|
|
pass
|
|
|
|
|
|
@pytest.fixture
|
|
def client(app, event_loop):
|
|
app.loop = event_loop
|
|
app.loop.run_until_complete(app.startup())
|
|
client = Client(app)
|
|
client.login()
|
|
yield client
|
|
app.loop.run_until_complete(app.shutdown())
|
|
|
|
|
|
@pytest.fixture
|
|
def app(): # Requested by Roll testing utilities.
|
|
traceback(copanier_app)
|
|
return copanier_app
|
|
|
|
|
|
@pytest.fixture
|
|
def delivery():
|
|
return Delivery(
|
|
name="CRAC d'automne",
|
|
contact="mister@me.me",
|
|
from_date=datetime.now() + timedelta(days=10),
|
|
to_date=datetime.now() + timedelta(days=10),
|
|
order_before=datetime.now() + timedelta(days=7),
|
|
products=[
|
|
Product(name="Lait", producer="ferme-du-coin", ref="lait", price=1.5,)
|
|
],
|
|
producers={"ferme-du-coin": Producer(name="Ferme du coin", id="ferme-du-coin")},
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def groups():
|
|
fractal_brocolis = Group(
|
|
id="fractal-brocolis", name="The Fractal Brocolis", members=["foo@bar.org"]
|
|
)
|
|
groups = Groups({"fractal-brocolis": fractal_brocolis})
|
|
groups.persist()
|
|
return groups
|
|
|
|
|
|
@pytest.fixture
|
|
def anothergroup():
|
|
return Group(id="another-group", name="Another Group", members=["another@bar.org"])
|
|
|
|
|
|
@pytest.fixture
|
|
def yaourt():
|
|
return Product(
|
|
ref="yaourt",
|
|
unit="pot 125ml",
|
|
name="Yaourt",
|
|
price="3.5",
|
|
packing=4,
|
|
producer="ferme-du-coin",
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def fromage():
|
|
return Product(
|
|
ref="fromage", name="Fromage", price="9.2", producer="ferme-du-coin",
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def person():
|
|
return Person(email="foo@bar.fr")
|