Armin Ronacher | 93e14c2 | 2007-03-20 16:17:48 +0100 | [diff] [blame] | 1 | # -*- coding: utf-8 -*- |
| 2 | """ |
| 3 | unit test for the loaders |
| 4 | ~~~~~~~~~~~~~~~~~~~~~~~~~ |
| 5 | |
| 6 | :copyright: 2007 by Armin Ronacher. |
| 7 | :license: BSD, see LICENSE for more details. |
| 8 | """ |
| 9 | |
Armin Ronacher | 963f97d | 2008-04-25 11:44:59 +0200 | [diff] [blame] | 10 | from py.test import raises |
Armin Ronacher | ccf284b | 2007-05-21 16:44:26 +0200 | [diff] [blame] | 11 | import time |
| 12 | import tempfile |
Christoph Hack | e9e43bb | 2008-04-13 23:35:48 +0200 | [diff] [blame] | 13 | from jinja2 import Environment, loaders |
| 14 | from jinja2.exceptions import TemplateNotFound |
Armin Ronacher | 93e14c2 | 2007-03-20 16:17:48 +0100 | [diff] [blame] | 15 | |
| 16 | |
| 17 | dict_loader = loaders.DictLoader({ |
| 18 | 'justdict.html': 'FOO' |
| 19 | }) |
| 20 | |
| 21 | package_loader = loaders.PackageLoader('loaderres', 'templates') |
| 22 | |
| 23 | filesystem_loader = loaders.FileSystemLoader('loaderres/templates') |
| 24 | |
Armin Ronacher | 9baa5ba | 2007-03-21 18:05:32 +0100 | [diff] [blame] | 25 | function_loader = loaders.FunctionLoader({'justfunction.html': 'FOO'}.get) |
| 26 | |
Armin Ronacher | 93e14c2 | 2007-03-20 16:17:48 +0100 | [diff] [blame] | 27 | choice_loader = loaders.ChoiceLoader([dict_loader, package_loader]) |
| 28 | |
| 29 | |
| 30 | def test_dict_loader(): |
| 31 | env = Environment(loader=dict_loader) |
| 32 | tmpl = env.get_template('justdict.html') |
| 33 | assert tmpl.render().strip() == 'FOO' |
| 34 | try: |
| 35 | env.get_template('missing.html') |
| 36 | except TemplateNotFound: |
| 37 | pass |
| 38 | else: |
| 39 | raise AssertionError('expected template exception') |
| 40 | |
| 41 | |
| 42 | def test_package_loader(): |
| 43 | env = Environment(loader=package_loader) |
Armin Ronacher | 963f97d | 2008-04-25 11:44:59 +0200 | [diff] [blame] | 44 | tmpl = env.get_template('test.html') |
| 45 | assert tmpl.render().strip() == 'BAR' |
| 46 | raises(TemplateNotFound, lambda: env.get_template('missing.html')) |
Armin Ronacher | 93e14c2 | 2007-03-20 16:17:48 +0100 | [diff] [blame] | 47 | |
| 48 | |
| 49 | def test_filesystem_loader(): |
| 50 | env = Environment(loader=filesystem_loader) |
| 51 | tmpl = env.get_template('test.html') |
| 52 | assert tmpl.render().strip() == 'BAR' |
Armin Ronacher | cb58db1 | 2007-05-30 00:05:24 +0200 | [diff] [blame] | 53 | tmpl = env.get_template('foo/test.html') |
| 54 | assert tmpl.render().strip() == 'FOO' |
Armin Ronacher | 963f97d | 2008-04-25 11:44:59 +0200 | [diff] [blame] | 55 | raises(TemplateNotFound, lambda: env.get_template('missing.html')) |
Armin Ronacher | 143a07e | 2007-06-02 01:47:34 +0200 | [diff] [blame] | 56 | |
| 57 | |
Armin Ronacher | 93e14c2 | 2007-03-20 16:17:48 +0100 | [diff] [blame] | 58 | def test_choice_loader(): |
| 59 | env = Environment(loader=choice_loader) |
| 60 | tmpl = env.get_template('justdict.html') |
| 61 | assert tmpl.render().strip() == 'FOO' |
| 62 | tmpl = env.get_template('test.html') |
| 63 | assert tmpl.render().strip() == 'BAR' |
| 64 | try: |
| 65 | env.get_template('missing.html') |
| 66 | except TemplateNotFound: |
| 67 | pass |
| 68 | else: |
| 69 | raise AssertionError('expected template exception') |
Armin Ronacher | 9baa5ba | 2007-03-21 18:05:32 +0100 | [diff] [blame] | 70 | |
| 71 | |
| 72 | def test_function_loader(): |
| 73 | env = Environment(loader=function_loader) |
| 74 | tmpl = env.get_template('justfunction.html') |
| 75 | assert tmpl.render().strip() == 'FOO' |
| 76 | try: |
| 77 | env.get_template('missing.html') |
| 78 | except TemplateNotFound: |
| 79 | pass |
| 80 | else: |
| 81 | raise AssertionError('expected template exception') |