"""Pytest configuration.""" import pytest from src.common.types import ApiRequestType from src.models.spotify import SpotifySearchAlbumByUPCRequest from src.models.spotify import SpotifySearchAlbumByUPCRequestPayload @pytest.fixture def spotify_search_album_by_upc_record(): """Spotify search album by UPC record fixture.""" return { 'api_request': { 'store': 'SPOTIFY', 'type': 'SEARCH_ALBUM_BY_UPC', }, 'payload': { 'product_id': 1, 'upc': 2, 'vendor_id': 3, 'subaccount_id': 4, } } @pytest.fixture def spotify_search_album_by_upc_request(spotify_search_album_by_upc_record): """Search request model mock.""" record = spotify_search_album_by_upc_record request_type = ApiRequestType( store=record['api_request']['store'], type=record['api_request']['type']) return SpotifySearchAlbumByUPCRequest( request_type=request_type, payload=SpotifySearchAlbumByUPCRequestPayload( product_id=record['payload']['product_id'], upc=record['payload']['upc'], vendor_id=record['payload']['vendor_id'], subaccount_id=record['payload']['subaccount_id'], ) )