diff --git a/tests/unit/test_path_security.py b/tests/unit/test_path_security.py index 78bbebcaf..b388130c4 100755 --- a/tests/unit/test_path_security.py +++ b/tests/unit/test_path_security.py @@ -117,7 +117,8 @@ class TestSafeResolvePath(unittest.TestCase): def test_extension_whitelist_pass(self): """白名单内的扩展名应该通过.""" result = safe_resolve_path( - "test.mp4", self.tmpdir, + "test.mp4", + self.tmpdir, allowed_extensions={".mp4", ".mov"}, ) self.assertEqual(result.suffix.lower(), ".mp4") @@ -126,7 +127,8 @@ class TestSafeResolvePath(unittest.TestCase): """白名单外的扩展名应该被拒绝.""" with self.assertRaises(PathSecurityError): safe_resolve_path( - "test.exe", self.tmpdir, + "test.exe", + self.tmpdir, allowed_extensions={".mp4", ".mov"}, )