Extracting integer

orlandos-nl · Dec 17, 2016 · 2af78ef · 2af78ef
1 parent e718fe1
commit 2af78ef
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/Sources/Document+Extraction.swift b/Sources/Document+Extraction.swift
@@ -6,6 +6,10 @@ extension Document {
             return nil
         }
 
+        if T.self == Int.self {
+            return response.int as? T
+        }
+
         return response.rawValue as? T
     }
 }
diff --git a/Tests/BSONTests/BSONPublicTests.swift b/Tests/BSONTests/BSONPublicTests.swift
@@ -551,6 +551,9 @@ final class BSONPublicTests: XCTestCase {
         let regex = kittenDocument.extract("regex") as RegularExpression?
         XCTAssertNotNil(regex)
         XCTAssertEqual(regex?.pattern, "[A-Z0-9._%+-]+@[A-Z0-9.-]+\\.[A-Z]{2,}")
+
+        XCTAssertEqual(kittenDocument.extract("cool32bitNumber") as Int?, 9001)
+        XCTAssertEqual(kittenDocument.extract("cool64bitNumber") as Int?, 21312153)
     }
 
     func testKeyDetection() {