Add tests for mapping behaviour in #22169

storybookjs · Apr 28, 2023 · 4ad3dff · 4ad3dff
1 parent 7090746
commit 4ad3dff
Showing 1 changed file with 45 additions and 0 deletions.
diff --git a/code/lib/preview-api/src/modules/store/csf/prepareStory.test.ts b/code/lib/preview-api/src/modules/store/csf/prepareStory.test.ts
@@ -510,6 +510,51 @@ describe('prepareStory', () => {
     });
   });
 
+  describe('mapping', () => {
+    it('maps labels to values in prepareContext', () => {
+      const story = prepareStory(
+        {
+          id,
+          name,
+          argTypes: {
+            one: { name: 'one', mapping: { 1: 'mapped-1' } },
+          },
+          moduleExport,
+        },
+        { id, title },
+        { render: jest.fn() }
+      );
+
+      const context = story.prepareContext({ args: { one: 1 }, ...story } as any);
+      expect(context).toMatchObject({
+        args: { one: 'mapped-1' },
+      });
+    });
+
+    it('maps arrays of labels to values in prepareContext', () => {
+      const story = prepareStory(
+        {
+          id,
+          name,
+          argTypes: {
+            one: { name: 'one', mapping: { 1: 'mapped-1' } },
+          },
+          moduleExport,
+        },
+        { id, title },
+        { render: jest.fn() }
+      );
+
+      const context = story.prepareContext({
+        args: { one: [1, 1] },
+        ...story,
+      } as any);
+      expect(context).toMatchObject({
+        args: { one: ['mapped-1', 'mapped-1'] },
+      });
+    });
+  });
+
   describe('with `FEATURES.argTypeTargetsV7`', () => {
     beforeEach(() => {
       global.FEATURES = { argTypeTargetsV7: true };