feat: load test files from examples

core/instrument/test/extract-component.test.ts

@@ -1,31 +1,22 @@
-import * as path from 'path';
-import * as fs from 'fs';
+import { loadTestFiles } from './loadTestFiles';
 import { defaultParserOptions, defaultResolveOptions } from '../src/index';
 import { extractComponent } from '../src/babel/extract-component';
 
 describe('extract-component', () => {
-  const folderName = path.join(__dirname, 'examples', 'extract-component');
-  const fileNames = fs.readdirSync(folderName);
-  // .filter(fn => fn === 'node-modules-source.js');
-  fileNames.forEach(file => {
-    const fileName = path.join(folderName, file);
-    it(file, async () => {
-      expect(
-        await extractComponent('Button', fileName, undefined, {
-          parser: defaultParserOptions,
-          resolve: defaultResolveOptions,
-          component: {
-            resolveFile: (componentName: string, filePath: string) => {
-              if (filePath.includes('/theme-ui/dist')) {
-                return `${
-                  filePath.split('/theme-ui/dist')[0]
-                }/@theme-ui/components/src/${componentName}.js`;
-              }
-              return filePath;
-            },
-          },
-        }),
-      ).toMatchSnapshot();
+  loadTestFiles(async (fileName: string) => {
+    return await extractComponent('Button', fileName, undefined, {
+      parser: defaultParserOptions,
+      resolve: defaultResolveOptions,
+      component: {
+        resolveFile: (componentName: string, filePath: string) => {
+          if (filePath.includes('/theme-ui/dist')) {
+            return `${
+              filePath.split('/theme-ui/dist')[0]
+            }/@theme-ui/components/src/${componentName}.js`;
+          }
+          return filePath;
+        },
+      },
     });
-  });
+  }, 'extract-component');
 });

core/instrument/test/loadTestFiles.ts

@@ -0,0 +1,17 @@
+import * as path from 'path';
+import * as fs from 'fs';
+
+export type LoadTestCallbackFn = (fileName: string) => any;
+
+export const loadTestFiles = (callback: LoadTestCallbackFn, ...rest) => {
+  const folderName = path.join(__dirname, 'examples', ...rest);
+  const fileNames = fs.readdirSync(folderName);
+
+  // .filter(fn => fn === 'node-modules-source.js');
+  fileNames.forEach(file => {
+    const fileName = path.join(folderName, file);
+    it(file, async () => {
+      expect(await callback(fileName)).toMatchSnapshot();
+    });
+  });
+};