Explicitely close another missing temporary directory

1 files changed, 21 insertions, 21 deletions

diff --git a/tests/test_commands.py b/tests/test_commands.py
index 9cd5c88..d52908a 100644
--- a/tests/test_commands.py
+++ b/tests/test_commands.py
@@ -145,27 +145,27 @@ class testCommandsSimple(unittest.TestCase, CommandsMixin):
         self.assertEqual(streams['stderr'].getvalue(), '')
 
     def test_export(self):
-        dest_tmpdir = tempfile.TemporaryDirectory()
-        utils.copytree(
-            'tests/data/simple',
-            dest_tmpdir.name,
-            dirs_exist_ok=True,
-        )
-        # TODO: make finding the templates less prone to breaking and
-        # then remove the cwd change from here
-        old_cwd = os.getcwd()
-        os.chdir(self.tmpdir.name)
-        args = {
-            'collection': self.tmpdir.name,
-            'git': True,
-            'template': 'templates/from_self.yaml',
-            'query': 'Another',
-            'destination': dest_tmpdir.name,
-        }
-        streams = self._run_command(command.Export(), args)
-        os.chdir(old_cwd)
-        self.assertEqual(streams['stdout'].getvalue(), '')
-        self.assertEqual(streams['stderr'].getvalue(), '')
+        with tempfile.TemporaryDirectory() as dest_tmpdir:
+            utils.copytree(
+                'tests/data/simple',
+                dest_tmpdir,
+                dirs_exist_ok=True,
+            )
+            # TODO: make finding the templates less prone to breaking and
+            # then remove the cwd change from here
+            old_cwd = os.getcwd()
+            os.chdir(self.tmpdir.name)
+            args = {
+                'collection': self.tmpdir.name,
+                'git': True,
+                'template': 'templates/from_self.yaml',
+                'query': 'Another',
+                'destination': dest_tmpdir,
+            }
+            streams = self._run_command(command.Export(), args)
+            os.chdir(old_cwd)
+            self.assertEqual(streams['stdout'].getvalue(), '')
+            self.assertEqual(streams['stderr'].getvalue(), '')
 
     def test_remove(self):
         args = {