在单元测试时快速获取UITableView中的数字行?

时间:2016-06-02 19:05:39

标签: swift uitableview count testcase

我正在为具有UITableView的UIViewController编写测试用例。我想问一下如何在UITableView中获取行数

 func testloadingDataIntoUiTableView()
    {      
      var  countRow:Int =  viewController.formListTableView.numberOfRowsInSection   
      XCTAssert(countRow == 4)  
    }

1 个答案:

答案 0 :(得分:8)

简介

请记住,数据模型会生成UI。但是你不应该查询UI来检索你的数据模型(除非我们讨论的是用户输入)。

让我们看看这个例子

class Controller:UITableViewController {

    let animals = ["Tiger", "Leopard", "Snow Leopard", "Lion", "Mountain Lion"]
    let places = ["Maveriks", "Yosemite", "El Capitan"];

    override func numberOfSectionsInTableView(tableView: UITableView) -> Int {
        return 2
    }

    override func tableView(tableView: UITableView, numberOfRowsInSection section: Int) -> Int {
        switch section {
        case 0: return animals.count
        case 1: return places.count
        default: fatalError()
        }
    }

    override func tableView(tableView: UITableView, cellForRowAtIndexPath indexPath: NSIndexPath) -> UITableViewCell {
        guard let cell = tableView.dequeueReusableCellWithIdentifier("MyCellID") else { fatalError("Who took the MyCellID cell???") }
        switch indexPath.section {
        case 0: cell.textLabel?.text = animals[indexPath.row]
        case 1: cell.textLabel?.text = places[indexPath.row]
        default: fatalError()
        }
        return cell
    }
}

丑陋的解决方案

在这种情况下,为了获得表中的总行数,我们应该查询模型(animalsplaces属性),所以

let controller: Controller = ...
let rows = controller.animals.count + controller.places.count

很好的解决方案

或者甚至更好,我们可以将animalsplaces属性设为私有,并添加像这样的计算属性

class Controller:UITableViewController {

    private let animals = ["Tiger", "Leopard", "Snow Leopard", "Lion", "Mountain Lion"]
    private let places = ["Maveriks", "Yosemite", "El Capitan"];

    var totalNumberOfRows: Int { return animals.count + places.count }

    ...

现在你可以使用这个

let controller: Controller = ...
let rows = controller.totalNumberOfRows