def test_delete_task(self):
"""After being deleted, search result for "John" should be 0"""
with patch('builtins.input', side_effect=['john', 'y', 'john']):
with patch('builtins.print') as mock_print:
# should find one match for "john"
employee_match = Search.by_name()
self.assertEqual(len(employee_match), 1)
# delete the task
work_log.delete_task(employee_match[0])
# length should now be zero
expected_output = Search.by_name()
self.assertEqual(len(expected_output), 0)
mock_print.assert_called_once_with('Task was deleted!')
def test_search_by_employee(self, MockInput):
"""Should be one entry found for the employee John"""
MockInput.side_effect = ['john']
expected_output = Search.by_name()
self.assertEqual(len(expected_output), 1)
self.assertEqual(expected_output[0].employee, self.task_2['employee'])
