import pytest

from drf_yasg.codecs import yaml_sane_load


def _get_versioned_schema(prefix, client, validate_schema):
    response = client.get(prefix + 'swagger.yaml')
    assert response.status_code == 200
    swagger = yaml_sane_load(response.content.decode('utf-8'))
    validate_schema(swagger)
    assert prefix + 'snippets/' in swagger['paths']
    return swagger


def _check_v1(swagger, prefix):
    assert swagger['info']['version'] == '1.0'
    versioned_post = swagger['paths'][prefix + 'snippets/']['post']
    assert versioned_post['responses']['201']['schema']['$ref'] == '#/definitions/Snippet'
    assert 'v2field' not in swagger['definitions']['Snippet']['properties']


def _check_v2(swagger, prefix):
    assert swagger['info']['version'] == '2.0'
    versioned_post = swagger['paths'][prefix + 'snippets/']['post']
    assert versioned_post['responses']['201']['schema']['$ref'] == '#/definitions/SnippetV2'
    assert 'v2field' in swagger['definitions']['SnippetV2']['properties']
    v2field = swagger['definitions']['SnippetV2']['properties']['v2field']
    assert v2field['description'] == 'version 2.0 field'


@pytest.mark.urls('urlconfs.url_versioning')
def test_url_v1(client, validate_schema):
    prefix = '/versioned/url/v1.0/'
    swagger = _get_versioned_schema(prefix, client, validate_schema)
    _check_v1(swagger, prefix)


@pytest.mark.urls('urlconfs.url_versioning')
def test_url_v2(client, validate_schema):
    prefix = '/versioned/url/v2.0/'
    swagger = _get_versioned_schema(prefix, client, validate_schema)
    _check_v2(swagger, prefix)


@pytest.mark.urls('urlconfs.ns_versioning')
def test_ns_v1(client, validate_schema):
    prefix = '/versioned/ns/v1.0/'
    swagger = _get_versioned_schema(prefix, client, validate_schema)
    _check_v1(swagger, prefix)


@pytest.mark.urls('urlconfs.ns_versioning')
def test_ns_v2(client, validate_schema):
    prefix = '/versioned/ns/v2.0/'
    swagger = _get_versioned_schema(prefix, client, validate_schema)
    _check_v2(swagger, prefix)