summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--yardstick/tests/unit/benchmark/core/test_task.py9
1 files changed, 9 insertions, 0 deletions
diff --git a/yardstick/tests/unit/benchmark/core/test_task.py b/yardstick/tests/unit/benchmark/core/test_task.py
index 9e8e4e9f7..e7d2c70b7 100644
--- a/yardstick/tests/unit/benchmark/core/test_task.py
+++ b/yardstick/tests/unit/benchmark/core/test_task.py
@@ -357,6 +357,11 @@ key2:
}
}
+ @staticmethod
+ def _remove_context(context):
+ if context:
+ context._delete_context()
+
def test__change_node_names(self):
ctx_attrs = {
@@ -371,6 +376,7 @@ key2:
}
my_context = dummy.DummyContext()
+ self.addCleanup(self._remove_context, my_context)
my_context.init(ctx_attrs)
expected_scenario = {
@@ -413,6 +419,7 @@ key2:
}
my_context = dummy.DummyContext()
+ self.addCleanup(self._remove_context, my_context)
my_context.init(ctx_attrs)
scenario = copy.deepcopy(self.scenario)
@@ -428,6 +435,7 @@ key2:
}
my_context = dummy.DummyContext()
+ self.addCleanup(self._remove_context, my_context)
my_context.init(ctx_attrs)
scenario = copy.deepcopy(self.scenario)
scenario['options'] = None
@@ -442,6 +450,7 @@ key2:
}
my_context = dummy.DummyContext()
+ self.addCleanup(self._remove_context, my_context)
my_context.init(ctx_attrs)
scenario = copy.deepcopy(self.scenario)
scenario['options']['server_name'] = None