From 21f21bbfbd1a27bc6f9e8777e10930b8a691106f Mon Sep 17 00:00:00 2001
From: Yanyan-Wang <yanyanwang93@gmail.com>
Date: Fri, 4 Jun 2021 13:46:25 +0800
Subject: [PATCH] test: refine test

---
 packages/pc/tests/unit/layout/circular-web-worker-spec.ts | 7 +++++++
 packages/pc/tests/unit/layout/force-web-worker-spec.ts    | 7 +++++++
 2 files changed, 14 insertions(+)

diff --git a/packages/pc/tests/unit/layout/circular-web-worker-spec.ts b/packages/pc/tests/unit/layout/circular-web-worker-spec.ts
index aa2d7c49e74..50ab302d18a 100644
--- a/packages/pc/tests/unit/layout/circular-web-worker-spec.ts
+++ b/packages/pc/tests/unit/layout/circular-web-worker-spec.ts
@@ -56,12 +56,19 @@ describe('circular layout(web worker)', () => {
     graph.data(data);
     graph.render();
 
+    let timeout = false;
     graph.on('afterlayout', () => {
+      if (timeout) return;
       const pos = (200 * Math.sqrt(2)) / 2;
       expect(mathEqual(data.nodes[0].x, 250 + pos)).toEqual(true);
       expect(mathEqual(data.nodes[0].y, 250 + pos)).toEqual(true);
       graph.destroy();
       done();
     });
+    setTimeout(() => {
+      timeout = true;
+      graph.destroy();
+      done();
+    }, 1000)
   });
 });
diff --git a/packages/pc/tests/unit/layout/force-web-worker-spec.ts b/packages/pc/tests/unit/layout/force-web-worker-spec.ts
index 49639f46e22..64eaf810255 100644
--- a/packages/pc/tests/unit/layout/force-web-worker-spec.ts
+++ b/packages/pc/tests/unit/layout/force-web-worker-spec.ts
@@ -35,7 +35,9 @@ describe('force layout(web worker)', function () {
       height: 500,
       defaultNode: { size: 10 },
     });
+    let timeout = false;
     graph.on('afterlayout', () => {
+      if (timeout) return;
       expect(node.x).not.toEqual(undefined);
       expect(node.y).not.toEqual(undefined);
       expect(count >= 1).toEqual(true);
@@ -45,5 +47,10 @@ describe('force layout(web worker)', function () {
     });
     graph.data(data);
     graph.render();
+    setTimeout(() => {
+      timeout = true;
+      graph.destroy();
+      done();
+    }, 4000)
   });
 });