Merge pull request #786 from DeluxeAlonso/feature/unit-tests

Feature/unit tests
DeluxeAlonso · Sep 10, 2023 · defa6a0 · defa6a0
2 parents a7dbe2c + 68f923c
commit defa6a0
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 1 deletion.
diff --git a/NetworkInfrastructure/NetworkInfrastructure/Core/FeedResults/AddToWatchlistResult.swift b/NetworkInfrastructure/NetworkInfrastructure/Core/FeedResults/AddToWatchlistResult.swift
@@ -5,7 +5,7 @@
 //  Created by Alonso on 6/09/21.
 //
 
-struct AddToWatchlistResult: Decodable {
+struct AddToWatchlistResult: Codable {
 
     let statusCode: Int
     let statusMessage: String

diff --git a/NetworkInfrastructure/NetworkInfrastructureTests/Services/AccountClientTests.swift b/NetworkInfrastructure/NetworkInfrastructureTests/Services/AccountClientTests.swift
@@ -312,4 +312,45 @@ final class AccountClientTests: XCTestCase {
         wait(for: [expectation], timeout: 1.0)
     }
 
+    func testAddToWatchlistSuccess() throws {
+        // Arrange
+        let data = try JSONEncoder().encode(AddToWatchlistResult(statusCode: 200, statusMessage: ""))
+        guard let url = URL(string: "www.google.com") else {
+            XCTFail("Invalid URL")
+            return
+        }
+        urlSession.dataTaskWithRequestCompletionHandler = (data, HTTPURLResponse(url: url, statusCode: 200, httpVersion: nil, headerFields: nil), nil)
+        let expectation = XCTestExpectation(description: "Add to watchlist success")
+        // Act
+        accountClient.addToWatchlist(1, sessionId: "", accountId: 1, watchlist: true) { result in
+            switch result {
+            case .success:
+                break
+            case .failure:
+                XCTFail("Add to watchlist error")
+            }
+            expectation.fulfill()
+        }
+        // Assert
+        wait(for: [expectation], timeout: 1.0)
+    }
+
+    func testAddToWatchlistError() throws {
+        // Arrange
+        urlSession.dataTaskWithRequestCompletionHandler = (nil, nil, nil)
+        let expectation = XCTestExpectation(description: "Add to watchlist error")
+        // Act
+        accountClient.addToWatchlist(1, sessionId: "", accountId: 1, watchlist: true) { result in
+            switch result {
+            case .success:
+                XCTFail("Add to watchlist success")
+            case .failure:
+                break
+            }
+            expectation.fulfill()
+        }
+        // Assert
+        wait(for: [expectation], timeout: 1.0)
+    }
+
 }