Adds unit test for cellForRow method for recently visited section

DeluxeAlonso · Sep 18, 2024 · a2e429e · a2e429e
1 parent baf7d01
commit a2e429e
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/UpcomingMoviesTests/Movies/SearchMovies/SearchOptions/SearchOptionsDataSourceTests.swift b/UpcomingMoviesTests/Movies/SearchMovies/SearchOptions/SearchOptionsDataSourceTests.swift
@@ -124,6 +124,16 @@ final class SearchOptionsDataSourceTests: XCTestCase {
         XCTAssertNotNil(cellForRow as? DefaultSearchOptionTableViewCell)
     }
 
+    func testCellForRowRecentlyVisited() {
+        // Arrange
+        let tableView = UITableView()
+        tableView.registerNib(cellType: RecentlyVisitedMoviesTableViewCell.self)
+        viewModel.sectionAtIndexResult = .recentlyVisited
+        // Act
+        let cellForRow = dataSource.tableView(tableView, cellForRowAt: IndexPath(row: 0, section: 0))
+        // Assert
+        XCTAssertNotNil(cellForRow as? RecentlyVisitedMoviesTableViewCell)
+    }
 
     func testDidSelectMovie() {
         // Act