def test_menu_loop_search_employee(self):
user_input = ['p', 'q']
with patch('builtins.input', side_effect=user_input):
with mock.patch('log.prep_employee_search') as mock_prep:
with mock.patch('log.search_employee') as mock_search:
log.menu_loop()
mock_prep.assert_called()
mock_search.assert_called()
def test_menu_loop_search_range(self):
user_input = ['r', 'q']
with patch('builtins.input', side_effect=user_input):
with mock.patch('log.search_range') as mock_range:
log.menu_loop()
mock_range.assert_called()
def test_menu_loop_search_mins(self):
user_input = ['m', 'q']
with patch('builtins.input', side_effect=user_input):
with mock.patch('log.search_minutes') as mock_min:
log.menu_loop()
mock_min.assert_called()
def test_menu_loop_view(self):
user_input = ['v', 'q']
with patch('builtins.input', side_effect=user_input):
with mock.patch('log.view_entries') as mock_view:
log.menu_loop()
mock_view.assert_called()
def test_menu_loop_add(self):
user_input = ['a', 'q']
with patch('builtins.input', side_effect=user_input):
with mock.patch('log.add_entry') as mock_add:
log.menu_loop()
mock_add.assert_called()