"""Pytest conftest module."""
import pytest


@pytest.fixture
def context_json_audio():
    """Context JSON received."""
    return """{
        "bucket": "test_bucket",
        "key": "test_path/AUDIO_TEST_01_0_0_0.json",
        "item_index": 0,
        "execution_name": "test_execution_name",
        "state_machine_name": "test_sfn_name",
        "correlation_id": "test_correlation_id"
    }"""


@pytest.fixture
def context_json_artwork():
    """Context JSON received."""
    return """{
        "bucket": "test_bucket",
        "key": "test_path/AUDIO_TEST_01_0_0_0.json",
        "item_index": 1,
        "execution_name": "test_execution_name",
        "state_machine_name": "test_sfn_name",
        "correlation_id": "test_correlation_id"
    }"""


@pytest.fixture
def test_fileobj():
    """Incoming file data."""
    return """[{"ASSET_TYPE" :"audio" ,"FILE_NAME" :"test_filename.wav"
     ,"FILE_SIZE" :27043352 ,"FOREIGN_REL_ID" :"123456"
     ,"ORCH_PROJECT_CODE" :"1234567890" ,"ORCH_RELEASE_ID" :1234567
     ,"ORCH_UPC" :123456789012 ,"PATH" :"test_path/"
     ,"S3_BUCKET" :"test-bucket" ,"SESSION_ID" :"abcdefgh123" ,"TRACK" :1
     ,"UPLOAD_DATE" :"2021-01-29" ,"VENDOR_ID" :12345 ,"VOLUME" :1},
    {"ASSET_TYPE": "cover", "FILE_NAME": "test_filename.tif",
     "FILE_SIZE": 27023716, "FOREIGN_REL_ID": "7890123",
     "ORCH_PROJECT_CODE": "1234567890", "ORCH_RELEASE_ID": 7654321,
     "ORCH_UPC": 123456789012, "PATH": "test_path/",
     "S3_BUCKET": "test-bucket", "SESSION_ID": "abcdefgh123", "TRACK": 0,
     "UPLOAD_DATE": "2021-01-29", "VENDOR_ID": 12345, "VOLUME": 0}]"""


@pytest.fixture
def test_graphql_response():
    """Mock response from GraphQL."""
    return {
        'upc': '123456789012',
        'vendorId': 12345,
        'subaccountId': None,
        'productId': 1234567,
        'productName': 'Product Name',
        'project': {
            'projectCode': '9876543210',
            'projectName': 'Project Name',
            'projectId': 123456789
        },
        'tracks': [
            {
                'isrc': 'ABCD12345678',
                'tuid': '76543210',
                'trackNumber': 1,
                'volumeNumber': 1,
                'trackName': 'Track Name 1'
            }
        ],
    }


@pytest.fixture
def artwork_response():
    """Create handler response for artwork."""
    return {
        'product': {
            'product_id': 1234567,
            'project_code': '9876543210',
            'project_id': 123456789,
            'project_name': 'Project Name',
            'release_name': 'Product Name',
            'upc': '123456789012',
            'vendor_id': 12345,
            'subaccount_id': None,
            'artwork': {
                'bucket': 'test-bucket',
                'filename': 'test_filename.tif',
                'key': 'test_path/test_filename.tif',
                'ows_assets_filename': None
            }
        },
        'asset_type': 'cover',
        'correlation_id': 'test_correlation_id',
        'execution_name': 'test_execution_name',
        'state_machine_name': 'test_sfn_name'
    }


@pytest.fixture
def audio_response():
    """Create handler response for audio."""
    return {
        'product': {
            'product_id': 1234567,
            'project_code': '9876543210',
            'project_id': 123456789,
            'project_name': 'Project Name',
            'release_name': 'Product Name',
            'upc': '123456789012',
            'vendor_id': 12345,
            'subaccount_id': None,
            'track': {
                'bucket': 'test-bucket',
                'key': 'test_path/test_filename.wav',
                'filename': 'test_filename.wav',
                'ows_assets_filename': None,
                'isrc': 'ABCD12345678',
                'tuid': '76543210',
                'track_sequence_number': 1,
                'track_volume_number': 1,
                'track_name': 'Track Name 1'
            },
        },
        'asset_type': 'audio',
        'correlation_id': 'test_correlation_id',
        'execution_name': 'test_execution_name',
        'state_machine_name': 'test_sfn_name'
    }


@pytest.fixture
def test_load_product_response():
    """Create response from load_graphql_product."""
    return {
        'product_id': 1234567,
        'upc': '123456789012',
        'vendor_id': 12345,
        'subaccount_id': None,
        'release_name': 'Product Name',
        'project_code': '9876543210',
        'project_name': 'Project Name',
        'project_id': 123456789
    }