fastapi-chameleon/tests/test_render.py

import asyncio
import os

import fastapi
# noinspection PyPackageRequirements
import pytest

import fastapi_chameleon as fc
from fastapi_chameleon.exceptions import FastAPIChameleonException

here = os.path.dirname(__file__)
folder = os.path.join(here, 'templates')


def test_cannot_decorate_missing_template():
    with pytest.raises(ValueError):
        @fc.template('home/missing.pt')
        def view_method():
            return {}

        view_method()


def test_requires_template_for_default_name():
    with pytest.raises(ValueError):
        @fc.template(None)
        def view_method():
            return {}

        view_method()


def test_default_template_name():
    @fc.template()
    def index(a, b, c):
        return {'a': a, 'b': b, 'c': c, 'world': 'WORLD'}

    resp = index(1, 2, 3)
    assert isinstance(resp, fastapi.Response)
    assert resp.status_code == 200
    html = resp.body.decode('utf-8')
    assert '<h1>Hello default WORLD!</h1>' in html


def test_can_decorate_dict_sync_method():
    @fc.template('home/index.pt')
    def view_method(a, b, c):
        return {'a': a, 'b': b, 'c': c}

    resp = view_method(1, 2, 3)
    assert isinstance(resp, fastapi.Response)
    assert resp.status_code == 200


def test_can_decorate_dict_async_method():
    @fc.template('home/index.pt')
    async def view_method(a, b, c):
        return {'a': a, 'b': b, 'c': c}

    resp = asyncio.run(view_method(1, 2, 3))
    assert isinstance(resp, fastapi.Response)
    assert resp.status_code == 200


def test_direct_response_pass_through():
    @fc.template('home/index.pt')
    def view_method(a, b, c):
        return fastapi.Response(content='abc', status_code=418)

    resp = view_method(1, 2, 3)
    assert isinstance(resp, fastapi.Response)
    assert resp.status_code == 418
    assert resp.body == b"abc"