Also update the lite client.

Author: wilhuff

packages/firestore/lite/src/api/reference.ts

@@ -307,7 +307,12 @@ export function setDoc<T>(
     ref.firestore._settings!
   );
 
-  const parsed = dataReader.parseSetData('setDoc', convertedValue, options);
+  const parsed = dataReader.parseSetData(
+    'setDoc',
+    ref._key,
+    convertedValue,
+    options
+  );
 
   return configureClient.then(datastore =>
     invokeCommitRpc(
@@ -349,12 +354,17 @@ export function updateDoc(
   ) {
     parsed = dataReader.parseUpdateVarargs(
       'updateDoc',
+      ref._key,
       fieldOrUpdateData,
       value,
       moreFieldsAndValues
     );
   } else {
-    parsed = dataReader.parseUpdateData('updateDoc', fieldOrUpdateData);
+    parsed = dataReader.parseUpdateData(
+      'updateDoc',
+      ref._key,
+      fieldOrUpdateData
+    );
   }
 
   return configureClient.then(datastore =>
@@ -406,7 +416,7 @@ export function addDoc<T>(
     collRef.firestore._databaseId,
     collRef.firestore._settings!
   );
-  const parsed = dataReader.parseSetData('addDoc', convertedValue);
+  const parsed = dataReader.parseSetData('addDoc', docRef._key, convertedValue);
 
   return configureClient
     .then(datastore =>

packages/firestore/lite/src/api/transaction.ts

@@ -108,6 +108,7 @@ export class Transaction implements firestore.Transaction {
     );
     const parsed = this._dataReader.parseSetData(
       'Transaction.set',
+      ref._key,
       convertedValue,
       options
     );
@@ -140,13 +141,15 @@ export class Transaction implements firestore.Transaction {
     ) {
       parsed = this._dataReader.parseUpdateVarargs(
         'Transaction.update',
+        ref._key,
         fieldOrUpdateData,
         value,
         moreFieldsAndValues
       );
     } else {
       parsed = this._dataReader.parseUpdateData(
         'Transaction.update',
+        ref._key,
         fieldOrUpdateData
       );
     }

packages/firestore/lite/src/api/write_batch.ts

@@ -73,6 +73,7 @@ export class WriteBatch implements firestore.WriteBatch {
 
     const parsed = this._dataReader.parseSetData(
       'WriteBatch.set',
+      ref._key,
       convertedValue,
       options
     );
@@ -109,13 +110,15 @@ export class WriteBatch implements firestore.WriteBatch {
     ) {
       parsed = this._dataReader.parseUpdateVarargs(
         'WriteBatch.update',
+        ref._key,
         fieldOrUpdateData,
         value,
         moreFieldsAndValues
       );
     } else {
       parsed = this._dataReader.parseUpdateData(
         'WriteBatch.update',
+        ref._key,
         fieldOrUpdateData
       );
     }