Fix for failing test

deephaven · Jan 30, 2025 · 5fdd22b · 5fdd22b
1 parent ebbc6a8
commit 5fdd22b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/extensions/parquet/table/src/test/java/io/deephaven/parquet/table/TestParquetTools.java b/extensions/parquet/table/src/test/java/io/deephaven/parquet/table/TestParquetTools.java
@@ -1093,9 +1093,10 @@ public void parquetWithNonUniqueFieldIds() {
             // notice this earlier on.
             try {
                 table.select();
-                failBecauseExceptionWasNotThrown(IllegalStateException.class);
-            } catch (IllegalArgumentException e) {
-                assertThat(e).hasMessageContaining(
+                failBecauseExceptionWasNotThrown(TableDataException.class);
+            } catch (TableDataException e) {
+                assertThat(e).hasCauseInstanceOf(IllegalArgumentException.class);
+                assertThat(e.getCause()).hasMessageContaining(
                         "Parquet columns can't be unambigously mapped. Bar -> 31337 has multiple paths [Foo], [Bar]");
             }
         }