mirror of
https://github.com/mealie-recipes/mealie.git
synced 2024-12-22 03:51:21 -08:00
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
import json
|
|
import random
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from mealie.schema.recipe.recipe import Recipe
|
|
from mealie.schema.recipe.recipe_step import IngredientReferences
|
|
from tests.utils import api_routes, jsonify
|
|
from tests.utils.fixture_schemas import TestUser
|
|
|
|
|
|
def test_associate_ingredient_with_step(api_client: TestClient, unique_user: TestUser, random_recipe: Recipe):
|
|
recipe: Recipe = random_recipe
|
|
|
|
# Associate an ingredient with a step
|
|
steps = {} # key=step_id, value=ingredient_id
|
|
|
|
for idx, step in enumerate(recipe.recipe_instructions or []):
|
|
ingredients = random.choices(recipe.recipe_ingredient, k=2)
|
|
|
|
step.ingredient_references = [
|
|
IngredientReferences(reference_id=ingredient.reference_id) for ingredient in ingredients
|
|
]
|
|
|
|
steps[idx] = [str(ingredient.reference_id) for ingredient in ingredients]
|
|
|
|
response = api_client.put(
|
|
api_routes.recipes_slug(recipe.slug),
|
|
json=jsonify(recipe.model_dump()),
|
|
headers=unique_user.token,
|
|
)
|
|
|
|
assert response.status_code == 200
|
|
|
|
# Get Recipe and check that the ingredient is associated with the step
|
|
|
|
response = api_client.get(api_routes.recipes_slug(recipe.slug), headers=unique_user.token)
|
|
assert response.status_code == 200
|
|
|
|
data: dict = json.loads(response.text)
|
|
|
|
for idx, stp in enumerate(data.get("recipeInstructions") or []):
|
|
all_refs = [ref["referenceId"] for ref in stp.get("ingredientReferences")]
|
|
|
|
assert len(all_refs) == 2
|
|
|
|
assert all(ref in steps[idx] for ref in all_refs)
|