"""Test xml splicing operations."""

import pytest

from src.main import select_packager


@pytest.mark.parametrize(
    'dms_id',
    [(1), (187), (399), (838), (1279), (1454)]
)
def test_transform(dms_id):
    """Test xml splicing for standard ddex38."""
    packager = select_packager(dms_id)
    fixtures_dir = 'tests/fixtures/transform_xml'
    input_filename = f'{fixtures_dir}/{dms_id}_input.xml'
    output_filename = f'{fixtures_dir}/{dms_id}_output.xml'
    with open(input_filename, 'r') as f:
        input_xml = f.read().rstrip()
    with open(output_filename, 'r') as f:
        output_xml = f.read().rstrip()
    stereo_metadata = {
        'productId': 4763731,
        'upc': '197190622188',
        'displayUpc': '197190622188',
        'notForDistribution': 'N',
        'tracks': [
            {'volumeNumber': 1, 'trackNumber': 1, 'isrc': 'INA3E2100726'},
            {'volumeNumber': 1, 'trackNumber': 2, 'isrc': 'INA3E2100728'}
        ]
    }
    spatial_metadata = {
        'productId': 4775852,
        'upc': '197190684056',
        'notForDistribution': 'NotforFurtherDistribution',
        'tracks': [
            {'volumeNumber': 1, 'trackNumber': 1, 'isrc': 'QMFMG2360977'}
        ]
    }
    spatial_audio_details = [
        (
            'qa-dolby-atmos-packager-alpha-input',
            '197190684056/197190684056_1_2_QMFMG2360978.wav',
            '2da8f8a950006ea6caa920bc62d607a7',
            136934
        )
    ]
    spatial_artwork_details = (
        '7c8400e5e5e0b9e6ddf0184be33353e0',
        300,
        500
    )

    result = packager.transform_xml(
        input_xml,
        stereo_metadata,
        spatial_metadata,
        spatial_audio_details,
        spatial_artwork_details
    )
    assert result == output_xml