✨ feat(component-asset): 优化 createTestAssetStore ，便于下游测试

ant-design · Jun 15, 2023 · e58eb84 · e58eb84
1 parent e8a61f3
commit e58eb84
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/ComponentAsset/store/createTestAssetStore.ts b/src/ComponentAsset/store/createTestAssetStore.ts
@@ -24,11 +24,13 @@ export const createTestAssetStore = <T>(createStore: CreateAssetStore<T>) => {
   const useAssetStore = create<T>(createStore);
   const { getConfig, setConfig } = useEditorStore.getState();
 
-  useAssetStore.setState({ getConfig, setConfig } as any);
+  const init = () => {
+    useAssetStore.setState({ getConfig, setConfig } as any);
 
-  useEditorStore.subscribe(({ config }) => {
-    useAssetStore.setState(config);
-  });
+    useEditorStore.subscribe(({ config }) => {
+      useAssetStore.setState(config);
+    });
+  };
 
-  return useAssetStore;
+  return { useAssetStore, init, useEditorStore };
 };