转载

Swift：使用自定义初始化方法进行依赖注入

作为我今年的第一个项目，我正在做 @catehstn’s iOS Unit Testing Workshop 的 Swift 版本 。如果感兴趣的话，可以在这里注册。

在我写代码和测试期间，我遇到了下面的问题…

假设我使用 Swift 创建了一个简单的计数器：

// Counter.swift
 
struct Counter {
    // `count` 在这里为一个变量
    // 但是我不想把 `count` 设置为外部可访问的 
    // 因此我把 `count` 设置为私有的（这里有小提示！https://twitter.com/mipstian/status/685489964403003393）
    private(set) var count: Int
    
    init() {
        // 我想让计数器默认从 0 开始
        count = 0
    }
    
    // 我只想让 `count` 通过此函数被修改
 // 通过像这样的函数 
    mutating func increment() {
        count += 1
    }
}

这个结构体超级简单，我也很容易地编写了一个 increment 测试用例。

// CounterTests.swift
 
 
import XCTest
@testable import Counter
 
class CounterTests: XCTestCase {
 
   func testIncrement() {
       var counter = Counter()
       XCTAssertEqual(counter.count, 0)
       counter.increment()
       XCTAssertEqual(counter.count, 1)
   }
  
}

测试正确，但是下面这种情况呢！

// Counter.swift
 
struct Counter {
    private(set) var count: Int
   
    init() {
        count = 0
    }
 
    // 代码缩减 (见上面)
 
    // 当用户进行内购
    // 提升了他们的分数！

    mutating func scaleBy(multiplier: Int) {
        count = count * multiplier
    }
}

因为 Counter 的值初始化为 0，乘法测试没有效果。

// CounterTests.swift
 
class CounterTests: XCTestCase {
 
   func testScaleBy() {
      var counter = Counter()
      XCTAssertEqual(counter.count, 0)
      counter.scaleBy(5)
      // 正如我们所知，scaleBy 可以做多次乘法运算和除法运算
      // 要么只是返回 count ，要么只是返回 0
      XCTAssertEqual(counter.count, 0)
   }
}

为了进行有效的测试,我们需要在 count 不为 0 的情况下测试我们的 scaleBy 方法，所以我们需要控制计数器的初始值。我们可以使用默认值和正确的初始值注入 init 函数（译者注：作者这里的意思是，如果初始化的时候，不给 init 函数参数，则使用默认值；反之则使用给定的初始值。）：

// Counter.swift
 
struct Counter {
    private(set) var count: Int
    
    // 依赖注入
    // 计数器的初始值是注入的
    init(count: Int = 0) {
        self.count = count
    }
    
    mutating func scaleBy(multiplier: Int) {
        count = count * multiplier
    }
}

现在测试如下！

// CounterTests.swift

class CounterTests: XCTestCase {

   func testScaleBy() {
      // 这次计数器从一个不为零的值开始
      var counter = Counter(count: 5)
      XCTAssertEqual(counter.count, 5)
      counter.scaleBy(5)
      XCTAssertEqual(counter.count, 25)
   }
}