932f4a72df
* ingredient parser hot fixes (float equality) * remove `get` in favor of `get_one` & `multi_query`
85 lines
2.3 KiB
Python
85 lines
2.3 KiB
Python
import pytest
|
|
import sqlalchemy
|
|
from pydantic import UUID4
|
|
|
|
from mealie.repos.repository_factory import AllRepositories
|
|
from mealie.schema.group.group_shopping_list import ShoppingListItemCreate, ShoppingListOut, ShoppingListSave
|
|
from tests.utils.factories import random_string
|
|
from tests.utils.fixture_schemas import TestUser
|
|
|
|
|
|
def create_item(list_id: UUID4) -> dict:
|
|
return {
|
|
"shopping_list_id": str(list_id),
|
|
"checked": False,
|
|
"position": 0,
|
|
"is_food": False,
|
|
"note": random_string(10),
|
|
"quantity": 1,
|
|
"unit_id": None,
|
|
"unit": None,
|
|
"food_id": None,
|
|
"food": None,
|
|
"recipe_id": None,
|
|
"label_id": None,
|
|
}
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def shopping_lists(database: AllRepositories, unique_user: TestUser):
|
|
|
|
models: list[ShoppingListOut] = []
|
|
|
|
for _ in range(3):
|
|
model = database.group_shopping_lists.create(
|
|
ShoppingListSave(name=random_string(10), group_id=unique_user.group_id),
|
|
)
|
|
|
|
models.append(model)
|
|
|
|
yield models
|
|
|
|
for model in models:
|
|
try:
|
|
database.group_shopping_lists.delete(model.id)
|
|
except Exception: # Entry Deleted in Test
|
|
pass
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def shopping_list(database: AllRepositories, unique_user: TestUser):
|
|
|
|
model = database.group_shopping_lists.create(
|
|
ShoppingListSave(name=random_string(10), group_id=unique_user.group_id),
|
|
)
|
|
|
|
yield model
|
|
|
|
try:
|
|
database.group_shopping_lists.delete(model.id)
|
|
except Exception: # Entry Deleted in Test
|
|
pass
|
|
|
|
|
|
@pytest.fixture(scope="function")
|
|
def list_with_items(database: AllRepositories, unique_user: TestUser):
|
|
list_model = database.group_shopping_lists.create(
|
|
ShoppingListSave(name=random_string(10), group_id=unique_user.group_id),
|
|
)
|
|
|
|
for _ in range(10):
|
|
database.group_shopping_list_item.create(
|
|
ShoppingListItemCreate(
|
|
**create_item(list_model.id),
|
|
)
|
|
)
|
|
|
|
# refresh model
|
|
list_model = database.group_shopping_lists.get_one(list_model.id)
|
|
|
|
yield list_model
|
|
|
|
try:
|
|
database.group_shopping_lists.delete(list_model.id)
|
|
except sqlalchemy.exc.NoResultFound: # Entry Deleted in Test
|
|
pass
|