[CG-10829] fix: import wildcard doesn't preserve import resolution (#502)

Importing from a file that contains a wildcard can break further resolution. This could occur depending on to_resolve ordering, if the outer file is processed first _wildcards will not be filled in time.

---------

Co-authored-by: tomcodegen <[email protected]>
Co-authored-by: tomcodgen <[email protected]>

Author: 3 people

src/codegen/sdk/core/import_resolution.py

@@ -607,6 +607,8 @@ def names(self) -> Generator[tuple[str, Self | WildcardImport[Self]], None, None
                     imp.file.invalidate()
 
                 return
+            elif self.resolved_symbol is None:
+                self._resolving_wildcards = False
         yield self.name, self
 
     @property

tests/unit/codegen/sdk/python/import_resolution/test_import_resolution.py

@@ -341,3 +341,30 @@ def some_func():
         # Verify usages are detected
         assert len(some_func.usages) > 0
         assert len(some_func.symbol_usages) > 0
+
+
+def test_import_wildcard_preserves_import_resolution(tmpdir: str) -> None:
+    """Tests importing from a file that contains a wildcard import doesn't break further resolution.
+    This could occur depending on to_resolve ordering, if the outer file is processed first _wildcards will not be filled in time.
+    """
+    # language=python
+    with get_codebase_session(
+        tmpdir,
+        files={
+            "testdir/sub/file.py": """
+                test_const=5
+                b=2
+            """,
+            "testdir/file.py": """
+            from testdir.sub.file import *
+            c=b
+            """,
+            "file.py": """
+            from testdir.file import test_const
+            test = test_const
+            """,
+        },
+    ) as codebase:
+        mainfile: SourceFile = codebase.get_file("file.py")
+
+        assert len(mainfile.ctx.edges) == 5