test(mocks): explicit jest spy instance

sabertazimi · Aug 10, 2021 · d5e0cfd · d5e0cfd
1 parent f19f668
commit d5e0cfd
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 36 deletions.
diff --git a/src/hooks/useBuildTime.test.ts b/src/hooks/useBuildTime.test.ts
@@ -4,18 +4,22 @@ import useBuildTime from './useBuildTime';
 const buildTime = new Date(2021, 0, 1, 8, 0, 0).toLocaleString('zh-CN');
 
 describe('useBuildTime', () => {
-  beforeAll(() => {
-    jest.spyOn(gatsby, 'useStaticQuery').mockImplementation(() => {
-      return {
-        site: {
-          buildTime,
-        },
-      };
-    });
+  let mockUseStaticQuery: jest.SpyInstance;
+
+  beforeEach(() => {
+    mockUseStaticQuery = jest
+      .spyOn(gatsby, 'useStaticQuery')
+      .mockImplementation(() => {
+        return {
+          site: {
+            buildTime,
+          },
+        };
+      });
   });
 
-  afterAll(() => {
-    (gatsby.useStaticQuery as unknown as jest.SpyInstance).mockRestore();
+  afterEach(() => {
+    mockUseStaticQuery.mockRestore();
   });
 
   test('should return date time', () => {

diff --git a/src/hooks/usePostsMetadata.test.ts b/src/hooks/usePostsMetadata.test.ts
@@ -28,13 +28,15 @@ describe('usePostsMetadata', () => {
   test.each(testIterator)(
     `should return correct [%i / ${testCount}] post metadata`,
     (index) => {
-      jest.spyOn(gatsby, 'useStaticQuery').mockImplementation(() => {
-        return {
-          allMarkdownRemark: {
-            edges: Posts,
-          },
-        };
-      });
+      const mockUseStaticQuery = jest
+        .spyOn(gatsby, 'useStaticQuery')
+        .mockImplementation(() => {
+          return {
+            allMarkdownRemark: {
+              edges: Posts,
+            },
+          };
+        });
 
       const { posts, tags } = usePostsMetadata();
 
@@ -73,20 +75,22 @@ describe('usePostsMetadata', () => {
       expect(tags['Frontend Development']).toBe(5);
       expect(tags['Web Development']).toBe(5);
 
-      (gatsby.useStaticQuery as unknown as jest.SpyInstance).mockRestore();
+      mockUseStaticQuery.mockRestore();
     }
   );
 
   test.each(testIterator)(
     `should return correct [%i / ${testCount}] post metadata with partial data`,
     (index) => {
-      jest.spyOn(gatsby, 'useStaticQuery').mockImplementation(() => {
-        return {
-          allMarkdownRemark: {
-            edges: BasePosts,
-          },
-        };
-      });
+      const mockUseStaticQuery = jest
+        .spyOn(gatsby, 'useStaticQuery')
+        .mockImplementation(() => {
+          return {
+            allMarkdownRemark: {
+              edges: BasePosts,
+            },
+          };
+        });
 
       const { posts, tags } = usePostsMetadata();
 
@@ -118,7 +122,7 @@ describe('usePostsMetadata', () => {
       // Check correct empty tags data of post are returned
       expect(tags).toMatchObject({});
 
-      (gatsby.useStaticQuery as unknown as jest.SpyInstance).mockRestore();
+      mockUseStaticQuery.mockRestore();
     }
   );
 });
diff --git a/src/hooks/useSiteMetadata.test.ts b/src/hooks/useSiteMetadata.test.ts
@@ -25,15 +25,25 @@ const siteMetadata = {
 };
 
 describe('useSiteMetadata', () => {
-  test(`should return correct site metadata`, () => {
-    jest.spyOn(gatsby, 'useStaticQuery').mockImplementation(() => {
-      return {
-        site: {
-          siteMetadata,
-        },
-      };
-    });
+  let mockUseStaticQuery: jest.SpyInstance;
+
+  beforeEach(() => {
+    mockUseStaticQuery = jest
+      .spyOn(gatsby, 'useStaticQuery')
+      .mockImplementation(() => {
+        return {
+          site: {
+            siteMetadata,
+          },
+        };
+      });
+  });
 
+  afterEach(() => {
+    mockUseStaticQuery.mockRestore();
+  });
+
+  test(`should return correct site metadata`, () => {
     const {
       title,
       author,
@@ -66,7 +76,5 @@ describe('useSiteMetadata', () => {
         siteMetadata['bookList'][index]['description']
       );
     });
-
-    (gatsby.useStaticQuery as unknown as jest.SpyInstance).mockRestore();
   });
 });