Armin Ronacher | ab45b84 | 2007-03-18 20:47:50 +0100 | [diff] [blame] | 1 | # -*- coding: utf-8 -*- |
| 2 | """ |
| 3 | conftest |
| 4 | ~~~~~~~~ |
| 5 | |
| 6 | Configure py.test for support stuff. |
| 7 | |
| 8 | :copyright: 2007 by Armin Ronacher, Alexander Schremmer. |
| 9 | :license: BSD, see LICENSE for more details. |
| 10 | """ |
| 11 | |
| 12 | import os |
| 13 | import sys |
| 14 | sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..')) |
| 15 | |
| 16 | import py |
Christoph Hack | e9e43bb | 2008-04-13 23:35:48 +0200 | [diff] [blame] | 17 | from jinja2 import Environment |
Armin Ronacher | f41d139 | 2008-04-18 16:41:52 +0200 | [diff] [blame] | 18 | from jinja2.loaders import BaseLoader |
| 19 | from jinja2.exceptions import TemplateNotFound |
Armin Ronacher | ab45b84 | 2007-03-18 20:47:50 +0100 | [diff] [blame] | 20 | |
Armin Ronacher | ccf284b | 2007-05-21 16:44:26 +0200 | [diff] [blame] | 21 | try: |
| 22 | # This code adds support for coverage.py (see |
| 23 | # http://nedbatchelder.com/code/modules/coverage.html). |
| 24 | # It prints a coverage report for the modules specified in all |
| 25 | # module globals (of the test modules) named "coverage_modules". |
| 26 | |
| 27 | import coverage, atexit |
| 28 | |
Christoph Hack | e9e43bb | 2008-04-13 23:35:48 +0200 | [diff] [blame] | 29 | IGNORED_MODULES = ['jinja2._speedups', 'jinja2.defaults', |
| 30 | 'jinja2.translators'] |
Armin Ronacher | ccf284b | 2007-05-21 16:44:26 +0200 | [diff] [blame] | 31 | |
| 32 | def report_coverage(): |
| 33 | coverage.stop() |
| 34 | module_list = [ |
| 35 | mod for name, mod in sys.modules.copy().iteritems() if |
| 36 | getattr(mod, '__file__', None) and |
Christoph Hack | e9e43bb | 2008-04-13 23:35:48 +0200 | [diff] [blame] | 37 | name.startswith('jinja2.') and |
Armin Ronacher | ccf284b | 2007-05-21 16:44:26 +0200 | [diff] [blame] | 38 | name not in IGNORED_MODULES |
| 39 | ] |
| 40 | module_list.sort() |
| 41 | coverage.report(module_list) |
| 42 | |
| 43 | def callback(option, opt_str, value, parser): |
| 44 | atexit.register(report_coverage) |
| 45 | coverage.erase() |
| 46 | coverage.start() |
| 47 | |
| 48 | py.test.config.addoptions('Test options', py.test.config.Option('-C', |
| 49 | '--coverage', action='callback', callback=callback, |
| 50 | help='Output information about code coverage (slow!)')) |
| 51 | |
| 52 | except ImportError: |
| 53 | coverage = None |
| 54 | |
Armin Ronacher | ab45b84 | 2007-03-18 20:47:50 +0100 | [diff] [blame] | 55 | |
Armin Ronacher | f41d139 | 2008-04-18 16:41:52 +0200 | [diff] [blame] | 56 | class GlobalLoader(BaseLoader): |
| 57 | scope = globals() |
Armin Ronacher | c1e3040 | 2007-03-18 22:41:09 +0100 | [diff] [blame] | 58 | |
Armin Ronacher | f41d139 | 2008-04-18 16:41:52 +0200 | [diff] [blame] | 59 | def get_source(self, environment, name): |
| 60 | try: |
| 61 | return self.scope[name.upper() + 'TEMPLATE'], None, None |
| 62 | except KeyError: |
| 63 | raise TemplateNotFound(name) |
Armin Ronacher | c1e3040 | 2007-03-18 22:41:09 +0100 | [diff] [blame] | 64 | |
| 65 | |
Armin Ronacher | f41d139 | 2008-04-18 16:41:52 +0200 | [diff] [blame] | 66 | loader = GlobalLoader(cache_size=0) |
Christoph Hack | e9e43bb | 2008-04-13 23:35:48 +0200 | [diff] [blame] | 67 | simple_env = Environment(trim_blocks=True, loader=loader) |
Armin Ronacher | ab45b84 | 2007-03-18 20:47:50 +0100 | [diff] [blame] | 68 | |
| 69 | |
| 70 | class Module(py.test.collect.Module): |
| 71 | |
| 72 | def __init__(self, *args, **kwargs): |
| 73 | self.env = simple_env |
| 74 | super(Module, self).__init__(*args, **kwargs) |
| 75 | |
Armin Ronacher | ccf284b | 2007-05-21 16:44:26 +0200 | [diff] [blame] | 76 | def makeitem(self, name, obj, usefilters=True): |
| 77 | if name.startswith('test_'): |
| 78 | if hasattr(obj, 'func_code'): |
| 79 | return JinjaTestFunction(name, parent=self) |
| 80 | elif isinstance(obj, basestring): |
| 81 | return JinjaDocTest(name, parent=self) |
Armin Ronacher | ab45b84 | 2007-03-18 20:47:50 +0100 | [diff] [blame] | 82 | |
| 83 | |
| 84 | class JinjaTestFunction(py.test.collect.Function): |
| 85 | |
| 86 | def execute(self, target, *args): |
Armin Ronacher | c1e3040 | 2007-03-18 22:41:09 +0100 | [diff] [blame] | 87 | loader.scope = target.func_globals |
Armin Ronacher | ab45b84 | 2007-03-18 20:47:50 +0100 | [diff] [blame] | 88 | co = target.func_code |
| 89 | if 'env' in co.co_varnames[:co.co_argcount]: |
| 90 | target(self.parent.env, *args) |
| 91 | else: |
| 92 | target(*args) |
Armin Ronacher | ccf284b | 2007-05-21 16:44:26 +0200 | [diff] [blame] | 93 | |
| 94 | |
| 95 | class JinjaDocTest(py.test.collect.Item): |
| 96 | |
| 97 | def run(self): |
| 98 | mod = py.std.types.ModuleType(self.name) |
| 99 | mod.__doc__ = self.obj |
| 100 | self.execute(mod) |
| 101 | |
| 102 | def execute(self, mod): |
| 103 | mod.env = self.parent.env |
| 104 | mod.MODULE = self.parent.obj |
| 105 | failed, tot = py.compat.doctest.testmod(mod, verbose=True) |
| 106 | if failed: |
| 107 | py.test.fail('doctest %s: %s failed out of %s' % ( |
| 108 | self.fspath, failed, tot)) |