test(asset-bundle): add "assetBundle asset node" and "load assetBundl…

…e" related test cases

src/core/composable_component/header/atom_component/asset_bundle/HeaderAssetBundleGenerateSingleRAB.re

@@ -394,7 +394,6 @@ let reducer = (action, state) =>
 
 let render =
     ({state, send}: ReasonReact.self('a, 'b, 'c), (closeFunc, submitFunc)) => {
-  Js.log("render sissssssss");
   let languageType =
     LanguageEditorService.unsafeGetType |> StateLogicService.getEditorState;
 

src/core/composable_component/mainEditor/composable_component/bottom_components/composable_component/project/composable_component/asset/composable_component/assetChildrenNode/ui/MainEditorAssetChildrenNode.re

@@ -108,13 +108,20 @@ module Method = {
       assetTreeChildrenNodeArr
       |> Js.Array.filter(node => node |> TextureNodeAssetService.isTextureNode);
 
+    let assetBundleAssetTreeChildrenNodeArr =
+      assetTreeChildrenNodeArr
+      |> Js.Array.filter(node =>
+           node |> AssetBundleNodeAssetService.isAssetBundleNode
+         );
+
     ArrayService.fastConcatArrays([|
       _sortByName(folderAssetTreeChildrenNodeArr, engineState),
       _sortByName(wdbAssetTreeChildrenNodeArr, engineState),
       _sortByName(materialAssetTreeChildrenNodeArr, engineState),
       _sortByName(scriptEventFunctionAssetTreeChildrenNodeArr, engineState),
       _sortByName(scriptAttributeAssetTreeChildrenNodeArr, engineState),
       _sortByName(textureAssetTreeChildrenNodeArr, engineState),
+      _sortByName(assetBundleAssetTreeChildrenNodeArr, engineState),
     |]);
   };
 

src/core/composable_component/mainEditor/composable_component/bottom_components/composable_component/project/composable_component/asset/composable_component/utils/AssetHeaderUtils.re

@@ -122,7 +122,6 @@ let _handleAssetWDBType =
        (editorState, engineState) |> resolve;
      });
 
