Python: Tests for field flow

python/ql/test/experimental/dataflow/fieldflow/dataflow.expected

@@ -0,0 +1,3 @@
+edges
+nodes
+#select

python/ql/test/experimental/dataflow/fieldflow/dataflow.ql

@@ -0,0 +1,10 @@
+/**
+ * @kind path-problem
+ */
+
+import experimental.dataflow.testConfig
+import DataFlow::PathGraph
+
+from TestConfiguration config, DataFlow::PathNode source, DataFlow::PathNode sink
+where config.hasFlowPath(source, sink)
+select sink.getNode(), source, sink, "<message>"

python/ql/test/experimental/dataflow/fieldflow/examples.py

@@ -0,0 +1,50 @@
+from python.ql.test.experimental.dataflow.testDefinitions import *
+
+# Preamble
+
+class MyObj(object):
+
+    def __init__(self, foo):
+        self.foo = foo
+
+
+class NestedObj(object):
+
+    def __init__(self):
+        self.obj = MyObj("OK")
+
+    def getObj(self):
+        return self.obj
+
+
+# Example 1
+def setFoo(obj, x):
+    SINK_F(obj.foo)
+    obj.foo = x
+
+myobj = MyObj("OK")
+
+setFoo(myobj, SOURCE)
+SINK(myobj.foo)
+
+# Example 2
+x = SOURCE
+
+a = NestedObj()
+
+a.obj.foo = x
+a.getObj().foo = x
+
+SINK(a.obj.foo)
+
+# Example 3
+obj = MyObj(SOURCE)
+SINK(obj.foo)
+
+# Local flow
+def fields_with_local_flow(x):
+  obj = MyObj(x)
+  a = obj.foo
+  return a
+
+SINK(fields_with_local_flow(SOURCE))

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

@@ -0,0 +1,56 @@
+from python.ql.test.experimental.dataflow.testDefinitions import *
+
+# Preamble
+
+
+class MyObj(object):
+
+    def __init__(self, foo):
+        self.foo = foo
+
+
+class NestedObj(object):
+
+    def __init__(self):
+        self.obj = MyObj("OK")
+
+    def getObj(self):
+        return self.obj
+
+
+def setFoo(obj, x):
+    SINK_F(obj.foo)
+    obj.foo = x
+
+
+def test_example1():
+    myobj = MyObj("OK")
+
+    setFoo(myobj, SOURCE)
+    SINK(myobj.foo)
+
+
+def test_example2():
+    x = SOURCE
+
+    a = NestedObj()
+
+    a.obj.foo = x
+    a.getObj().foo = x
+
+    SINK(a.obj.foo)
+
+
+def test_example3():
+    obj = MyObj(SOURCE)
+    SINK(obj.foo)
+
+
+def fields_with_local_flow(x):
+    obj = MyObj(x)
+    a = obj.foo
+    return a
+
+
+def test_fields():
+    SINK(fields_with_local_flow(SOURCE))