Python: Add list-comprehension taint test

RasmusWL · RasmusWL · commit e2d3fa7093cd · 2021-07-22T11:59:46.000+02:00
diff --git a/python/ql/test/experimental/dataflow/tainttracking/defaultAdditionalTaintStep/test_collections.py b/python/ql/test/experimental/dataflow/tainttracking/defaultAdditionalTaintStep/test_collections.py
@@ -52,6 +52,8 @@ def test_access(x, y, z):
         reversed(tainted_list), # $ tainted
         iter(tainted_list), # $ tainted
         next(iter(tainted_list)), # $ tainted
+        [i for i in tainted_list], # $ MISSING: tainted
+        [tainted_list for _i in [1,2,3]], # $ MISSING: tainted
     )
 
     a, b, c = tainted_list[0:3]

Original file line number	Diff line number	Diff line change
`@@ -52,6 +52,8 @@ def test_access(x, y, z):`
`52`	`52`	`reversed(tainted_list), # $ tainted`
`53`	`53`	`iter(tainted_list), # $ tainted`
`54`	`54`	`next(iter(tainted_list)), # $ tainted`
	`55`	`+ [i for i in tainted_list], # $ MISSING: tainted`
	`56`	`+ [tainted_list for _i in [1,2,3]], # $ MISSING: tainted`
`55`	`57`	`)`
`56`	`58`
`57`	`59`	`a, b, c = tainted_list[0:3]`