test(ui): fix falling test

kestra-io · Jan 27, 2023 · ec20485 · ec20485
1 parent 9e02166
commit ec20485
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 20 deletions.
diff --git a/ui/tests/local.js b/ui/tests/local.js
@@ -6,6 +6,7 @@ import {extendMoment} from "moment-range";
 import ElementPlus from "element-plus";
 import filters from "../src/utils/filters";
 import translations from "../src/translations.json";
+import stores from "../src/stores/store"
 import "../src/utils/global"
 
 
@@ -17,21 +18,12 @@ let i18n = createI18n({
     warnHtmlMessage: false,
 });
 
-const store = createStore({
-    modules: {
-        plugin: {
-            state: {
-                icons: []
-            },
-            namespaced: true
-        }
-    }
-});
+const store = createStore(stores);
 
 moment.locale("en");
 
-export default (component, options) => {
-    return mount(
+export default (component, options, callback) => {
+    const app =  mount(
         component,
         {
             ...{
@@ -48,4 +40,8 @@ export default (component, options) => {
             ...options
         }
     )
+
+    callback(store);
+
+    return app;
 }
diff --git a/ui/tests/unit/components/graph/TreeNode.spec.js b/ui/tests/unit/components/graph/TreeNode.spec.js
@@ -6,14 +6,20 @@ import EACH_SEQUENTIAL_EXECUTION from "../../../fixtures/executions/each-sequent
 import mount from "../../../local.js";
 
 const localMount = (n, execution) => {
-    return mount(TreeNode, {
-        props: {
-            n: n,
-            execution: execution,
-            flowId: "flowId",
-            namespace: "namespace",
+    return mount(
+        TreeNode,
+        {
+            props: {
+                n: n,
+                execution: execution,
+                flowId: "flowId",
+                namespace: "namespace",
+            }
+        },
+        (store) => {
+            store.commit("execution/setExecution", execution)
         }
-    })
+    )
 }
 
 describe("TreeNode", () => {
@@ -26,7 +32,10 @@ describe("TreeNode", () => {
         expect(wrapper.vm.task.id).toBe("1-2");
         expect(wrapper.vm.state).toBe("SUCCESS");
         expect(wrapper.vm.taskRuns).toHaveLength(3);
-        expect(wrapper.vm.duration).toBe(0.000633852);
+        expect(wrapper.vm.histories[0].state).toBe("CREATED");
+        expect(wrapper.vm.histories[0].date.toISOString()).toBe("2020-12-26T20:38:16.001Z");
+        expect(wrapper.vm.histories[1].state).toBe("SUCCESS");
+        expect(wrapper.vm.histories[1].date.toISOString()).toBe("2020-12-26T20:38:16.002Z");
     })
 
     it("sorting state", () => {