def test_collapse_all_from_start_node(self):
"""
test listbox collapse all nested nodes when press key `c`.
"""
data = {
"key": {
"key": 'val'
}
}
node = self._node_factory.create_root_node(data)
# act
walker = JSONListWalker(start_from=node)
listbox = JSONListBox(walker)
size = (18, 5)
listbox.keypress(size, 'c')
content = [[t[2] for t in row] for row in
listbox.render(size).content()]
# verify
expected = [
[B("{\xe2\x80\xa6} ")],
[B(" ")],
[B(" ")],
[B(" ")],
[B(" ")]
]
self.assertEqual(expected, content)
def test_expand_all_from_start_node(self):
"""
test listbox expand all nested nodes when press key `e`.
"""
data = {
"key": {
"key": 'val'
}
}
node = self._node_factory.create_root_node(data)
# act
walker = JSONListWalker(start_from=node)
listbox = JSONListBox(walker)
listbox.keypress((18, 5), 'e')
content = [[t[2] for t in row] for row in
listbox.render((18, 5)).content()]
# verify
expected = [
[B("{ ")],
[B(" "), B('"key"'), B(": "), B("{ ")],
[B(" "), B('"key"'), B(": "), B('"val"')],
[B(" "), B("} ")],
[B("} ")]
]
self.assertEqual(expected, content)
def test_toggle_expanded_on_start_widget(self):
"""
test listbox collapse expandable widgets on start widget
"""
data = [
"item"
]
node = NodeFactory.create_node("", data, parent=None, display_key=False)
# act
walker = JSONListWalker(start_from=node)
listbox = JSONListBox(walker)
listbox.keypress((18, 18), 'enter')
collapse_content = [[t[2] for t in row] for row in listbox.render((18, 1)).content()]
# verify
expected_collapse_content = [[B("[\xe2\x80\xa6] ")]]
self.assertEqual(expected_collapse_content, collapse_content)
def test_toggle_expanded_on_start_widget(self):
"""
test listbox collapse expandable widgets on start widget
"""
data = [
'item'
]
node = self._node_factory.create_root_node(data)
# act
walker = JSONListWalker(start_from=node)
listbox = JSONListBox(walker)
listbox.keypress((18, 18), 'enter')
collapse_content = [[t[2] for t in row]
for row in listbox.render((18, 1)).content()]
# verify
expected_collapse_content = [[B('[\xe2\x80\xa6] ')]]
self.assertEqual(expected_collapse_content, collapse_content)
def test_expand_after_collapse_all(self):
"""
test listbox will collapse all nested nodes after keypress 'c'
"""
"""
test listbox collapse all nested nodes when press key `c`.
"""
data = [
{
"key": [
"1",
"2"
]
},
{
"key": [
"2"
]
}
]
node = self._node_factory.create_root_node(data)
# act
walker = JSONListWalker(start_from=node)
listbox = JSONListBox(walker)
size = (18, 13) # 13 rows after expanded in total
listbox.expand_all(size) # expand all first to expand all nodes
listbox.keypress(size, "c") # collapse all
listbox.toggle_collapse_on_focused_parent(size)
content = [[t[2] for t in row] for row in
listbox.render((18, 4)).content()]
# verify
expected = [
[B("[ ")],
[B(" "), B("{\xe2\x80\xa6} ")],
[B(" "), B("{\xe2\x80\xa6} ")],
[B("] ")]
]
self.assertEqual(expected, content)
def test_expand_all_from_middle_node(self):
"""
test listbox expand all nested nodes when press key `e` and focus is at
the middle of a node.
"""
data = [
{
"key": [
"1",
"2"
]
},
{
"key": [
"2"
]
}
]
node = self._node_factory.create_root_node(data)
# act
walker = JSONListWalker(start_from=node)
listbox = JSONListBox(walker)
size = (18, 13) # 13 rows after expanded in total
# move to focus to the second item, thus move down twice, as the view
# should be expanded at the first level
listbox.move_focus_to_next_line(size)
listbox.move_focus_to_next_line(size)
listbox.keypress(size, 'e')
content = [[t[2] for t in row] for row in
listbox.render(size).content()]
# verify
expected = [
[B("[ ")],
[B(" "), B("{ ")],
[B(" "), B('"key"'), B(": "), B("[ ")],
[B(" "), B('"1"'), B(' ')],
[B(" "), B('"2"'), B(' ')],
[B(" "), B("] ")],
[B(" "), B("} ")],
[B(" "), B("{ ")],
[B(" "), B('"key"'), B(": "), B("[ ")],
[B(" "), B('"2"'), B(' ')],
[B(" "), B("] ")],
[B(" "), B("} ")],
[B("] ")]
]
self.assertEqual(expected, content)
def test_list_box_with_simple_object(self):
"""
test listbox rendering simple object with collapse pressing key `enter`
and moving focus with pressing key `ctrl n`.
"""
data = {
'key': 'value'
}
node = self._node_factory.create_root_node(data)
# act
walker = JSONListWalker(start_from=node)
listbox = JSONListBox(walker)
contents_until_moving_to_end = []
prev_widget = None
cur_widget, cur_node = listbox.get_focus()
while prev_widget != cur_widget:
if not cur_node.is_expanded():
listbox.keypress((18, 18), 'enter')
cur_widget, cur_node = listbox.get_focus()
contents_until_moving_to_end.append(
cur_widget.render((18,)).content())
listbox.keypress((18, 18), 'down')
prev_widget = cur_widget
cur_widget, cur_node = listbox.get_focus()
texts_until_moving_to_end = [[[t[2] for t in row] for row in content]
for content in
contents_until_moving_to_end]
# moving up until the start and collapse all expanded widgets
contents_until_moving_to_start = []
prev_widget = None
cur_widget, cur_node = listbox.get_focus()
while prev_widget != cur_widget:
if cur_node.is_expanded():
listbox.keypress((18, 18), 'enter')
cur_widget, cur_node = listbox.get_focus()
listbox.keypress((18, 18), 'up')
prev_widget = cur_widget
cur_widget, cur_node = listbox.get_focus()
contents_until_moving_to_start.append(
listbox.render((18, 1)).content()
)
texts_until_moving_to_start = [[[t[2] for t in row] for row in content]
for content in
contents_until_moving_to_start]
# verify
# verify that after moving to the end, all the expandable lines have
# been expanded
self.assertEqual(3, len(texts_until_moving_to_end))
expected = [
[[B('{ ')]],
[[B(' '), B('"key"'), B(': '), B('"value"'), B(' ')]],
[[B('} ')]],
]
self.assertEqual(expected, texts_until_moving_to_end)
# verify that after moving to the start, all the expanded lines is
# collapsed
self.assertEqual(1, len(texts_until_moving_to_start))
expected = [[[B('{\xe2\x80\xa6} ')]]]
self.assertEqual(expected, texts_until_moving_to_start)
def test_collapse_all_from_middle_node(self):
"""
test listbox collapse all nested nodes when press key `c`.
"""
data = [
{
"key": [
"1",
"2"
]
},
{
"key": [
"2"
]
}
]
node = self._node_factory.create_root_node(data)
# act
walker = JSONListWalker(start_from=node)
listbox = JSONListBox(walker)
size = (18, 13) # 13 rows after expanded in total
# move to focus to the second item, thus move down twice, as the view
# should be expanded at the first level
listbox.move_focus_to_next_line(size)
listbox.move_focus_to_next_line(
size) # move to the second item in the list
listbox.toggle_collapse_on_focused_parent(
size) # expand the second item
listbox.move_focus_to_next_line(
size) # move to the first child of the object
listbox.toggle_collapse_on_focused_parent(size) # expand the object
listbox.move_focus_to_next_line(
size) # move to the first child of the object
listbox.keypress(size, 'c')
content = [[t[2] for t in row] for row in
listbox.render((18, 2)).content()]
# verify
expected = [
[B("[\xe2\x80\xa6] ")],
[B(" ")]
]
self.assertEqual(expected, content)
def test_list_box_with_simple_object(self):
"""
test listbox rendering simple object with collapse pressing key `enter` and
moving focus with pressing key `ctrl n`.
"""
data = {
"key": "value"
}
node = NodeFactory.create_node("", data, parent=None, display_key=False)
# act
walker = JSONListWalker(start_from=node)
listbox = JSONListBox(walker)
contents_until_moving_to_end = []
prev_widget, prev_node = None, None
cur_widget, cur_node = listbox.get_focus()
while prev_widget != cur_widget:
if not cur_node.is_expanded():
listbox.keypress((18, 18), "enter")
cur_widget, cur_node = listbox.get_focus()
contents_until_moving_to_end.append(cur_widget.render((18,)).content())
listbox.keypress((18, 18), "down")
prev_widget, prev_node = cur_widget, cur_node
cur_widget, cur_node = listbox.get_focus()
texts_until_moving_to_end = [[[t[2] for t in row] for row in content]
for content in contents_until_moving_to_end]
# moving up until the start and collapse all expanded widgets
contents_until_moving_to_start = []
prev_widget, prev_node = None, None
cur_widget, cur_node = listbox.get_focus()
while prev_widget != cur_widget:
if cur_node.is_expanded():
listbox.keypress((18, 18), "enter")
cur_widget, cur_node = listbox.get_focus()
listbox.keypress((18, 18), "up")
prev_widget, prev_node = cur_widget, cur_node
cur_widget, cur_node = listbox.get_focus()
contents_until_moving_to_start.append(listbox.render((18, 1)).content())
texts_until_moving_to_start = [[[t[2] for t in row] for row in content]
for content in contents_until_moving_to_start]
# verify
# verify that after moving to the end, all the expandable lines have been expanded
self.assertEqual(3, len(texts_until_moving_to_end))
expected = [
[[B("{ ")]],
[[B(" "), B("key"), B(": "), B("value ")]],
[[B("} ")]],
]
self.assertEqual(expected, texts_until_moving_to_end)
# verify that after moving to the start, all the expanded lines is collapsed
self.assertEqual(1, len(texts_until_moving_to_start))
expected = [[[B("{\xe2\x80\xa6} ")]]]
self.assertEqual(expected, texts_until_moving_to_start)