YQL-20381: Infer columns from indirect named node

commit_hash:05d53dee40119ad1d756d90775e6056fbebd99d2

1 files changed, 16 insertions, 0 deletions

diff --git a/yql/essentials/sql/v1/complete/sql_complete_ut.cpp b/yql/essentials/sql/v1/complete/sql_complete_ut.cpp
index ee19e89b23e..7af98365e36 100644
--- a/yql/essentials/sql/v1/complete/sql_complete_ut.cpp
+++ b/yql/essentials/sql/v1/complete/sql_complete_ut.cpp
@@ -1687,6 +1687,22 @@ Y_UNIT_TEST_SUITE(SqlCompleteTests) {
         UNIT_ASSERT_VALUES_EQUAL(CompleteTop(3, engine, input[2]), expected);
     }
 
+    Y_UNIT_TEST(ColumnFromIndirectNamedNode) {
+        auto engine = MakeSqlCompletionEngineUT();
+
+        TString query = R"sql(
+            $x = (SELECT 1 AS a);
+            $y = $x;
+            SELECT # FROM $y;
+        )sql";
+
+        TVector<TCandidate> expected = {
+            {ColumnName, "a"},
+        };
+
+        UNIT_ASSERT_VALUES_EQUAL(CompleteTop(expected.size(), engine, query), expected);
+    }
+
     Y_UNIT_TEST(ColumnQuoted) {
         auto engine = MakeSqlCompletionEngineUT();