Firebase mock refactoring

__mocks__/gatsby-plugin-firebase.js

@@ -1,21 +1,51 @@
 import path from 'path';
 import fs from 'fs';
 
-const demoResumeId = 'demore';
-const emptyResumeId = 'mtre01';
+let onAuthStateChangedObservers = [];
 let resumesDictionary = {};
 let useDemoResume = false;
 
-let onAuthStateChangedObservers = [];
-const testUser = {
+const auth = () => {
+  const testUser = {
     displayName: 'Test User',
     email: 'noemail@noemail.com',
+  };
+
+  const __init = () => {
+    onAuthStateChangedObservers = [];
+  };
+
+  const onAuthStateChanged = (observer) => {
+    onAuthStateChangedObservers.push(observer);
+
+    return () => {
+      onAuthStateChangedObservers = onAuthStateChangedObservers.filter(
+        (observer) => observer !== observer,
+      );
+    };
+  };
+
+  const signInAnonymously = async () => {
+    onAuthStateChangedObservers.forEach((observer) => observer(testUser));
+
+    var result = await Promise.resolve(testUser);
+    return result;
+  };
+
+  return {
+    __init,
+    onAuthStateChanged,
+    signInAnonymously,
+  };
 };
 
-const __init = () => {
+const database = () => {
+  const demoResumeId = 'demore';
+  const emptyResumeId = 'mtre01';
+
+  const __init = () => {
     resumesDictionary = {};
     useDemoResume = false;
-  onAuthStateChangedObservers = [];
 
     const demoResume = __readFile('../src/data/demoState.json');
     resumesDictionary[demoResumeId] = demoResume;
@@ -32,58 +62,38 @@ const __init = () => {
       date.setMonth(date.getMonth() - 2);
       resume.createdAt = date.valueOf();
     }
-};
+  };
 
-const __readFile = (fileRelativePath) => {
+  const __readFile = (fileRelativePath) => {
     const fileAbsolutePath = path.resolve(__dirname, fileRelativePath);
     const fileBuffer = fs.readFileSync(fileAbsolutePath);
     const fileData = JSON.parse(fileBuffer);
     return fileData;
-};
+  };
 
-const __getResume = () => {
+  const __getResume = () => {
     return useDemoResume
       ? resumesDictionary[demoResumeId]
       : resumesDictionary[emptyResumeId];
-};
-
-const __useDemoResume = (value) => {
-  useDemoResume = value;
-};
-
-const __getResumeId = () => {
-  return __getResume().id;
-};
-
-const onAuthStateChanged = (observer) => {
-  onAuthStateChangedObservers.push(observer);
-
-  return () => {
-    onAuthStateChangedObservers = onAuthStateChangedObservers.filter(
-      (observer) => observer !== observer,
-    );
   };
-};
 
-const signInAnonymously = async () => {
-  onAuthStateChangedObservers.forEach((observer) => observer(testUser));
+  const __useDemoResume = (value) => {
+    useDemoResume = value;
+  };
 
-  var result = await Promise.resolve(testUser);
-  return result;
-};
-
-export default {
-  database: jest.fn().mockReturnValue({
+  return {
+    __init,
+    __getResume,
+    __useDemoResume,
     ref: jest.fn().mockReturnValue({
       once: jest.fn().mockResolvedValue({
         val: jest.fn(__getResume),
       }),
     }),
-  }),
-  auth: jest.fn().mockReturnValue({
-    onAuthStateChanged: onAuthStateChanged,
-    signInAnonymously: signInAnonymously,
-  }),
+  };
 };
 
-export { __init, __useDemoResume, __getResumeId };
+export default {
+  auth,
+  database,
+};

src/pages/app/__tests__/builder.mockFirebase.test.js

@@ -1,11 +1,7 @@
 import React from 'react';
-import { render, cleanup } from '@testing-library/react';
+import { act, render, cleanup } from '@testing-library/react';
 
-import firebase, {
-  __init as firebaseMockInit,
-  __useDemoResume as firebaseMockUseDemoResume,
-  __getResumeId as firebaseMockGetResumeId,
-} from 'gatsby-plugin-firebase';
+import firebaseMock from 'gatsby-plugin-firebase';
 
 import { UserProvider } from '../../../contexts/UserContext';
 import { DatabaseProvider } from '../../../contexts/DatabaseContext';
@@ -15,29 +11,34 @@ import { StorageProvider } from '../../../contexts/StorageContext';
 import Builder from '../builder';
 
 beforeEach(() => {
-  firebaseMockInit();
+  firebaseMock.auth().__init();
+  firebaseMock.database().__init();
 });
 
 afterEach(cleanup);
 
 it('renders correctly', async () => {
-  firebaseMockUseDemoResume(true);
-  const resumeId = firebaseMockGetResumeId();
+  firebaseMock.database().__useDemoResume(true);
+  const resume = firebaseMock.database().__getResume();
 
-  const container = render(
+  const element = (
     <UserProvider>
       <DatabaseProvider>
         <ResumeProvider>
           <StorageProvider>
-            <Builder id={resumeId} />
+            <Builder id={resume.id} />
           </StorageProvider>
         </ResumeProvider>
       </DatabaseProvider>
-    </UserProvider>,
+    </UserProvider>
   );
+  let container = null;
 
-  await firebase.auth().signInAnonymously();
+  act(() => {
+    container = render(element);
+  });
+
+  await firebaseMock.auth().signInAnonymously();
 
   expect(container).toBeTruthy();
-  //expect(container).toBeInTheDocument();
 });