Adding Unit test

Author: mansisampat

packages/auth/src/core/auth/auth_impl.ts

@@ -460,9 +460,7 @@ export class AuthImpl implements AuthInternal, _FirebaseService {
   async _updateFirebaseToken(
     firebaseToken: FirebaseToken | null
   ): Promise<void> {
-    if (firebaseToken) {
       this.firebaseToken = firebaseToken;
-    }
   }
 
   async signOut(): Promise<void> {

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

@@ -20,7 +20,11 @@ import { expect, use } from 'chai';
 import * as sinon from 'sinon';
 import chaiAsPromised from 'chai-as-promised';
 
-import { regionalTestAuth, 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';
@@ -38,8 +42,8 @@ describe('core/auth/firebase_internal', () => {
   afterEach(() => {
     sinon.restore();
     delete (auth as unknown as Record<string, unknown>)[
-        '_initializationPromise'
-      ];
+      '_initializationPromise'
+    ];
   });
 
   context('getUid', () => {
@@ -222,9 +226,12 @@ describe('core/auth/firebase_internal', () => {
 describe('core/auth/firebase_internal - Regional Firebase Auth', () => {
   let regionalAuth: AuthInternal;
   let regionalAuthInternal: AuthInterop;
+  let now: number;
   beforeEach(async () => {
     regionalAuth = await regionalTestAuth();
     regionalAuthInternal = new AuthInterop(regionalAuth);
+    now = Date.now();
+    sinon.stub(Date, 'now').returns(now);
   });
 
   afterEach(() => {
@@ -235,5 +242,37 @@ describe('core/auth/firebase_internal - Regional Firebase Auth', () => {
     it('returns null if firebase token is undefined', async () => {
       expect(await regionalAuthInternal.getFirebaseToken()).to.be.null;
     });
+
+    it('returns the id token correctly', async () => {
+      await regionalAuth._updateFirebaseToken({
+        token: 'access-token',
+        expirationTime: now + 300_000
+      });
+      expect(await regionalAuthInternal.getFirebaseToken()).to.eql({
+        accessToken: 'access-token'
+      });
+    });
+
+    it('logs out the the id token expires in next 30 seconds', async () => {
+      expect(await regionalAuthInternal.getFirebaseToken()).to.be.null;
+    });
+
+    it('logs out if token has expired', async () => {
+      await regionalAuth._updateFirebaseToken({
+        token: 'access-token',
+        expirationTime: now - 5_000
+      });
+      expect(await regionalAuthInternal.getFirebaseToken()).to.null;
+      expect(regionalAuth.firebaseToken).to.null;
+    });
+
+    it('logs out if token is expiring in next 5 seconds', async () => {
+      await regionalAuth._updateFirebaseToken({
+        token: 'access-token',
+        expirationTime: now + 5_000
+      });
+      expect(await regionalAuthInternal.getFirebaseToken()).to.null;
+      expect(regionalAuth.firebaseToken).to.null;
+    });
   });
 });

packages/auth/test/helpers/mock_auth.ts

@@ -121,7 +121,8 @@ export async function testAuth(
 export async function regionalTestAuth(
   popupRedirectResolver?: PopupRedirectResolver,
   persistence = new MockPersistenceLayer(),
-  skipAwaitOnInit?: boolean): Promise<TestAuth> {
+  skipAwaitOnInit?: boolean
+): Promise<TestAuth> {
   const tenantConfig = { 'location': 'us', 'tenantId': 'tenant-1' };
   const auth: TestAuth = new AuthImpl(
     FAKE_APP,