理解 RXSwift:单元测试(四)

1,299 阅读4分钟

理解 RxSwift:为什么要使用 RxSwift(一)

理解 RxSwift:实现原理(二)

理解 RxSwift:MVVM(三)

理解 RxSwift:单元测试(四)

这篇文章是阅读 RxSwift: Reactive Programming with Swift 时记的笔记,介绍如何使用RxTest和RxBlocking对RxSwift编写单元测试。

书上使用RxTest和RxBlocking对一个 demo 进行了完整的单元测试和讲解,在这个过程中,我们可以熟悉一下RxSwift 单元测试的使用方法,了解单元测试的一般使用技巧和注意事项。

Demo 介绍

Demo 是一个转换颜色的小应用,根据用户输入的16进制颜色代码自动转换为红、绿、蓝色值和颜色名字(若有)。

Demo 是使用MVVM模式实现的,逻辑代码放在了ViewModel中,ViewController管理和控制View。ViewModel的主要代码如下:

// Convert hex text to color
color = hexString.asObservable()
  .map { hex in
    guard hex.characters.count == 7 else { return .clear }
    let color = UIColor(hex: hex)
    return color
  }
  .asDriver(onErrorJustReturn: .clear)

// Convert the color to an rgb tuple
rgb = color.asObservable()
  .map { color in
    var red: CGFloat = 0.0
    var green: CGFloat = 0.0
    var blue: CGFloat = 0.0

    color.getRed(&red, green: &green, blue: &blue, alpha: nil)
    let rgb = (Int(red * 255.0), Int(green * 255.0), Int(blue * 255.0))
    return rgb
  }
  .asDriver(onErrorJustReturn: (0, 0, 0))

// Convert the hex text to a matching name
colorName = hexString.asObservable()
  .map { hexString in
    let hex = String(hexString.characters.dropFirst())

    if let color = ColorName(rawValue: hex) {
      return "\(color)"
    } else {
      return "--"
    }
  }
  .asDriver(onErrorJustReturn: "")
  

接下来,主要针对ViewModel编写单元测试。

编写单元测试

首先介绍两个RxSwift单元测试的库,RxTest 和 RxBlocking,它们让编写测试用例更简单。

RxTest 是一个独立的库,需要通过pod单独安装。RxTest提供了很多有用的功能,例如TestScheduler,它是一个虚拟时间scheduler,可以在指定时间发出事件,包括next、completed和error。

RxBlocking 也是独立的库,也需要通过pod安装。它的作用是通过它的toBlocking(timeout:)方法将observable 转换成 BlockingObservable。阻塞当前线程,从而将异步操作转换为同步操作,使测试变得容易。

打开TestingTests组中的TestingViewModel.swift,在类的顶部定义两个属性:

var scheduler: ConcurrentDispatchQueueScheduler!
var disposeBag: Disposable!

ConcurrentDispatchQueueScheduler 是并行队列调度器,封装了GCD的并行队列。

setUp()方法,每个测试用例在开始前都会调用它。这里初始化一个新的 scheduler 和 viewModel。tearDown(),在每个测试完成时调用,一般执行一些销毁操作。

func testColorIsRedWhenHexStringIsFF0000_async() {
  let disposeBag = DisposeBag()
  // 1
  let expect = expectation(description: #function)
  // 2
  let expectedColor = UIColor(red: 1.0, green: 0.0, blue: 0.0, alpha:
    1.0)
  // 3
  var result: UIColor!
}

验证异步操作是否按预期运行,我们在测试中创建期望 expectation,然后在异步操作成功后调用fulfill()方法,表示满足期望 expectation。

  1. 创建一个异步事件期望。
  2. 创建 expectedColor 表示预期的测试结果。
  3. 定义实际结果变量。
// 1
viewModel.color.asObservable()
  .skip(1)
  .subscribe(onNext: {
    // 2
    result = $0
    expect.fulfill()
  })
  .disposed(by: disposeBag)
// 3
viewModel.hexString.value = "#ff0000"
// 4
waitForExpectations(timeout: 1.0) { error in
  guard error == nil else {
    XCTFail(error!.localizedDescription)
    return
  }
  // 5
  XCTAssertEqual(expectedColor, result)
}
  1. 创建一个订阅,追踪 viewModel 的color 变化。这略过了第一个元素,因为第一个元素通常是初始化值。
  2. 分配.next事件,把结果赋值给result,并在expect上调用fulfill(),表示此异步事件已完成。
  3. 给viewModel的hexString赋一个新值
  4. 超时1秒来等待expectation的完成,并提供error处理
  5. 断言期望的color等于实际的result。

现在使用RxBlocking来实现同样的事情:

func testColorIsRedWhenHexStringIsFF0000() {
  // 1
  let colorObservable =
    viewModel.color.asObservable().subscribeOn(scheduler)
  // 2
  viewModel.hexString.value = "#ff0000"
  // 3
  do {
    guard let result = try colorObservable.toBlocking(timeout:
      1.0).first() else { return }
    XCTAssertEqual(result, .red)
  } catch {
    print(error)
  }
}
  1. 创建colorObservable来订阅并发scheduler上的observable结果。
  2. 在viewModel的hexString上赋一个新值。
  3. 调用toBlocking(),并设置1秒的超时时间,如果抛出,捕获并打印错误,然后断言实际的结果与预期的匹配。
func testRgbIs010WhenHexStringIs00FF00() {
  // 1
  let rgbObservable =
    viewModel.rgb.asObservable().subscribeOn(scheduler)
  // 2
  viewModel.hexString.value = "#00ff00"
  // 3
  let result = try! rgbObservable.toBlocking().first()!
  XCTAssertEqual(0 * 255, result.0)
  XCTAssertEqual(1 * 255, result.1)
  XCTAssertEqual(0 * 255, result.2)
}
  1. 创建rgbObservable来订阅viewModel的rgb值。
  2. 在viewModel的hexString上赋一个新值。
  3. 检索在rgbObservable上调用toBlocking的第一个结果,然后断言每个值与期望的相等。
func testColorNameIsRayWenderlichGreenWhenHexStringIs006636() {
  // 1
  let colorNameObservable =
    viewModel.colorName.asObservable().subscribeOn(scheduler)
  // 2
  viewModel.hexString.value = "#006636"
  // 3
  XCTAssertEqual("rayWenderlichGreen", try!
    colorNameObservable.toBlocking().first()!)
}
  1. 创建observable
  2. 在viewModel的hexString上赋一个新值。
  3. 断言实际的结果来匹配期望的结果。