mealie/tests/multitenant_tests/case_abc.py

from abc import ABC, abstractmethod
from typing import Tuple

from fastapi import Response
from fastapi.testclient import TestClient

from mealie.repos.repository_factory import AllRepositories


class ABCMultiTenantTestCase(ABC):
    def __init__(self, database: AllRepositories, client: TestClient) -> None:
        self.database = database
        self.client = client
        self.items = []

    @abstractmethod
    def seed_action(repos: AllRepositories, group_id: str) -> set[int] | set[str]:
        ...

    def seed_multi(self, group1_id: str, group2_id: str) -> Tuple[set[int], set[int]]:
        pass

    @abstractmethod
    def get_all(token: str) -> Response:
        ...

    @abstractmethod
    def cleanup(self) -> None:
        ...

    def __enter__(self):
        pass

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.cleanup()