Merge pull request #21 from alickbass/fix-decimal-encoding

Fix handling decimal value
alickbass · Feb 11, 2018 · b0e8686 · b0e8686
2 parents df9dfe1 + 0c60f70
commit b0e8686
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/CodableFirebase/Encoder.swift b/CodableFirebase/Encoder.swift
@@ -383,6 +383,8 @@ extension _FirebaseEncoder {
             return try self.box((value as! Data))
         } else if T.self == URL.self || T.self == NSURL.self {
             return self.box((value as! URL).absoluteString)
+        } else if T.self == Decimal.self || T.self == NSDecimalNumber.self {
+            return (value as! NSDecimalNumber)
         } else if options.skipFirestoreTypes && (value is FirestoreEncodable) {
             guard let value = value as? NSObject else {
                 throw DocumentReferenceError.typeIsNotNSObject

diff --git a/CodableFirebaseTests/TestCodableFirestore.swift b/CodableFirebaseTests/TestCodableFirestore.swift
@@ -109,6 +109,11 @@ class TestCodableFirestore: XCTestCase {
         _testRoundTrip(of: TopLevelWrapper(date), expected: ["value": date])
     }
 
+    func testDecimalValue() {
+        let value = Decimal(2)
+        _testRoundTrip(of: TopLevelWrapper(value), expected: ["value": value])
+    }
+
     // MARK: - GeoPoint & Document Reference
     func testEncodingGeoPoint() {
         let point = GeoPoint(latitude: 2, longitude: 2)