aboutsummaryrefslogtreecommitdiffstats
path: root/lib/python2.7/site-packages/Twisted-12.2.0-py2.7-linux-x86_64.egg/twisted/python/test/test_htmlizer.py
blob: 38e607a7be7cc0b7483b07f2fa1e07381e4051f4 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
# Copyright (c) Twisted Matrix Laboratories.
# See LICENSE for details.

"""
Tests for L{twisted.python.htmlizer}.
"""

from StringIO import StringIO

from twisted.trial.unittest import TestCase
from twisted.python.htmlizer import filter


class FilterTests(TestCase):
    """
    Tests for L{twisted.python.htmlizer.filter}.
    """
    def test_empty(self):
        """
        If passed an empty input file, L{filter} writes a I{pre} tag containing
        only an end marker to the output file.
        """
        input = StringIO("")
        output = StringIO()
        filter(input, output)
        self.assertEqual(output.getvalue(), '<pre><span class="py-src-endmarker"></span></pre>\n')


    def test_variable(self):
        """
        If passed an input file containing a variable access, L{filter} writes
        a I{pre} tag containing a I{py-src-variable} span containing the
        variable.
        """
        input = StringIO("foo\n")
        output = StringIO()
        filter(input, output)
        self.assertEqual(
            output.getvalue(),
            '<pre><span class="py-src-variable">foo</span><span class="py-src-newline">\n'
            '</span><span class="py-src-endmarker"></span></pre>\n')