Fix testConvertComparison test in TestPostgreSqlClient

trinodb · Apr 8, 2024 · b6742fb · b6742fb
1 parent a54d768
commit b6742fb
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/plugin/trino-postgresql/src/test/java/io/trino/plugin/postgresql/TestPostgreSqlClient.java b/plugin/trino-postgresql/src/test/java/io/trino/plugin/postgresql/TestPostgreSqlClient.java
@@ -285,20 +285,16 @@ public void testConvertComparison()
             switch (operator) {
                 case EQUAL:
                 case NOT_EQUAL:
-                    assertThat(converted).isPresent();
-                    assertThat(converted.get().expression()).isEqualTo(format("(\"c_bigint\") %s (?)", operator.getValue()));
-                    assertThat(converted.get().parameters()).isEqualTo(List.of(new QueryParameter(BIGINT, Optional.of(42L))));
-                    return;
                 case LESS_THAN:
                 case LESS_THAN_OR_EQUAL:
                 case GREATER_THAN:
                 case GREATER_THAN_OR_EQUAL:
                 case IS_DISTINCT_FROM:
-                    // Not supported yet, even for bigint
-                    assertThat(converted).isEmpty();
-                    return;
+                    assertThat(converted).isPresent();
+                    assertThat(converted.get().expression()).isEqualTo(format("(\"c_bigint\") %s (?)", operator.getValue()));
+                    assertThat(converted.get().parameters()).isEqualTo(List.of(new QueryParameter(BIGINT, Optional.of(42L))));
+                    break;
             }
-            throw new UnsupportedOperationException("Unsupported operator: " + operator);
         }
     }