Python: Expand setdefault tests

Author: RasmusWL

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

@@ -31,22 +31,27 @@ def SINK_F(x):
 # Actual tests
 # ------------------------------------------------------------------------------
 
-@expects(3) # $ unresolved_call=expects(..) unresolved_call=expects(..)(..)
+@expects(2) # $ unresolved_call=expects(..) unresolved_call=expects(..)(..)
 def test_dict_literal():
     d = {"key": SOURCE}
     SINK(d["key"])  # $ flow="SOURCE, l:-1 -> d['key']"
     SINK(d.get("key"))  # $ flow="SOURCE, l:-2 -> d.get(..)"
-    SINK(d.setdefault("key", NONSOURCE))  # $ flow="SOURCE, l:-3 -> d.setdefault(..)"
 
 
-@expects(3) # $ unresolved_call=expects(..) unresolved_call=expects(..)(..)
+@expects(2) # $ unresolved_call=expects(..) unresolved_call=expects(..)(..)
 def test_dict_update():
     d = {}
     d["key"] = SOURCE
     SINK(d["key"])  # $ flow="SOURCE, l:-1 -> d['key']"
     SINK(d.get("key"))  # $ flow="SOURCE, l:-2 -> d.get(..)"
-    SINK(d.setdefault("key", NONSOURCE))  # $ flow="SOURCE, l:-3 -> d.setdefault(..)"
 
+@expects(3) # $ unresolved_call=expects(..) unresolved_call=expects(..)(..)
+def test_setdefault():
+    d = {}
+    x = d.setdefault("key", SOURCE)
+    SINK(x)  # $ MISSING: flow="SOURCE, l:-1 -> d.setdefault(..)"
+    SINK(d["key"])  # $ flow="SOURCE, l:-2 -> d['key']"
+    SINK(d.setdefault("key", NONSOURCE))  # $ flow="SOURCE, l:-3 -> d.setdefault(..)"
 
 @expects(2) # $ unresolved_call=expects(..) unresolved_call=expects(..)(..)
 def test_dict_override():