import requests

from tests.integration import config


def test_lookup_company_brands_by_uuid():
    """POST /lookup/company-brands/uuids/."""
    response = requests.post(
        url='{qa_base_url}/lookup/company-brands/uuids/'.format(qa_base_url=config.QA_BASE_URL),
        json={
            'uuids': ['d25a4cd1-e820-45f2-be5c-56edcfeb8298'],
        },
    )
    assert response.status_code == 200
    response_json = response.json()
    assert len(response_json['company_brands']) == 1
    assert response_json['company_brands'][0] == {
        'company_brand_id': 2,
        'uuid': 'd25a4cd1-e820-45f2-be5c-56edcfeb8298',
        'parent_company_uuid': '955a1bbd-b623-4ea1-ab5f-8d6620c442fb',
    }


def test_lookup_company_brands_by_uuid_only_returns_existent_cases():
    """POST /lookup/company-brands/uuids/ UUID does not exist."""
    response = requests.post(
        url='{qa_base_url}/lookup/company-brands/uuids/'.format(qa_base_url=config.QA_BASE_URL),
        json={
            'uuids': [
                '00000000-0000-0000-0000-000000000000',
                'd25a4cd1-e820-45f2-be5c-56edcfeb8298',
            ]
        },
    )
    assert response.status_code == 200
    assert response.json() == {
        'company_brands': [
            None,
            {
                'company_brand_id': 2,
                'uuid': 'd25a4cd1-e820-45f2-be5c-56edcfeb8298',
                'parent_company_uuid': '955a1bbd-b623-4ea1-ab5f-8d6620c442fb',
            },
        ]
    }


def test_lookup_company_brands_by_uuid_empty_list():
    """POST /lookup/company-brands/uuids/ with empty list."""
    response = requests.post(
        url='{qa_base_url}/lookup/company-brands/uuids/'.format(qa_base_url=config.QA_BASE_URL),
        json={'uuids': []},
    )
    assert response.status_code == 400
    assert response.json()['code'] == 'invalid_input'


def test_lookup_company_brands_by_ids():
    """POST /lookup/company-brands/company-brand-ids/."""
    response = requests.post(
        url='{qa_base_url}/lookup/company-brands/company-brand-ids/'.format(
            qa_base_url=config.QA_BASE_URL
        ),
        json={
            'company_brand_ids': [2],
        },
    )
    assert response.status_code == 200
    response_json = response.json()
    assert len(response_json['company_brands']) == 1
    assert response_json['company_brands'][0] == {
        'company_brand_id': 2,
        'uuid': 'd25a4cd1-e820-45f2-be5c-56edcfeb8298',
        'parent_company_uuid': '955a1bbd-b623-4ea1-ab5f-8d6620c442fb',
    }


def test_lookup_company_brands_by_ids_only_returns_existent_cases():
    """POST /lookup/company-brands/company-brand-ids/ company_brand_id does not exist."""
    response = requests.post(
        url='{qa_base_url}/lookup/company-brands/company-brand-ids/'.format(
            qa_base_url=config.QA_BASE_URL
        ),
        json={
            'company_brand_ids': [999999, 2],
        },
    )
    assert response.status_code == 200
    assert response.json() == {
        'company_brands': [
            None,
            {
                'company_brand_id': 2,
                'uuid': 'd25a4cd1-e820-45f2-be5c-56edcfeb8298',
                'parent_company_uuid': '955a1bbd-b623-4ea1-ab5f-8d6620c442fb',
            },
        ]
    }


def test_lookup_company_brands_by_ids_empty_list():
    """POST /lookup/company-brands/company-brand-ids/ with empty list."""
    response = requests.post(
        url='{qa_base_url}/lookup/company-brands/company-brand-ids/'.format(
            qa_base_url=config.QA_BASE_URL
        ),
        json={'company_brand_ids': []},
    )
    assert response.status_code == 400
    assert response.json()['code'] == 'invalid_input'