Python: Fix format for pandas.read_pickle

RasmusWL · RasmusWL · commit d1caa7505396 · 2023-09-25T10:24:27.000+02:00
diff --git a/python/ql/lib/semmle/python/frameworks/Pandas.qll b/python/ql/lib/semmle/python/frameworks/Pandas.qll
@@ -32,6 +32,6 @@ private module Pandas {
 
     override DataFlow::Node getOutput() { result = this }
 
-    override string getFormat() { result = "pandas" }
+    override string getFormat() { result = "pickle" }
   }
 }
diff --git a/python/ql/test/library-tests/frameworks/pandas/Decoding.py b/python/ql/test/library-tests/frameworks/pandas/Decoding.py
@@ -1,4 +1,4 @@
 import pandas
 
-pandas.read_pickle(file_)  # $ decodeInput=file_ decodeOutput=pandas.read_pickle(..) decodeFormat=pandas decodeMayExecuteInput
-pandas.read_pickle(filepath_or_buffer=file_)  # $ decodeInput=file_ decodeOutput=pandas.read_pickle(..) decodeFormat=pandas decodeMayExecuteInput
+pandas.read_pickle(file_)  # $ decodeInput=file_ decodeOutput=pandas.read_pickle(..) decodeFormat=pickle decodeMayExecuteInput
+pandas.read_pickle(filepath_or_buffer=file_)  # $ decodeInput=file_ decodeOutput=pandas.read_pickle(..) decodeFormat=pickle decodeMayExecuteInput

Original file line number	Diff line number	Diff line change
`@@ -32,6 +32,6 @@ private module Pandas {`
`32`	`32`
`33`	`33`	`override DataFlow::Node getOutput() { result = this }`
`34`	`34`
`35`		`- override string getFormat() { result = "pandas" }`
	`35`	`+ override string getFormat() { result = "pickle" }`
`36`	`36`	`}`
`37`	`37`	`}`