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'])