stress-ng: fix teardown

Fix the teardown function to prevent AttributeError by adding checks to see if the attribute and path exists before trying to remove it. Before fix: ERROR: 'Stressng' object has no attribute 'loop_dev' ERROR: 'Stressng' object has no attribute 'stressmnt' After fix: CANCEL: Build Failed, Please check the build logs for details !! Signed-off-by: Disha Goel <[email protected]>
vaishnavibhat · Oct 23, 2024 · 8b8766c · 8b8766c
1 parent d752a5a
commit 8b8766c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/generic/stress-ng.py b/generic/stress-ng.py
@@ -198,11 +198,12 @@ def test(self):
                       "\n".join(ERROR))
 
     def tearDown(self):
-        if 'filesystem' in self.class_type:
+        if hasattr(self, 'loop_dev') and os.path.exists(self.loop_dev):
             process.run("umount %s" % self.loop_dev, ignore_status=True,
                         sudo=True)
             process.run("losetup -d %s" % self.loop_dev, ignore_status=True,
                         sudo=True)
+        if os.path.exists('/tmp/blockfile'):
             process.run("rm -rf /tmp/blockfile", ignore_status=True, sudo=True)
-            if (os.path.exists(self.stressmnt)):
-                process.run(f"rm -rf {self.stressmnt}")
+        if hasattr(self, 'stressmnt') and os.path.exists(self.stressmnt):
+            process.run(f"rm -rf {self.stressmnt}", ignore_status=True)