diff --git a/tools/marvin/marvin/lib/base.py b/tools/marvin/marvin/lib/base.py index 71aa3e67a67..6356914f13f 100755 --- a/tools/marvin/marvin/lib/base.py +++ b/tools/marvin/marvin/lib/base.py @@ -2637,6 +2637,22 @@ class StoragePool: cmd.id = self.id return apiclient.enableStorageMaintenance(cmd) + @classmethod + def enableMaintenance(cls, apiclient, id): + """enables maintenance mode Storage pool""" + + cmd = enableStorageMaintenance.enableStorageMaintenanceCmd() + cmd.id = id + return apiclient.enableStorageMaintenance(cmd) + + @classmethod + def cancelMaintenance(cls, apiclient, id): + """Cancels maintenance mode Storage pool""" + + cmd = cancelStorageMaintenance.cancelStorageMaintenanceCmd() + cmd.id = id + return apiclient.cancelStorageMaintenance(cmd) + @classmethod def list(cls, apiclient, **kwargs): """List all storage pools matching criteria"""