fix: adapt to API updates

modelon-community · Jan 16, 2023 · 6e97505 · 6e97505
1 parent 1a68f11
commit 6e97505
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/jest.config.js b/jest.config.js
@@ -2,4 +2,5 @@
 module.exports = {
     preset: 'ts-jest',
     testEnvironment: 'node',
+    verbose: true,
 }
diff --git a/src/Client.ts b/src/Client.ts
@@ -461,7 +461,7 @@ export class Client {
                                 },
                             }
                         )
-                        .then((res) => resolve(res.data))
+                        .then((res) => resolve(res.data.data.items))
                         .catch((e) => reject(e))
                 })
                 .catch((e) => reject(e))

diff --git a/tests/integration/integration.test.ts b/tests/integration/integration.test.ts
@@ -56,7 +56,7 @@ test(
 
         const client = getClient()
 
-        const experimentId = await client.executeExperiment({
+        const experimentId = await client.executeExperimentSync({
             caseIds: ['case_1', 'case_2'],
             experiment,
             workspaceId,
@@ -76,8 +76,8 @@ test(
             workspaceId,
         })
         expect(trajectories.length).toBe(2)
-        expect(trajectories[0].length).toBe(502)
-        expect(trajectories[1].length).toBe(502)
+        expect(trajectories[0].trajectory.length).toBe(102)
+        expect(trajectories[1].trajectory.length).toBe(102)
 
         trajectories = await client.getCaseTrajectories({
             caseId: 'case_2',
@@ -86,8 +86,8 @@ test(
             workspaceId,
         })
         expect(trajectories.length).toBe(2)
-        expect(trajectories[0].length).toBe(502)
-        expect(trajectories[1].length).toBe(502)
+        expect(trajectories[0].trajectory.length).toBe(102)
+        expect(trajectories[1].trajectory.length).toBe(102)
     },
     20 * 1000
 )