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

Feature/code cleanup
DeluxeAlonso · May 8, 2024 · 71855f5 · 71855f5
2 parents dbb3a03 + 86f66c2
commit 71855f5
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/UpcomingMoviesTests/Account/SavedMovies/SavedMoviesViewModelTests.swift b/UpcomingMoviesTests/Account/SavedMovies/SavedMoviesViewModelTests.swift
@@ -8,7 +8,6 @@
 
 import XCTest
 @testable import UpcomingMovies
-import UpcomingMoviesDomain
 
 final class SavedMoviesViewModelTests: XCTestCase {
 
@@ -38,7 +37,7 @@ final class SavedMoviesViewModelTests: XCTestCase {
 
     func testGetCollectionListPopulated() {
         // Arrange
-        let moviestoTest = [Movie.with(id: 1), Movie.with(id: 2)]
+        let moviestoTest = [MockMovieProtocol(id: 1), MockMovieProtocol(id: 2)]
         var statesToReceive: [SimpleViewState<MovieProtocol>] = [.paging(moviestoTest, next: 2), .populated(moviestoTest)]
 
         let expectation = XCTestExpectation(description: "Should get populated state after a paging state")
@@ -57,7 +56,7 @@ final class SavedMoviesViewModelTests: XCTestCase {
 
     func testGetCollectionListPaging() {
         // Arrange
-        let moviesToTest = [Movie.with(id: 1), Movie.with(id: 2)]
+        let moviesToTest = [MockMovieProtocol(id: 1), MockMovieProtocol(id: 2)]
         let expectation = XCTestExpectation(description: "Should get paging state")
         // Act
         viewModelToTest.viewState.bind { state in
@@ -72,7 +71,7 @@ final class SavedMoviesViewModelTests: XCTestCase {
 
     func testGetCollectionListEmpty() {
         // Arrange
-        let moviesToTest: [UpcomingMoviesDomain.Movie] = []
+        let moviesToTest: [MovieProtocol] = []
         let expectation = XCTestExpectation(description: "Should get empty state")
         // Act
         viewModelToTest.viewState.bind { state in
@@ -102,7 +101,7 @@ final class SavedMoviesViewModelTests: XCTestCase {
 
     func testMovieCellsCount() {
         // Arrange
-        let moviesToTest = [Movie.with(id: 1), Movie.with(id: 2)]
+        let moviesToTest = [MockMovieProtocol(id: 1), MockMovieProtocol(id: 2)]
         // Act
         mockInteractor.getSavedMoviesResult = Result.success(moviesToTest)
         viewModelToTest.getCollectionList()
@@ -113,7 +112,7 @@ final class SavedMoviesViewModelTests: XCTestCase {
 
     func testMovieCellsZeroCount() {
         // Arrange
-        let moviesToTest: [UpcomingMoviesDomain.Movie] = []
+        let moviesToTest: [MovieProtocol] = []
         // Act
         mockInteractor.getSavedMoviesResult = Result.success(moviesToTest)
         viewModelToTest.getCollectionList()
@@ -124,7 +123,7 @@ final class SavedMoviesViewModelTests: XCTestCase {
 
     func testMovieIndex() {
         // Arrange
-        let moviesToTest = [Movie.with(id: 1), Movie.with(id: 2)]
+        let moviesToTest = [MockMovieProtocol(id: 1), MockMovieProtocol(id: 2)]
         let indexToTest = Int.random(in: 0...moviesToTest.count - 1)
         // Act
         mockInteractor.getSavedMoviesResult = Result.success(moviesToTest)