Fix a bug during generate clean up steps

Signed-off-by: Luyao Huang <[email protected]>
iccaszhulili · Aug 16, 2019 · 7ad22a8 · 7ad22a8
1 parent 5c1cb75
commit 7ad22a8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/depend_test_framework/case_generator.py b/depend_test_framework/case_generator.py
@@ -77,7 +77,8 @@ def gen_cases(self, test_func, random_cleanup=False, need_cleanup=False, src_env
                 continue
             cleanup_steps = None
             if need_cleanup:
-                cleanups = self.compute_route_permutations(src_env, tgt_env, True)
+                new_tgt_env = tgt_env.gen_transfer_env(test_func)
+                cleanups = self.compute_route_permutations(src_env, new_tgt_env, True)
                 if cleanups:
                     if random_cleanup:
                         cleanup_steps = random.choice(cleanups)