diff --git a/tests/test_bmap_helpers.py b/tests/test_bmap_helpers.py index f224731..2cefad6 100644 --- a/tests/test_bmap_helpers.py +++ b/tests/test_bmap_helpers.py @@ -64,13 +64,13 @@ def test_get_file_system_type_symlink(self): """Check a file system type is returned when used with a symlink""" with TemporaryDirectory(prefix="testdir_", dir=".") as directory: - fobj = tempfile.NamedTemporaryFile( + with tempfile.NamedTemporaryFile( "r", prefix="testfile_", delete=False, dir=directory, suffix=".img" - ) - lnk = os.path.join(directory, "test_symlink") - os.symlink(fobj.name, lnk) - fstype = BmapHelpers.get_file_system_type(lnk) - self.assertTrue(fstype) + ) as fobj: + lnk = os.path.join(directory, "test_symlink") + os.symlink(fobj.name, lnk) + fstype = BmapHelpers.get_file_system_type(lnk) + self.assertTrue(fstype) def test_is_zfs_configuration_compatible_enabled(self): """Check compatibility check is true when zfs param is set correctly"""