# fixtures/test_fetch_by_key.yml

--- 

test_fetch_by_key:
  storage: 
    accounts:
      - { key: 1, name: "Visa Classic" }
      - { key: 2, name: "Visa Gold"}

test_success:
  - id: First test case
    account_key: 1
    expected: { key: 1, name: "Visa Classic" }
  - id: Second test case
    account_key: 2
    expected: { key: 2, name: "Visa Gold" }

test_missing:
  - id: Try fetch missing account
    account_key: 3
    expected_exc: AccountNotFound

# conftest.py

from typing import Any

import pytest

from tests.accounts import Account, AccountRepo, Storage


@pytest.fixture(scope="function")
def account(request) -> Account:
    """Account entity."""
    return Account(**request.param)


@pytest.fixture(scope="function")
def storage(request) -> Storage:
    """Entity storage."""
    storage = Storage()

    if "accounts" in request.param:
        accounts = [Account(**account) for account in request.param["accounts"]]
        storage.accounts = AccountRepo(accounts=accounts)

    return storage


@pytest.fixture(scope="function")
def expected(request) -> Any:
    """Expected test case result."""
    return request.param

from typing import Type

import pytest

from tests.accounts import Account, AccountNotFound, Storage


@pytest.fixture
def account_key(request) -> int:
    """Test account identifier."""
    return request.param


@pytest.fixture
def expected(request) -> Account:
    """Expected test case result."""
    return Account(**request.param)


@pytest.mark.integration
def test_success(storage: Storage, account_key: int, expected: Account) -> None:
    """Successful fetch account by key from storage."""
    result = storage.accounts.fetch_by_key(account_key)

    assert result == expected


@pytest.fixture
def expected_exc(request) -> Type[AccountNotFound]:
    """Expected exception when account not found in storage."""
    return AccountNotFound


@pytest.mark.integration
def test_missing(storage: Storage, account_key: int, expected_exc: Type[AccountNotFound]) -> None:
    """Try to fetch missing account by key from storage."""
    with pytest.raises(expected_exc):
        storage.accounts.fetch_by_key(account_key)

(env) $ python3 -m pip install pytest-fixture-maker

Initial release.