class TestSearchWidget(unittest.TestCase):
def setUp(self):
self.search_func = mock.Mock()
self.clear_func = mock.Mock()
self.widget = SearchWidget(
self.search_func,
self.clear_func
)
def test_invoke_search(self):
self.widget.keypress((10,), "a")
self.widget.keypress((10,), "b")
self.widget.keypress((10,), "c")
self.widget.keypress((10,), "enter")
self.assertEqual(
self.widget.get_keyword(),
"abc")
self.search_func.assert_called_with("abc")
def test_invoke_clear(self):
self.widget.keypress((10,), "a")
self.widget.keypress((10,), "b")
self.widget.keypress((10,), "c")
self.widget.keypress((10,), "esc")
self.assertEqual(
self.widget.get_keyword(),
"")
self.clear_func.assert_called_with()
class DetailWidget(BasicWidget):
""" Display content for message
Attributes:
message: message generate by DataStore
displayer_context: DisplayerContext instance
index: view's index
"""
def __init__(self, message, displayer_context, index=0, **kwargs):
super(DetailWidget, self).__init__(**kwargs)
self.index = index
self.message = message
self.displayer_context = displayer_context
self.views = self.displayer_context.displayer.get_views()
self.name, view_callable = self.views[index]
self.view = view_callable.__call__(self.message)
self.content_widget = self.view.widget(
self.message, **kwargs)
_verify_keys(self.view.actions)
self.search_widget = SearchWidget(self._search, self._clear_search)
self.help_widget = HelpWidget(
HelpContent(
[HelpCategory("Basic", self.context.config.keys),
HelpCategory("Advanced", _ADVANCED_KEYS),
make_category_with_actions("Custom", self.view.actions)]),
**kwargs
)
if len(self.views) > 1:
header = Tabs([k for k, _ in self.views], self.index)
else:
header = None
widget = urwid.Frame(self.content_widget, header=header)
self.display(widget)
def _open(self, index):
self._clear_search()
try:
self.controller.open_view(DetailWidget(
self.message, self.displayer_context, index=index,
controller=self.controller, context=self.context),
push_prev=False)
except:
self.controller.open_error()
def _next_view(self):
if len(self.views) == 1:
return
if len(self.views) > self.index + 1:
next_index = self.index + 1
else:
next_index = 0
self._open(next_index)
def _prev_view(self):
if len(self.views) == 1:
return
next_index = len(self.views) - 1 if self.index == 0 else self.index - 1
self._open(next_index)
def _open_search(self):
self.search_widget.clear()
self.content_widget.clear_prev_search()
self.controller.open_edit(self.search_widget)
def _search(self, keyword):
self.controller._focus_body()
self.content_widget.search_next(keyword)
self.controller.close_edit()
def _clear_search(self):
self.controller.close_edit()
def _export(self): # pragma: no cover
file_name = "export-%13d" % (time.time() * 1000)
with open(file_name, "w") as f:
f.write(str(self.view))
self.controller.notify("Export to file {0}".format(file_name))
def update_info(self):
return self.controller._update_info(self, self.name)
def keypress(self, size, key):
if key == "q":
self._clear_search()
self.controller.back()
return None
if key == "tab":
self._next_view()
return None
if key == "shift tab":
self._prev_view()
return None
if key == "/":
self._open_search()
return None
if key == "n":
if self.search_widget.get_keyword():
self.content_widget.search_next(
self.search_widget.get_keyword()
)
return None
if key == "N":
if self.search_widget.get_keyword():
self.content_widget.search_prev(
self.search_widget.get_keyword()
)
return None
if key == "e": # pragma: no cover
self._export()
return None
if key == "?": # pragma: no cover
self.controller.open_view(self.help_widget)
return None
return super(DetailWidget, self).keypress(size, key) # pragma: no cover
