From 4223c519dd8c8a09eb577171a2e9075b01e6ede4 Mon Sep 17 00:00:00 2001
From: Sowmya Krishnan <sowmya.krishnan@citrix.com>
Date: Mon, 17 Aug 2015 15:51:48 +0530
Subject: [PATCH] CLOUDSTACK-8738 - Added two methods which were missed out
 earlier in base.py for Storage Pool maintenance

---
 tools/marvin/marvin/lib/base.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

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"""