summaryrefslogtreecommitdiff
path: root/test/utils.py
blob: de9bc234a0b2a295a35faa59e0ede3901b855393 (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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
# copyright 2003-2011 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
#
# This file is part of logilab-common.
#
# logilab-common is free software: you can redistribute it and/or modify it under
# the terms of the GNU Lesser General Public License as published by the Free
# Software Foundation, either version 2.1 of the License, or (at your option) any
# later version.
#
# logilab-common is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
# details.
#
# You should have received a copy of the GNU Lesser General Public License along
# with logilab-common.  If not, see <http://www.gnu.org/licenses/>.
"""unit tests utilities for ureports
"""

from __future__ import print_function

import sys
from io import StringIO

buffers = [StringIO]
if sys.version_info < (3, 0):
    from cStringIO import StringIO as cStringIO
    from StringIO import StringIO as pStringIO

    buffers += [cStringIO, pStringIO]

from logilab.common.ureports.nodes import *


class WriterTC:
    def _test_output(self, test_id, layout, msg=None):
        for buffercls in buffers:
            buffer = buffercls()
            self.writer.format(layout, buffer)
            got = buffer.getvalue()
            expected = getattr(self, test_id)
            try:
                self.assertMultiLineEqual(got, expected)
            except:
                print("**** using a %s" % buffer.__class__)
                print("**** got for %s" % test_id)
                print(got)
                print("**** while expected")
                print(expected)
                print("****")
                raise

    def test_section(self):
        layout = Section("Section title", "Section's description.\nBlabla bla")
        self._test_output("section_base", layout)
        layout.append(Section("Subsection", "Sub section description"))
        self._test_output("section_nested", layout)

    def test_verbatim(self):
        layout = VerbatimText("blablabla")
        self._test_output("verbatim_base", layout)

    def test_list(self):
        layout = List(children=("item1", "item2", "item3", "item4"))
        self._test_output("list_base", layout)

    def test_nested_list(self):
        layout = List(
            children=(Paragraph(("blabla", List(children=("1", "2", "3")))), "an other point")
        )
        self._test_output("nested_list", layout)

    def test_table(self):
        layout = Table(cols=2, children=("head1", "head2", "cell1", "cell2"))
        self._test_output("table_base", layout)

    def test_field_table(self):
        table = Table(cols=2, klass="field", id="mytable")
        for field, value in (("f1", "v1"), ("f22", "v22"), ("f333", "v333")):
            table.append(Text(field))
            table.append(Text(value))
        self._test_output("field_table", table)

    def test_advanced_table(self):
        table = Table(cols=2, klass="whatever", id="mytable", rheaders=1)
        for field, value in (("field", "value"), ("f1", "v1"), ("f22", "v22"), ("f333", "v333")):
            table.append(Text(field))
            table.append(Text(value))
        table.append(Link("http://www.perdu.com", "toi perdu ?"))
        table.append(Text(""))
        self._test_output("advanced_table", table)


##     def test_image(self):
##         layout = Verbatim('blablabla')
##         self._test_output('verbatim_base', layout)