X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=functest%2Fci%2Ftier_handler.py;h=6b4864b5bebfb148eb20c7dcb25b718d8544b860;hb=e6f875c4a756245c47211483247fe3674fd73834;hp=1eadfba50bee90498b749eaa9d9e02e2ab41db6e;hpb=d9cecf6576de1115252f78e78ff987f3a1ebb340;p=functest.git

diff --git a/functest/ci/tier_handler.py b/functest/ci/tier_handler.py
index 1eadfba50..6b4864b5b 100644
--- 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