kytos-ng / kytos

    assert KytosConfig().options["daemon"].api_traceback_on_500

async def test_get_json_or_400(controller, api_client) -> None:

    """Test get_json_or_400."""

    controller.loop = asyncio.get_running_loop()

    def handler(request: Request) -> JSONResponse:

        body = get_json_or_400(request, controller.loop)

        return JSONResponse(body)

    endpoint = "prefix/resource"

    controller.api_server.register_core_endpoint(endpoint, handler,

                                                 methods=["POST"])

    response = await api_client.post(f"kytos/core/{endpoint}")

    assert response.status_code == 400

    body = {"some_key": "some_value"}

    response = await api_client.post(f"kytos/core/{endpoint}", json=body)

    assert response.status_code == 200

    assert response.json() == body

async def test_error_msg():

    assert response.json() == "response"

async def test_aget_json_or_400(controller, api_client) -> None:

    """Test aget_json_or_400."""

    async def handler(request: Request) -> JSONResponse:

        body = await aget_json_or_400(request)

        return JSONResponse(body)

    endpoint = "prefix/resource"

    controller.api_server.register_core_endpoint(endpoint, handler,

                                                 methods=["POST"])

    response = await api_client.post(f"kytos/core/{endpoint}")

    assert response.status_code == 400

    body = {"some_key": "some_value"}

    response = await api_client.post(f"kytos/core/{endpoint}", json=body)

    assert response.status_code == 200

    assert response.json() == body

def test_api_500_traceback_by_default() -> None: