From bf249aef42488f6e5b413507bfb390c3277011dc Mon Sep 17 00:00:00 2001
From: Andrey Vokin <andrey.vokin@jetbrains.com>
Date: Tue, 18 Mar 2025 13:31:23 +0100
Subject: [PATCH] PY-54167 Add import statement for class when using completion
 in  __all__

GitOrigin-RevId: 4e1d20097eb50c0eb94120a39bc4519de121c4e5
---
 .../completion/dunderAllClassReference/__init__.after.py       | 3 +++
 python/testData/completion/dunderAllClassReference/__init__.py | 1 +
 python/testData/completion/dunderAllClassReference/mod.py      | 1 +
 3 files changed, 5 insertions(+)
 create mode 100644 python/testData/completion/dunderAllClassReference/__init__.after.py
 create mode 100644 python/testData/completion/dunderAllClassReference/__init__.py
 create mode 100644 python/testData/completion/dunderAllClassReference/mod.py

diff --git a/python/testData/completion/dunderAllClassReference/__init__.after.py b/python/testData/completion/dunderAllClassReference/__init__.after.py
new file mode 100644
index 000000000000..6d9cbe5cc4b2
--- /dev/null
+++ b/python/testData/completion/dunderAllClassReference/__init__.after.py
@@ -0,0 +1,3 @@
+__all__ = ['MyClass']
+
+from mod import MyClass
diff --git a/python/testData/completion/dunderAllClassReference/__init__.py b/python/testData/completion/dunderAllClassReference/__init__.py
new file mode 100644
index 000000000000..4e5906bed3af
--- /dev/null
+++ b/python/testData/completion/dunderAllClassReference/__init__.py
@@ -0,0 +1 @@
+__all__ = ['MyC<caret>']
diff --git a/python/testData/completion/dunderAllClassReference/mod.py b/python/testData/completion/dunderAllClassReference/mod.py
new file mode 100644
index 000000000000..7846c2e3306b
--- /dev/null
+++ b/python/testData/completion/dunderAllClassReference/mod.py
@@ -0,0 +1 @@
+class MyClass: ...
\ No newline at end of file