Unit Testing

MVVM 구조 Testing 하기

• View와 ViewModel이 Binding 되어 있는 구조로 View에서 발생한 특정 이벤트로 인해 ViewModel의 State 값이 바뀌면 그 값에 의해 View가 업데이트 됨
  • 따라서, View가 그려지는 로직 역시 testing이 가능함
• 객체 간 의존관계가 높아지면 testing이 힘들어지므로 추상화가 필요함. 현재 프로젝트에서 의존관계를 살펴보면 ViewModel - UseCase(+ EventLogger) - NetworkService 이므로,
  • ViewModel을 test 하기 위해선 UseCase가 반드시 필요하고
  • UseCase를 test 하기 위해선 NetworkService가 반드시 필요함.
  • 따라서 UseCase와 NetworkService를 추상화하여 Unit Test를 진행할 때는 모의 객체를 만들어 주입하는 방식으로 테스트
• 각각의 객체의 역할을 명확하게 나누고, 의존성을 가지는 객체를 프로토콜로 추상화하여 모의 객체를 통해 객체간 interaction에 대한 독립적인 테스트가 가능

Test 예시 : 특정 트랙 '좋아요' 하기 (성공)

TrackUseCase Testing

• UseCaes는 NetworkService에 의존성을 가지므로 추상화된 NetworkServiceType을 채택하여 무조건적으로 성공적인 response를 돌려주는 MockSuccessNetworkService 구현

struct MockSuccessfulNetworkService: NetworkServiceType {
   let data: Data
   
   func request(url: String, request type: RequestType, body: Data?) -> AnyPublisher<Data, NetworkError> {
      return Just(data)
            .setFailureType(to: NetworkError.self)
            .eraseToAnyPublisher()
   }
}

• UseCase가 '좋아요' 요청을 네트워크를 통해 보내고 돌아온 응답이 예상 했던 응답과 일치하고 최종적으로 실패하지 않는 것이 보장되어야 함

func test_like_track_success() {
   let expectations = XCTestExpectation(description: "track like success test")
   defer { wait(for: [expectations], timeout: 5) }
        
   let response = TrackUseCase.LikeTrackResponse(success: true)
   let data = try? JSONEncoder().encode(likeTrack)
   let usecase = TrackUseCase(network: MockSuccessNetworkService(data: data!))

   usecase.likeTrack(id: 0)
      .sink { result in
         switch result {
         case .finished:
            expectations.fulfill() // 성공했으므로 expectation fulfill
         case let .failure(error):
            XCTFail(error.localizedDescription) // 성공해야 될 요청이 실패한다면 XCTFail
         }
      } receiveValue: { receivedResult in
         XCTAssertEqual(receivedResult, response.success) // 받아온 값이 예상했던 success = true와 일치해야 함
      }
      .store(in: &cancellables)
}

TrackViewModel Testing

• ViewModel은 UseCase와 EventLogger에 의존성을 가짐
• TrackUseCase를 추상화한 프로토콜 TrackuseCaseType을 채택하여 의도한 결과를 방출해주는 MockTrackUseCase 구현

struct MockTrackuseCase: TrackuseCaseType {
   let track: TrackInfo
   ...
   func likeTrack(id: Int) -> AnyPublisher<Bool, UseCaseError> {
      Just(true) // UseCase를 통한 요청 및 응답이 성공적이었음
         .setFailureType(to: UseCaseError.self)
         .eraseToAnyPublisher()
   }
}

• EventLogger를 추상화한 프로토콜 EventLoggerType을 채택하여 send메서드가 잘 호출되었는지 확인할 수 있는 MockEventLogger 구현

struct MockEventLogger: EventLoggerType {
    
    var handler: ((EventLogType) -> Void)?

    func send<T>(_ event: T) where T: EventLogType {
        handler?(event)
    }
}

• ViewModel의 State 값이 바뀔 수 있도록 View에서 일어날 이벤트를 넣어주어 값의 변화가 올바르게 되었는지 관찰하고, 이벤트 발생에 따라 EventLogger를 통해 이벤트를 잘 전송하는지도 테스트

func test_like_track() {
   let expectation = XCTestExpectation(description: "Like track test")
   expectation.expectedFulfillmentCount = 2     // 이벤트가 제대로 전송되었는지, State가 정상적으로 변경되었는지 확인을 위해 expectedFulfillmentCount를 2로 설정
   defer { wait(for: [expectation], timeout: 5) }

   let usecase = MockTrackuseCase(track: track) // track.liked == false
   let eventLogger = MockEventLogger { data in
       XCTAssertEqual(data.event, "Like")       // Like 이벤트가 제대로 전송되었는지 확인
       expectation.fulfill()
   }
   let viewModel = TrackViewModel(track: track, usecase: usecase, eventLogger: eventLogger)

   viewModel.$state
      .sink { state in
         if state.track.liked {                 // track.liked == true 로 바뀌어야 expectation fulfill
            expectation.fulfill()
         }
      }
      .store(in: &cancellables)

   viewModel.send(.like)                        // View에서 발생할 좋아요 이벤트 전달
}