Add support to Encodable types, fix ClientTests

Author: ppamorim

Sources/MeiliSearch/Client.swift

@@ -143,7 +143,36 @@ public struct MeiliSearch {
      For a partial update of the document see `updateDocument`.
 
      - parameter UID:        The unique identifier for the Document's index to be found.
-     - parameter documents:  The documents data (JSON) to be processed.
+     - parameter documents:  The documents to be processed.
+     - parameter completion: The completion closure used to notify when the server
+     completes the update request, it returns a `Result` object that contains `Update`
+     value. If the request was sucessful or `Error` if a failure occured.
+     */
+    public func addDocuments<T>(
+        UID: String,
+        documents: [T],
+        encoder: JSONEncoder? = nil,
+        primaryKey: String?,
+        _ completion: @escaping (Result<Update, Swift.Error>) -> Void) where T: Encodable {
+        self.documents.add(
+            UID,
+            documents,
+            encoder,
+            primaryKey,
+            completion)
+    }
+
+    /**
+     Add a list of documents as data or replace them if they already exist.
+
+     If you send an already existing document (same id) the whole existing document will
+     be overwritten by the new document. Fields previously in the document not present in
+     the new document are removed.
+
+     For a partial update of the document see `updateDocument`.
+
+     - parameter UID:        The unique identifier for the Document's index to be found.
+     - parameter documents:  The  data to be processed.
      - parameter completion: The completion closure used to notify when the server
      completes the update request, it returns a `Result` object that contains `Update`
      value. If the request was sucessful or `Error` if a failure occured.

Sources/MeiliSearch/Constants.swift

@@ -7,4 +7,11 @@ struct Constants {
         decoder.dateDecodingStrategy = .formatted(Formatter.iso8601)
         return decoder
     }()
+
+    static let customJSONEecoder: JSONEncoder = {
+        let encoder = JSONEncoder()
+        encoder.dateEncodingStrategy = .formatted(Formatter.iso8601)
+        return encoder
+    }()
+
 }

Sources/MeiliSearch/Documents.swift

@@ -75,6 +75,7 @@ struct Documents {
         _ document: Data,
         _ primaryKey: String?,
         _ completion: @escaping (Result<Update, Swift.Error>) -> Void) {
+
         var query: String = "/indexes/\(UID)/documents"
         if let primaryKey: String = primaryKey {
             query += "?primaryKey=\(primaryKey)"
@@ -94,6 +95,42 @@ struct Documents {
         }
     }
 
+    func add<T>(
+        _ UID: String,
+        _ documents: [T],
+        _ encoder: JSONEncoder? = nil,
+        _ primaryKey: String?,
+        _ completion: @escaping (Result<Update, Swift.Error>) -> Void) where T: Encodable {
+
+        var query: String = "/indexes/\(UID)/documents"
+        if let primaryKey: String = primaryKey {
+            query += "?primaryKey=\(primaryKey)"
+        }
+
+        let data: Data!
+
+        switch encodeJSON(documents, encoder) {
+        case .success(let documentData):
+          data = documentData
+        case .failure(let error):
+          completion(.failure(error))
+          return
+        }
+
+        request.post(api: query, data) { result in
+
+            switch result {
+            case .success(let data):
+
+                Documents.decodeJSON(data, completion: completion)
+
+            case .failure(let error):
+                completion(.failure(error))
+            }
+
+        }
+    }
+
     func update(
         _ UID: String,
         _ document: Data,
@@ -217,4 +254,16 @@ struct Documents {
         }
     }
 
+    private func encodeJSON<T: Encodable>(
+        _ documents: [T],
+        _ encoder: JSONEncoder?) -> Result<Data, Swift.Error> {
+        do {
+            let data: Data = try (encoder ?? Constants.customJSONEecoder)
+              .encode(documents)
+            return .success(data)
+        } catch {
+            return .failure(error)
+        }
+    }
+
 }

Tests/MeiliSearchUnitTests/ClientTests.swift

@@ -3,8 +3,11 @@ import XCTest
 
 class ClientTests: XCTestCase {
 
+  private let session = MockURLSession()
+
   func testValidHostURL() {
-    XCTAssertNotNil(try? MeiliSearch(Config(hostURL: "http://localhost:7700", apiKey: "masterKey")))
+    session.pushEmpty(code: 200)
+    XCTAssertNotNil(try? MeiliSearch(Config.default(apiKey: "masterKey", session: session)))
   }
 
   func testEmptyHostURL() {

Tests/MeiliSearchUnitTests/DocumentsTests.swift

@@ -47,6 +47,51 @@ class DocumentsTests: XCTestCase {
 
         let uid: String = "Movies"
 
+        let movie = Movie(
+            id: 287947,
+            title: "Shazam",
+            overview: "A boy is given the ability to become an adult superhero in times of need with a single magic word.",
+            releaseDate: Date(timeIntervalSince1970: TimeInterval(1553299200)))
+
+        let expectation = XCTestExpectation(description: "Add or replace Movies document")
+
+        self.client.addDocuments(
+            UID: uid,
+            documents: [movie],
+            primaryKey: "") { result in
+
+            switch result {
+            case .success(let update):
+                XCTAssertEqual(stubUpdate, update)
+                expectation.fulfill()
+            case .failure:
+                XCTFail("Failed to add or replace Movies document")
+            }
+
+        }
+
+        self.wait(for: [expectation], timeout: 1.0)
+
+    }
+
+    func testAddDataDocuments() {
+
+        //Prepare the mock server
+
+        let jsonString = """
+        {"updateId":0}
+        """
+
+        let decoder: JSONDecoder = JSONDecoder()
+        let jsonData = jsonString.data(using: .utf8)!
+        let stubUpdate: Update = try! decoder.decode(Update.self, from: jsonData)
+
+        session.pushData(jsonString, code: 202)
+
+        // Start the test with the mocked server
+
+        let uid: String = "Movies"
+
         let documentJsonString = """
         [{
             "id": 287947,