Python: Update test file

tausbn · tausbn · commit 5974af661e1c · 2021-02-03T22:43:21.000+01:00
Makes the `a.b.c.d` test more sensible.

Also adds a test that shows a case where we're currently _not_ getting
the right flow.
diff --git a/python/ql/test/experimental/dataflow/ApiGraphs/test.py b/python/ql/test/experimental/dataflow/ApiGraphs/test.py
@@ -22,13 +22,46 @@
 
 abcd = abc.d #$ use=moduleImport("a").getMember("b").getMember("c").getMember("d")
 
-x5 = abcd() #$ use=moduleImport("a").getMember("b").getMember("c").getMember("d").getReturn()
+x5 = abcd.method() #$ use=moduleImport("a").getMember("b").getMember("c").getMember("d").getMember("method").getReturn()
 
-y5 = x5.method() #$ use=moduleImport("a").getMember("b").getMember("c").getMember("d").getReturn().getMember("method").getReturn()
+from a6 import m6 #$ use=moduleImport("a6").getMember("m6")
+
+x6 = m6().foo().bar() #$ use=moduleImport("a6").getMember("m6").getReturn().getMember("foo").getReturn().getMember("bar").getReturn()
 
 
 # Relative imports. These are ignored
 
 from .foo import bar
 
-from ..foobar import baz
+from ..foobar import baz
+
+
+# Use of imports across scopes
+
+def use_m4():
+    x = m4.blah4 #$ use=moduleImport("a4").getMember("b4").getMember("c4").getMember("blah4")
+
+def local_import_use():
+    from foo import bar #$ use=moduleImport("foo").getMember("bar")
+
+    x = bar() #$ use=moduleImport("foo").getMember("bar").getReturn()
+
+from eggs import ham as spam #$ use=moduleImport("eggs").getMember("ham")
+
+def bbb():
+    f = spam #$ use=moduleImport("eggs").getMember("ham")
+
+from danger import SOURCE #$ use=moduleImport("danger").getMember("SOURCE")
+
+foo = SOURCE #$ use=moduleImport("danger").getMember("SOURCE")
+
+def change_foo():
+    global foo
+    foo = SOURCE #$ use=moduleImport("danger").getMember("SOURCE")
+
+def f():
+    global foo
+    sink(foo) #$ use=moduleImport("danger").getMember("SOURCE")
+    foo = NONSOURCE
+    change_foo()
+    sink(foo) #$ MISSING: use=moduleImport("danger").getMember("SOURCE")
