Python: Add some module initialization tests

Author: yoff

python/ql/test/experimental/dataflow/module-initialization/base.py

@@ -0,0 +1 @@
+foo = 3

python/ql/test/experimental/dataflow/module-initialization/m1.py

@@ -0,0 +1,10 @@
+# constant
+foo = 42
+
+import base
+
+def passOn(x):
+    return x
+
+# depends on other constant
+bar = passOn(base.foo)

python/ql/test/experimental/dataflow/module-initialization/test.py

@@ -0,0 +1,33 @@
+# These are defined so that we can evaluate the test code.
+NONSOURCE = "not a source"
+SOURCE = "source"
+
+
+def is_source(x):
+    return x == "source" or x == b"source" or x == 42 or x == 42.0 or x == 42j
+
+
+def SINK(x):
+    if is_source(x):
+        print("OK")
+    else:
+        print("Unexpected flow", x)
+
+
+def SINK_F(x):
+    if is_source(x):
+        print("Unexpected flow", x)
+    else:
+        print("OK")
+
+import base
+
+base.foo = 42
+
+import m1
+
+def test_const():
+    SINK(m1.foo)
+
+def test_overwritten():
+    SINK(m1.bar)

python/ql/test/experimental/dataflow/module-initialization/testOnce.py

@@ -0,0 +1,33 @@
+# These are defined so that we can evaluate the test code.
+NONSOURCE = "not a source"
+SOURCE = "source"
+
+
+def is_source(x):
+    return x == "source" or x == b"source" or x == 42 or x == 42.0 or x == 42j
+
+
+def SINK(x):
+    if is_source(x):
+        print("OK")
+    else:
+        print("Unexpected flow", x)
+
+
+def SINK_F(x):
+    if is_source(x):
+        print("Unexpected flow", x)
+    else:
+        print("OK")
+
+import m1
+
+import base
+
+base.foo = 42
+
+def test_const():
+    SINK(m1.foo)
+
+def test_unoverwritten():
+    SINK_F(m1.bar)

python/ql/test/experimental/dataflow/validTest.py

@@ -56,3 +56,6 @@ def check_tests_valid(testFile):
     check_tests_valid("variable-capture.in")
     check_tests_valid("variable-capture.nonlocal")
     check_tests_valid("variable-capture.dict")
+    # The below fails when trying to import modules
+    # check_tests_valid("module-initialization.test")
+    # check_tests_valid("module-initialization.testOnce")