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)