def test_create_employee_exception(self, mock_validation, mock_db, mock_print, mock_input): mock_db().commit_data.side_effect = Exception admin = Admin() admin.create_employee() self.assertEqual(1, mock_print.call_count) self.assertEqual(1, mock_db().create_employee.call_count) mock_db().rollback_data.assert_called_once()
def test_create_employee(self, mock_validation, mock_db, mock_print, mock_input): mock_db().commit_data.side_effect = [sqlite3.IntegrityError, ''] admin = Admin() admin.create_employee() self.assertEqual(2, mock_print.call_count) self.assertEqual(2, mock_db().create_employee.call_count)