mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-10-27 08:14:30 -04:00
c9cc7a93c8
* fixed type/abc errors in tests * fixed false positive multitentant tests * fix tools not allowing unique slugs across groups * fixed alembic refs --------- Co-authored-by: Hayden <64056131+hay-kot@users.noreply.github.com>
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
from requests import Response
|
|
|
|
from mealie.schema.recipe.recipe_ingredient import IngredientUnit, SaveIngredientUnit
|
|
from tests import utils
|
|
from tests.multitenant_tests.case_abc import ABCMultiTenantTestCase
|
|
from tests.utils import api_routes
|
|
|
|
|
|
class UnitsTestCase(ABCMultiTenantTestCase):
|
|
items: list[IngredientUnit]
|
|
|
|
def seed_action(self, group_id: str) -> set[str]:
|
|
unit_ids: set[str] = set()
|
|
for _ in range(10):
|
|
unit = self.database.ingredient_units.create(
|
|
SaveIngredientUnit(
|
|
group_id=group_id,
|
|
name=utils.random_string(10),
|
|
)
|
|
)
|
|
|
|
unit_ids.add(str(unit.id))
|
|
self.items.append(unit)
|
|
|
|
return unit_ids
|
|
|
|
def seed_multi(self, group1_id: str, group2_id: str) -> tuple[set[str], set[str]]:
|
|
g1_item_ids: set[str] = set()
|
|
g2_item_ids: set[str] = set()
|
|
|
|
for _ in range(10):
|
|
name = utils.random_string(10)
|
|
for group_id, item_ids in [(group1_id, g1_item_ids), (group2_id, g2_item_ids)]:
|
|
food = self.database.ingredient_units.create(
|
|
SaveIngredientUnit(
|
|
group_id=group_id,
|
|
name=name,
|
|
)
|
|
)
|
|
item_ids.add(str(food.id))
|
|
self.items.append(food)
|
|
|
|
return g1_item_ids, g2_item_ids
|
|
|
|
def get_all(self, token: str) -> Response:
|
|
return self.client.get(api_routes.units, headers=token)
|
|
|
|
def cleanup(self) -> None:
|
|
for item in self.items:
|
|
self.database.ingredient_units.delete(item.id)
|