diff options
Diffstat (limited to 'websdk/werkzeug/testsuite/debug.py')
| -rw-r--r-- | websdk/werkzeug/testsuite/debug.py | 163 |
1 files changed, 163 insertions, 0 deletions
diff --git a/websdk/werkzeug/testsuite/debug.py b/websdk/werkzeug/testsuite/debug.py new file mode 100644 index 0000000..ffe9a2e --- /dev/null +++ b/websdk/werkzeug/testsuite/debug.py @@ -0,0 +1,163 @@ +# -*- coding: utf-8 -*- +""" + werkzeug.testsuite.debug + ~~~~~~~~~~~~~~~~~~~~~~~~ + + Tests some debug utilities. + + :copyright: (c) 2011 by Armin Ronacher. + :license: BSD, see LICENSE for more details. +""" +import unittest +import sys +import re + +from werkzeug.testsuite import WerkzeugTestCase + +from werkzeug.debug.repr import debug_repr, DebugReprGenerator, \ + dump, helper +from werkzeug.debug.console import HTMLStringO + + +class DebugReprTestCase(WerkzeugTestCase): + + def test_basic_repr(self): + assert debug_repr([]) == u'[]' + assert debug_repr([1, 2]) == \ + u'[<span class="number">1</span>, <span class="number">2</span>]' + assert debug_repr([1, 'test']) == \ + u'[<span class="number">1</span>, <span class="string">\'test\'</span>]' + assert debug_repr([None]) == \ + u'[<span class="object">None</span>]' + + def test_sequence_repr(self): + assert debug_repr(list(range(20))) == ( + u'[<span class="number">0</span>, <span class="number">1</span>, ' + u'<span class="number">2</span>, <span class="number">3</span>, ' + u'<span class="number">4</span>, <span class="number">5</span>, ' + u'<span class="number">6</span>, <span class="number">7</span>, ' + u'<span class="extended"><span class="number">8</span>, ' + u'<span class="number">9</span>, <span class="number">10</span>, ' + u'<span class="number">11</span>, <span class="number">12</span>, ' + u'<span class="number">13</span>, <span class="number">14</span>, ' + u'<span class="number">15</span>, <span class="number">16</span>, ' + u'<span class="number">17</span>, <span class="number">18</span>, ' + u'<span class="number">19</span></span>]' + ) + + def test_mapping_repr(self): + assert debug_repr({}) == u'{}' + assert debug_repr({'foo': 42}) == \ + u'{<span class="pair"><span class="key"><span class="string">\'foo\''\ + u'</span></span>: <span class="value"><span class="number">42' \ + u'</span></span></span>}' + assert debug_repr(dict(zip(range(10), [None] * 10))) == \ + u'{<span class="pair"><span class="key"><span class="number">0</span></span>: <span class="value"><span class="object">None</span></span></span>, <span class="pair"><span class="key"><span class="number">1</span></span>: <span class="value"><span class="object">None</span></span></span>, <span class="pair"><span class="key"><span class="number">2</span></span>: <span class="value"><span class="object">None</span></span></span>, <span class="pair"><span class="key"><span class="number">3</span></span>: <span class="value"><span class="object">None</span></span></span>, <span class="extended"><span class="pair"><span class="key"><span class="number">4</span></span>: <span class="value"><span class="object">None</span></span></span>, <span class="pair"><span class="key"><span class="number">5</span></span>: <span class="value"><span class="object">None</span></span></span>, <span class="pair"><span class="key"><span class="number">6</span></span>: <span class="value"><span class="object">None</span></span></span>, <span class="pair"><span class="key"><span class="number">7</span></span>: <span class="value"><span class="object">None</span></span></span>, <span class="pair"><span class="key"><span class="number">8</span></span>: <span class="value"><span class="object">None</span></span></span>, <span class="pair"><span class="key"><span class="number">9</span></span>: <span class="value"><span class="object">None</span></span></span></span>}' + assert debug_repr((1, 'zwei', u'drei')) ==\ + u'(<span class="number">1</span>, <span class="string">\'' \ + u'zwei\'</span>, <span class="string">u\'drei\'</span>)' + + def test_custom_repr(self): + class Foo(object): + def __repr__(self): + return '<Foo 42>' + assert debug_repr(Foo()) == '<span class="object"><Foo 42></span>' + + def test_list_subclass_repr(self): + class MyList(list): + pass + assert debug_repr(MyList([1, 2])) == \ + u'<span class="module">werkzeug.testsuite.debug.</span>MyList([' \ + u'<span class="number">1</span>, <span class="number">2</span>])' + + def test_regex_repr(self): + assert debug_repr(re.compile(r'foo\d')) == \ + u're.compile(<span class="string regex">r\'foo\\d\'</span>)' + assert debug_repr(re.compile(ur'foo\d')) == \ + u're.compile(<span class="string regex">ur\'foo\\d\'</span>)' + + def test_set_repr(self): + assert debug_repr(frozenset('x')) == \ + u'frozenset([<span class="string">\'x\'</span>])' + assert debug_repr(set('x')) == \ + u'set([<span class="string">\'x\'</span>])' + + def test_recursive_repr(self): + a = [1] + a.append(a) + assert debug_repr(a) == u'[<span class="number">1</span>, [...]]' + + def test_broken_repr(self): + class Foo(object): + def __repr__(self): + 1/0 + + assert debug_repr(Foo()) == \ + u'<span class="brokenrepr"><broken repr (ZeroDivisionError: ' \ + u'integer division or modulo by zero)></span>' + + +class DebugHelpersTestCase(WerkzeugTestCase): + + def test_object_dumping(self): + class Foo(object): + x = 42 + y = 23 + def __init__(self): + self.z = 15 + + drg = DebugReprGenerator() + out = drg.dump_object(Foo()) + assert re.search('Details for werkzeug.testsuite.debug.Foo object at', out) + assert re.search('<th>x.*<span class="number">42</span>(?s)', out) + assert re.search('<th>y.*<span class="number">23</span>(?s)', out) + assert re.search('<th>z.*<span class="number">15</span>(?s)', out) + + out = drg.dump_object({'x': 42, 'y': 23}) + assert re.search('Contents of', out) + assert re.search('<th>x.*<span class="number">42</span>(?s)', out) + assert re.search('<th>y.*<span class="number">23</span>(?s)', out) + + out = drg.dump_object({'x': 42, 'y': 23, 23: 11}) + assert not re.search('Contents of', out) + + out = drg.dump_locals({'x': 42, 'y': 23}) + assert re.search('Local variables in frame', out) + assert re.search('<th>x.*<span class="number">42</span>(?s)', out) + assert re.search('<th>y.*<span class="number">23</span>(?s)', out) + + def test_debug_dump(self): + old = sys.stdout + sys.stdout = HTMLStringO() + try: + dump([1, 2, 3]) + x = sys.stdout.reset() + dump() + y = sys.stdout.reset() + finally: + sys.stdout = old + + assert 'Details for list object at' in x + assert '<span class="number">1</span>' in x + assert 'Local variables in frame' in y + assert '<th>x' in y + assert '<th>old' in y + + def test_debug_help(self): + old = sys.stdout + sys.stdout = HTMLStringO() + try: + helper([1, 2, 3]) + x = sys.stdout.reset() + finally: + sys.stdout = old + + assert 'Help on list object' in x + assert '__delitem__' in x + + +def suite(): + suite = unittest.TestSuite() + suite.addTest(unittest.makeSuite(DebugReprTestCase)) + suite.addTest(unittest.makeSuite(DebugHelpersTestCase)) + return suite |