From: grakiss <grakiss.wanglei@huawei.com>
Date: Thu, 31 Dec 2015 01:31:31 +0000 (+0800)
Subject: Update cinder volume quota for new tenant 'vIMS'
X-Git-Tag: 0.2~1848
X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=commitdiff_plain;h=3a65e3e4e78e00d9d46a0909b6713a726f9a9602;p=functest-xtesting.git

Update cinder volume quota for new tenant 'vIMS'

Change-Id: I26aad4f3989c89a62f66add85e0d88b7fd274425
Signed-off-by: grakiss <grakiss.wanglei@huawei.com>
---

diff --git a/testcases/functest_utils.py b/testcases/functest_utils.py
index 88c00858..fe0b3be7 100644
--- a/testcases/functest_utils.py
+++ b/testcases/functest_utils.py
@@ -321,6 +321,19 @@ def update_sg_quota(neutron_client, tenant_id, sg_quota, sg_rule_quota):
         print "Error:", sys.exc_info()[0]
         return False
 
+def update_cinder_quota(cinder_client, tenant_id, vols_quota, snapshots_quota,gigabytes_quota):
+    quotas_values = {
+            "volumes": vols_quota,
+            "snapshots": snapshots_quota,
+            "gigabytes": gigabytes_quota
+    }
+
+    try:
+        quotas_default=cinder_client.quotas.update(tenant_id,**quotas_values)
+        return True
+    except:
+        print "Error:", sys.exc_info()[0]
+        return False
 
 def get_private_net(neutron_client):
     # Checks if there is an existing private network
diff --git a/testcases/vIMS/CI/vIMS.py b/testcases/vIMS/CI/vIMS.py
index 12fb775a..5a3003fe 100644
--- a/testcases/vIMS/CI/vIMS.py
+++ b/testcases/vIMS/CI/vIMS.py
@@ -242,6 +242,18 @@ def main():
         logger.error("Failed to update security group quota for tenant %s" %TENANT_NAME)
         exit(-1)
 
+    logger.info("Update cinder quota for this tenant")
+    from cinderclient import client as cinderclient
+
+    creds_cinder = functest_utils.get_credentials("cinder")
+    cinder_client = cinderclient.Client('1',creds_cinder['username'],
+                                        creds_cinder['api_key'],
+                                        creds_cinder['project_id'],
+                                        creds_cinder['auth_url'],
+                                        service_type="volume")
+    if not functest_utils.update_cinder_quota(cinder_client,tenant_id,20,50,1500):
+        logger.error("Failed to update cinder quota for tenant %s" % TENANT_NAME)
+        exit(-1)
 
     ################ CLOUDIFY INITIALISATION ################