Added tests for raw_display.Screen and treetools.Tree*unittests-1

2 files changed, 140 insertions, 2 deletions

diff --git a/urwid/raw_display.py b/urwid/raw_display.py
index 9eadc0e..94c8080 100644
--- a/urwid/raw_display.py
+++ b/urwid/raw_display.py
@@ -539,6 +539,20 @@ class Screen(BaseScreen, RealTerminal):
         return ready
 
     def _getch(self, timeout):
+        """
+        Wait for character on _term_input_file.
+
+        >>> import os
+        >>> r, w = os.pipe()
+        >>> r = os.fdopen(r)
+        >>> w = os.fdopen(w, 'w')
+        >>> s = Screen(input=r)
+        >>> _ = w.write('x')
+        >>> w.close()
+        >>> chr(s._getch(1))
+        'x'
+
+        """
         ready = self._wait_for_input_ready(timeout)
         if self.gpm_mev is not None:
             if self.gpm_mev.stdout.fileno() in ready:
@@ -1020,6 +1034,32 @@ class Screen(BaseScreen, RealTerminal):
 
         0 <= index < 256 (some terminals will only have 16 or 88 colors)
         0 <= red, green, blue < 256
+
+        >>> import os
+        >>> r, w = os.pipe()
+        >>> r = os.fdopen(r)
+        >>> w = os.fdopen(w, 'w')
+        >>> s = Screen(output=w)
+        >>> s.modify_terminal_palette([
+        ...     (1, 10, 20, 30),
+        ...     (2, 40, 50, 60)
+        ... ])
+
+        >>> import os
+        >>> r, w = os.pipe()
+        >>> r = os.fdopen(r)
+        >>> w = os.fdopen(w, 'w')
+        >>> os.environ['TERM'] = 'fbterm'
+        >>> s = Screen(output=w)
+        >>> s.modify_terminal_palette([
+        ...     (1, 10, 20, 30),
+        ...     (2, 40, 50, 60)
+        ... ])
+        >>> s.write('foo\\n')
+        >>> w.close()
+        >>> r.read(128)
+        '\\x1b[3;1;10;20;30;2;40;50;60}foo\\n'
+
         """
 
         if self.term == 'fbterm':
@@ -1038,9 +1078,9 @@ class Screen(BaseScreen, RealTerminal):
     AttrSpec = lambda self, fg, bg: AttrSpec(fg, bg, self.colors)
 
 
-def _test():
+def _test():  # pragma: no cover
     import doctest
     doctest.testmod()
 
-if __name__=='__main__':
+if __name__=='__main__':  # pragma: no cover
     _test()
diff --git a/urwid/tests/test_treetools.py b/urwid/tests/test_treetools.py
new file mode 100644
index 0000000..1defde8
--- /dev/null
+++ b/urwid/tests/test_treetools.py
@@ -0,0 +1,98 @@
+# -*- coding: utf-8 -*-
+import unittest
+from textwrap import dedent
+
+from urwid.compat import B
+import urwid
+
+
+class ExampleTreeWidget(urwid.TreeWidget):
+    """ Display widget for leaf nodes """
+    def get_display_text(self):
+        return self.get_node().get_value()['name']
+
+
+class ExampleNode(urwid.TreeNode):
+    """ Data storage object for leaf nodes """
+    def load_widget(self):
+        return ExampleTreeWidget(self)
+
+
+class ExampleParentNode(urwid.ParentNode):
+    """ Data storage object for interior/parent nodes """
+    def load_widget(self):
+        return ExampleTreeWidget(self)
+
+    def load_child_keys(self):
+        data = self.get_value()
+        return range(len(data['children']))
+
+    def load_child_node(self, key):
+        """Return either an ExampleNode or ExampleParentNode"""
+        childdata = self.get_value()['children'][key]
+        childdepth = self.get_depth() + 1
+        if 'children' in childdata:
+            childclass = ExampleParentNode
+        else:
+            childclass = ExampleNode
+        return childclass(childdata, parent=self, key=key, depth=childdepth)
+
+
+class TreeToolsTest(unittest.TestCase):
+    def setUp(self):
+        self.data = dict(
+            name='Countries',
+            children=[
+                dict(
+                    name='Ukraine',
+                    children=[
+                        dict(
+                            name='Lviv'
+                        ),
+                        dict(
+                            name='Kyiv'
+                        )
+                    ]
+                ),
+                dict(
+                    name='United States of America',
+                    children=[
+                        dict(
+                            name='New York'
+                        ),
+                        dict(
+                            name='Los Angeles'
+                        )
+                    ]
+                )
+            ]
+        )
+        self.node = ExampleParentNode(self.data)
+        # self.tree = ExampleTreeWidget(self.node)
+        # self.tree.expanded = True
+        self.listbox = urwid.TreeListBox(urwid.TreeWalker(self.node))
+
+        self.expected_tree = dedent('''\
+            - Countries
+               - Ukraine
+                  Lviv
+                  Kyiv
+               - United States of America
+                  New York
+                  Los Angeles
+        ''')
+
+    def test_render_listbox(self):
+        canvas = self.listbox.render((30, 8))
+        lines = [
+            ''.join([
+                (block[2] if isinstance(block, tuple) else block).decode('utf-8')
+                for block
+                in line
+            ]).rstrip()
+            for line
+            in canvas.content()
+        ]
+        result = '\n'.join(lines)
+        self.assertEqual(result, self.expected_tree)
+