56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
try:
|
|
import uasyncio as asyncio
|
|
except ImportError:
|
|
import asyncio
|
|
|
|
import unittest
|
|
from microdot import Microdot, Request
|
|
from microdot_asyncio import Microdot as MicrodotAsync, Request as RequestAsync
|
|
from microdot_utemplate import render_template, init_templates
|
|
from tests.mock_socket import get_request_fd, get_async_request_fd
|
|
|
|
init_templates('tests/templates')
|
|
|
|
|
|
def _run(coro):
|
|
return asyncio.get_event_loop().run_until_complete(coro)
|
|
|
|
|
|
class TestUTemplate(unittest.TestCase):
|
|
def test_render_template(self):
|
|
s = list(render_template('hello.utemplate.txt', name='foo'))
|
|
self.assertEqual(s, ['Hello, ', 'foo', '!\n'])
|
|
|
|
def test_render_template_in_app(self):
|
|
app = Microdot()
|
|
|
|
@app.route('/')
|
|
def index(req):
|
|
return render_template('hello.utemplate.txt', name='foo')
|
|
|
|
req = Request.create(app, get_request_fd('GET', '/'), 'addr')
|
|
res = app.dispatch_request(req)
|
|
self.assertEqual(res.status_code, 200)
|
|
self.assertEqual(list(res.body_iter()), ['Hello, ', 'foo', '!\n'])
|
|
|
|
def test_render_template_in_app_async(self):
|
|
app = MicrodotAsync()
|
|
|
|
@app.route('/')
|
|
async def index(req):
|
|
return render_template('hello.utemplate.txt', name='foo')
|
|
|
|
req = _run(RequestAsync.create(
|
|
app, get_async_request_fd('GET', '/'), 'writer', 'addr'))
|
|
res = _run(app.dispatch_request(req))
|
|
self.assertEqual(res.status_code, 200)
|
|
|
|
async def get_result():
|
|
result = []
|
|
async for chunk in res.body_iter():
|
|
result.append(chunk)
|
|
return result
|
|
|
|
result = _run(get_result())
|
|
self.assertEqual(result, ['Hello, ', 'foo', '!\n'])
|