Make partition overwrite assertion use exact rows

trinodb · Nov 8, 2021 · 3935f5f · 3935f5f
1 parent f7c87f8
commit 3935f5f
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/plugin/trino-hive/src/test/java/io/trino/plugin/hive/BaseTestHiveInsertOverwrite.java b/plugin/trino-hive/src/test/java/io/trino/plugin/hive/BaseTestHiveInsertOverwrite.java
@@ -192,19 +192,20 @@ protected void assertOverwritePartition(String testTable)
                         "('POLAND', 'Test Data', 25, 5), " +
                         "('CZECH', 'Test Data', 26, 5)",
                 testTable));
-        query(format("SELECT count(*) FROM %s WHERE regionkey = 5", testTable))
+        query(format("SELECT name, comment, nationkey, regionkey FROM %s WHERE regionkey = 5", testTable))
                 .assertThat()
                 .skippingTypesCheck()
                 .containsAll(resultBuilder(getSession())
-                        .row(2L)
+                        .row("POLAND", "Test Data", 25L, 5L)
+                        .row("CZECH", "Test Data", 26L, 5L)
                         .build());
 
         computeActual(format("INSERT INTO %s values('POLAND', 'Overwrite', 25, 5)", testTable));
-        query(format("SELECT count(*) FROM %s WHERE regionkey = 5", testTable))
+        query(format("SELECT name, comment, nationkey, regionkey FROM %s WHERE regionkey = 5", testTable))
                 .assertThat()
                 .skippingTypesCheck()
                 .containsAll(resultBuilder(getSession())
-                        .row(1L)
+                        .row("POLAND", "Overwrite", 25L, 5L)
                         .build());
         computeActual(format("DROP TABLE %s", testTable));
     }