YQL-20379: Fix columns at subquery

Thet top-level source was always selected for
column inference, even when the cursor is at a
subquery.
commit_hash:c0eb00b98b23d52079aea5eb8d646887ff2fbbdd

1 files changed, 18 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 5ff13ad91f2..ee19e89b23e 100644
--- a/yql/essentials/sql/v1/complete/sql_complete_ut.cpp
+++ b/yql/essentials/sql/v1/complete/sql_complete_ut.cpp
@@ -1778,6 +1778,24 @@ Y_UNIT_TEST_SUITE(SqlCompleteTests) {
         UNIT_ASSERT_VALUES_EQUAL(CompleteTop(expected.size(), engine, queries[1]), expected);
     }
 
+    Y_UNIT_TEST(ColumnAtWhereInSubquery) {
+        auto engine = MakeSqlCompletionEngineUT();
+
+        TString query = R"sql(
+            SELECT * FROM (
+                SELECT a AS b
+                FROM (SELECT 1 AS a) AS x
+                WHERE x.#
+            );
+        )sql";
+
+        TVector<TCandidate> expected = {
+            {ColumnName, "a"},
+        };
+
+        UNIT_ASSERT_VALUES_EQUAL(CompleteTop(expected.size(), engine, query), expected);
+    }
+
     Y_UNIT_TEST(NoBindingAtQuoted) {
         auto engine = MakeSqlCompletionEngineUT();