Update import map when new modules added (#20271)

ilevkivskyi · web-flow · commit a087a5894935 · 2025-11-20T11:58:01.000Z
Fixes #20209
diff --git a/mypy/server/update.py b/mypy/server/update.py
@@ -631,6 +631,8 @@ def restore(ids: list[str]) -> None:
 
     # Find any other modules brought in by imports.
     changed_modules = [(st.id, st.xpath) for st in new_modules]
+    for m in new_modules:
+        manager.import_map[m.id] = set(m.dependencies + m.suppressed)
 
     # If there are multiple modules to process, only process one of them and return
     # the remaining ones to the caller.
diff --git a/test-data/unit/fine-grained-modules.test b/test-data/unit/fine-grained-modules.test
@@ -2244,3 +2244,19 @@ undefined
 a.py:1: error: Cannot find implementation or library stub for module named "foobar"
 a.py:1: note: See https://mypy.readthedocs.io/en/stable/running_mypy.html#missing-imports
 ==
+
+[case testDaemonImportMapRefresh]
+# cmd: mypy main.py
+[file main.py]
+[file main.py.2]
+import a.b
+reveal_type(a.b.foo())
+[file a/__init__.pyi]
+[file a/b.pyi]
+import a.c
+def foo() -> a.c.C: ...
+[file a/c.pyi]
+class C: ...
+[out]
+==
+main.py:2: note: Revealed type is "a.c.C"
