mealie/tests/integration_tests/user_household_tests/test_household_perferences.py
Michael Genson fd0257c1b8
feat: Additional Household Permissions (#4158)
Co-authored-by: Kuchenpirat <24235032+Kuchenpirat@users.noreply.github.com>
2024-09-17 10:48:14 -05:00

64 lines
2.5 KiB
Python

from fastapi.testclient import TestClient
from mealie.schema.household.household_preferences import UpdateHouseholdPreferences
from tests.utils import api_routes
from tests.utils.assertion_helpers import assert_ignore_keys
from tests.utils.factories import random_bool
from tests.utils.fixture_schemas import TestUser
def test_get_preferences(api_client: TestClient, unique_user: TestUser) -> None:
response = api_client.get(api_routes.households_preferences, headers=unique_user.token)
assert response.status_code == 200
preferences = response.json()
assert preferences["recipePublic"] in {True, False}
assert preferences["recipeShowNutrition"] in {True, False}
def test_preferences_in_household(api_client: TestClient, unique_user: TestUser) -> None:
response = api_client.get(api_routes.households_self, headers=unique_user.token)
assert response.status_code == 200
household = response.json()
assert household["preferences"] is not None
# Spot Check
assert household["preferences"]["recipePublic"] in {True, False}
assert household["preferences"]["recipeShowNutrition"] in {True, False}
def test_update_preferences_no_permission(api_client: TestClient, user_tuple: list[TestUser]) -> None:
unique_user, other_user = user_tuple
user = other_user.repos.users.get_one(unique_user.user_id)
assert user
user.can_manage_household = False
other_user.repos.users.update(user.id, user)
new_data = UpdateHouseholdPreferences(recipe_public=random_bool(), recipe_show_nutrition=random_bool())
response = api_client.put(api_routes.households_preferences, json=new_data.model_dump(), headers=unique_user.token)
assert response.status_code == 403
def test_update_preferences(api_client: TestClient, user_tuple: list[TestUser]) -> None:
unique_user, other_user = user_tuple
user = other_user.repos.users.get_one(unique_user.user_id)
assert user
user.can_manage_household = True
other_user.repos.users.update(user.id, user)
new_data = UpdateHouseholdPreferences(recipe_public=random_bool(), recipe_show_nutrition=random_bool())
response = api_client.put(api_routes.households_preferences, json=new_data.model_dump(), headers=unique_user.token)
assert response.status_code == 200
preferences = response.json()
assert preferences is not None
assert preferences["recipePublic"] == new_data.recipe_public
assert preferences["recipeShowNutrition"] == new_data.recipe_show_nutrition
assert_ignore_keys(new_data.model_dump(by_alias=True), preferences, ["id", "householdId"])