[swift] RxDataSources의 테스트 방법

28 Nov 2023

swift

RxDataSources는 Swift에서 TableView와 CollectionView를 쉽게 구현할 수 있는 라이브러리입니다. 이 라이브러리를 사용하여 데이터를 표시하고 제어하는 것은 간단하지만, 테스트는 조금 어려울 수 있습니다.

이 글에서는 RxDataSources의 테스트 방법을 알아보겠습니다. 아래는 테스트하기 위한 예제 코드입니다.

import XCTest
import RxSwift
import RxDataSources

class MyDataSourceTests: XCTestCase {
    var disposeBag: DisposeBag!
    var dataSource: RxTableViewSectionedReloadDataSource<MySection>!
    var tableView: UITableView!
    
    override func setUp() {
        super.setUp()
        disposeBag = DisposeBag()
        
        dataSource = RxTableViewSectionedReloadDataSource<MySection> { (_, tableView, indexPath, item) -> UITableViewCell in
            let cell = tableView.dequeueReusableCell(withIdentifier: "MyCell", for: indexPath) as! MyCell
            cell.configure(item)
            return cell
        }
        
        tableView = UITableView()
        tableView.register(MyCell.self, forCellReuseIdentifier: "MyCell")
    }
    
    func testTableView() {
        let items = Observable.just([
            MySection(header: "Section 1", items: [
                "Item 1",
                "Item 2",
                "Item 3"
            ]),
            MySection(header: "Section 2", items: [
                "Item 4",
                "Item 5",
                "Item 6"
            ])
        ])
        
        items
            .bind(to: tableView.rx.items(dataSource: dataSource))
            .disposed(by: disposeBag)
        
        tableView.reloadData()
        
        let cell = tableView.cellForRow(at: IndexPath(row: 0, section: 0)) as? MyCell
        XCTAssertNotNil(cell)
        
        let expectedTitle = "Item 1"
        XCTAssertEqual(cell?.textLabel?.text, expectedTitle)
    }
}

struct MySection {
    var header: String
    var items: [Item]
}

extension MySection: SectionModelType {
    typealias Item = String
    
    init(original: MySection, items: [Item]) {
        self = original
        self.items = items
    }
}

class MyCell: UITableViewCell {
    func configure(_ text: String) {
        textLabel?.text = text
    }
}

이 코드는 RxDataSources를 사용하여 TableView에 데이터를 표시하는 예제입니다. 테스트하려는 부분은 testTableView 메서드입니다. 이 메서드는 예제 데이터를 생성하고 TableView에 데이터를 바인딩한 뒤, 첫 번째 셀의 텍스트를 확인하는 테스트를 수행합니다.

위의 코드는 테스트 케이스의 기본 설정과 예제를 포함하고 있습니다. 테스트할 때 필요한 데이터소스와 TableView를 초기화하고, 예제 데이터를 생성하는 부분이 포함되어 있습니다.

이제 testTableView 메서드에서 TableView를 초기화하고 데이터를 바인딩하고, 첫 번째 셀의 텍스트를 확인하는 테스트를 수행할 수 있습니다.

테스트를 실행하려면 XCTest 프레임워크를 사용해야 합니다. 이 예제는 XCTest를 활용하고 있으므로 해당 프레임워크를 함께 사용해야 합니다.

테스트 실행 결과 정상적으로 통과하면 RxDataSources를 올바르게 사용한 것입니다. 만약 테스트가 실패한다면, 데이터소스나 셀 구현부에 버그가 있을 수 있습니다.

RxDataSources의 테스트를 위한 예제 코드를 소개했습니다. 이를 참고하여 RxDataSources를 테스트해보시기 바랍니다.

참고 자료