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.
@@ -814,11 +814,11 @@ export class Transaction implements PublicTransaction {
 export class WriteBatch implements PublicWriteBatch {
   private _mutations = [] as Mutation[];
   private _committed = false;
+  private _firestoreClient: FirestoreClient;
 
-  constructor(
-    private _firestore: Firestore,
-    private _firestoreClient: FirestoreClient
-  ) {}
+  constructor(private _firestore: Firestore) {
+    this._firestoreClient = _firestore._ensureClientConfigured();
+  }
 
   set<T>(
     documentRef: DocumentReference<T>,