from http import HTTPStatus

import pytest
from connexion import AsyncApp


async def fails():
    return {"data": []}, HTTPStatus.CREATED


async def ok():
    return {"data": []}, 201


@pytest.fixture
def app():
    app = AsyncApp(__name__)
    app.add_api(
        validate_responses=True,
        specification={
            "openapi": "3.0.0",
            "info": {"title": "test", "version": "v1"},
            "paths": {
                "/fails": {
                    "get": {
                        "x-openapi-router-controller": __name__,
                        "operationId": "fails",
                        "responses": {
                            "201": {
                                "description": "ok",
                            },
                            "default": {
                                "description": "default",
                                "content": {
                                    "application/json": {
                                        "schema": {
                                            "type": "object",
                                            "properties": {
                                                "error_code": {"type": "integer"},
                                            },
                                            "required": ["error_code"],
                                        },
                                    }
                                },
                            },
                        },
                    }
                },
                "/ok": {
                    "get": {
                        "x-openapi-router-controller": __name__,
                        "operationId": "ok",
                        "responses": {
                            "201": {
                                "description": "ok",
                            }
                        },
                    }
                },
            },
        },
    )
    return app


def test_ok(app):
    with app.test_client() as client:
        res = client.get("/ok")
        assert res.status_code == HTTPStatus.CREATED


def test_fails(app):
    # SHOULD NOT FAIL
    with app.test_client() as client:
        res = client.get("/fails")
        assert res.status_code == HTTPStatus.CREATED