1 files changed, 14 insertions, 5 deletions
diff --git a/functest/ci/tier_handler.py b/functest/ci/tier_handler.py
index 1eadfba5..6b4864b5 100755
--- a/functest/ci/tier_handler.py
+++ b/functest/ci/tier_handler.py
@@ -28,7 +28,7 @@ def split_text(text, max_len):
     return lines
 
 
-class Tier:
+class Tier(object):
 
     def __init__(self, name, order, ci_loop, description=""):
         self.tests_array = []
@@ -102,14 +102,20 @@ class Tier:
         return out
 
 
-class TestCase:
+class TestCase(object):
 
-    def __init__(self, name, dependency, criteria, blocking, description=""):
+    def __init__(self, name,
+                 dependency,
+                 criteria,
+                 blocking,
+                 clean_flag,
+                 description=""):
         self.name = name
         self.dependency = dependency
-        self.description = description
         self.criteria = criteria
         self.blocking = blocking
+        self.clean_flag = clean_flag
+        self.description = description
 
     @staticmethod
     def is_none(item):
@@ -138,6 +144,9 @@ class TestCase:
     def is_blocking(self):
         return self.blocking
 
+    def needs_clean(self):
+        return self.clean_flag
+
     def __str__(self):
         lines = split_text(self.description, LINE_LENGTH - 6)
 
@@ -160,7 +169,7 @@ class TestCase:
         return out
 
 
-class Dependency:
+class Dependency(object):
 
     def __init__(self, installer, scenario):
         self.installer = installer