fix(client-js): workflow tests

polywrap · Jun 30, 2022 · fb1a828 · fb1a828
1 parent 07ce9e4
commit fb1a828
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/packages/js/client/src/__tests__/e2e/workflow.spec.ts b/packages/js/client/src/__tests__/e2e/workflow.spec.ts
@@ -6,7 +6,7 @@ import {
   ensAddresses,
   providers
 } from "@polywrap/test-env-js";
-import { createPolywrapClient, PolywrapClient, PolywrapClientConfig } from "../..";
+import { createPolywrapClient, JobResult, PolywrapClient, PolywrapClientConfig } from "../..";
 import { outPropWorkflow, sanityWorkflow } from "./workflow-test-cases";
 
 jest.setTimeout(200000);
@@ -89,17 +89,17 @@ describe("workflow", () => {
     test("sanity workflow", async () => {
       await client.run({
         workflow: sanityWorkflow,
-        onExecution: async (id: string, data: unknown, error: unknown) => {
-          await tests[id](data, error);
+        onExecution: async (id: string, jobResult: JobResult) => {
+          await tests[id](jobResult.data, jobResult.error);
         },
       });
     });
 
     test("workflow with output propagation", async () => {
       await client.run({
         workflow: outPropWorkflow,
-        onExecution: async (id: string, data: unknown, error: unknown) => {
-          await tests[id](data, error);
+        onExecution: async (id: string, jobResult: JobResult) => {
+          await tests[id](jobResult.data, jobResult.error);
         },
       });
     });