fix(storybook): fix FakeStore api (#1030)

Co-authored-by: Black-Hole <158blackhole@gmail.com>
netless-io · Oct 25, 2021 · 5ed7415 · 5ed7415
1 parent 72e8d1f
commit 5ed7415
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...es/flat-components/src/containers/CloudStorageContainer/CloudStorageContainer.stories.tsx b/...es/flat-components/src/containers/CloudStorageContainer/CloudStorageContainer.stories.tsx
@@ -28,6 +28,7 @@ const fakeStoreImplProps = [
     "onItemMenuClick",
     "onItemTitleClick",
     "onNewFileName",
+    "addOnlineH5",
 ] as const;
 
 type FakeStoreImplProps = typeof fakeStoreImplProps[number];
@@ -43,6 +44,7 @@ class FakeStore extends CloudStorageStore {
     public onItemMenuClick: FakeStoreConfig["onItemMenuClick"];
     public onItemTitleClick;
     public onNewFileName: FakeStoreConfig["onNewFileName"];
+    public addOnlineH5;
 
     public pendingUploadTasks: CloudStorageStore["pendingUploadTasks"] = [];
     public uploadingUploadTasks: CloudStorageStore["uploadingUploadTasks"] = [];
@@ -102,6 +104,7 @@ class FakeStore extends CloudStorageStore {
         this.onUploadCancel = config.onUploadCancel;
         this.onUploadPanelClose = config.onUploadPanelClose;
         this.onUploadRetry = config.onUploadRetry;
+        this.addOnlineH5 = config.addOnlineH5;
         this.onItemMenuClick = (fileUUID, menuKey) => {
             switch (menuKey) {
                 case "download": {
@@ -153,7 +156,7 @@ class FakeStore extends CloudStorageStore {
 
 function fakeStoreArgTypes(): ArgTypes {
     return fakeStoreImplProps.reduce((o, k) => {
-        o[k] = { table: { disable: true } };
+        o[k] = { table: { disable: true }, action: k };
         return o;
     }, {} as ArgTypes);
 }