fix - BQ TableResult is now abstract, constructed via builder

DataBiosphere · Aug 7, 2024 · 654f178 · 654f178
1 parent b0fbbf3
commit 654f178
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/src/test/java/bio/terra/common/BQTestUtils.java b/src/test/java/bio/terra/common/BQTestUtils.java
@@ -66,8 +66,11 @@ public static void mockBQQuery(
 
   private static Answer<TableResult> mockAnswer(Schema schema, List<Map<String, String>> results) {
     return a ->
-        new TableResult(
-            schema, results.size(), new PageImpl<>(null, null, convertValues(results, schema)));
+        TableResult.newBuilder()
+            .setSchema(schema)
+            .setTotalRows((long) results.size())
+            .setPageNoSchema(new PageImpl<>(null, null, convertValues(results, schema)))
+            .build();
   }
 
   private static List<FieldValueList> convertValues(

diff --git a/src/test/java/bio/terra/service/tabulardata/google/bigquery/BigQueryPdaoUnitTest.java b/src/test/java/bio/terra/service/tabulardata/google/bigquery/BigQueryPdaoUnitTest.java
@@ -1061,7 +1061,8 @@ void aggregateSnapshotTableTest() {
 
     Page<FieldValueList> page = mockPage(listOfFieldValueList);
 
-    TableResult table = new TableResult(schema, 10, page);
+    TableResult table =
+        TableResult.newBuilder().setSchema(schema).setTotalRows(10L).setPageNoSchema(page).build();
 
     List<BigQueryDataResultModel> result = BigQueryPdao.aggregateTableData(table);
 
@@ -1091,7 +1092,8 @@ void testAggregateTextColumnStats() {
                             "2"))))); // For some reason, it wants the numeric value passed in as a
     // string
 
-    TableResult table = new TableResult(schema, 10, page);
+    TableResult table =
+        TableResult.newBuilder().setSchema(schema).setTotalRows(10L).setPageNoSchema(page).build();
 
     List<ColumnStatisticsTextValue> result =
         BigQueryPdao.aggregateTextColumnStats(table, columnName);