pyr0ball
1a5fb23dfd
Extends AssemblyTemplate dataclass with slug, icon, descriptor, and role_hints fields. Updates all 13 template instantiations with appropriate values. Adds _TEMPLATE_BY_SLUG lookup dict and get_templates_for_api() serialiser for the templates endpoint.
135 lines
4.1 KiB
Python
135 lines
4.1 KiB
Python
"""Tests for Build Your Own recipe assembly schemas."""
|
|
import pytest
|
|
from app.models.schemas.recipe import (
|
|
AssemblyRoleOut,
|
|
AssemblyTemplateOut,
|
|
RoleCandidateItem,
|
|
RoleCandidatesResponse,
|
|
BuildRequest,
|
|
)
|
|
|
|
|
|
def test_assembly_role_out_schema():
|
|
"""Test AssemblyRoleOut schema creation and field access."""
|
|
role = AssemblyRoleOut(
|
|
display="protein",
|
|
required=True,
|
|
keywords=["chicken"],
|
|
hint="Main ingredient"
|
|
)
|
|
assert role.display == "protein"
|
|
assert role.required is True
|
|
assert role.keywords == ["chicken"]
|
|
assert role.hint == "Main ingredient"
|
|
|
|
|
|
def test_assembly_template_out_schema():
|
|
"""Test AssemblyTemplateOut schema with nested roles."""
|
|
tmpl = AssemblyTemplateOut(
|
|
id="burrito_taco",
|
|
title="Burrito / Taco",
|
|
icon="🌯",
|
|
descriptor="Protein, veg, and sauce in a tortilla or over rice",
|
|
role_sequence=[
|
|
AssemblyRoleOut(
|
|
display="base",
|
|
required=True,
|
|
keywords=["tortilla"],
|
|
hint="The wrap"
|
|
),
|
|
],
|
|
)
|
|
assert tmpl.id == "burrito_taco"
|
|
assert tmpl.title == "Burrito / Taco"
|
|
assert tmpl.icon == "🌯"
|
|
assert len(tmpl.role_sequence) == 1
|
|
assert tmpl.role_sequence[0].display == "base"
|
|
|
|
|
|
def test_role_candidate_item_schema():
|
|
"""Test RoleCandidateItem schema with tags."""
|
|
item = RoleCandidateItem(
|
|
name="bell pepper",
|
|
in_pantry=True,
|
|
tags=["sweet", "vegetable"]
|
|
)
|
|
assert item.name == "bell pepper"
|
|
assert item.in_pantry is True
|
|
assert "sweet" in item.tags
|
|
|
|
|
|
def test_role_candidates_response_schema():
|
|
"""Test RoleCandidatesResponse with compatible and other candidates."""
|
|
resp = RoleCandidatesResponse(
|
|
compatible=[
|
|
RoleCandidateItem(name="bell pepper", in_pantry=True, tags=["sweet"])
|
|
],
|
|
other=[
|
|
RoleCandidateItem(
|
|
name="corn",
|
|
in_pantry=False,
|
|
tags=["sweet", "starchy"]
|
|
)
|
|
],
|
|
available_tags=["sweet", "starchy"],
|
|
)
|
|
assert len(resp.compatible) == 1
|
|
assert resp.compatible[0].name == "bell pepper"
|
|
assert len(resp.other) == 1
|
|
assert "sweet" in resp.available_tags
|
|
assert "starchy" in resp.available_tags
|
|
|
|
|
|
def test_build_request_schema():
|
|
"""Test BuildRequest schema with template and role overrides."""
|
|
req = BuildRequest(
|
|
template_id="burrito_taco",
|
|
role_overrides={"protein": "chicken", "sauce": "verde"}
|
|
)
|
|
assert req.template_id == "burrito_taco"
|
|
assert req.role_overrides["protein"] == "chicken"
|
|
assert req.role_overrides["sauce"] == "verde"
|
|
|
|
|
|
def test_role_candidates_response_defaults():
|
|
"""Test RoleCandidatesResponse with default factory fields."""
|
|
resp = RoleCandidatesResponse()
|
|
assert resp.compatible == []
|
|
assert resp.other == []
|
|
assert resp.available_tags == []
|
|
|
|
|
|
def test_build_request_defaults():
|
|
"""Test BuildRequest with default role_overrides."""
|
|
req = BuildRequest(template_id="test_template")
|
|
assert req.template_id == "test_template"
|
|
assert req.role_overrides == {}
|
|
|
|
|
|
def test_get_templates_for_api_returns_13():
|
|
from app.services.recipe.assembly_recipes import get_templates_for_api
|
|
templates = get_templates_for_api()
|
|
assert len(templates) == 13
|
|
|
|
|
|
def test_get_templates_for_api_shape():
|
|
from app.services.recipe.assembly_recipes import get_templates_for_api
|
|
templates = get_templates_for_api()
|
|
t = next(t for t in templates if t["id"] == "burrito_taco")
|
|
assert t["title"] == "Burrito / Taco"
|
|
assert t["icon"] == "🌯"
|
|
assert isinstance(t["role_sequence"], list)
|
|
assert len(t["role_sequence"]) >= 1
|
|
role = t["role_sequence"][0]
|
|
assert "display" in role
|
|
assert "required" in role
|
|
assert "keywords" in role
|
|
assert "hint" in role
|
|
|
|
|
|
def test_get_templates_for_api_all_have_slugs():
|
|
from app.services.recipe.assembly_recipes import get_templates_for_api
|
|
templates = get_templates_for_api()
|
|
slugs = {t["id"] for t in templates}
|
|
assert len(slugs) == 13
|
|
assert all(isinstance(s, str) and len(s) > 3 for s in slugs)
|