← Back to branch summary

~ubuntu-branches/ubuntu/natty/moin/natty-updates

« back to all changes in this revision

Viewing changes to MoinMoin/support/werkzeug/contrib/testtools.py

  • Committer: Bazaar Package Importer
  • Author(s): Jonas Smedegaard
  • Date: 2008-06-22 21:17:13 UTC
  • mto: This revision was merged to the branch mainline in revision 18.
  • Revision ID: james.westby@ubuntu.com-20080622211713-inlv5k4eifxckelr
Import upstream version 1.7.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
MoinMoin/support/werkzeug/contrib/testtools.py
1
 
# -*- coding: utf-8 -*-
2
 
"""
3
 
    werkzeug.contrib.testtools
4
 
    ~~~~~~~~~~~~~~~~~~~~~~~~~~
5
 
 
6
 
    This module implements extended wrappers for simplified testing.
7
 
 
8
 
    `TestResponse`
9
 
        A response wrapper which adds various cached attributes for
10
 
        simplified assertions on various content types.
11
 
 
12
 
    :copyright: (c) 2009 by the Werkzeug Team, see AUTHORS for more details.
13
 
    :license: BSD, see LICENSE for more details.
14
 
"""
15
 
from werkzeug import Response, cached_property, import_string
16
 
 
17
 
 
18
 
class ContentAccessors(object):
19
 
    """
20
 
    A mixin class for response objects that provides a couple of useful
21
 
    accessors for unittesting.
22
 
    """
23
 
 
24
 
    def xml(self):
25
 
        """Get an etree if possible."""
26
 
        if 'xml' not in self.mimetype:
27
 
            raise AttributeError(
28
 
                'Not a XML response (Content-Type: %s)'
29
 
                % self.mimetype)
30
 
        for module in ['xml.etree.ElementTree', 'ElementTree',
31
 
                       'elementtree.ElementTree']:
32
 
            etree = import_string(module, silent=True)
33
 
            if etree is not None:
34
 
                return etree.XML(self.body)
35
 
        raise RuntimeError('You must have ElementTree installed '
36
 
                           'to use TestResponse.xml')
37
 
    xml = cached_property(xml)
38
 
 
39
 
    def lxml(self):
40
 
        """Get an lxml etree if possible."""
41
 
        if ('html' not in self.mimetype and 'xml' not in self.mimetype):
42
 
            raise AttributeError('Not an HTML/XML response')
43
 
        from lxml import etree
44
 
        try:
45
 
            from lxml.html import fromstring
46
 
        except ImportError:
47
 
            fromstring = etree.HTML
48
 
        if self.mimetype=='text/html':
49
 
            return fromstring(self.data)
50
 
        return etree.XML(self.data)
51
 
    lxml = cached_property(lxml)
52
 
 
53
 
    def json(self):
54
 
        """Get the result of simplejson.loads if possible."""
55
 
        if 'json' not in self.mimetype:
56
 
            raise AttributeError('Not a JSON response')
57
 
        try:
58
 
            from simplejson import loads
59
 
        except:
60
 
            from json import loads
61
 
        return loads(self.data)
62
 
    json = cached_property(json)
63
 
 
64
 
 
65
 
class TestResponse(Response, ContentAccessors):
66
 
    """Pass this to `werkzeug.test.Client` for easier unittesting."""