Add unit test

Author: mansisampat

packages/auth/src/core/auth/firebase_internal.test.ts

@@ -20,7 +20,7 @@ import { expect, use } from 'chai';
 import * as sinon from 'sinon';
 import chaiAsPromised from 'chai-as-promised';
 
-import { testAuth, testUser } from '../../../test/helpers/mock_auth';
+import { regionalTestAuth, testAuth, testUser } from '../../../test/helpers/mock_auth';
 import { AuthInternal } from '../../model/auth';
 import { UserInternal } from '../../model/user';
 import { AuthInterop } from './firebase_internal';
@@ -37,6 +37,9 @@ describe('core/auth/firebase_internal', () => {
 
   afterEach(() => {
     sinon.restore();
+    delete (auth as unknown as Record<string, unknown>)[
+        '_initializationPromise'
+      ];
   });
 
   context('getUid', () => {
@@ -215,3 +218,22 @@ describe('core/auth/firebase_internal', () => {
     });
   });
 });
+
+describe('core/auth/firebase_internal - Regional Firebase Auth', () => {
+  let regionalAuth: AuthInternal;
+  let regionalAuthInternal: AuthInterop;
+  beforeEach(async () => {
+    regionalAuth = await regionalTestAuth();
+    regionalAuthInternal = new AuthInterop(regionalAuth);
+  });
+
+  afterEach(() => {
+    sinon.restore();
+  });
+
+  context('getFirebaseToken', () => {
+    it('returns null if firebase token is undefined', async () => {
+      expect(await regionalAuthInternal.getFirebaseToken()).to.be.null;
+    });
+  });
+});

packages/auth/src/core/auth/firebase_internal.ts

@@ -54,6 +54,7 @@ export class AuthInterop implements FirebaseAuthInternal {
 
   async getFirebaseToken(): Promise<{ accessToken: string } | null> {
     this.assertAuthConfigured();
+    await this.auth._initializationPromise;
     this.assertRegionalAuthConfigured();
     if (!this.auth.firebaseToken) {
       return null;

packages/auth/test/helpers/mock_auth.ts

@@ -118,7 +118,10 @@ export async function testAuth(
   return auth;
 }
 
-export async function regionalTestAuth(): Promise<TestAuth> {
+export async function regionalTestAuth(
+  popupRedirectResolver?: PopupRedirectResolver,
+  persistence = new MockPersistenceLayer(),
+  skipAwaitOnInit?: boolean): Promise<TestAuth> {
   const tenantConfig = { 'location': 'us', 'tenantId': 'tenant-1' };
   const auth: TestAuth = new AuthImpl(
     FAKE_APP,
@@ -135,6 +138,15 @@ export async function regionalTestAuth(): Promise<TestAuth> {
     },
     tenantConfig
   ) as TestAuth;
+  if (skipAwaitOnInit) {
+    // This is used to verify scenarios where auth flows (like signInWithRedirect) are invoked before auth is fully initialized.
+    // eslint-disable-next-line @typescript-eslint/no-floating-promises
+    auth._initializeWithPersistence([persistence], popupRedirectResolver);
+  } else {
+    await auth._initializeWithPersistence([persistence], popupRedirectResolver);
+  }
+  auth.persistenceLayer = persistence;
+  auth.settings.appVerificationDisabledForTesting = true;
   return auth;
 }