diff options
Diffstat (limited to 'tests/test_loader.py')
| -rw-r--r-- | tests/test_loader.py | 242 |
1 files changed, 124 insertions, 118 deletions
diff --git a/tests/test_loader.py b/tests/test_loader.py index 68c1384..44000e1 100644 --- a/tests/test_loader.py +++ b/tests/test_loader.py @@ -30,15 +30,15 @@ from jinja2.loaders import split_template_path class TestLoaders(object): def test_dict_loader(self, dict_loader): env = Environment(loader=dict_loader) - tmpl = env.get_template('justdict.html') - assert tmpl.render().strip() == 'FOO' - pytest.raises(TemplateNotFound, env.get_template, 'missing.html') + tmpl = env.get_template("justdict.html") + assert tmpl.render().strip() == "FOO" + pytest.raises(TemplateNotFound, env.get_template, "missing.html") def test_package_loader(self, package_loader): env = Environment(loader=package_loader) - tmpl = env.get_template('test.html') - assert tmpl.render().strip() == 'BAR' - pytest.raises(TemplateNotFound, env.get_template, 'missing.html') + tmpl = env.get_template("test.html") + assert tmpl.render().strip() == "BAR" + pytest.raises(TemplateNotFound, env.get_template, "missing.html") def test_filesystem_loader_overlapping_names(self, filesystem_loader): res = os.path.dirname(filesystem_loader.searchpath[0]) @@ -52,91 +52,94 @@ class TestLoaders(object): def test_choice_loader(self, choice_loader): env = Environment(loader=choice_loader) - tmpl = env.get_template('justdict.html') - assert tmpl.render().strip() == 'FOO' - tmpl = env.get_template('test.html') - assert tmpl.render().strip() == 'BAR' - pytest.raises(TemplateNotFound, env.get_template, 'missing.html') + tmpl = env.get_template("justdict.html") + assert tmpl.render().strip() == "FOO" + tmpl = env.get_template("test.html") + assert tmpl.render().strip() == "BAR" + pytest.raises(TemplateNotFound, env.get_template, "missing.html") def test_function_loader(self, function_loader): env = Environment(loader=function_loader) - tmpl = env.get_template('justfunction.html') - assert tmpl.render().strip() == 'FOO' - pytest.raises(TemplateNotFound, env.get_template, 'missing.html') + tmpl = env.get_template("justfunction.html") + assert tmpl.render().strip() == "FOO" + pytest.raises(TemplateNotFound, env.get_template, "missing.html") def test_prefix_loader(self, prefix_loader): env = Environment(loader=prefix_loader) - tmpl = env.get_template('a/test.html') - assert tmpl.render().strip() == 'BAR' - tmpl = env.get_template('b/justdict.html') - assert tmpl.render().strip() == 'FOO' - pytest.raises(TemplateNotFound, env.get_template, 'missing') + tmpl = env.get_template("a/test.html") + assert tmpl.render().strip() == "BAR" + tmpl = env.get_template("b/justdict.html") + assert tmpl.render().strip() == "FOO" + pytest.raises(TemplateNotFound, env.get_template, "missing") def test_caching(self): changed = False class TestLoader(loaders.BaseLoader): def get_source(self, environment, template): - return u'foo', None, lambda: not changed + return u"foo", None, lambda: not changed + env = Environment(loader=TestLoader(), cache_size=-1) - tmpl = env.get_template('template') - assert tmpl is env.get_template('template') + tmpl = env.get_template("template") + assert tmpl is env.get_template("template") changed = True - assert tmpl is not env.get_template('template') + assert tmpl is not env.get_template("template") changed = False def test_no_cache(self): - mapping = {'foo': 'one'} + mapping = {"foo": "one"} env = Environment(loader=loaders.DictLoader(mapping), cache_size=0) - assert env.get_template('foo') is not env.get_template('foo') + assert env.get_template("foo") is not env.get_template("foo") def test_limited_size_cache(self): - mapping = {'one': 'foo', 'two': 'bar', 'three': 'baz'} + mapping = {"one": "foo", "two": "bar", "three": "baz"} loader = loaders.DictLoader(mapping) env = Environment(loader=loader, cache_size=2) - t1 = env.get_template('one') - t2 = env.get_template('two') - assert t2 is env.get_template('two') - assert t1 is env.get_template('one') - t3 = env.get_template('three') + t1 = env.get_template("one") + t2 = env.get_template("two") + assert t2 is env.get_template("two") + assert t1 is env.get_template("one") + t3 = env.get_template("three") loader_ref = weakref.ref(loader) - assert (loader_ref, 'one') in env.cache - assert (loader_ref, 'two') not in env.cache - assert (loader_ref, 'three') in env.cache + assert (loader_ref, "one") in env.cache + assert (loader_ref, "two") not in env.cache + assert (loader_ref, "three") in env.cache def test_cache_loader_change(self): - loader1 = loaders.DictLoader({'foo': 'one'}) - loader2 = loaders.DictLoader({'foo': 'two'}) + loader1 = loaders.DictLoader({"foo": "one"}) + loader2 = loaders.DictLoader({"foo": "two"}) env = Environment(loader=loader1, cache_size=2) - assert env.get_template('foo').render() == 'one' + assert env.get_template("foo").render() == "one" env.loader = loader2 - assert env.get_template('foo').render() == 'two' + assert env.get_template("foo").render() == "two" def test_dict_loader_cache_invalidates(self): - mapping = {'foo': "one"} + mapping = {"foo": "one"} env = Environment(loader=loaders.DictLoader(mapping)) - assert env.get_template('foo').render() == "one" - mapping['foo'] = "two" - assert env.get_template('foo').render() == "two" + assert env.get_template("foo").render() == "one" + mapping["foo"] = "two" + assert env.get_template("foo").render() == "two" def test_split_template_path(self): - assert split_template_path('foo/bar') == ['foo', 'bar'] - assert split_template_path('./foo/bar') == ['foo', 'bar'] - pytest.raises(TemplateNotFound, split_template_path, '../foo') + assert split_template_path("foo/bar") == ["foo", "bar"] + assert split_template_path("./foo/bar") == ["foo", "bar"] + pytest.raises(TemplateNotFound, split_template_path, "../foo") @pytest.mark.loaders @pytest.mark.filesystemloader class TestFileSystemLoader(object): - searchpath = os.path.dirname(os.path.abspath(__file__)) + '/res/templates' + searchpath = os.path.join( + os.path.dirname(os.path.abspath(__file__)), "res", "templates" + ) @staticmethod def _test_common(env): - tmpl = env.get_template('test.html') - assert tmpl.render().strip() == 'BAR' - tmpl = env.get_template('foo/test.html') - assert tmpl.render().strip() == 'FOO' - pytest.raises(TemplateNotFound, env.get_template, 'missing.html') + tmpl = env.get_template("test.html") + assert tmpl.render().strip() == "BAR" + tmpl = env.get_template("foo/test.html") + assert tmpl.render().strip() == "FOO" + pytest.raises(TemplateNotFound, env.get_template, "missing.html") def test_searchpath_as_str(self): filesystem_loader = loaders.FileSystemLoader(self.searchpath) @@ -144,9 +147,10 @@ class TestFileSystemLoader(object): env = Environment(loader=filesystem_loader) self._test_common(env) - @pytest.mark.skipif(PY2, reason='pathlib is not available in Python 2') + @pytest.mark.skipif(PY2, reason="pathlib is not available in Python 2") def test_searchpath_as_pathlib(self): import pathlib + searchpath = pathlib.Path(self.searchpath) filesystem_loader = loaders.FileSystemLoader(searchpath) @@ -154,12 +158,13 @@ class TestFileSystemLoader(object): env = Environment(loader=filesystem_loader) self._test_common(env) - @pytest.mark.skipif(PY2, reason='pathlib is not available in Python 2') + @pytest.mark.skipif(PY2, reason="pathlib is not available in Python 2") def test_searchpath_as_list_including_pathlib(self): import pathlib + searchpath = pathlib.Path(self.searchpath) - filesystem_loader = loaders.FileSystemLoader(['/tmp/templates', searchpath]) + filesystem_loader = loaders.FileSystemLoader(["/tmp/templates", searchpath]) env = Environment(loader=filesystem_loader) self._test_common(env) @@ -168,26 +173,23 @@ class TestFileSystemLoader(object): filesystem_loader = loaders.FileSystemLoader(self.searchpath) env = Environment(loader=filesystem_loader) - tmpl1 = env.get_template('test.html') - tmpl2 = env.get_template('test.html') + tmpl1 = env.get_template("test.html") + tmpl2 = env.get_template("test.html") assert tmpl1 is tmpl2 - os.utime( - os.path.join(self.searchpath, "test.html"), - (time.time(), time.time()) - ) - tmpl3 = env.get_template('test.html') + os.utime(os.path.join(self.searchpath, "test.html"), (time.time(), time.time())) + tmpl3 = env.get_template("test.html") assert tmpl1 is not tmpl3 - @pytest.mark.parametrize('encoding, expected_text', [ - ('utf-8', u'tech'), - ('utf-16', u'整档'), - ]) - def test_uses_specified_encoding(self, encoding, expected_text): - filesystem_loader = loaders.FileSystemLoader(self.searchpath, encoding=encoding) - env = Environment(loader=filesystem_loader) - tmpl = env.get_template('variable_encoding.txt') - assert tmpl.render().strip() == expected_text + @pytest.mark.parametrize( + ("encoding", "expect"), + [("utf-8", u"文字化け"), ("iso-8859-1", u"æ\x96\x87å\xad\x97å\x8c\x96ã\x81\x91"),], + ) + def test_uses_specified_encoding(self, encoding, expect): + loader = loaders.FileSystemLoader(self.searchpath, encoding=encoding) + e = Environment(loader=loader) + t = e.get_template("mojibake.txt") + assert t.render() == expect @pytest.mark.loaders @@ -195,22 +197,22 @@ class TestFileSystemLoader(object): class TestModuleLoader(object): archive = None - def compile_down(self, prefix_loader, zip='deflated', py_compile=False): + def compile_down(self, prefix_loader, zip="deflated", py_compile=False): log = [] self.reg_env = Environment(loader=prefix_loader) if zip is not None: - fd, self.archive = tempfile.mkstemp(suffix='.zip') + fd, self.archive = tempfile.mkstemp(suffix=".zip") os.close(fd) else: self.archive = tempfile.mkdtemp() - self.reg_env.compile_templates(self.archive, zip=zip, - log_function=log.append, - py_compile=py_compile) + self.reg_env.compile_templates( + self.archive, zip=zip, log_function=log.append, py_compile=py_compile + ) self.mod_env = Environment(loader=loaders.ModuleLoader(self.archive)) - return ''.join(log) + return "".join(log) def teardown(self): - if hasattr(self, 'mod_env'): + if hasattr(self, "mod_env"): if os.path.isfile(self.archive): os.remove(self.archive) else: @@ -219,27 +221,31 @@ class TestModuleLoader(object): def test_log(self, prefix_loader): log = self.compile_down(prefix_loader) - assert 'Compiled "a/foo/test.html" as ' \ - 'tmpl_a790caf9d669e39ea4d280d597ec891c4ef0404a' in log - assert 'Finished compiling templates' in log - assert 'Could not compile "a/syntaxerror.html": ' \ - 'Encountered unknown tag \'endif\'' in log + assert ( + 'Compiled "a/foo/test.html" as ' + "tmpl_a790caf9d669e39ea4d280d597ec891c4ef0404a" in log + ) + assert "Finished compiling templates" in log + assert ( + 'Could not compile "a/syntaxerror.html": ' + "Encountered unknown tag 'endif'" in log + ) def _test_common(self): - tmpl1 = self.reg_env.get_template('a/test.html') - tmpl2 = self.mod_env.get_template('a/test.html') + tmpl1 = self.reg_env.get_template("a/test.html") + tmpl2 = self.mod_env.get_template("a/test.html") assert tmpl1.render() == tmpl2.render() - tmpl1 = self.reg_env.get_template('b/justdict.html') - tmpl2 = self.mod_env.get_template('b/justdict.html') + tmpl1 = self.reg_env.get_template("b/justdict.html") + tmpl2 = self.mod_env.get_template("b/justdict.html") assert tmpl1.render() == tmpl2.render() def test_deflated_zip_compile(self, prefix_loader): - self.compile_down(prefix_loader, zip='deflated') + self.compile_down(prefix_loader, zip="deflated") self._test_common() def test_stored_zip_compile(self, prefix_loader): - self.compile_down(prefix_loader, zip='stored') + self.compile_down(prefix_loader, zip="stored") self._test_common() def test_filesystem_compile(self, prefix_loader): @@ -248,8 +254,8 @@ class TestModuleLoader(object): def test_weak_references(self, prefix_loader): self.compile_down(prefix_loader) - tmpl = self.mod_env.get_template('a/test.html') - key = loaders.ModuleLoader.get_template_key('a/test.html') + tmpl = self.mod_env.get_template("a/test.html") + key = loaders.ModuleLoader.get_template_key("a/test.html") name = self.mod_env.loader.module.__name__ assert hasattr(self.mod_env.loader.module, key) @@ -260,6 +266,7 @@ class TestModuleLoader(object): try: import gc + gc.collect() except: pass @@ -268,65 +275,64 @@ class TestModuleLoader(object): # This test only makes sense on non-pypy python 2 @pytest.mark.skipif( - not (PY2 and not PYPY), - reason='This test only makes sense on non-pypy python 2') + not (PY2 and not PYPY), reason="This test only makes sense on non-pypy python 2" + ) def test_byte_compilation(self, prefix_loader): log = self.compile_down(prefix_loader, py_compile=True) assert 'Byte-compiled "a/test.html"' in log - tmpl1 = self.mod_env.get_template('a/test.html') - mod = self.mod_env.loader.module. \ - tmpl_3c4ddf650c1a73df961a6d3d2ce2752f1b8fd490 - assert mod.__file__.endswith('.pyc') + tmpl1 = self.mod_env.get_template("a/test.html") + mod = self.mod_env.loader.module.tmpl_3c4ddf650c1a73df961a6d3d2ce2752f1b8fd490 + assert mod.__file__.endswith(".pyc") def test_choice_loader(self, prefix_loader): log = self.compile_down(prefix_loader) - self.mod_env.loader = loaders.ChoiceLoader([ - self.mod_env.loader, - loaders.DictLoader({'DICT_SOURCE': 'DICT_TEMPLATE'}) - ]) + self.mod_env.loader = loaders.ChoiceLoader( + [self.mod_env.loader, loaders.DictLoader({"DICT_SOURCE": "DICT_TEMPLATE"})] + ) - tmpl1 = self.mod_env.get_template('a/test.html') - assert tmpl1.render() == 'BAR' - tmpl2 = self.mod_env.get_template('DICT_SOURCE') - assert tmpl2.render() == 'DICT_TEMPLATE' + tmpl1 = self.mod_env.get_template("a/test.html") + assert tmpl1.render() == "BAR" + tmpl2 = self.mod_env.get_template("DICT_SOURCE") + assert tmpl2.render() == "DICT_TEMPLATE" def test_prefix_loader(self, prefix_loader): log = self.compile_down(prefix_loader) - self.mod_env.loader = loaders.PrefixLoader({ - 'MOD': self.mod_env.loader, - 'DICT': loaders.DictLoader({'test.html': 'DICT_TEMPLATE'}) - }) + self.mod_env.loader = loaders.PrefixLoader( + { + "MOD": self.mod_env.loader, + "DICT": loaders.DictLoader({"test.html": "DICT_TEMPLATE"}), + } + ) - tmpl1 = self.mod_env.get_template('MOD/a/test.html') - assert tmpl1.render() == 'BAR' - tmpl2 = self.mod_env.get_template('DICT/test.html') - assert tmpl2.render() == 'DICT_TEMPLATE' + tmpl1 = self.mod_env.get_template("MOD/a/test.html") + assert tmpl1.render() == "BAR" + tmpl2 = self.mod_env.get_template("DICT/test.html") + assert tmpl2.render() == "DICT_TEMPLATE" - @pytest.mark.skipif(PY2, reason='pathlib is not available in Python 2') + @pytest.mark.skipif(PY2, reason="pathlib is not available in Python 2") def test_path_as_pathlib(self, prefix_loader): self.compile_down(prefix_loader) mod_path = self.mod_env.loader.module.__path__[0] import pathlib + mod_loader = loaders.ModuleLoader(pathlib.Path(mod_path)) self.mod_env = Environment(loader=mod_loader) self._test_common() - @pytest.mark.skipif(PY2, reason='pathlib is not available in Python 2') + @pytest.mark.skipif(PY2, reason="pathlib is not available in Python 2") def test_supports_pathlib_in_list_of_paths(self, prefix_loader): self.compile_down(prefix_loader) mod_path = self.mod_env.loader.module.__path__[0] import pathlib - mod_loader = loaders.ModuleLoader([ - pathlib.Path(mod_path), - '/tmp/templates' - ]) + + mod_loader = loaders.ModuleLoader([pathlib.Path(mod_path), "/tmp/templates"]) self.mod_env = Environment(loader=mod_loader) self._test_common() |