Fix validation for non object types

coda · Oct 7, 2024 · 96ccb7b · 96ccb7b
1 parent 3bcf351
commit 96ccb7b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/dist/testing/execution.js b/dist/testing/execution.js
diff --git a/testing/execution.ts b/testing/execution.ts
@@ -133,16 +133,14 @@ async function findAndExecutePackFunction<T extends FormulaSpecification>(
   if (formula) {
     const resultToNormalize =
       formulaSpec.type === FormulaType.Sync ? (result as GenericSyncFormulaResult).result : result;
+    let resultToValidate = resultToNormalize;
 
     // Matches legacy behavior within handler_templates:generateObjectResponseHandler where we never
     // called transform body on non-object responses.
     if (typeof resultToNormalize === 'object') {
       const schema = executionContext?.sync?.schema ?? formula.schema;
       let normalizedResult = transformBody(resultToNormalize, schema);
-
-      if (shouldValidateResult) {
-        validateResult(formula, normalizedResult);
-      }
+      resultToValidate = normalizedResult;
 
       if (!useDeprecatedResultNormalization) {
         normalizedResult = untransformBody(normalizedResult, schema);
@@ -154,6 +152,10 @@ async function findAndExecutePackFunction<T extends FormulaSpecification>(
         result = normalizedResult;
       }
     }
+
+    if (shouldValidateResult) {
+      validateResult(formula, resultToValidate);
+    }
   }
 
   return result;