import json

import pytest
from ruamel import yaml

from drf_yasg import openapi, codecs
from drf_yasg.generators import OpenAPISchemaGenerator


def test_schema_generates_without_errors(generator):
    generator.get_schema(None, True)


def test_schema_is_valid(generator, codec_yaml):
    swagger = generator.get_schema(request=None, public=False)
    codec_yaml.encode(swagger)


def test_invalid_schema_fails(codec_json):
    # noinspection PyTypeChecker
    bad_generator = OpenAPISchemaGenerator(
        info=openapi.Info(
            title="Test generator", default_version="v1",
            contact=openapi.Contact(name=69, email=[])
        ),
        version="v2",
    )

    swagger = bad_generator.get_schema(None, True)
    with pytest.raises(codecs.SwaggerValidationError):
        codec_json.encode(swagger)


def test_json_codec_roundtrip(codec_json, generator, validate_schema):
    swagger = generator.get_schema(None, True)
    json_bytes = codec_json.encode(swagger)
    validate_schema(json.loads(json_bytes.decode('utf-8')))


def test_yaml_codec_roundtrip(codec_yaml, generator, validate_schema):
    swagger = generator.get_schema(None, True)
    yaml_bytes = codec_yaml.encode(swagger)
    assert b'omap' not in yaml_bytes
    validate_schema(yaml.safe_load(yaml_bytes.decode('utf-8')))


def test_basepath_only():
    generator = OpenAPISchemaGenerator(
        info=openapi.Info(title="Test generator", default_version="v1"),
        version="v2",
        url='/basepath/',
    )

    swagger = generator.get_schema(None, public=True)
    assert 'host' not in swagger
    assert 'schemes' not in swagger
    assert swagger['basePath'] == '/'  # base path is not implemented for now
    assert swagger['info']['version'] == 'v2'