Add support for looking up into DictUnpack's value in Dict.getitem

2 files changed, 15 insertions, 0 deletions

diff --git a/astroid/node_classes.py b/astroid/node_classes.py
index 744d88f..ebe4fd8 100644
--- a/astroid/node_classes.py
+++ b/astroid/node_classes.py
@@ -810,6 +810,12 @@ class Dict(bases.NodeNG, bases.Instance):
 
     def getitem(self, lookup_key, context=None):
         for key, value in self.items:
+            # TODO(cpopa): no support for overriding yet, {1:2, **{1: 3}}.
+            if isinstance(key, DictUnpack):
+                try:
+                    return value.getitem(lookup_key, context)
+                except IndexError:
+                    continue
             for inferredkey in key.infer(context):
                 if inferredkey is util.YES:
                     continue
diff --git a/astroid/tests/unittest_python3.py b/astroid/tests/unittest_python3.py
index f98d220..6e08776 100644
--- a/astroid/tests/unittest_python3.py
+++ b/astroid/tests/unittest_python3.py
@@ -241,6 +241,15 @@ class Python3TC(unittest.TestCase):
         node = extract_node(code)
         self.assertEqual(node.as_string(), code)
 
+    @require_version('3.5')
+    def test_unpacking_in_dict_getitem(self):
+        node = extract_node('{1:2, **{2:3, 3:4}, **{5: 6}}')
+        for key, expected in ((1, 2), (2, 3), (3, 4), (5, 6)):
+            value = node.getitem(key)
+            self.assertIsInstance(value, nodes.Const)
+            self.assertEqual(value.value, expected)
+        
+
 
 if __name__ == '__main__':
     unittest.main()