Files
foodplanner/tests/test_weekly_menu.py
2025-09-29 09:31:01 -07:00

63 lines
2.0 KiB
Python

"""
Tests for Weekly Menu operations
"""
import pytest
class TestWeeklyMenuRoutes:
"""Test weekly menu-related routes"""
def test_get_weekly_menu_page(self, client):
"""Test GET /weeklymenu page"""
response = client.get("/weeklymenu")
assert response.status_code == 200
assert b"Weekly" in response.content or b"weekly" in response.content or b"Menu" in response.content
class TestWeeklyMenuCRUD:
"""Test weekly menu CRUD operations"""
def test_create_weekly_menu(self, client, db_session, sample_template):
"""Test creating a weekly menu"""
from main import WeeklyMenu, WeeklyMenuDay
weekly_menu = WeeklyMenu(name="Test Weekly Menu")
db_session.add(weekly_menu)
db_session.commit()
db_session.refresh(weekly_menu)
# Add days to weekly menu
for day in range(7):
menu_day = WeeklyMenuDay(
weekly_menu_id=weekly_menu.id,
day_of_week=day,
template_id=sample_template.id
)
db_session.add(menu_day)
db_session.commit()
assert weekly_menu.id is not None
assert len(weekly_menu.weekly_menu_days) == 7
def test_weekly_menu_relationships(self, client, db_session, sample_template):
"""Test weekly menu relationships"""
from main import WeeklyMenu, WeeklyMenuDay
weekly_menu = WeeklyMenu(name="Relationship Test Menu")
db_session.add(weekly_menu)
db_session.commit()
db_session.refresh(weekly_menu)
menu_day = WeeklyMenuDay(
weekly_menu_id=weekly_menu.id,
day_of_week=0, # Monday
template_id=sample_template.id
)
db_session.add(menu_day)
db_session.commit()
# Verify relationships
assert menu_day.weekly_menu.id == weekly_menu.id
assert menu_day.template.id == sample_template.id