mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-10-31 02:03:35 -04:00
3c2744a3da
* add basic pre-commit file * add flake8 * add isort * add pep585-upgrade (typing upgrades) * use namespace for import * add mypy * update ci for backend * flake8 scope * fix version format * update makefile * disable strict option (temporary) * fix mypy issues * upgrade type hints (pre-commit) * add vscode typing check * add types to dev deps * remote container draft * update setup script * update compose version * run setup on create * dev containers update * remove unused pages * update setup tips * expose ports * Update pre-commit to include flask8-print (#1053) * Add in flake8-print to pre-commit * pin version of flake8-print * formatting * update getting strated docs * add mypy to pre-commit * purge .mypy_cache on clean * drop mypy Co-authored-by: zackbcom <zackbcom@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 routes
|
|
|
|
|
|
class UnitsTestCase(ABCMultiTenantTestCase):
|
|
items: list[IngredientUnit]
|
|
|
|
def seed_action(self, group_id: str) -> set[int]:
|
|
unit_ids: set[int] = 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()
|
|
g2_item_ids = set()
|
|
|
|
for group_id, item_ids in [(group1_id, g1_item_ids), (group2_id, g2_item_ids)]:
|
|
for _ in range(10):
|
|
name = utils.random_string(10)
|
|
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(routes.RoutesUnits.base, headers=token)
|
|
|
|
def cleanup(self) -> None:
|
|
for item in self.items:
|
|
self.database.ingredient_units.delete(item.id)
|