diff --git a/tests/integration_tests/user_group_tests/test_group_seeder.py b/tests/integration_tests/user_group_tests/test_group_seeder.py index 87c58475b..8788f154e 100644 --- a/tests/integration_tests/user_group_tests/test_group_seeder.py +++ b/tests/integration_tests/user_group_tests/test_group_seeder.py @@ -15,14 +15,12 @@ def test_seed_foods(api_client: TestClient, unique_user: TestUser): CREATED_FOODS = 2687 database = unique_user.repos - # Check that the foods was created foods = database.ingredient_foods.page_all(PaginationQuery(page=1, per_page=-1)).items assert len(foods) == 0 resp = api_client.post(api_routes.groups_seeders_foods, json={"locale": "en-US"}, headers=unique_user.token) assert resp.status_code == 200 - # Check that the foods was created foods = database.ingredient_foods.page_all(PaginationQuery(page=1, per_page=-1)).items assert len(foods) == CREATED_FOODS @@ -31,29 +29,37 @@ def test_seed_units(api_client: TestClient, unique_user: TestUser): CREATED_UNITS = 24 database = unique_user.repos - # Check that the foods was created units = database.ingredient_units.page_all(PaginationQuery(page=1, per_page=-1)).items assert len(units) == 0 resp = api_client.post(api_routes.groups_seeders_units, json={"locale": "en-US"}, headers=unique_user.token) assert resp.status_code == 200 - # Check that the foods was created units = database.ingredient_units.page_all(PaginationQuery(page=1, per_page=-1)).items assert len(units) == CREATED_UNITS + # Check that the "pint" unit was created and includes standardized data + pint_found = False + for unit in units: + if unit.name != "pint": + continue + + pint_found = True + assert unit.standard_quantity == 2 + assert unit.standard_unit == "cup" + + assert pint_found + def test_seed_labels(api_client: TestClient, unique_user: TestUser): CREATED_LABELS = 32 database = unique_user.repos - # Check that the foods was created labels = database.group_multi_purpose_labels.page_all(PaginationQuery(page=1, per_page=-1)).items assert len(labels) == 0 resp = api_client.post(api_routes.groups_seeders_labels, json={"locale": "en-US"}, headers=unique_user.token) assert resp.status_code == 200 - # Check that the foods was created labels = database.group_multi_purpose_labels.page_all(PaginationQuery(page=1, per_page=-1)).items assert len(labels) == CREATED_LABELS