Accept FIRTimestamp where NSDate is currently accepted as a parameter (#823)

1 files changed, 7 insertions, 0 deletions

diff --git a/Firestore/Source/API/FSTUserDataConverter.mm b/Firestore/Source/API/FSTUserDataConverter.mm
index f97aea1..5f89b8e 100644
--- a/Firestore/Source/API/FSTUserDataConverter.mm
+++ b/Firestore/Source/API/FSTUserDataConverter.mm
@@ -539,6 +539,13 @@ typedef NS_ENUM(NSInteger, FSTUserDataSource) {
   } else if ([input isKindOfClass:[NSDate class]]) {
     return [FSTTimestampValue timestampValue:[FIRTimestamp timestampWithDate:input]];
 
+  } else if ([input isKindOfClass:[FIRTimestamp class]]) {
+    FIRTimestamp *originalTimestamp = (FIRTimestamp *)input;
+    FIRTimestamp *truncatedTimestamp =
+        [FIRTimestamp timestampWithSeconds:originalTimestamp.seconds
+                               nanoseconds:originalTimestamp.nanoseconds / 1000 * 1000];
+    return [FSTTimestampValue timestampValue:truncatedTimestamp];
+
   } else if ([input isKindOfClass:[FIRGeoPoint class]]) {
     return [FSTGeoPointValue geoPointValue:input];