From 1e4695b9aa5e42f26b84b47349fd81e567deb7e9 Mon Sep 17 00:00:00 2001
From: sailajamada <sailaja.mada@citrix.com>
Date: Fri, 18 Jul 2014 09:55:22 +0530
Subject: [PATCH] Fixing vGPU BVT testcase to skip the test on unsupported
 hypervisors

---
 .../smoke/test_deploy_vgpu_enabled_vm.py      | 45 ++++++++++++++++---
 1 file changed, 38 insertions(+), 7 deletions(-)

diff --git a/test/integration/smoke/test_deploy_vgpu_enabled_vm.py b/test/integration/smoke/test_deploy_vgpu_enabled_vm.py
index 2188c7f2165..aca597142da 100644
--- a/test/integration/smoke/test_deploy_vgpu_enabled_vm.py
+++ b/test/integration/smoke/test_deploy_vgpu_enabled_vm.py
@@ -42,14 +42,45 @@ class TestDeployvGPUenabledVM(cloudstackTestCase):
     Test deploy a vGPU enabled VM into a user account
     """
     @classmethod
-    def setUpClass(cls):
-        testClient = super(TestDeployvGPUenabledVM, cls).getClsTestClient()
+    def setUpClass(self):
+        testClient = super(TestDeployvGPUenabledVM, self).getClsTestClient()
+        self.apiclient = testClient.getApiClient()
+        self.testdata = self.testClient.getParsedTestDataConfig()
         #Need to add check whether zone containing the xen hypervisor or not as well
-        hypervisor = testClient.getHypervisorInfo()
-        if hypervisor.lower() != XEN_SERVER.lower():
-            raise unittest.skipTest("GPU feature is supported only on XenServer")
-
+        hosts = list_hosts(
+               self.apiclient,
+               hypervisor="XenServer"
+               )
+        if hosts is None:
+            raise unittest.SkipTest("There are no XenServers available. GPU feature is supported only on XenServer.Check listhosts response")
+        else: 
+             gpuhosts=0
+             for ghost in hosts :
+                    if ghost.hypervisorversion >= "6.2.0":
+                       sshClient = SshClient(host=ghost.ipaddress, port=22, user='root',passwd=self.testdata["xen_host_password"]) 
+                       if ghost.hypervisorversion == "6.2.0":
+                          res = sshClient.execute("xe patch-list uuid=0850b186-4d47-11e3-a720-001b2151a503")
+                          if len(res) == 0:
+                              continue
+                       res = sshClient.execute("xe vgpu-type-list model-name=\"GRID K120Q\"")
+                       if len(res) != 0 :
+                           gpuhosts=gpuhosts+1 
+                       else:        
+                           continue
+        if gpuhosts == 0:
+           raise unittest.SkipTest("No XenServer available with GPU Drivers installed")
 
+        self.domain = get_domain(self.apiclient)
+        self.zone = get_zone(self.apiclient, self.testClient.getZoneForTests())
+        #Creating Account 
+        self.account = Account.create(
+            self.apiclient,
+            self.testdata["account"],
+            domainid=self.domain.id
+            )
+        self._cleanup = [
+                         self.account
+                        ]
     def setUp(self):
         self.testdata = self.testClient.getParsedTestDataConfig()["vgpu"]
         self.apiclient = self.testClient.getApiClient()
@@ -75,7 +106,7 @@ class TestDeployvGPUenabledVM(cloudstackTestCase):
 
         self.testdata["vgpu140q"]["zoneid"] = self.zone.id
         self.testdata["vgpu140q"]["template"] = self.template.id
-        self.testdata["service_offerings"]["vgpu260qwin"]["serviceofferingdetails"] = [{'pciDevice': 'VGPU'},
+        self.testdata["service_offerings"]["vgpu260qwin"]["serviceofferingdetails"] = [{'pciDevice': 'Group of NVIDIA Corporation GK107GL [GRID K1] GPUs'},
                                                                                        {'vgpuType':'GRID K120Q'}]
         #create a service offering
         self.service_offering = ServiceOffering.create(