"""Set up unit test environment."""

from typing import Any, Generator

import pytest

from carveouts import api

from tests.unit.utils import db_operations


@pytest.fixture
def db_fixture() -> Generator[None, Any, None]:
    """Set up the db fixtures."""
    db_operations.seed_customer_master_master()
    db_operations.seed_customer_master()
    db_operations.seed_country()
    db_operations.seed_releases()
    db_operations.seed_release_default_dms_master_restriction()
    db_operations.seed_release_dms_master_restriction()
    db_operations.seed_release_dms_restriction()
    db_operations.seed_release_territory_restriction()
    db_operations.seed_subaccount()
    db_operations.seed_subaccount_dms_master_restriction()
    db_operations.seed_subaccount_dms_restriction()
    db_operations.seed_subaccount_territory_restriction()
    db_operations.seed_vendor()
    db_operations.seed_vendor_contract_table()
    db_operations.seed_vendor_contract_distribution_type()
    db_operations.seed_vendor_dms_master_restriction()
    db_operations.seed_vendor_contract_distribution_type()
    yield


@pytest.fixture
def app_context() -> Generator[None, Any, None]:
    """Create an api test app fixture."""
    with api.app.app_context():
        yield