[ReactorKit] 테스트 코드 작성해보기 (feat.expectation)

ReactorKit을 공부하기 위해서 잠시 진행했던 프로젝트다.

GitHub - 9oHigh/usket.RandomUser: Random User Generator를 활용한 ReactorKit & RxDataSource 학습

 

간단한 앱이다.

랜덤 유저 API를 통해서 100명의 사람들의 데이터를 가져오고

해당 인물들을 특정한 카테고리로 분류해 보여주는 앱이다.

그렇기에 테스트해야 하는 것도 간단하다.

*UI 테스트는 진행하지 않았다.

 

테스트 하기 전

가장 먼저 어떤 걸 테스트해야 할지 정해보자.

해당 프로젝트에는 리액터가 두가지다.

하나는 메인화면에서 사용되는 리액터

하나는 각 카테고리에 진입했을 때 사용되는 리액터

두 리액터에서 특정한 액션을 취했을 경우, 원하는 대로 State가 변경되는지 확인하면 된다.

 

메인 리액터

메인 리액터는 3가지의 액션을 가지고 있다.

1. 유저 데이터 로드

2. 디테일 화면 푸시

3. 푸시된 뷰컨트롤러를 nil 처리하기

진행할 모든 테스트에서는 데이터를 로드해야 하는 공통된 작업이 필요하다.

로드하기가 번거롭다면 mock 데이터를 만들어도 좋다.

아무튼,

이를 위해서 setUp 메서드에 로드작업을 미리 해야 한다.

setUp은 각 테스트 메서드가 진행되기 이전에 호출되는 메서드이다.

( ↔️ tearDown은 테스트 메서드가 마무리된 후 호출된다 )

여기서 추가적으로 로드될 시간을 벌기 위해서 expectaion을 이용했다.

override func setUp() {
    mainReactor = MainReactor()
    let expectation = self.expectation(description: "Initial data load")

    // 데이터가 로드된 후에 테스트를 진행할 수 있도록 구독 설정
    mainReactor?.state
        .map { $0.sectionData }
        .distinctUntilChanged()
        .skip(1) // 첫 번째는 초기 값이므로 skip
        .subscribe(onNext: { [weak self] sectionData in
            guard let self = self, sectionData.count > 0 else {
                return
            }

            expectation.fulfill()
        })
        .disposed(by: disposeBag)

    mainReactor?.action.onNext(.load(100))
    wait(for: [expectation], timeout: 3.0)
}

 

✅ wait 메서드는 expectation 배열에 있는 모든 expectation이 fulfill 될 때까지 기다린다.

✅ timeout을 3초로 제한해 두면 3초가 되는 시점에 fulfill 되지 않은 모든 expectation이 fulfill 되게 된다.

(1) load

특정한 카운트만큼의 데이터를 요청했을 경우, 정상적으로 해당 개수만큼의 데이터가 들어오고 SectionData가 변경되었는지 확인해야 한다. ( SectionData는 각 섹션별 카테고리 이름에 따른 사람들을 정리한 딕셔너리다 ) 

아래와 같이 코드를 작성했다. 딕셔너리의 데이터에는 중복된 데이터가 존재하므로 set을 통해 중복을 제거해야 했다.

func testMainReactorLoad() throws {
    let expectation = self.expectation(description: "Section Data is changed")
    var counts = 0
    mainReactor?.state
        .map { $0.sectionData }
        .distinctUntilChanged()
        .subscribe(onNext: { [weak self] sectionData in
            guard let self = self, sectionData.count > 0 else {
                return
            }

            var values: Set<PersonInfoDetail> = []
            for (_, value) in sectionData {
                values.formUnion(value)
            }

            let totalCount = values.count
            counts = totalCount
            
            expectation.fulfill()
        })
        .disposed(by: disposeBag)

    wait(for: [expectation], timeout: 3.0)
    XCTAssertEqual(counts, 100)
}

 (2) toSectionDetail

load를 통해 들어온 데이터 중 일부 사람들 데이터를 전달했을 경우, 상태의 pushingViewController가 변경되었는지 즉, nil이 아니게 되었는지 확인해야 한다. XCAssertNotNil을 통해 간편하게 nil이 아닌지 확인할 수 있다.

func testMainReactorToSectionDetail() throws {
    let expectation = self.expectation(description: "PushingViewController is changed")

    mainReactor?.action.onNext(.toSectionDetail(mainReactor?.currentState.sectionData["20대"] ?? []))

    mainReactor?.state
        .map { $0.pushingViewController }
        .subscribe(onNext: { value in
            expectation.fulfill()
        })
        .disposed(by: disposeBag)

    wait(for: [expectation], timeout: 3.0)
    XCTAssertNotNil(mainReactor?.currentState.pushingViewController)
}

 (3) removePushed

pushingViewController의 값이 있는지 확인하고, removePushed 액션을 요청했을 때, 정상적으로 pushingViewController가 nil이 되어있는지 확인해야 한다.

func testMainReactorRemovePushed() throws {
    let expectation = self.expectation(description: "PushingViewController is changed")
    mainReactor?.action.onNext(.removePushed)

    mainReactor?.state
        .map { $0.pushingViewController }
        .subscribe(onNext: { _ in
            expectation.fulfill()
        })
        .disposed(by: disposeBag)

    wait(for: [expectation], timeout: 3.0)
    XCTAssertNil(mainReactor?.currentState.pushingViewController)
}

 

디테일 리액터

디테일 리액터에는 하나의 액션이 있다.

특정한 사람을 클릭할 경우, 그 근처에 있는 사람을 확인할 수 있는 액션이 있다.

따라서, State에는 expandedIndexPath를 가지고 있다.

toggle(IndexPath)

 특정한 IndexPath를 전달했을 경우, 상태의 expandedIndexPath가 변경되었는지 확인해봐야 한다.

func testDetailReactor() throws {
    let expectation = self.expectation(description: "IndexPath is changed")
    let detailReactor: DetailReactor = DetailReactor(people: mainReactor!.currentState.sectionData["20대"]!)
    let indexPath: IndexPath = IndexPath(row: 0, section: 0)
    detailReactor.action.onNext(.toggle(indexPath))

    detailReactor.state
        .map { $0.expandedIndexPath }
        .distinctUntilChanged()
        .subscribe(onNext: { _ in
            expectation.fulfill()
        })
        .disposed(by: disposeBag)

    wait(for: [expectation], timeout: 3.0)
    XCTAssertNotNil(detailReactor.currentState.expandedIndexPath)
}