68f7efc177
* fix #1144 * fix type checks * refactor test routes package * fix #1208 * unify test routes into module
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
from requests import Response
|
|
|
|
from mealie.schema.recipe.recipe import RecipeTool
|
|
from mealie.schema.recipe.recipe_tool import RecipeToolSave
|
|
from tests import utils
|
|
from tests.multitenant_tests.case_abc import ABCMultiTenantTestCase
|
|
from tests.utils import routes
|
|
|
|
|
|
class ToolsTestCase(ABCMultiTenantTestCase):
|
|
items: list[RecipeTool]
|
|
|
|
def seed_action(self, group_id: str) -> set[int]:
|
|
tool_ids: set[int] = set()
|
|
for _ in range(10):
|
|
tool = self.database.tools.create(
|
|
RecipeToolSave(
|
|
group_id=group_id,
|
|
name=utils.random_string(10),
|
|
)
|
|
)
|
|
|
|
tool_ids.add(str(tool.id))
|
|
self.items.append(tool)
|
|
|
|
return tool_ids
|
|
|
|
def seed_multi(self, group1_id: str, group2_id: str) -> tuple[set[int], set[int]]:
|
|
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)
|
|
tool = self.database.tools.create(
|
|
RecipeToolSave(
|
|
group_id=group_id,
|
|
name=name,
|
|
)
|
|
)
|
|
item_ids.add(str(tool.id))
|
|
self.items.append(tool)
|
|
|
|
return g1_item_ids, g2_item_ids
|
|
|
|
def get_all(self, token: str) -> Response:
|
|
return self.client.get(routes.organizers.Tools.base, headers=token)
|
|
|
|
def cleanup(self) -> None:
|
|
for item in self.items:
|
|
self.database.tools.delete(item.id)
|