-/* TODO test */
 let _handleAssetAssetBundleType =
     (
       (fileName, assetBundleArrayBuffer),
@@ -135,7 +134,12 @@ let _handleAssetAssetBundleType =
         selectedFolderNodeInAssetTree,
         AssetBundleNodeAssetService.buildNode(
           ~nodeId=assetBundleNodeId,
-          ~name=FileNameService.getBaseName(fileName),
+          ~name=
+            OperateTreeAssetLogicService.getUniqueNodeName(
+              FileNameService.getBaseName(fileName),
+              selectedFolderNodeInAssetTree,
+              engineState,
+            ),
           ~assetBundle=assetBundleArrayBuffer,
           ~type_=
             switch (FileNameService.getExtName(fileName)) {

src/service/record/editor/asset/AssetBundleNodeAssetService.re

@@ -11,6 +11,20 @@ let getTypeStr = ({type_}: assetBundleNodeData) =>
   | WAB => "WAB"
   };
 
+let getNodeData = node =>
+  switch (node) {
+  | AssetBundleNode(_, nodeData) => nodeData
+  | _ =>
+    WonderLog.Log.fatal(
+      LogUtils.buildFatalMessage(
+        ~description={j|should be assetBundle node|j},
+        ~reason="",
+        ~solution={j||j},
+        ~params={j||j},
+      ),
+    )
+  };
+
 let buildNodeData = (~name, ~assetBundle, ~type_) => {
   name,
   assetBundle,
@@ -23,4 +37,12 @@ let buildNode = (~nodeId, ~name, ~assetBundle, ~type_) =>
 let buildNodeByNodeData = (~nodeId, ~nodeData) =>
   AssetBundleNode(nodeId, nodeData);
 
-let rename = (~name, ~nodeData): assetBundleNodeData => {...nodeData, name};
+let rename = (~name, ~nodeData): assetBundleNodeData => {...nodeData, name};
+
+let isAssetBundleNode = node =>
+  switch (node) {
+  | AssetBundleNode(_, _) => true
+  | _ => false
+  };
+
+let getAssetBundle = node => getNodeData(node).assetBundle;

test/integration/asset/composable_component/header/__snapshots__/mainEditorAssetHeaderLoadAssetBundle_test.js.snap

@@ -0,0 +1,75 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`MainEditorAssetHeader->load assetBundle test load asset bundle fix bug if load the same asset bundle in the same dir, should handle their name to be unique 1`] = `
+<article
+  className="wonder-asset-assetChildren"
+>
+  <article
+    className="wonder-asset-fileBox "
+    onClick={[Function]}
+  >
+    <div
+      className="box-image"
+    >
+      <img
+        onDragStart={[Function]}
+        src="./public/img/assetBundle.png"
+      />
+    </div>
+    <div
+      className="item-text "
+    >
+      <span>
+        A
+      </span>
+    </div>
+  </article>
+  <article
+    className="wonder-asset-fileBox "
+    onClick={[Function]}
+  >
+    <div
+      className="box-image"
+    >
+      <img
+        onDragStart={[Function]}
+        src="./public/img/assetBundle.png"
+      />
+    </div>
+    <div
+      className="item-text "
+    >
+      <span>
+        A 1
+      </span>
+    </div>
+  </article>
+</article>
+`;
+
+exports[`MainEditorAssetHeader->load assetBundle test load asset bundle should add to asset children 1`] = `
+<article
+  className="wonder-asset-assetChildren"
+>
+  <article
+    className="wonder-asset-fileBox "
+    onClick={[Function]}
+  >
+    <div
+      className="box-image"
+    >
+      <img
+        onDragStart={[Function]}
+        src="./public/img/assetBundle.png"
+      />
+    </div>
+    <div
+      className="item-text "
+    >
+      <span>
+        A
+      </span>
+    </div>
+  </article>
+</article>
+`;

test/integration/asset/composable_component/header/__snapshots__/mainEditorAssetHeaderRemoveAssetBundle_test.js.snap

@@ -0,0 +1,10 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`MainEditorAssetHeader->remove assetBundle 
+        select assetBundle;
+        click remove-button;
+             should remove it from assetTreeRoot 1`] = `
+<article
+  className="wonder-asset-assetChildren"
+/>
+`;

test/integration/asset/composable_component/header/mainEditorAssetHeaderLoadAssetBundle_test.re

@@ -0,0 +1,71 @@
+open Wonder_jest;
+
+open Expect;
+
+open Expect.Operators;
+
+open Sinon;
+
+open Js.Promise;
+
+let _ =
+  describe("MainEditorAssetHeader->load assetBundle", () => {
+    let sandbox = getSandboxDefaultVal();
+
+    beforeEach(() => {
+      sandbox := createSandbox();
+
+      MainEditorSceneTool.initState(~sandbox, ());
+      MainEditorAssetTool.buildFakeFileReader();
+    });
+
+    afterEach(() => restoreSandbox(refJsObjToSandbox(sandbox^)));
+
+    describe("test load asset bundle", () => {
+      testPromise("should add to asset children", () =>
+        MainEditorAssetUploadTool.loadOneAssetBundle(~fileName="A.rab", ())
+        |> then_(uploadedAssetBundleNodeId =>
+             BuildComponentTool.buildAssetChildrenNode()
+             |> ReactTestTool.createSnapshotAndMatch
+             |> resolve
+           )
+      );
+
+      testPromise("should store assetBundle arrayBuffer", () => {
+        let assetBundle = Js.Typed_array.ArrayBuffer.make(20);
+
+        MainEditorAssetUploadTool.loadOneAssetBundle(
+          ~fileName="A.rab",
+          ~assetBundle,
+          (),
+        )
+        |> then_(uploadedAssetBundleNodeId =>
+             MainEditorAssetAssetBundleNodeTool.getAssetBundle(
+               uploadedAssetBundleNodeId,
+             )
+             |> StateLogicService.getEditorState
+             |> expect == assetBundle
+             |> resolve
+           );
+      });
+
+      describe("fix bug", () =>
+        testPromise(
+          "if load the same asset bundle in the same dir, should handle their name to be unique",
+          () =>
+          MainEditorAssetUploadTool.loadOneAssetBundle(~fileName="A.rab", ())
+          |> then_(uploadedAssetBundleNodeId =>
+               MainEditorAssetUploadTool.loadOneAssetBundle(
+                 ~fileName="A.rab",
+                 (),
+               )
+               |> then_(uploadedAssetBundleNodeId =>
+                    BuildComponentTool.buildAssetChildrenNode()
+                    |> ReactTestTool.createSnapshotAndMatch
+                    |> resolve
+                  )
+             )
+        )
+      );
+    });
+  });

test/integration/asset/composable_component/header/mainEditorAssetHeaderRemoveAssetBundle_test.re

@@ -0,0 +1,43 @@
+open Wonder_jest;
+
+open Expect;
+
+open Expect.Operators;
+
+open Sinon;
+
+open Js.Promise;
+
+let _ =
+  describe("MainEditorAssetHeader->remove assetBundle", () => {
+    let sandbox = getSandboxDefaultVal();
+
+    beforeEach(() => {
+      sandbox := createSandbox();
+      MainEditorSceneTool.initState(~sandbox, ());
+      MainEditorAssetTool.buildFakeFileReader();
+    });
+
+    afterEach(() => restoreSandbox(refJsObjToSandbox(sandbox^)));
+
+    describe(
+      {|
+        select assetBundle;
+        click remove-button;
+            |},
+      () =>
+      testPromise("should remove it from assetTreeRoot", () =>
+        MainEditorAssetUploadTool.loadOneAssetBundle()
+        |> then_(uploadedAssetBundleNodeId => {
+             MainEditorAssetHeaderOperateNodeTool.removeAssetBundleNode(
+               ~assetBundleNodeId=uploadedAssetBundleNodeId,
+               (),
+             );
+
+             BuildComponentTool.buildAssetChildrenNode()
+             |> ReactTestTool.createSnapshotAndMatch
+             |> resolve;
+           })
+      )
+    );
+  });

test/integration/asset/tool/MainEditorAssetAssetBundleNodeTool.re

@@ -0,0 +1,10 @@
+let getName = (nodeId, editorState) =>
+  AssetBundleNodeAssetService.getNodeName(
+    OperateTreeAssetEditorService.unsafeFindNodeById(nodeId, editorState)
+    |> AssetBundleNodeAssetService.getNodeData,
+  );
+
+let getAssetBundle = (nodeId, editorState) =>
+  AssetBundleNodeAssetService.getAssetBundle(
+    OperateTreeAssetEditorService.unsafeFindNodeById(nodeId, editorState),
+  );

test/integration/asset/tool/MainEditorAssetChildrenNodeTool.re

@@ -8,4 +8,8 @@ let selectMaterialNode =
 
 let selectFolderNode =
     (~nodeId, ~event=Obj.magic(-1), ~dispatchFunc=TestTool.getDispatch(), ()) =>
+  FileBox.Method.onSelect(nodeId, dispatchFunc, event);
+
+let selectAssetBundleNode =
+    (~nodeId, ~event=Obj.magic(-1), ~dispatchFunc=TestTool.getDispatch(), ()) =>
   FileBox.Method.onSelect(nodeId, dispatchFunc, event);

test/integration/asset/tool/MainEditorAssetHeaderOperateNodeTool.re

@@ -50,6 +50,15 @@ let removeScriptAttributeNode =
     ) =>
   removeNode(~dispatchFunc, ~uiState, ~nodeId=scriptAttributeNodeId, ());
 
