mealie/tests/integration_tests/user_recipe_tests/test_recipe_owner.py

from fastapi.testclient import TestClient

from tests.utils.factories import random_string
from tests.utils.fixture_schemas import TestUser


class Routes:
    base = "/api/recipes"
    user = "/api/users/self"


GROUP_ID = 1
ADMIN_ID = 1
USER_ID = 2


def test_ownership_on_new_with_admin(api_client: TestClient, admin_token):
    recipe_name = random_string()
    response = api_client.post(Routes.base, json={"name": recipe_name}, headers=admin_token)
    assert response.status_code == 201

    recipe = api_client.get(Routes.base + f"/{recipe_name}", headers=admin_token).json()

    assert recipe["userId"] == ADMIN_ID
    assert recipe["groupId"] == GROUP_ID


def test_ownership_on_new_with_user(api_client: TestClient, g2_user: TestUser):
    recipe_name = random_string()

    response = api_client.post(Routes.base, json={"name": recipe_name}, headers=g2_user.token)

    assert response.status_code == 201

    response = api_client.get(Routes.base + f"/{recipe_name}", headers=g2_user.token)

    assert response.status_code == 200

    recipe = response.json()

    assert recipe["userId"] == g2_user.user_id
    assert recipe["groupId"] == g2_user.group_id


def test_get_all_only_includes_group_recipes(api_client: TestClient, unique_user: TestUser):
    for _ in range(5):
        recipe_name = random_string()
        response = api_client.post(Routes.base, json={"name": recipe_name}, headers=unique_user.token)

    response = api_client.get(Routes.base, headers=unique_user.token)

    assert response.status_code == 200

    recipes = response.json()

    assert len(recipes) == 5

    for recipe in recipes:
        assert recipe["groupId"] == unique_user.group_id
        assert recipe["userId"] == unique_user.user_id


def test_unique_slug_by_group(api_client: TestClient, unique_user: TestUser, g2_user: TestUser) -> None:
    create_data = {"name": random_string()}

    response = api_client.post(Routes.base, json=create_data, headers=unique_user.token)
    assert response.status_code == 201

    response = api_client.post(Routes.base, json=create_data, headers=g2_user.token)
    assert response.status_code == 201

    # Try to create a recipe again with the same name
    response = api_client.post(Routes.base, json=create_data, headers=g2_user.token)
    assert response.status_code == 400