fix(): also update brick tree before build

Closes NEXT_BUILDER-3913
easyops-cn · Oct 16, 2023 · 2433cf8 · 2433cf8
1 parent 31544e4
commit 2433cf8
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/bricks/next-builder/src/shared/workbench/WorkbenchBackend.spec.ts b/bricks/next-builder/src/shared/workbench/WorkbenchBackend.spec.ts
@@ -177,7 +177,14 @@ describe("WorkbenchBackend should work", () => {
     jest.runAllTimers();
     await (global as any).flushPromises();
 
-    expect(handleBackendMessage).toHaveBeenNthCalledWith(3, "message", {
+    expect(handleBackendMessage).toHaveBeenNthCalledWith(
+      3,
+      "message",
+      expect.objectContaining({
+        action: "update-graph-data",
+      })
+    );
+    expect(handleBackendMessage).toHaveBeenNthCalledWith(4, "message", {
       action: "build-start",
     });
 
@@ -192,7 +199,7 @@ describe("WorkbenchBackend should work", () => {
       },
     });
 
-    expect(handleBackendMessage).toHaveBeenNthCalledWith(4, "message", {
+    expect(handleBackendMessage).toHaveBeenNthCalledWith(5, "message", {
       action: "build-success",
       data: {
         storyboard: {
@@ -360,7 +367,7 @@ describe("WorkbenchBackend should work", () => {
       instanceId: "abc",
     });
 
-    expect(handleBackendMessage).toBeCalledTimes(10);
+    expect(handleBackendMessage).toBeCalledTimes(11);
     backendInstance.unsubscribe(listener);
 
     backendInstance.push({
@@ -373,7 +380,7 @@ describe("WorkbenchBackend should work", () => {
     });
     await (global as any).flushPromises();
 
-    expect(handleBackendMessage).toBeCalledTimes(10);
+    expect(handleBackendMessage).toBeCalledTimes(11);
 
     // @ts-ignore
     backendInstance.mockNodeIdCache.set("mock_id_1", "new-id");

diff --git a/bricks/next-builder/src/shared/workbench/WorkbenchBackend.ts b/bricks/next-builder/src/shared/workbench/WorkbenchBackend.ts
@@ -470,6 +470,7 @@ export default class WorkbenchBackend {
     if (this.isBuilding) return;
     this.isBuilding = true;
     try {
+      await this.updateBrickTree();
       // eslint-disable-next-line no-console
       console.log("=== building ===");
       this.publish("message", {