from os import path

from slz_check_complete.dsp_complete_criteria import Service, load
from . import FIXTURES_PATH


def test_min_required_apple(logger_test):
    cfg = load(path.join(FIXTURES_PATH, 'apple.json'))
    srv = Service(logger_test, cfg)
    result = srv.min_required('apple', 'sme', 'amStreams')
    expected = set(['80026921', '80030905', '80031738'])
    assert expected == result


def test_required_apple(logger_test):
    cfg = load(path.join(FIXTURES_PATH, 'apple.json'))
    srv = Service(logger_test, cfg)
    result = srv.required('apple', 'sme', 'amStreams')
    expected = set(['80030469', '80030906', '80030905', '80031738', '80026921', '80026923'])
    assert expected == result


def test_min_required_spotify(logger_test):
    cfg = load(path.join(FIXTURES_PATH, 'spotify.json'))
    srv = Service(logger_test, cfg)
    result = srv.min_required('spotify', 'sme', 'streams')
    expected = set(['AL', 'AM', 'AE', 'AF'])
    assert expected == result


def test_required_spotify(logger_test):
    cfg = load(path.join(FIXTURES_PATH, 'spotify.json'))
    srv = Service(logger_test, cfg)
    result = srv.required('spotify', 'sme', 'streams')
    expected = set(['AI', 'AE', 'AM', 'AD', 'AG', 'AL', 'AF'])
    assert expected == result


def test_all_required(logger_test):
    cfg = load(path.join(FIXTURES_PATH, 'all_required.json'))
    srv = Service(logger_test, cfg)
    result = srv.min_required('apple', 'smejp', 'amContentDemographics')
    expected = set(['86780702', '86875390'])
    assert expected == result

    result_all = srv.required('apple', 'smejp', 'amContentDemographics')
    assert result_all == result


def test_no_required(logger_test):
    cfg = load(path.join(FIXTURES_PATH, 'no_required.json'))
    srv = Service(logger_test, cfg)
    result = srv.min_required('apple', 'smejp', 'amContentDemographics')
    expected = set([])
    assert expected == result

    result = srv.required('apple', 'smejp', 'amContentDemographics')
    expected = set(['86780702', '86875390'])
    assert expected == result


def test_empty_complete_criteria(logger_test):
    cfg = load(path.join(FIXTURES_PATH, 'no_required.json'))
    srv = Service(logger_test, cfg)
    result = srv.min_required('apple', 'smejp', 'amContentDemographics')
    expected = set([])
    assert expected == result