CLOUDSTACK-9005: Modifying tearDown function

Priti Sarap · Priti Sarap · commit d793c7f50ff5 · 2015-10-29T10:58:15.000+05:30
diff --git a/test/integration/testpaths/testpath_snapshot_limits.py b/test/integration/testpaths/testpath_snapshot_limits.py
@@ -123,24 +123,25 @@ def setUp(self):
                 "Snapshots are not supported on %s" %
                 self.hypervisor)
 
-
     def tearDown(self):
         try:
-            data_volumes_list = Volume.list(
-                self.userapiclient,
-                id=self.data_volume_created.id,
-                virtualmachineid=self.vm.id
-            )
-            status = validateList(data_volumes_list)
-            self.assertEqual(
-                status[0],
-                PASS,
-                "DATA Volume List Validation Failed")
-
-            self.vm.detach_volume(
-                self.userapiclient,
-                data_volumes_list[0]
-            )
+            if hasattr(self, "data_volume_created"):
+                data_volumes_list = Volume.list(
+                    self.userapiclient,
+                    id=self.data_volume_created.id,
+                    virtualmachineid=self.vm.id
+                )
+                status = validateList(data_volumes_list)
+                self.assertEqual(
+                    status[0],
+                    PASS,
+                    "DATA Volume List Validation Failed")
+
+            if data_volumes_list:
+                self.vm.detach_volume(
+                    self.userapiclient,
+                    data_volumes_list[0]
+                )
 
             cleanup_resources(self.apiclient, self.cleanup)
         except Exception as e:
