Add an extension-comparison test to QueryDslSqlSelectTest

Author: momosetkn

doma-core/src/test/java/org/seasar/doma/jdbc/criteria/MyExtension.java

@@ -0,0 +1,35 @@
+/*
+ * Copyright Doma Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package org.seasar.doma.jdbc.criteria;
+
+import org.seasar.doma.jdbc.criteria.declaration.UserDefinedCriteriaContext;
+import org.seasar.doma.jdbc.criteria.expression.Expressions;
+import org.seasar.doma.jdbc.criteria.metamodel.PropertyMetamodel;
+
+record MyExtension(UserDefinedCriteriaContext context) {
+  public void likeMultiple(PropertyMetamodel<?> entityMetamodel, String... patterns) {
+    context.add(
+        (b) -> {
+          for (String pattern : patterns) {
+            b.appendExpression(entityMetamodel);
+            b.appendSql(" like ");
+            b.appendExpression(Expressions.literal("%" + pattern + "%"));
+            b.appendSql(" or ");
+          }
+          b.cutBackSql(4);
+        });
+  }
+}

doma-core/src/test/java/org/seasar/doma/jdbc/criteria/QueryDslSqlSelectTest.java

@@ -1775,6 +1775,126 @@ void userDefinedExpression_countDistinctMultiple() {
         sql.getFormattedSql());
   }
 
+  @Test
+  void extension() {
+    var e = new Emp_();
+    var stmt =
+        dsl.from(e)
+            .where(
+                c -> {
+                  c.extension(
+                      MyExtension::new,
+                      (ext) -> {
+                        ext.likeMultiple(e.name, "SA");
+                      });
+                })
+            .select(e.id);
+
+    var sql = stmt.asSql();
+    assertEquals("select t0_.ID from EMP t0_ where t0_.NAME like '%SA%'", sql.getFormattedSql());
+  }
+
+  @Test
+  void extension_multiple_condition() {
+    var e = new Emp_();
+    var stmt =
+        dsl.from(e)
+            .where(
+                c -> {
+                  c.extension(
+                      MyExtension::new,
+                      (ext) -> {
+                        ext.likeMultiple(e.name, "SA");
+                        ext.likeMultiple(e.name, "LE");
+                      });
+                })
+            .select(e.id);
+
+    var sql = stmt.asSql();
+    assertEquals(
+        "select t0_.ID from EMP t0_ where t0_.NAME like '%SA%' and t0_.NAME like '%LE%'",
+        sql.getFormattedSql());
+  }
+
+  @Test
+  void extension_and() {
+    var e = new Emp_();
+    var stmt =
+        dsl.from(e)
+            .where(
+                c -> {
+                  c.extension(
+                      MyExtension::new,
+                      (ext) -> {
+                        ext.likeMultiple(e.name, "SA");
+                        c.and(
+                            () -> {
+                              ext.likeMultiple(e.name, "LE");
+                            });
+                      });
+                })
+            .select(e.id);
+
+    var sql = stmt.asSql();
+    assertEquals(
+        "select t0_.ID from EMP t0_ where t0_.NAME like '%SA%' and (t0_.NAME like '%LE%')",
+        sql.getFormattedSql());
+  }
+
+  @Test
+  void extension_or() {
+    var e = new Emp_();
+    var stmt =
+        dsl.from(e)
+            .where(
+                c -> {
+                  c.extension(
+                      MyExtension::new,
+                      (ext) -> {
+                        ext.likeMultiple(e.name, "CC");
+                        c.or(
+                            () -> {
+                              ext.likeMultiple(e.name, "SA");
+                            });
+                      });
+                })
+            .select(e.id);
+
+    var sql = stmt.asSql();
+    assertEquals(
+        "select t0_.ID from EMP t0_ where t0_.NAME like '%CC%' or (t0_.NAME like '%SA%')",
+        sql.getFormattedSql());
+  }
+
+  @Test
+  void extension_or_and() {
+    var e = new Emp_();
+    var stmt =
+        dsl.from(e)
+            .where(
+                c -> {
+                  c.extension(
+                      MyExtension::new,
+                      (ext) -> {
+                        ext.likeMultiple(e.name, "CC");
+                        c.or(
+                            () -> {
+                              ext.likeMultiple(e.name, "SA");
+                              c.and(
+                                  () -> {
+                                    ext.likeMultiple(e.name, "LE");
+                                  });
+                            });
+                      });
+                })
+            .select(e.id);
+
+    var sql = stmt.asSql();
+    assertEquals(
+        "select t0_.ID from EMP t0_ where t0_.NAME like '%CC%' or (t0_.NAME like '%SA%' and (t0_.NAME like '%LE%'))",
+        sql.getFormattedSql());
+  }
+
   private static UserDefinedExpression<Long> countDistinctMultiple(
       PropertyMetamodel<?>... propertyMetamodels) {
     return userDefined(