[SLT] Add missing casts in output values

mihaibudiu · mihaibudiu · commit f9b29d4c7ee6 · 2025-07-23T16:39:44.000Z
Signed-off-by: Mihai Budiu &lt;mbudiu@feldera.com&gt;
diff --git a/sql-to-dbsp-compiler/slt/src/main/java/org/dbsp/sqllogictest/executors/DBSPExecutor.java b/sql-to-dbsp-compiler/slt/src/main/java/org/dbsp/sqllogictest/executors/DBSPExecutor.java
@@ -39,6 +39,7 @@
 import org.dbsp.sqlCompiler.compiler.frontend.TableContents;
 import org.dbsp.sqlCompiler.compiler.frontend.calciteCompiler.ProgramIdentifier;
 import org.dbsp.sqlCompiler.compiler.frontend.calciteObject.CalciteObject;
+import org.dbsp.sqlCompiler.compiler.visitors.inner.CreateRuntimeErrorWrappers;
 import org.dbsp.sqlCompiler.ir.DBSPFunction;
 import org.dbsp.sqlCompiler.ir.DBSPNode;
 import org.dbsp.sqlCompiler.ir.expression.DBSPApplyExpression;
@@ -216,7 +217,8 @@ boolean runBatch(TestStatistics result, boolean cleanup) {
     }
 
     /** Convert a description of the data in the SLT format to a ZSet. */
-    public static DBSPZSetExpression convert(@Nullable List<String> data, DBSPTypeZSet outputType) {
+    public static DBSPZSetExpression convert(
+            DBSPCompiler compiler, @Nullable List<String> data, DBSPTypeZSet outputType) {
         if (data == null)
             data = Linq.list();
         DBSPZSetExpression result;
@@ -272,6 +274,8 @@ else if (colType.is(DBSPTypeBool.class))
             throw new RuntimeException("Could not assign all query output values to rows. " +
                     "I have " + col + " leftover values in the last row");
         }
+        CreateRuntimeErrorWrappers cw = new CreateRuntimeErrorWrappers(compiler);
+        result = cw.apply(result).to(DBSPZSetExpression.class);
         return result;
     }
 
@@ -311,7 +315,7 @@ ProgramAndTester generateTestCase(
         DBSPTypeZSet outputType = sink.outputType.to(DBSPTypeZSet.class);
         DBSPZSetExpression expectedOutput = null;
         if (testQuery.outputDescription.hash == null) {
-            expectedOutput = DBSPExecutor.convert(testQuery.outputDescription.getQueryResults(), outputType);
+            expectedOutput = DBSPExecutor.convert(compiler, testQuery.outputDescription.getQueryResults(), outputType);
         }
 
         return createTesterCode(
