"""Pytest configuration."""

import pytest
from src.common.types import ApiRequestType
from src.models.apple_music import AppleMusicSearchAlbumByUPCRequest
from src.models.apple_music import AppleMusicSearchAlbumByUPCRequestPayload


@pytest.fixture
def apple_music_search_album_by_upc_record():
    """Apple Music search album by UPC record fixture."""
    return {
        'api_request': {
            'store': 'APPLE_MUSIC',
            'type': 'SEARCH_ALBUM_BY_UPC',
        },
        'payload': {
            'product_id': 1,
            'upc': 2,
            'vendor_id': 3,
            'subaccount_id': 4,
        }
    }


@pytest.fixture
def apple_music_search_album_by_upc_request(
        apple_music_search_album_by_upc_record):
    """Search request model mock."""
    record = apple_music_search_album_by_upc_record
    request_type = ApiRequestType(
        store=record['api_request']['store'],
        type=record['api_request']['type'])
    return AppleMusicSearchAlbumByUPCRequest(
        request_type=request_type,
        payload=AppleMusicSearchAlbumByUPCRequestPayload(
            product_id=record['payload']['product_id'],
            upc=record['payload']['upc'],
            vendor_id=record['payload']['vendor_id'],
            subaccount_id=record['payload']['subaccount_id'],
        )
    )