Adds unit tests for watchlist in account client

DeluxeAlonso · Sep 3, 2023 · 8701e61 · 8701e61
1 parent 6d8d0ac
commit 8701e61
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/NetworkInfrastructure/NetworkInfrastructureTests/Services/AccountClientTests.swift b/NetworkInfrastructure/NetworkInfrastructureTests/Services/AccountClientTests.swift
@@ -67,4 +67,45 @@ final class AccountClientTests: XCTestCase {
         wait(for: [expectation], timeout: 1.0)
     }
 
+    func testGetWatchlistSuccess() throws {
+        // Arrange
+        let data = try JSONEncoder().encode(MovieResult(results: [], currentPage: 1, totalPages: 1))
+        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: "Get watchlist success")
+        // Act
+        accountClient.getWatchlist(page: 1, sortBy: .createdAtAsc, sessionId: "", accountId: 1) { result in
+            switch result {
+            case .success:
+                break
+            case .failure:
+                XCTFail("Get watchlist error")
+            }
+            expectation.fulfill()
+        }
+        // Assert
+        wait(for: [expectation], timeout: 1.0)
+    }
+
+    func testGetWatchlistError() throws {
+        // Arrange
+        urlSession.dataTaskWithRequestCompletionHandler = (nil, nil, nil)
+        let expectation = XCTestExpectation(description: "Get watchlist error")
+        // Act
+        accountClient.getWatchlist(page: 1, sortBy: .createdAtAsc, sessionId: "", accountId: 1) { result in
+            switch result {
+            case .success:
+                XCTFail("Get watchlist success")
+            case .failure:
+                break
+            }
+            expectation.fulfill()
+        }
+        // Assert
+        wait(for: [expectation], timeout: 1.0)
+    }
+
 }