Enable firestore sdk to talk to emulator

packages/firestore/src/api/credentials.ts

@@ -18,7 +18,7 @@ import { User } from '../auth/user';
 import { assert, fail } from '../util/assert';
 import { Code, FirestoreError } from '../util/error';
 import { FirebaseApp } from '@firebase/app-types';
-import { _FirebaseApp } from '@firebase/app-types/private';
+import { _FirebaseApp, FirebaseAuthTokenData } from '@firebase/app-types/private';
 
 // TODO(mikelehen): This should be split into multiple files and probably
 // moved to an auth/ folder to match other platforms.
@@ -145,6 +145,8 @@ export class FirebaseCredentialsProvider implements CredentialsProvider {
 
   private forceRefresh = false;
 
+  private tokenOverride: string = null;
+
   constructor(private readonly app: FirebaseApp) {
     // We listen for token changes but all we really care about is knowing when
     // the uid may have changed.
@@ -160,6 +162,7 @@ export class FirebaseCredentialsProvider implements CredentialsProvider {
     };
 
     this.userCounter = 0;
+    this.tokenOverride = app.options.tokenOverride;
 
     // Will fire at least once where we set this.currentUser
     (this.app as _FirebaseApp).INTERNAL.addAuthTokenListener(
@@ -179,7 +182,13 @@ export class FirebaseCredentialsProvider implements CredentialsProvider {
     const initialUserCounter = this.userCounter;
     const forceRefresh = this.forceRefresh;
     this.forceRefresh = false;
-    return (this.app as _FirebaseApp).INTERNAL.getToken(forceRefresh).then(
+    var token: Promise<FirebaseAuthTokenData>;
+    if (this.tokenOverride != null) {
+      token = Promise.resolve({ accessToken: this.tokenOverride });
+    } else {
+      token = (this.app as _FirebaseApp).INTERNAL.getToken(forceRefresh);
+    }
+    return token.then(
       tokenData => {
         // Cancel the request since the user changed while the request was
         // outstanding so the response is likely for a previous user (which

packages/testing/index.ts

@@ -26,5 +26,6 @@ export {
   assertSucceeds,
   initializeAdminApp,
   initializeTestApp,
+  initializeFirestoreTestApp,
   loadDatabaseRules
 } from './src/api';

packages/testing/src/api/index.ts

@@ -14,9 +14,12 @@
  * limitations under the License.
  */
 
+import { firebase } from '@firebase/app';
 import * as admin from 'firebase-admin';
 import request from 'request-promise';
 import * as fs from 'fs';
+import { FirebaseApp } from '@firebase/app-types';
+import * as util from '@firebase/util';
 
 const DBURL = 'http://localhost:9000';
 
@@ -64,6 +67,30 @@ export function initializeTestApp(options: any): admin.app.App {
   );
 }
 
+export function initializeFirestoreTestApp(options: any): FirebaseApp {
+  if (!('projectId' in options)) {
+    throw new Error('projectId not specified');
+  }
+  if (typeof options.auth != 'object') {
+    throw new Error('auth must be an object');
+  }
+  var header = {
+    alg: "RS256",
+    kid: "fakekid"
+  };
+  var fakeToken = [
+    util.base64.encodeString(JSON.stringify(header),/*webSafe=*/true),
+    util.base64.encodeString(JSON.stringify(options.auth),/*webSafe=*/true),
+    "fakesignature"
+  ].join(".");
+  return firebase.initializeApp({
+      projectId: options.projectId,
+      tokenOverride: fakeToken
+    },
+    'app-' + new Date().getTime() + "-" + Math.random()
+  );
+}
+
 export type LoadDatabaseRulesOptions = {
   databaseName: String;
   rules: String;

packages/testing/test/database.test.ts

@@ -84,6 +84,32 @@ describe('Testing Module Tests', function() {
     );
   });
 
+  it('initializeFirestoreTestApp() throws if no projectId', function() {
+    expect(firebase.initializeFirestoreTestApp.bind(null, { auth: {} })).to.throw(
+      /projectId not specified/
+    );
+    expect(
+      firebase.initializeFirestoreTestApp.bind(null, { projectId: 'foo', auth: {} })
+    ).to.not.throw();
+  });
+
+  it('initializeFirestoreTestApp() throws if auth is not an object', function() {
+    expect(firebase.initializeFirestoreTestApp.bind(null, { projectId: 'a', auth: 'b' })).to.throw(
+      /auth must be an object/
+    );
+    expect(
+      firebase.initializeFirestoreTestApp.bind(null, { projectId: 'a', auth: {} })
+    ).to.not.throw();
+  });
+
+  it('initializeFirestoreTestApp() uses specified auth.', function() {
+    let app = firebase.initializeFirestoreTestApp({ projectId: 'foo', auth: {} });
+    expect(app.options).to.have.any.keys('tokenOverride');
+
+    app = firebase.initializeFirestoreTestApp({ projectId: 'foo', auth: { uid: 'alice' }});
+    expect(app.options).to.have.any.keys('tokenOverride');
+  });
+
   it('loadDatabaseRules() throws if no databaseName or rulesPath', async function() {
     expect(firebase.loadDatabaseRules.bind(null, {})).to.throw(
       /databaseName not specified/