[EasyCloset] 🧑‍🔬 뷰모델 유닛 테스트

ViewModel 유닛 테스트, CacheManager 유닛 테스트

배경

• 사용자의 액션의 로직을 담고 있는 ViewModel 의 unit-test 를 구현하고자 했음

유닛테스트를 위한 Mock Repository 구현

• ViewModel은 Repository에 의존하고 있으므로, 실제 Repository가 아닌 가상의 시나리오로 동작하는 Mock Repository를 구현하여 ViewModel을 테스트

  드디어 protocol 추상화, 의존성 주입이 빛을 발할 때가 되었다..!😲🥂

• 기존에 만들어놓은 Mock 객체들을 return 해주도록 구현함

  일단은 Repository가 성공의 시나리오만 발생시키도록 구현. 실패 시나리오가 필요할 시, 각 응답을 property 로 만들어 테스트 직전 원하는 결과를 주입시켜줄 수 있을 것 같았음

final class MockClothesRepository: ClothesRepositoryProtocol {
  func fetchClothesList() -> AnyPublisher<ClothesList, RepositoryError> {
    return Just(ClothesList.mocks)
      .setFailureType(to: RepositoryError.self)
      .eraseToAnyPublisher()
  }
...

ClothesViewModel 테스트

• 이처럼 ClothesRepositoryProtocol 를 구현한 MockRepository를 ViewModel 에 주입해서 sut를 초기화

var sut: ClothesViewModel!

override func setUpWithError() throws {
  let mockRepository = MockClothesRepository()
  sut = ClothesViewModel(repository: mockRepository)
...

계절 filter가 적용되는지 테스트

• 사용자가 필터 검색을 입력한 시나리오인 filter르 걸었을 때,
• 기대하는 출력값이 나오는지 테스트

func test_계절_filter가_적용되는지_테스트() {
  // given
  let categories = ClothesCategory.allCases
  let weatherFilterType: WeatherType = .fall
  sut.searchFilters.send([.weather(weatherFilterType)]) // 계절 filter 검색
  
  // 각각의 카테고리에 대한 expectation
  var expectations: [XCTestExpectation] = []
  
  categories.forEach { category in
    let expectation = XCTestExpectation(description: category.korean)
    expectations.append(expectation)
    
    // when
    sut.clothes(of: category)
      .sink { clothes in     
        // then
        XCTAssertTrue(clothes.allSatisfy {
	 // 모든 결과값이 해당 filter와 동일한지 체크
          $0.weatherType == weatherFilterType
        })
        expectation.fulfill()
      }
      .store(in: &cancellables)
...

이미지 캐시매니저 유닛 테스트

• 메모리 캐시인 NSCache 를 활용한 ImageCacheManager 를 테스트
• 싱글턴 형태이긴 했지만, 메모리 캐시이기 때문에 각각의 케이스 후 removeAll 만 호출 해 주면 문제없이 테스트를 진행할 수 있을 것이라 판단
• 특히, 구현했던 갯수제한과 용량제한이 적절히 이뤄지는지를 테스트하고 싶었음

  다만, 해당 테스트가 반드시 성공하리라는 보장은 할 수 없었는데,
  NSCache의 특성상 countLimit과 totalCostLimit이 제공하는 limit이 imprecise 하다고 명시되어 있었기 때문.
  다행히, 어느 정도까지는 예상한 대로 동작하는 것을 확인할 수 있었음

• 저장시 갯수제한이 적용되는지 확인

func test_저장시_갯수제한이_적용되는지_확인() {
  // given
  let ids = (0...10).map { _ in UUID() }
  let countLimit = 3
  
  // when
  sut.countLimit = countLimit // 캐싱 저장 이미지 수를 3개로 제한
  ids.forEach { id in
    sut.store(UIImage(), for: id)
  }
  
  // then
  let storedImages = ids
    .compactMap { sut.get(for: $0) }

  // 갯수 제한을 준 갯수보다 같거나 적게 저장되었는지 확인
  XCTAssertGreaterThanOrEqual(countLimit, storedImages.count)

• 저장시 용량제한이 적용되는지 확인
  • SF Symbol의 기본 이미지 5장을 저장하려고 시도하며,
  • 5장의 이미지 중 가장 작은 용량 * 3 으로 제약을 줌

func test_저장시_용량제한이_적용되는지_확인() {
  // given
  let images = [
    UIImage(systemName: "pencil")!,
    UIImage(systemName: "pencil.slash")!,
    UIImage(systemName: "pencil.circle")!,
    UIImage(systemName: "pencil.circle.fill")!,
    UIImage(systemName: "pencil.line")!
  ]
  // 가장 작은 이미지 용량
  let imageDataSize = images.compactMap { $0.pngData()?.count }.min() ?? 0
  let ids = (0..<5).map { _ in UUID() }
  
  // when
  sut.byteLimit = imageDataSize * 3 // 대략 이미지 3개 정도의 사이즈만큼 용량 제한을 줌
  zip(images, ids).forEach { image, id in
    sut.store(image, for: id)
  }
  
  // then
  let storedImages = ids
    .compactMap { sut.get(for: $0) }

  // 용량 제한을 준 3개로 주었기에, 그보다 같거나 적게 저장되었는지 확인
  XCTAssertGreaterThanOrEqual(3, storedImages.count)