improve error checks

apache · Oct 18, 2023 · a51761b · a51761b
1 parent d4f510d
commit a51761b
Showing 1 changed file with 27 additions and 22 deletions.
diff --git a/core/src/test/java/org/apache/iceberg/view/ViewCatalogTests.java b/core/src/test/java/org/apache/iceberg/view/ViewCatalogTests.java
@@ -36,7 +36,6 @@
 import org.apache.iceberg.exceptions.NoSuchTableException;
 import org.apache.iceberg.exceptions.NoSuchViewException;
 import org.apache.iceberg.relocated.com.google.common.collect.ImmutableList;
-import org.apache.iceberg.rest.RESTCatalog;
 import org.apache.iceberg.types.Types;
 import org.assertj.core.api.Assumptions;
 import org.junit.jupiter.api.Test;
@@ -1284,14 +1283,16 @@ public void updateViewPropertiesConflict() {
     catalog().dropView(identifier);
     assertThat(catalog().viewExists(identifier)).as("View should not exist").isFalse();
 
-    String expectedMessage =
-        catalog() instanceof RESTCatalog ? "View does not exist: ns.view" : "Cannot commit";
-    Class<?> expectedException =
-        catalog() instanceof RESTCatalog ? NoSuchViewException.class : CommitFailedException.class;
-
     assertThatThrownBy(() -> updateViewProperties.set("key1", "val1").commit())
-        .isInstanceOf(expectedException)
-        .hasMessageContaining(expectedMessage);
+        .satisfiesAnyOf(
+            x ->
+                assertThat(x)
+                    .isInstanceOf(NoSuchViewException.class)
+                    .hasMessageContaining("View does not exist: ns.view"),
+            x ->
+                assertThat(x)
+                    .isInstanceOf(CommitFailedException.class)
+                    .hasMessageContaining("Cannot commit"));
   }
 
   @Test
@@ -1319,20 +1320,22 @@ public void replaceViewVersionConflict() {
     catalog().dropView(identifier);
     assertThat(catalog().viewExists(identifier)).as("View should not exist").isFalse();
 
-    String expectedMessage =
-        catalog() instanceof RESTCatalog ? "View does not exist: ns.view" : "Cannot commit";
-    Class<?> expectedException =
-        catalog() instanceof RESTCatalog ? NoSuchViewException.class : CommitFailedException.class;
-
     assertThatThrownBy(
             () ->
                 replaceViewVersion
                     .withQuery("trino", "select * from ns.tbl")
                     .withSchema(SCHEMA)
                     .withDefaultNamespace(identifier.namespace())
                     .commit())
-        .isInstanceOf(expectedException)
-        .hasMessageContaining(expectedMessage);
+        .satisfiesAnyOf(
+            x ->
+                assertThat(x)
+                    .isInstanceOf(NoSuchViewException.class)
+                    .hasMessageContaining("View does not exist: ns.view"),
+            x ->
+                assertThat(x)
+                    .isInstanceOf(CommitFailedException.class)
+                    .hasMessageContaining("Cannot commit"));
   }
 
   @Test
@@ -1522,14 +1525,16 @@ public void updateViewLocationConflict() {
     catalog().dropView(identifier);
     assertThat(catalog().viewExists(identifier)).as("View should not exist").isFalse();
 
-    String expectedMessage =
-        catalog() instanceof RESTCatalog ? "View does not exist: ns.view" : "Cannot commit";
-    Class<?> expectedException =
-        catalog() instanceof RESTCatalog ? NoSuchViewException.class : CommitFailedException.class;
-
     // the view was already dropped concurrently
     assertThatThrownBy(() -> updateViewLocation.setLocation("new-location").commit())
-        .isInstanceOf(expectedException)
-        .hasMessageContaining(expectedMessage);
+        .satisfiesAnyOf(
+            x ->
+                assertThat(x)
+                    .isInstanceOf(NoSuchViewException.class)
+                    .hasMessageContaining("View does not exist: ns.view"),
+            x ->
+                assertThat(x)
+                    .isInstanceOf(CommitFailedException.class)
+                    .hasMessageContaining("Cannot commit"));
   }
 }