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()
