Merge pull request #974 from DeluxeAlonso/feature/code-cleanup

Feature/code cleanup
DeluxeAlonso · Sep 14, 2024 · 8c0d96e · 8c0d96e
2 parents f0bcecd + 035a95e
commit 8c0d96e
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/UpcomingMoviesTests/Movies/SearchMovies/SearchOptions/SearchOptionsDataSourceTests.swift b/UpcomingMoviesTests/Movies/SearchMovies/SearchOptions/SearchOptionsDataSourceTests.swift
@@ -44,4 +44,29 @@ final class SearchOptionsDataSourceTests: XCTestCase {
         XCTAssertEqual(numberOfSections, 2)
     }
 
+    func testTitleForHeaderInSectionPopulatedState() {
+        // Arrange
+        viewModel.viewState = .init(.populatedMovieVisits)
+        // Act
+        let title = dataSource.tableView(UITableView(), titleForHeaderInSection: 0)
+        // Assert
+        XCTAssertEqual(title, "Recently visited")
+    }
+
+    func testTitleForHeaderInSectionEmptyState() {
+        // Arrange
+        viewModel.viewState = .init(.emptyMovieVisits)
+        // Act
+        let title = dataSource.tableView(UITableView(), titleForHeaderInSection: 0)
+        // Assert
+        XCTAssertNil(title)
+    }
+
+    func testDidSelectMovie() {
+        // Act
+        dataSource.recentlyVisitedMoviesTableViewCell(RecentlyVisitedMoviesTableViewCell(), didSelectMovieAt: IndexPath(row: 0, section: 0))
+        // Assert
+        XCTAssertEqual(viewModel.getRecentlyVisitedMovieSelectionCallCount, 1)
+    }
+
 }