Python: Add some tests for pathlib

yoff · yoff · commit 848940305122 · 2021-04-15T09:40:23.000+02:00
diff --git a/python/ql/test/library-tests/frameworks/stdlib/FileSystemAccess.py b/python/ql/test/library-tests/frameworks/stdlib/FileSystemAccess.py
@@ -16,3 +16,18 @@
 
 io.open("filepath")  # $getAPathArgument="filepath"
 io.open(file="filepath")  # $getAPathArgument="filepath"
+
+from pathlib import Path, PosixPath, WindowsPath
+
+p = Path("filepath")
+posix = PosixPath("posix/filepath")
+windows = WindowsPath("windows/filepath")
+
+p.chmod(0o777)  # MISSING: $getAPathArgument=p
+posix.chmod(0o777)  # MISSING: $getAPathArgument=posix
+windows.chmod(0o777)  # MISSING: $getAPathArgument=windows
+
+with p.open() as f:  # MISSING: $getAPathArgument=p
+    f.read()
+
+p.write_bytes(b"hello")  # MISSING: $getAPathArgument=p
diff --git a/python/ql/test/library-tests/frameworks/stdlib/SafeAccessCheck.py b/python/ql/test/library-tests/frameworks/stdlib/SafeAccessCheck.py
@@ -3,6 +3,6 @@
 if s.startswith("tainted"):  # $checks=s branch=true
     pass
 
-sw = s.startswith  # $ MISSING: checks=s branch=true
-if sw("safe"):
+sw = s.startswith
+if sw("safe"):  # $ MISSING: checks=s branch=true
     pass
