Add test for aggregating BigQuery view more than once

ebyhr · ebyhr · commit 152abe9c1651 · 2021-01-25T14:35:09.000+09:00
diff --git a/plugin/trino-bigquery/src/test/java/io/trino/plugin/bigquery/TestBigQueryIntegrationSmokeTest.java b/plugin/trino-bigquery/src/test/java/io/trino/plugin/bigquery/TestBigQueryIntegrationSmokeTest.java
@@ -28,6 +28,7 @@
 import static io.trino.spi.type.VarcharType.VARCHAR;
 import static io.trino.testing.MaterializedResult.resultBuilder;
 import static io.trino.testing.assertions.Assert.assertEquals;
+import static io.trino.testing.sql.TestTable.randomTableSuffix;
 import static java.lang.String.format;
 import static org.assertj.core.api.Assertions.assertThat;
 
@@ -133,6 +134,25 @@ public void testCountAggregationView()
                 "VALUES (1)");
     }
 
+    /**
+     * regression test for https://github.com/trinodb/trino/issues/6696
+     */
+    @Test
+    public void testRepeatCountAggregationView()
+    {
+        BigQuery client = createBigQueryClient();
+
+        String viewName = "test.repeat_count_aggregation_view_" + randomTableSuffix();
+
+        executeBigQuerySql(client, "DROP VIEW IF EXISTS " + viewName);
+        executeBigQuerySql(client, "CREATE VIEW " + viewName + " AS SELECT 1 AS col1");
+
+        assertQuery("SELECT count(*) FROM " + viewName, "VALUES (1)");
+        assertQuery("SELECT count(*) FROM " + viewName, "VALUES (1)");
+
+        executeBigQuerySql(client, "DROP VIEW " + viewName);
+    }
+
     private static void executeBigQuerySql(BigQuery bigquery, String query)
     {
         QueryJobConfiguration queryConfig = QueryJobConfiguration.newBuilder(query)
