fix(cubesql): Rollup doesn't work over aliased columns (#8334)

Author: waralexrom

packages/cubejs-backend-native/Cargo.lock

@@ -152,6 +152,9 @@
     "---------------------------------------",
     "SQL API: Simple Rollup",
     "SQL API: Complex Rollup",
-    "SQL API: Nested Rollup"
+    "SQL API: Nested Rollup",
+    "SQL API: Rollup with aliases",
+    "SQL API: Rollup over exprs",
+    "SQL API: Nested Rollup with aliases"
   ]
 }

packages/cubejs-testing-drivers/fixtures/athena.json

@@ -159,6 +159,7 @@
     "SKIPPED SQL API (Need work)",
     "---------------------------------------",
     "SQL API: reuse params",
-    "SQL API: Complex Rollup"
+    "SQL API: Complex Rollup",
+    "SQL API: Rollup with aliases"
   ]
 }

packages/cubejs-testing-drivers/fixtures/bigquery.json

@@ -160,6 +160,7 @@
 
     "SKIPPED SQL API (Need work)",
     "---------------------------------------",
-    "SQL API: Nested Rollup"
+    "SQL API: Nested Rollup",
+    "SQL API: Nested Rollup with aliases"
   ]
 }

packages/cubejs-testing-drivers/fixtures/databricks-jdbc.json

@@ -141,6 +141,9 @@
     "SQL API: powerbi min max push down",
     "SQL API: Simple Rollup",
     "SQL API: Complex Rollup",
-    "SQL API: Nested Rollup"
+    "SQL API: Nested Rollup",
+    "SQL API: Rollup with aliases",
+    "SQL API: Rollup over exprs",
+    "SQL API: Nested Rollup with aliases"
   ]
 }

packages/cubejs-testing-drivers/fixtures/mssql.json

@@ -133,6 +133,9 @@
     "SQL API: reuse params",
     "SQL API: Simple Rollup",
     "SQL API: Complex Rollup",
-    "SQL API: Nested Rollup"
+    "SQL API: Nested Rollup",
+    "SQL API: Rollup with aliases",
+    "SQL API: Rollup over exprs",
+    "SQL API: Nested Rollup with aliases"
   ]
 }

packages/cubejs-testing-drivers/fixtures/mysql.json

@@ -1706,6 +1706,34 @@ from
       expect(res.rows).toMatchSnapshot('complex_rollup');
     });
 
+    executePg('SQL API: Rollup with aliases', async (connection) => {
+      const res = await connection.query(`
+    select
+        rowId as "row", orderId as "order", orderDate as "orderData", city as "city", sum(count)
+    from
+      "ECommerce" as "ECommerce"
+    group by
+      ROLLUP(rowId, 2), 3, ROLLUP(4) 
+    order by 1, 2, 3, 4
+
+  `);
+      expect(res.rows).toMatchSnapshot('rollup_with_aliases');
+    });
+
+    executePg('SQL API: Rollup over exprs', async (connection) => {
+      const res = await connection.query(`
+    select
+        rowId + sales * 2 as "order", orderDate as "orderData", city as "city", sum(count)
+    from
+      "ECommerce" as "ECommerce"
+    group by
+      ROLLUP(1, 2, 3) 
+    order by 1, 2, 3
+
+  `);
+      expect(res.rows).toMatchSnapshot('rollup_over_exprs');
+    });
+
     executePg('SQL API: Nested Rollup', async (connection) => {
       const res = await connection.query(`
     select rowId, orderId, orderDate, sum(cnt)
@@ -1724,5 +1752,24 @@ from
   `);
       expect(res.rows).toMatchSnapshot('nested_rollup');
     });
+
+    executePg('SQL API: Nested Rollup with aliases', async (connection) => {
+      const res = await connection.query(`
+    select rowId as "row", orderId as "order", orderDate as "date", sum(cnt)
+    from (
+        select
+            rowId, orderId, orderDate, sum(count) as cnt
+        from
+        "ECommerce" as "ECommerce"
+        group by 1, 2, 3
+
+    ) a
+    group by
+      ROLLUP(1, 2, 3)
+    order by 1, 2, 3
+
+  `);
+      expect(res.rows).toMatchSnapshot('nested_rollup_with_aliases');
+    });
   });
 }