Parcourir la source

tests: add some ctx managers for better cleanup

Thomas Waldmann il y a 1 mois
Parent
commit
4aa63a7866

+ 7 - 5
src/borg/testsuite/archiver/__init__.py

@@ -514,10 +514,12 @@ def fuse_mount(archiver, mountpoint=None, *options, fork=True, os_fork=False, **
             # with the call to `cmd`, above.
             yield
             return
-    wait_for_mountstate(mountpoint, mounted=True)
-    yield
-    umount(mountpoint)
-    wait_for_mountstate(mountpoint, mounted=False)
-    os.rmdir(mountpoint)
+    try:
+        wait_for_mountstate(mountpoint, mounted=True)
+        yield
+    finally:
+        umount(mountpoint)
+        wait_for_mountstate(mountpoint, mounted=False)
+        os.rmdir(mountpoint)
     # Give the daemon some time to exit
     time.sleep(0.2)

+ 5 - 5
src/borg/testsuite/platform/freebsd_test.py

@@ -86,11 +86,11 @@ def test_access_acl():
 
 @skipif_acls_not_working
 def test_default_acl():
-    tmpdir = tempfile.mkdtemp()
-    assert get_acl(tmpdir) == {}
-    set_acl(tmpdir, access=ACCESS_ACL, default=DEFAULT_ACL)
-    assert get_acl(tmpdir)["acl_access"] == ACCESS_ACL
-    assert get_acl(tmpdir)["acl_default"] == DEFAULT_ACL
+    with tempfile.TemporaryDirectory() as tmpdir:
+        assert get_acl(tmpdir) == {}
+        set_acl(tmpdir, access=ACCESS_ACL, default=DEFAULT_ACL)
+        assert get_acl(tmpdir)["acl_access"] == ACCESS_ACL
+        assert get_acl(tmpdir)["acl_default"] == DEFAULT_ACL
 
 
 # nfs4 acls testing not implemented.

+ 5 - 5
src/borg/testsuite/platform/linux_test.py

@@ -72,11 +72,11 @@ def test_access_acl():
 
 @skipif_acls_not_working
 def test_default_acl():
-    tmpdir = tempfile.mkdtemp()
-    assert get_acl(tmpdir) == {}
-    set_acl(tmpdir, access=ACCESS_ACL, default=DEFAULT_ACL)
-    assert get_acl(tmpdir)["acl_access"] == ACCESS_ACL
-    assert get_acl(tmpdir)["acl_default"] == DEFAULT_ACL
+    with tempfile.TemporaryDirectory() as tmpdir:
+        assert get_acl(tmpdir) == {}
+        set_acl(tmpdir, access=ACCESS_ACL, default=DEFAULT_ACL)
+        assert get_acl(tmpdir)["acl_access"] == ACCESS_ACL
+        assert get_acl(tmpdir)["acl_default"] == DEFAULT_ACL
 
 
 @skipif_acls_not_working