Add clear_envs to these tests

Change-Id: If7dd0d6d1a19a3c112c1e13d7d744b1423a4ce28

1 files changed, 32 insertions, 0 deletions

diff --git a/tests/test_version_traversal.py b/tests/test_version_traversal.py
index 16cb1a2..5d22d1c 100644
--- a/tests/test_version_traversal.py
+++ b/tests/test_version_traversal.py
@@ -360,6 +360,10 @@ class BranchFromMergepointTest(MigrationTest):
             splice=True,
         )
 
+    @classmethod
+    def teardown_class(cls):
+        clear_staging_env()
+
     def test_mergepoint_to_only_one_side_upgrade(self):
 
         self._assert_upgrade(
@@ -440,6 +444,10 @@ class BranchFrom3WayMergepointTest(MigrationTest):
             splice=True,
         )
 
+    @classmethod
+    def teardown_class(cls):
+        clear_staging_env()
+
     def test_mergepoint_to_only_one_side_upgrade(self):
 
         self._assert_upgrade(
@@ -533,6 +541,10 @@ class TwinMergeTest(MigrationTest):
 
         cls.d2 = env.generate_revision("d2", "d2", head=cls.c2.revision)
 
+    @classmethod
+    def teardown_class(cls):
+        clear_staging_env()
+
     def test_upgrade(self):
         head = HeadMaintainer(mock.Mock(), [self.a.revision])
 
@@ -592,6 +604,10 @@ class NotQuiteTwinMergeTest(MigrationTest):
 
         cls.d2 = env.generate_revision("d2", "d2", head=cls.c2.revision)
 
+    @classmethod
+    def teardown_class(cls):
+        clear_staging_env()
+
     def test_upgrade(self):
         head = HeadMaintainer(mock.Mock(), [self.a.revision])
 
@@ -650,6 +666,10 @@ class DependsOnBranchTestOne(MigrationTest):
             util.rev_id(), "e1->f1", head=cls.e1.revision
         )
 
+    @classmethod
+    def teardown_class(cls):
+        clear_staging_env()
+
     def test_downgrade_to_dependency(self):
         heads = [self.c2.revision, self.d1.revision]
         head = HeadMaintainer(mock.Mock(), heads)
@@ -728,6 +748,10 @@ class DependsOnBranchTestTwo(MigrationTest):
             depends_on=[cls.a3.revision, cls.b2.revision, cls.c1.revision],
         )
 
+    @classmethod
+    def teardown_class(cls):
+        clear_staging_env()
+
     def test_kaboom(self):
         # here's the upgrade path:
         # ['->c1', '->b2', '->a3', 'overmerge', '->c3', '->c2', 'cmerge',
@@ -837,6 +861,10 @@ class DependsOnBranchTestThree(MigrationTest):
             "a3", "->a3", head="a2", depends_on="b1"
         )
 
+    @classmethod
+    def teardown_class(cls):
+        clear_staging_env()
+
     def test_downgrade_over_crisscross(self):
         # this state was not possible prior to
         # #377.  a3 would be considered half of a merge point
@@ -882,6 +910,10 @@ class DependsOnBranchLabelTest(MigrationTest):
             util.rev_id(), "e1->f1", head=cls.e1.revision
         )
 
+    @classmethod
+    def teardown_class(cls):
+        clear_staging_env()
+
     def test_upgrade_path(self):
         self._assert_upgrade(
             self.c2.revision,