Improved async test

diff --git a/tests/test_async.py b/tests/test_async.py
index 02160403d0b3d75e9fa9575702a88957546d2a77..00dfffe13dbbfa2f9017b421af3ac802c2280103 100644 (file)
--- a/tests/test_async.py
+++ b/tests/test_async.py
@@ -44,11 +44,14 @@ def test_await_on_calls():
 
 @pytest.mark.skipif(not have_async_gen, reason='No async generators')
 def test_await_and_macros():
-    t = Template('{% macro foo(x) %}[{{ x }}]{% endmacro %}{{ foo(42) }}',
-                 enable_async=True)
+    t = Template('{% macro foo(x) %}[{{ x }}][{{ async_func() }}]'
+                 '{% endmacro %}{{ foo(42) }}', enable_async=True)
+
+    async def async_func():
+        return 42
 
     async def func():
-        return await t.render_async()
+        return await t.render_async(async_func=async_func)
 
     rv = run(func)
-    assert rv == '[42]'
+    assert rv == '[42][42]'