+let removeAssetBundleNode =
+    (
+      ~dispatchFunc=TestTool.getDispatch(),
+      ~uiState=TestTool.buildEmptyAppState(),
+      ~assetBundleNodeId,
+      (),
+    ) =>
+  removeNode(~dispatchFunc, ~uiState, ~nodeId=assetBundleNodeId, ());
+
 let removeFolderNode =
     (
       ~dispatchFunc=TestTool.getDispatch(),

test/integration/asset/tool/MainEditorAssetUploadTool.re

@@ -76,4 +76,22 @@ let loadOneGLTFZip =
     BaseEventTool.buildGLTFZipFileEvent(fileName),
   )
   |> then_(() => uploadedWDBNodeId |> resolve);
+};
+
+let loadOneAssetBundle =
+    (
+      ~uiState=TestTool.buildEmptyAppState(),
+      ~dispatchFunc=TestTool.getDispatch(),
+      ~fileName="A.rab",
+      ~assetBundle=Js.Typed_array.ArrayBuffer.make(10),
+      (),
+    ) => {
+  let uploadedWDBNodeId = MainEditorAssetIdTool.getNewAssetId();
+
+  AssetHeaderFileLoadEventHandler.MakeEventHandler.pushUndoStackWithNoCopyEngineState(
+    (uiState, dispatchFunc),
+    WonderBsJszip.Zip.create,
+    BaseEventTool.buildAssetBundleFileEvent(fileName, assetBundle),
+  )
+  |> then_(() => uploadedWDBNodeId |> resolve);
 };