mirror of
https://github.com/sstent/foodplanner.git
synced 2025-12-05 23:51:46 +00:00
246 lines
9.7 KiB
Python
246 lines
9.7 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy.pool import StaticPool
|
|
|
|
from main import app
|
|
from app.database import Base, get_db, Food, Meal, MealFood, TrackedDay, TrackedMeal, TrackedMealFood
|
|
from datetime import date
|
|
|
|
# Setup for in-memory SQLite database for testing
|
|
SQLALCHEMY_DATABASE_URL = "sqlite:///:memory:"
|
|
engine = create_engine(
|
|
SQLALCHEMY_DATABASE_URL,
|
|
connect_args={"check_same_thread": False},
|
|
poolclass=StaticPool,
|
|
)
|
|
TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
|
|
|
@pytest.fixture(name="session")
|
|
def session_fixture():
|
|
Base.metadata.create_all(engine)
|
|
db = TestingSessionLocal()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
Base.metadata.drop_all(engine)
|
|
|
|
@pytest.fixture(name="client")
|
|
def client_fixture(session):
|
|
def override_get_db():
|
|
yield session
|
|
app.dependency_overrides[get_db] = override_get_db
|
|
yield TestClient(app)
|
|
app.dependency_overrides.clear()
|
|
|
|
def create_test_data(session: TestingSessionLocal):
|
|
food1 = Food(name="Apple", serving_size=100, serving_unit="g", calories=52, protein=0.3, carbs=14, fat=0.2, fiber=2.4, sugar=10.4, sodium=1)
|
|
food2 = Food(name="Banana", serving_size=100, serving_unit="g", calories=89, protein=1.1, carbs=23, fat=0.3, fiber=2.6, sugar=12.2, sodium=1)
|
|
session.add_all([food1, food2])
|
|
session.commit()
|
|
session.refresh(food1)
|
|
session.refresh(food2)
|
|
|
|
meal1 = Meal(name="Fruit Salad", meal_type="custom", meal_time="Breakfast")
|
|
session.add(meal1)
|
|
session.commit()
|
|
session.refresh(meal1)
|
|
|
|
meal_food1 = MealFood(meal_id=meal1.id, food_id=food1.id, quantity=150)
|
|
meal_food2 = MealFood(meal_id=meal1.id, food_id=food2.id, quantity=100)
|
|
session.add_all([meal_food1, meal_food2])
|
|
session.commit()
|
|
|
|
tracked_day = TrackedDay(person="Sarah", date=date.today(), is_modified=False)
|
|
session.add(tracked_day)
|
|
session.commit()
|
|
session.refresh(tracked_day)
|
|
|
|
tracked_meal = TrackedMeal(tracked_day_id=tracked_day.id, meal_id=meal1.id, meal_time="Breakfast")
|
|
session.add(tracked_meal)
|
|
session.commit()
|
|
session.refresh(tracked_meal)
|
|
|
|
return food1, food2, meal1, tracked_day, tracked_meal
|
|
|
|
def test_get_tracked_meal_foods_endpoint(client: TestClient, session: TestingSessionLocal):
|
|
"""Test retrieving foods for a tracked meal"""
|
|
food1, food2, meal1, tracked_day, tracked_meal = create_test_data(session)
|
|
|
|
response = client.get(f"/tracker/get_tracked_meal_foods/{tracked_meal.id}")
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["status"] == "success"
|
|
assert len(data["meal_foods"]) == 2
|
|
|
|
# Check if food details are correct
|
|
food_names = [f["food_name"] for f in data["meal_foods"]]
|
|
assert "Apple" in food_names
|
|
assert "Banana" in food_names
|
|
|
|
# Check quantities
|
|
for food_data in data["meal_foods"]:
|
|
if food_data["food_name"] == "Apple":
|
|
assert food_data["quantity"] == 150.0
|
|
elif food_data["food_name"] == "Banana":
|
|
assert food_data["quantity"] == 100.0
|
|
|
|
def test_edit_tracked_meal_with_override_flow(client: TestClient, session: TestingSessionLocal):
|
|
"""
|
|
Test the full flow of editing a tracked meal, overriding a food, and then retrieving its foods.
|
|
This test aims to reproduce the "Error loading tracked meal foods" bug.
|
|
"""
|
|
food1, food2, meal1, tracked_day, tracked_meal = create_test_data(session)
|
|
|
|
# 1. Simulate adding a meal (already done by create_test_data, so tracked_meal exists)
|
|
# 2. Simulate updating a food in the tracked meal to create an override
|
|
# This will call /tracker/update_tracked_meal_foods
|
|
|
|
# Get the original MealFood for food1
|
|
original_meal_food1 = session.query(MealFood).filter(
|
|
MealFood.meal_id == meal1.id,
|
|
MealFood.food_id == food1.id
|
|
).first()
|
|
assert original_meal_food1 is not None
|
|
|
|
# Prepare update data: update food1 quantity (should create a TrackedMealFood and delete original MealFood)
|
|
updated_foods_data = [
|
|
{"id": original_meal_food1.id, "food_id": food1.id, "grams": 175.0, "is_custom": False}, # Original MealFood, but quantity changed
|
|
{"id": None, "food_id": food2.id, "grams": 100.0, "is_custom": False} # Unchanged original MealFood
|
|
]
|
|
|
|
response_update = client.post(
|
|
"/tracker/update_tracked_meal_foods",
|
|
json={
|
|
"tracked_meal_id": tracked_meal.id,
|
|
"foods": updated_foods_data
|
|
}
|
|
)
|
|
assert response_update.status_code == 200
|
|
assert response_update.json()["status"] == "success"
|
|
|
|
session.expire_all() # Ensure a fresh load from the database
|
|
|
|
# Verify original MealFood for food1 is deleted
|
|
deleted_meal_food1 = session.query(MealFood).filter(MealFood.id == original_meal_food1.id).first()
|
|
assert deleted_meal_food1 is None
|
|
|
|
# Verify a TrackedMealFood for food1 now exists
|
|
overridden_tracked_food1 = session.query(TrackedMealFood).filter(
|
|
TrackedMealFood.tracked_meal_id == tracked_meal.id,
|
|
TrackedMealFood.food_id == food1.id
|
|
).first()
|
|
assert overridden_tracked_food1 is not None
|
|
assert overridden_tracked_food1.quantity == 175.0
|
|
|
|
# 3. Now, try to get the tracked meal foods again, which is where the bug occurs
|
|
# This will call /tracker/get_tracked_meal_foods
|
|
response_get = client.get(f"/tracker/get_tracked_meal_foods/{tracked_meal.id}")
|
|
assert response_get.status_code == 200
|
|
data_get = response_get.json()
|
|
assert data_get["status"] == "success"
|
|
assert len(data_get["meal_foods"]) == 2
|
|
|
|
# Verify the contents of the returned meal_foods
|
|
food_names = [f["food_name"] for f in data_get["meal_foods"]]
|
|
assert "Apple" in food_names
|
|
assert "Banana" in food_names
|
|
|
|
for food_data in data_get["meal_foods"]:
|
|
if food_data["food_name"] == "Apple":
|
|
assert food_data["quantity"] == 175.0
|
|
assert food_data["is_custom"] == True
|
|
elif food_data["food_name"] == "Banana":
|
|
assert food_data["quantity"] == 100.0
|
|
assert food_data["is_custom"] == False
|
|
|
|
|
|
def test_update_tracked_meal_foods_endpoint(client: TestClient, session: TestingSessionLocal):
|
|
"""Test updating quantities of foods in a tracked meal"""
|
|
food1, food2, meal1, tracked_day, tracked_meal = create_test_data(session)
|
|
|
|
# Add a tracked meal food for food1 to allow updates
|
|
tracked_meal_food1 = TrackedMealFood(tracked_meal_id=tracked_meal.id, food_id=food1.id, quantity=150.0)
|
|
session.add(tracked_meal_food1)
|
|
session.commit()
|
|
session.refresh(tracked_meal_food1)
|
|
|
|
# Prepare update data
|
|
updated_foods = [
|
|
{"id": tracked_meal_food1.id, "food_id": food1.id, "grams": 200.0, "is_custom": True},
|
|
{"id": None, "food_id": food2.id, "grams": 50.0, "is_custom": False} # This represents original meal food
|
|
]
|
|
|
|
response = client.post(
|
|
"/tracker/update_tracked_meal_foods",
|
|
json={
|
|
"tracked_meal_id": tracked_meal.id,
|
|
"foods": updated_foods
|
|
}
|
|
)
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["status"] == "success"
|
|
session.expire_all() # Expire all objects in the session to ensure a fresh load
|
|
|
|
# Verify updates in the database
|
|
updated_meal_foods = session.query(TrackedMealFood).filter(TrackedMealFood.tracked_meal_id == tracked_meal.id).all()
|
|
assert len(updated_meal_foods) == 2
|
|
|
|
for tmf in updated_meal_foods:
|
|
if tmf.food_id == food1.id:
|
|
assert tmf.quantity == 200.0
|
|
elif tmf.food_id == food2.id:
|
|
assert tmf.quantity == 50.0
|
|
|
|
def test_add_food_to_tracked_meal_endpoint(client: TestClient, session: TestingSessionLocal):
|
|
"""Test adding a new food to an existing tracked meal"""
|
|
food1, food2, meal1, tracked_day, tracked_meal = create_test_data(session)
|
|
|
|
# Create a new food to add
|
|
food3 = Food(name="Orange", serving_size=130, serving_unit="g", calories=62, protein=1.2, carbs=15, fat=0.2, fiber=3.1, sugar=12, sodium=0)
|
|
session.add(food3)
|
|
session.commit()
|
|
session.refresh(food3)
|
|
|
|
response = client.post(
|
|
"/tracker/add_food_to_tracked_meal",
|
|
json={
|
|
"tracked_meal_id": tracked_meal.id,
|
|
"food_id": food3.id,
|
|
"grams": 200
|
|
}
|
|
)
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["status"] == "success"
|
|
|
|
# Verify the food was added to the meal associated with the tracked meal
|
|
updated_meal_foods = session.query(MealFood).filter(MealFood.meal_id == meal1.id).all()
|
|
assert len(updated_meal_foods) == 3 # Original 2 + new 1
|
|
|
|
# Check the new food's quantity
|
|
orange_meal_food = next(mf for mf in updated_meal_foods if mf.food_id == food3.id)
|
|
assert orange_meal_food.quantity == 200
|
|
|
|
def test_remove_food_from_tracked_meal_endpoint(client: TestClient, session: TestingSessionLocal):
|
|
"""Test removing a food from a tracked meal"""
|
|
food1, food2, meal1, tracked_day, tracked_meal = create_test_data(session)
|
|
|
|
# Get the meal_food_id for food1
|
|
meal_food_to_remove = session.query(MealFood).filter(
|
|
MealFood.meal_id == meal1.id,
|
|
MealFood.food_id == food1.id
|
|
).first()
|
|
|
|
response = client.delete(f"/tracker/remove_food_from_tracked_meal/{meal_food_to_remove.id}")
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert data["status"] == "success"
|
|
|
|
# Verify the food was removed from the meal associated with the tracked meal
|
|
updated_meal_foods = session.query(MealFood).filter(MealFood.meal_id == meal1.id).all()
|
|
assert len(updated_meal_foods) == 1 # Original 2 - removed 1
|
|
assert updated_meal_foods[0].food_id == food2.id # Only food2 should remain |