[CALCITE-7345] Quidem test support for Field Trimmer

Author: NobiGo

core/src/main/java/org/apache/calcite/tools/Programs.java

@@ -286,6 +286,11 @@ public static Program standard() {
     return standard(DefaultRelMetadataProvider.INSTANCE, true);
   }
 
+  /** Returns the standard program with enableFieldTrimming config. */
+  public static Program standard(Boolean enableFieldTrimming) {
+    return standard(DefaultRelMetadataProvider.INSTANCE, enableFieldTrimming);
+  }
+
   /** Returns the standard program with user metadata provider. */
   public static Program standard(RelMetadataProvider metadataProvider) {
     return standard(metadataProvider, true);

core/src/test/resources/sql/join.iq

@@ -1137,4 +1137,30 @@ SELECT * FROM (VALUES (NULLIF(5, 5)), (NULLIF(5, 5))) a, (VALUES (NULLIF(5, 5)),
 
 !ok
 
+# [CALCITE-7345] Quidem test support for Field Trimmer
+!use scott
+!set trimfields false
+
+select distinct dept.deptno, emp.deptno
+from "scott".emp join "scott".dept using (deptno);
+
+EnumerableAggregate(group=[{0, 10}])
+  EnumerableHashJoin(condition=[=($0, $10)], joinType=[inner])
+    EnumerableTableScan(table=[[scott, DEPT]])
+    EnumerableTableScan(table=[[scott, EMP]])
+!plan
+
+!set trimfields true
+
+select distinct dept.deptno, emp.deptno
+from "scott".emp join "scott".dept using (deptno);
+
+EnumerableAggregate(group=[{0, 2}])
+  EnumerableHashJoin(condition=[=($0, $2)], joinType=[inner])
+    EnumerableCalc(expr#0..2=[{inputs}], DEPTNO=[$t0])
+      EnumerableTableScan(table=[[scott, DEPT]])
+    EnumerableCalc(expr#0..7=[{inputs}], EMPNO=[$t0], DEPTNO=[$t7])
+      EnumerableTableScan(table=[[scott, EMP]])
+!plan
+
 # End join.iq

testkit/src/main/java/org/apache/calcite/test/QuidemTest.java

@@ -39,6 +39,7 @@
 import org.apache.calcite.sql.parser.SqlParserImplFactory;
 import org.apache.calcite.sql.pretty.SqlPrettyWriter;
 import org.apache.calcite.sql.type.SqlTypeName;
+import org.apache.calcite.sql2rel.SqlToRelConverter;
 import org.apache.calcite.test.schemata.catchall.CatchallSchema;
 import org.apache.calcite.tools.Frameworks;
 import org.apache.calcite.tools.Planner;
@@ -246,6 +247,17 @@ protected void checkRun(String path) throws Exception {
               int thresholdValue = ((BigDecimal) value).intValue();
               closer.add(Prepare.THREAD_INSUBQUERY_THRESHOLD.push(thresholdValue));
             }
+            if (propertyName.equals("trimfields")) {
+              final boolean b = value instanceof Boolean
+                  && (Boolean) value;
+              closer.add(
+                  Hook.SQL2REL_CONVERTER_CONFIG_BUILDER.addThread(
+                  (Consumer<Holder<SqlToRelConverter.Config>>) configHolder ->
+                      configHolder.set(configHolder.get().withTrimUnusedFields(b))));
+              closer.add(
+                  Hook.PROGRAM.addThread((Consumer<Holder<Program>>)
+                      holder -> holder.set(Programs.standard(b))));
+            }
             // Configures query planner rules via "!set planner-rules" command.
             // The value can be set as follows:
             // - Add rule:       "+EnumerableRules.ENUMERABLE_INTERSECT_RULE"