Fix a fixup bug for nested class references.

This has plagued me for weeks, but I could not repro it until now.
gitter-badger · Sep 28, 2016 · bec9454 · bec9454
1 parent 5f0d02c
commit bec9454
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 1 deletion.
diff --git a/mypy/fixup.py b/mypy/fixup.py
@@ -233,9 +233,9 @@ def lookup_qualified_stnode(modules: Dict[str, MypyFile], name: str) -> SymbolTa
     while True:
         assert '.' in head, "Cannot find %s" % (name,)
         head, tail = head.rsplit('.', 1)
+        rest.append(tail)
         mod = modules.get(head)
         if mod is not None:
-            rest.append(tail)
             break
     names = mod.names
     while True:

diff --git a/test-data/unit/check-incremental.test b/test-data/unit/check-incremental.test
@@ -1590,3 +1590,28 @@ pass
 [rechecked a, a.c]
 [stale a, a.c]
 [out]
+
+[case testIncrementalNestedClassRef]
+import top
+
+[file top.py]
+from funcs import callee
+from classes import Outer
+def caller(a: Outer.Inner) -> None:
+    callee(a)
+
+[file funcs.py]
+from classes import Outer
+def callee(a: Outer.Inner) -> None:
+    pass
+
+[file classes.py]
+class Outer:
+    class Inner:
+        pass
+
+[file top.py.next]
+from funcs import callee
+from classes import Outer
+def caller(a: Outer.Inner) -> int:
+    callee(a)