Fix/test faking for new methods

3lvis · Jul 19, 2024 · dba6e3a · dba6e3a
1 parent d7dad91
commit dba6e3a
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/Sources/Networking/Networking+New.swift b/Sources/Networking/Networking+New.swift
@@ -18,6 +18,13 @@ extension Networking {
                     if T.self == Data.self {
                         logger.info("Successfully processed fake request to \(path, privacy: .public)")
                         return .success(Data() as! T)
+                    } else if T.self == NetworkingResponse.self {
+                        let headers = Dictionary(uniqueKeysWithValues: response.headers.compactMap { key, value in
+                            (key as? String).map { ($0, AnyCodable(value)) }
+                        })
+                        let body = try JSONDecoder().decode([String: AnyCodable].self, from: response.data)
+                        let networkingJSON = NetworkingResponse(headers: headers, body: body)
+                        return .success(networkingJSON as! T)
                     } else {
                         let decoder = JSONDecoder()
                         decoder.dateDecodingStrategy = .iso8601

diff --git a/Tests/NetworkingTests/FakeRequestTests.swift b/Tests/NetworkingTests/FakeRequestTests.swift
@@ -683,4 +683,37 @@ extension FakeRequestTests {
             XCTFail(response.error.localizedDescription)
         }
     }
+
+    func testNewPostWithFakeHeaders() async {
+        let networking = Networking(baseURL: baseURL)
+
+        networking.fakePOST("/auth/verify_confirmation_code", response: [
+            "phone_number": "phoneNumber"
+        ], headerFields: [
+            "client": "aClient",
+            "access-token": "anAccessToken",
+            "uid": "aUID",
+            "Authorization": "authorization",
+        ])
+
+        let parameters: [String: Any] = [
+            "phone_number": "phoneNumber",
+            "confirmation_code": "confirmationCode"
+        ]
+
+        let result: Result<NetworkingResponse, NetworkingError> = await networking.newPost("/auth/verify_confirmation_code", parameters: parameters)
+        switch result {
+        case .success(let response):
+            let headers = response.headers
+            XCTAssertEqual(headers.string(for: "access-token"), "anAccessToken")
+            XCTAssertEqual(headers.string(for: "client"), "aClient")
+            XCTAssertEqual(headers.string(for: "uid"), "aUID")
+            XCTAssertEqual(headers.string(for: "Authorization"), "authorization")
+
+            let body = response.body
+            XCTAssertEqual(body.string(for: "phone_number"), "phoneNumber")
+        case .failure (let response):
+            XCTFail(response.localizedDescription)
+        }
+    }
 }