import pytest
from urllib import request
from zipfile import ZipFile


states = {}


@pytest.fixture(scope='session')
def r128_test_data(tmp_path_factory):
    '''Download and extract the test WAV files.'''

    # The latest data can be found here:
    # https://tech.ebu.ch/publications/ebu_loudness_test_set

    # I used to point to the official dataset hosted by the EBU, but they
    # either throttled me or are getting hammered by other requests. For now, I
    # am hosting a copy to get around the issue. If anyone has suggestions on
    # dataset hosting for integration tests, I'm all ears.
    url = 'https://jodh.us/export/ebu-loudness-test-setv05.zip'
    z = tmp_path_factory.getbasetemp() / 'ebu-loudness-test-setv05.zip'

    data = request.urlopen(url)
    with open(z, 'wb') as fp:
        fp.write(data.read())

    with ZipFile(z, 'r') as zp:
        zp.extractall(tmp_path_factory.getbasetemp())

    return tmp_path_factory.getbasetemp()


@pytest.fixture(scope='session')
def r128_states():
    return states