From: Armin Ronacher <armin.ronacher@active-4.com>
Date: Sun, 26 Jul 2009 09:48:41 +0000 (+0200)
Subject: Added unittests for meta.
X-Git-Tag: 2.2~10
X-Git-Url: http://git.tremily.us/?a=commitdiff_plain;h=61b5960d7fadd62103821baa6e6ed23faacc9a30;p=jinja2.git

Added unittests for meta.

--HG--
branch : trunk
---

diff --git a/tests/test_meta.py b/tests/test_meta.py
new file mode 100644
index 0000000..a8c600e
--- /dev/null
+++ b/tests/test_meta.py
@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+"""
+    unit test for the meta module
+    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+    :copyright: (c) 2009 by the Jinja Team.
+    :license: BSD, see LICENSE for more details.
+"""
+from jinja2 import Environment, meta
+
+
+def test_find_undeclared_variables():
+    env = Environment()
+    ast = env.parse('{% set foo = 42 %}{{ bar + foo }}')
+    x = meta.find_undeclared_variables(ast)
+    assert x == set(['bar'])
+
+    ast = env.parse('{% set foo = 42 %}{{ bar + foo }}'
+                    '{% macro meh(x) %}{{ x }}{% endmacro %}'
+                    '{% for item in seq %}{{ muh(item) + meh(seq) }}{% endfor %}')
+    x = meta.find_undeclared_variables(ast)
+    assert x == set(['bar', 'seq', 'muh'])
+
+
+def test_find_refererenced_templates():
+    env = Environment()
+    ast = env.parse('{% extends "layout.html" %}{% include helper %}')
+    i = meta.find_referenced_templates(ast)
+    assert i.next() == 'layout.html'
+    assert i.next() is None
+    assert list(i) == []
+
+    ast = env.parse('{% extends "layout.html" %}'
+                    '{% from "test.html" import a, b as c %}'
+                    '{% import "meh.html" as meh %}'
+                    '{% include "muh.html" %}')
+    i = meta.find_referenced_templates(ast)
+    assert list(i) == ['layout.html', 'test.html', 'meh.html', 'muh.html']