Match DocumentReference

Author: schmidt-sebastian

packages/firestore/src/api/database.ts

@@ -649,7 +649,7 @@ export class Firestore
 
   batch(): PublicWriteBatch {
     this._ensureClientConfigured();
-    return new WriteBatch(this, this._firestoreClient!);
+    return new WriteBatch(this);
   }
 
   // Visible for testing.
@@ -815,10 +815,7 @@ export class WriteBatch implements PublicWriteBatch {
   private _mutations = [] as Mutation[];
   private _committed = false;
 
-  constructor(
-    private _firestore: Firestore,
-    private _firestoreClient: FirestoreClient
-  ) {}
+  constructor(private _firestore: Firestore) {}
 
   set<T>(
     documentRef: DocumentReference<T>,
@@ -932,7 +929,8 @@ export class WriteBatch implements PublicWriteBatch {
     this.verifyNotCommitted();
     this._committed = true;
     if (this._mutations.length > 0) {
-      return firestoreClientWrite(this._firestoreClient, this._mutations);
+      const firestoreClient = this._firestore._ensureClientConfigured();
+      return firestoreClientWrite(firestoreClient, this._mutations);
     }
 
     return Promise.resolve();