30 lines
648 B
Python
30 lines
648 B
Python
import pytest
|
|
from drf_swagger import openapi, codecs
|
|
from drf_swagger.generators import OpenAPISchemaGenerator
|
|
from ruamel import yaml
|
|
|
|
|
|
@pytest.fixture
|
|
def generator():
|
|
return OpenAPISchemaGenerator(
|
|
info=openapi.Info("Test generator", "v1"),
|
|
version="v2",
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def codec_json():
|
|
return codecs.OpenAPICodecJson(['flex', 'ssv'])
|
|
|
|
|
|
@pytest.fixture
|
|
def codec_yaml():
|
|
return codecs.OpenAPICodecYaml(['ssv', 'flex'])
|
|
|
|
|
|
@pytest.fixture
|
|
def swagger_dict():
|
|
swagger = generator().get_schema(None, True)
|
|
json_bytes = codec_yaml().encode(swagger)
|
|
return yaml.safe_load(json_bytes.decode('utf-8'))
|