feat: refresh tests on update snapshot folder

ccontrols · Apr 23, 2021 · bd03dcd · bd03dcd
1 parent 1fd4a93
commit bd03dcd
Show file tree

Hide file tree

Showing 4 changed files with 297 additions and 258 deletions.
diff --git a/core/instrument/src/misc/chached-file.ts b/core/instrument/src/misc/chached-file.ts
@@ -40,6 +40,16 @@ export class CachedFileResource<T> {
         .digest('hex'),
     );
   };
+
+  private getKey = () => {
+    const version = '2';
+    const files = Array.isArray(this.filePath)
+      ? this.filePath
+      : [this.filePath];
+    return files
+      .reduce((acc, f) => acc.update(f), createHash('md5').update(version))
+      .digest('hex');
+  };
   /**
    *
    * @returns if the data is in the cache or undefined
@@ -56,7 +66,12 @@ export class CachedFileResource<T> {
       if (cacheModified >= fileModified) {
         const fileData = fs.readFileSync(cachedFileName, 'utf8');
         const json = JSON.parse(fileData);
-        return Object.keys(json).length ? json : undefined;
+        if (json.hasOwnProperty('key') && json.hasOwnProperty('data')) {
+          if (json.key === this.getKey()) {
+            return json['data'];
+          }
+        }
+        return undefined;
       }
     }
     return undefined;
@@ -68,6 +83,10 @@ export class CachedFileResource<T> {
    */
   set = (data: T | undefined): void => {
     const cachedFileName = this.getCachedFile();
-    fs.writeFileSync(cachedFileName, JSON.stringify(data || {}));
+    const json = {
+      key: this.getKey(),
+      data,
+    };
+    fs.writeFileSync(cachedFileName, JSON.stringify(json));
   };
 }
diff --git a/core/instrument/src/misc/jest-tests.ts b/core/instrument/src/misc/jest-tests.ts
@@ -1,4 +1,5 @@
 import path from 'path';
+import fs from 'fs';
 import {
   JestConfig,
   runProjectTests,
@@ -27,19 +28,31 @@ export const extractTests = async (
       acc: {
         testFiles: string[];
         coverageFiles: string[];
+        snapshotFiles: string[];
       },
       file,
     ) => {
       acc.testFiles.push(...getRelatedTests(file));
-      acc.coverageFiles.push(file);
+      const snapshotFolder = path.join(path.dirname(file), '__snapshots__');
+      if (fs.existsSync(snapshotFolder)) {
+        acc.snapshotFiles.push(
+          ...fs
+            .readdirSync(snapshotFolder)
+            .map(f => path.resolve(snapshotFolder, f)),
+        );
+      }
+      if (!acc.coverageFiles.includes(file)) {
+        acc.coverageFiles.push(file);
+      }
       return acc;
     },
-    { testFiles: [], coverageFiles: [] },
+    { testFiles: [], coverageFiles: [], snapshotFiles: [] },
   );
   if (tests.testFiles.length) {
     const cached = new CachedFileResource<JestTests>('jest-tests', files[0], [
       ...tests.testFiles,
       ...tests.coverageFiles,
+      ...tests.snapshotFiles,
     ]);
     const cachedResults = cached.get();
     if (cachedResults) {

diff --git a/core/loader/src/loader.ts b/core/loader/src/loader.ts
@@ -32,24 +32,31 @@ async function loader(this: WebpackLoaderContext): Promise<string> {
     if (store?.doc) {
       log('loaded: ', filePath);
       if (store.stories && store.components && store.packages) {
-        const dependencies: string[] = [];
+        const fileDependencies: string[] = [];
+        const folderDependencies: string[] = [];
         Object.values(store.components).forEach(component => {
           if (component.request) {
-            dependencies.push(component.request as string);
+            fileDependencies.push(component.request as string);
             if (component.jest) {
               const componentFolder = path.dirname(component.request);
               component.jest.results.forEach(r => {
-                dependencies.push(
-                  path.resolve(componentFolder, r.testFilePath),
+                const testFilePath = path.resolve(
+                  componentFolder,
+                  r.testFilePath,
+                );
+                fileDependencies.push(testFilePath);
+                folderDependencies.push(
+                  path.join(path.dirname(testFilePath), '__snapshots__'),
                 );
               });
               Object.keys(component.jest.coverage).forEach(f => {
-                dependencies.push(path.resolve(componentFolder, f));
+                fileDependencies.push(path.resolve(componentFolder, f));
               });
             }
           }
         });
-        new Set(dependencies).forEach(d => this.addDependency(d));
+        new Set(fileDependencies).forEach(d => this.addDependency(d));
+        new Set(folderDependencies).forEach(d => this.addContextDependency(d));
         addStoriesDoc(filePath, this._compilation.records.hash, {
           stories: store.stories,
           components: store.components,