Test for experimental_TEST_PROVIDER addon type

storybookjs · Sep 13, 2024 · b423331 · b423331
1 parent 40ca30b
commit b423331
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/code/core/src/manager-api/tests/addons.test.js b/code/core/src/manager-api/tests/addons.test.js
@@ -19,11 +19,21 @@ const PANELS = {
   },
 };
 
+const TEST_PROVIDERS = {
+  'storybook/test/test-provider': {
+    id: 'storybook/test/test-provider',
+    title: 'Component tests',
+  },
+}
+
 const provider = {
   getElements(type) {
     if (type === types.PANEL) {
       return PANELS;
     }
+    if (type === types.experimental_TEST_PROVIDER) {
+      return TEST_PROVIDERS;
+    }
     return null;
   },
 };
@@ -38,14 +48,13 @@ const store = {
 describe('Addons API', () => {
   describe('#getElements', () => {
     it('should return provider elements', () => {
-      // given
       const { api } = initAddons({ provider, store });
 
-      // when
       const panels = api.getElements(types.PANEL);
-
-      // then
       expect(panels).toBe(PANELS);
+
+      const testProviders = api.getElements(types.experimental_TEST_PROVIDER);
+      expect(testProviders).toBe(TEST_PROVIDERS);
     });
   });