mirror of
https://github.com/mealie-recipes/mealie.git
synced 2024-12-21 19:41:22 -08:00
365 lines
16 KiB
Python
365 lines
16 KiB
Python
import inspect
|
|
import time
|
|
from functools import cached_property
|
|
from uuid import uuid4
|
|
|
|
import pytest
|
|
from pydantic import UUID4
|
|
from sqlalchemy.orm import Session
|
|
|
|
from mealie.db.models._model_base import SqlAlchemyBase
|
|
from mealie.repos._utils import NOT_SET, NotSet
|
|
from mealie.repos.all_repositories import get_repositories
|
|
from mealie.repos.repository_generic import GroupRepositoryGeneric, HouseholdRepositoryGeneric, RepositoryGeneric
|
|
from mealie.schema._mealie.mealie_model import MealieModel
|
|
from mealie.schema.household.group_shopping_list import ShoppingListCreate
|
|
from mealie.schema.household.webhook import SaveWebhook
|
|
from mealie.schema.recipe.recipe import Recipe
|
|
from mealie.schema.recipe.recipe_ingredient import SaveIngredientFood
|
|
from mealie.schema.response.pagination import PaginationQuery
|
|
from mealie.services.household_services.shopping_lists import ShoppingListService
|
|
from tests.utils.factories import random_email, random_string
|
|
|
|
|
|
@pytest.mark.parametrize("group_id", [uuid4(), None, NOT_SET])
|
|
@pytest.mark.parametrize("household_id", [uuid4(), None, NOT_SET])
|
|
def test_get_repositories_sets_ids(
|
|
session: Session, group_id: UUID4 | None | NotSet, household_id: UUID4 | None | NotSet
|
|
):
|
|
kwargs = {}
|
|
if not isinstance(group_id, NotSet):
|
|
kwargs["group_id"] = group_id
|
|
if not isinstance(household_id, NotSet):
|
|
kwargs["household_id"] = household_id
|
|
|
|
repositories = get_repositories(session, **kwargs)
|
|
assert repositories.group_id == group_id
|
|
assert repositories.household_id == household_id
|
|
|
|
# test that sentinel is used correctly
|
|
if isinstance(group_id, NotSet):
|
|
assert repositories.group_id is NOT_SET
|
|
if isinstance(household_id, NotSet):
|
|
assert repositories.household_id is NOT_SET
|
|
|
|
|
|
def test_repository_generic_constructor(session: Session):
|
|
RepositoryGeneric(session, "id", MealieModel, SqlAlchemyBase)
|
|
|
|
|
|
def test_repository_group_constructor(session: Session):
|
|
BASE_ARGS = (session, "id", MealieModel, SqlAlchemyBase)
|
|
|
|
with pytest.raises(ValueError):
|
|
GroupRepositoryGeneric(*BASE_ARGS, group_id=NOT_SET)
|
|
|
|
GroupRepositoryGeneric(*BASE_ARGS, group_id=None)
|
|
GroupRepositoryGeneric(*BASE_ARGS, group_id=uuid4())
|
|
|
|
|
|
def test_repository_household_constructor(session: Session):
|
|
BASE_ARGS = (session, "id", MealieModel, SqlAlchemyBase)
|
|
|
|
with pytest.raises(ValueError):
|
|
HouseholdRepositoryGeneric(*BASE_ARGS, group_id=NOT_SET, household_id=NOT_SET)
|
|
HouseholdRepositoryGeneric(*BASE_ARGS, group_id=uuid4(), household_id=NOT_SET)
|
|
HouseholdRepositoryGeneric(*BASE_ARGS, group_id=NOT_SET, household_id=uuid4())
|
|
|
|
HouseholdRepositoryGeneric(*BASE_ARGS, group_id=None, household_id=None)
|
|
HouseholdRepositoryGeneric(*BASE_ARGS, group_id=uuid4(), household_id=None)
|
|
HouseholdRepositoryGeneric(*BASE_ARGS, group_id=None, household_id=uuid4())
|
|
HouseholdRepositoryGeneric(*BASE_ARGS, group_id=uuid4(), household_id=uuid4())
|
|
|
|
|
|
@pytest.mark.parametrize("use_group_id", [True, False])
|
|
@pytest.mark.parametrize("use_household_id", [True, False])
|
|
def test_all_repositories_constructors(session: Session, use_group_id: bool, use_household_id: bool):
|
|
kwargs = {}
|
|
if use_group_id:
|
|
kwargs["group_id"] = uuid4()
|
|
if use_household_id:
|
|
kwargs["household_id"] = uuid4()
|
|
repositories = get_repositories(session, **kwargs)
|
|
for name, member in inspect.getmembers(repositories.__class__):
|
|
if not isinstance(member, cached_property):
|
|
continue
|
|
signature = inspect.signature(member.func)
|
|
repo_type = signature.return_annotation
|
|
try:
|
|
if not issubclass(repo_type, RepositoryGeneric):
|
|
continue
|
|
except TypeError:
|
|
continue
|
|
|
|
if issubclass(repo_type, HouseholdRepositoryGeneric):
|
|
if not (use_group_id and use_household_id):
|
|
with pytest.raises(ValueError):
|
|
getattr(repositories, name)
|
|
else:
|
|
repo = getattr(repositories, name)
|
|
assert repo.group_id == kwargs["group_id"]
|
|
assert repo.household_id == kwargs["household_id"]
|
|
elif issubclass(repo_type, GroupRepositoryGeneric):
|
|
if not use_group_id:
|
|
with pytest.raises(ValueError):
|
|
getattr(repositories, name)
|
|
else:
|
|
repo = getattr(repositories, name)
|
|
assert repo.group_id == kwargs["group_id"]
|
|
assert repo.household_id is None
|
|
else:
|
|
repo = getattr(repositories, name)
|
|
assert repo.group_id is None
|
|
assert repo.household_id is None
|
|
|
|
|
|
def test_group_repositories_filter_by_group(session: Session):
|
|
unfiltered_repos = get_repositories(session, group_id=None, household_id=None)
|
|
group_1 = unfiltered_repos.groups.create({"name": random_string()})
|
|
group_2 = unfiltered_repos.groups.create({"name": random_string()})
|
|
|
|
group_1_repos = get_repositories(session, group_id=group_1.id, household_id=None)
|
|
group_2_repos = get_repositories(session, group_id=group_2.id, household_id=None)
|
|
food_1 = group_1_repos.ingredient_foods.create(
|
|
SaveIngredientFood(id=uuid4(), group_id=group_1.id, name=random_string())
|
|
)
|
|
food_2 = group_2_repos.ingredient_foods.create(
|
|
SaveIngredientFood(id=uuid4(), group_id=group_2.id, name=random_string())
|
|
)
|
|
|
|
# unfiltered_repos should find both foods
|
|
assert food_1 == unfiltered_repos.ingredient_foods.get_one(food_1.id)
|
|
assert food_2 == unfiltered_repos.ingredient_foods.get_one(food_2.id)
|
|
all_foods = unfiltered_repos.ingredient_foods.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
assert food_1 in all_foods
|
|
assert food_2 in all_foods
|
|
|
|
# group_repos should only find foods with the correct group_id
|
|
assert food_1 == group_1_repos.ingredient_foods.get_one(food_1.id)
|
|
assert group_1_repos.ingredient_foods.get_one(food_2.id) is None
|
|
assert [food_1] == group_1_repos.ingredient_foods.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
|
|
assert group_2_repos.ingredient_foods.get_one(food_1.id) is None
|
|
assert food_2 == group_2_repos.ingredient_foods.get_one(food_2.id)
|
|
assert [food_2] == group_2_repos.ingredient_foods.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
|
|
|
|
def test_household_repositories_filter_by_household(session: Session):
|
|
unfiltered_repos = get_repositories(session, group_id=None, household_id=None)
|
|
group = unfiltered_repos.groups.create({"name": random_string()})
|
|
group_repos = get_repositories(session, group_id=group.id, household_id=None)
|
|
household_1 = group_repos.households.create({"name": random_string(), "group_id": group.id})
|
|
household_2 = group_repos.households.create({"name": random_string(), "group_id": group.id})
|
|
|
|
household_1_repos = get_repositories(session, group_id=group.id, household_id=household_1.id)
|
|
household_2_repos = get_repositories(session, group_id=group.id, household_id=household_2.id)
|
|
webhook_1 = household_1_repos.webhooks.create(
|
|
SaveWebhook(group_id=group.id, household_id=household_1.id, scheduled_time=time.time())
|
|
)
|
|
webhook_2 = household_2_repos.webhooks.create(
|
|
SaveWebhook(group_id=group.id, household_id=household_2.id, scheduled_time=time.time())
|
|
)
|
|
|
|
# unfiltered_repos and group_repos should find both webhooks
|
|
for repos in [unfiltered_repos, group_repos]:
|
|
assert webhook_1 == repos.webhooks.get_one(webhook_1.id)
|
|
assert webhook_2 == repos.webhooks.get_one(webhook_2.id)
|
|
all_webhooks = repos.webhooks.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
assert webhook_1 in all_webhooks
|
|
assert webhook_2 in all_webhooks
|
|
|
|
# household_repos should only find webhooks with the correct household_id
|
|
assert webhook_1 == household_1_repos.webhooks.get_one(webhook_1.id)
|
|
assert household_1_repos.webhooks.get_one(webhook_2.id) is None
|
|
assert [webhook_1] == household_1_repos.webhooks.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
|
|
assert household_2_repos.webhooks.get_one(webhook_1.id) is None
|
|
assert webhook_2 == household_2_repos.webhooks.get_one(webhook_2.id)
|
|
assert [webhook_2] == household_2_repos.webhooks.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
|
|
# a different group's repos shouldn't find anything
|
|
other_group = unfiltered_repos.groups.create({"name": random_string()})
|
|
for household_id in [household_1.id, household_2.id]:
|
|
other_group_repos = get_repositories(session, group_id=other_group.id, household_id=household_id)
|
|
assert other_group_repos.webhooks.get_one(webhook_1.id) is None
|
|
assert other_group_repos.webhooks.get_one(webhook_2.id) is None
|
|
assert other_group_repos.webhooks.page_all(PaginationQuery(page=1, per_page=-1)).items == []
|
|
|
|
|
|
def test_recipe_repo_filter_by_household_with_proxy(session: Session):
|
|
unfiltered_repos = get_repositories(session, group_id=None, household_id=None)
|
|
group = unfiltered_repos.groups.create({"name": random_string()})
|
|
group_repos = get_repositories(session, group_id=group.id, household_id=None)
|
|
household_1 = group_repos.households.create({"name": random_string(), "group_id": group.id})
|
|
household_2 = group_repos.households.create({"name": random_string(), "group_id": group.id})
|
|
|
|
user_1 = group_repos.users.create(
|
|
{
|
|
"username": random_string(),
|
|
"email": random_email(),
|
|
"group": group.name,
|
|
"household": household_1.name,
|
|
"full_name": random_string(),
|
|
"password": random_string(),
|
|
"admin": False,
|
|
}
|
|
)
|
|
user_2 = group_repos.users.create(
|
|
{
|
|
"username": random_string(),
|
|
"email": random_email(),
|
|
"group": group.name,
|
|
"household": household_2.name,
|
|
"full_name": random_string(),
|
|
"password": random_string(),
|
|
"admin": False,
|
|
}
|
|
)
|
|
|
|
household_1_repos = get_repositories(session, group_id=group.id, household_id=household_1.id)
|
|
household_2_repos = get_repositories(session, group_id=group.id, household_id=household_2.id)
|
|
recipe_1 = household_1_repos.recipes.create(
|
|
Recipe(
|
|
user_id=user_1.id,
|
|
group_id=group.id,
|
|
name=random_string(),
|
|
)
|
|
)
|
|
recipe_2 = household_2_repos.recipes.create(
|
|
Recipe(
|
|
user_id=user_2.id,
|
|
group_id=group.id,
|
|
name=random_string(),
|
|
)
|
|
)
|
|
assert recipe_1.id and recipe_2.id
|
|
|
|
assert household_1_repos.recipes.get_one(recipe_1.slug) == recipe_1
|
|
assert household_1_repos.recipes.get_one(recipe_2.slug) is None
|
|
result = household_1_repos.recipes.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
assert len(result) == 1
|
|
assert result[0].id == recipe_1.id
|
|
|
|
assert household_2_repos.recipes.get_one(recipe_1.slug) is None
|
|
assert household_2_repos.recipes.get_one(recipe_2.slug) == recipe_2
|
|
result = household_2_repos.recipes.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
assert len(result) == 1
|
|
assert result[0].id == recipe_2.id
|
|
|
|
|
|
def test_generic_repo_filter_by_household_with_proxy(session: Session):
|
|
unfiltered_repos = get_repositories(session, group_id=None, household_id=None)
|
|
group = unfiltered_repos.groups.create({"name": random_string()})
|
|
group_repos = get_repositories(session, group_id=group.id, household_id=None)
|
|
household_1 = group_repos.households.create({"name": random_string(), "group_id": group.id})
|
|
household_2 = group_repos.households.create({"name": random_string(), "group_id": group.id})
|
|
|
|
user_1 = group_repos.users.create(
|
|
{
|
|
"username": random_string(),
|
|
"email": random_email(),
|
|
"group": group.name,
|
|
"household": household_1.name,
|
|
"full_name": random_string(),
|
|
"password": random_string(),
|
|
"admin": False,
|
|
}
|
|
)
|
|
user_2 = group_repos.users.create(
|
|
{
|
|
"username": random_string(),
|
|
"email": random_email(),
|
|
"group": group.name,
|
|
"household": household_2.name,
|
|
"full_name": random_string(),
|
|
"password": random_string(),
|
|
"admin": False,
|
|
}
|
|
)
|
|
|
|
household_1_repos = get_repositories(session, group_id=group.id, household_id=household_1.id)
|
|
household_2_repos = get_repositories(session, group_id=group.id, household_id=household_2.id)
|
|
shopping_list_service_1 = ShoppingListService(household_1_repos)
|
|
shopping_list_service_2 = ShoppingListService(household_2_repos)
|
|
shopping_list_1 = shopping_list_service_1.create_one_list(ShoppingListCreate(name=random_string()), user_1.id)
|
|
shopping_list_2 = shopping_list_service_2.create_one_list(ShoppingListCreate(name=random_string()), user_2.id)
|
|
|
|
assert household_1_repos.group_shopping_lists.get_one(shopping_list_1.id) == shopping_list_1
|
|
assert household_1_repos.group_shopping_lists.get_one(shopping_list_2.id) is None
|
|
result = household_1_repos.group_shopping_lists.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
assert len(result) == 1
|
|
assert result[0].id == shopping_list_1.id
|
|
|
|
assert household_2_repos.group_shopping_lists.get_one(shopping_list_1.id) is None
|
|
assert household_2_repos.group_shopping_lists.get_one(shopping_list_2.id) == shopping_list_2
|
|
result = household_2_repos.group_shopping_lists.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
assert len(result) == 1
|
|
assert result[0].id == shopping_list_2.id
|
|
|
|
|
|
def test_changing_user_changes_household(session: Session):
|
|
unfiltered_repos = get_repositories(session, group_id=None, household_id=None)
|
|
group = unfiltered_repos.groups.create({"name": random_string()})
|
|
group_repos = get_repositories(session, group_id=group.id, household_id=None)
|
|
household_1 = group_repos.households.create({"name": random_string(), "group_id": group.id})
|
|
household_2 = group_repos.households.create({"name": random_string(), "group_id": group.id})
|
|
|
|
user_1 = group_repos.users.create(
|
|
{
|
|
"username": random_string(),
|
|
"email": random_email(),
|
|
"group": group.name,
|
|
"household": household_1.name,
|
|
"full_name": random_string(),
|
|
"password": random_string(),
|
|
"admin": False,
|
|
}
|
|
)
|
|
user_2 = group_repos.users.create(
|
|
{
|
|
"username": random_string(),
|
|
"email": random_email(),
|
|
"group": group.name,
|
|
"household": household_2.name,
|
|
"full_name": random_string(),
|
|
"password": random_string(),
|
|
"admin": False,
|
|
}
|
|
)
|
|
|
|
household_1_repos = get_repositories(session, group_id=group.id, household_id=household_1.id)
|
|
household_2_repos = get_repositories(session, group_id=group.id, household_id=household_2.id)
|
|
|
|
# create shopping list with user_1/household_1
|
|
shopping_list = ShoppingListService(household_1_repos).create_one_list(
|
|
ShoppingListCreate(name=random_string()), user_1.id
|
|
)
|
|
|
|
# only household_1_repos should find the list
|
|
response = household_1_repos.group_shopping_lists.get_one(shopping_list.id)
|
|
assert response
|
|
assert response.user_id == user_1.id
|
|
response = household_2_repos.group_shopping_lists.get_one(shopping_list.id)
|
|
assert response is None
|
|
|
|
items = household_1_repos.group_shopping_lists.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
assert len(items) == 1
|
|
assert items[0].id == shopping_list.id
|
|
assert household_2_repos.group_shopping_lists.page_all(PaginationQuery(page=1, per_page=-1)).items == []
|
|
|
|
# update shopping list to user_2/household_2 using household_1_repos
|
|
shopping_list.user_id = user_2.id
|
|
household_1_repos.group_shopping_lists.update(shopping_list.id, shopping_list)
|
|
|
|
# now only household_2_repos should find the list
|
|
response = household_1_repos.group_shopping_lists.get_one(shopping_list.id)
|
|
assert response is None
|
|
response = household_2_repos.group_shopping_lists.get_one(shopping_list.id)
|
|
assert response
|
|
assert response.user_id == user_2.id
|
|
|
|
items = household_2_repos.group_shopping_lists.page_all(PaginationQuery(page=1, per_page=-1)).items
|
|
assert len(items) == 1
|
|
assert items[0].id == shopping_list.id
|
|
assert household_1_repos.group_shopping_lists.page_all(PaginationQuery(page=1, per_page=-1)).items == []
|