如何使用 XCTest 测试线程安全

How to test thread-safety with XCTest

假设我们有以下 class 具有可变状态:

class Machine {
   var state = 0
}

现在,假设有一些控制状态的内部机制。但是,任何线程或队列都可能发生状态更改,因此必须在线程安全环境中读取和写入 state 属性。为此,我们将在 dispatch_queue_t 上使用简单的 sync(:_) 方法来同步对 state 变量的访问。 (这不是唯一的方法,但这是一个例子)

现在,我们可以创建一个保存状态值的私有变量和另一个使用 dispatch_sync(_:) 方法的自定义设置器和获取器的 public 变量。

class Machine {
    private var internalState = 0

    var state: Int {
        get {
            var value: Int?
            dispatch_sync(dispatch_get_main_queue()) {
                value = self.internalState
            }
            return value!
        }

        set(newState) {
            dispatch_sync(dispatch_get_main_queue()) {
                self.internalState = newState
            }
        }
    }
}

state 现在可以从任何队列或线程进行安全的同步访问 - 它是线程安全的。

问题来了。

如何使用 XCTest 测试此行为?

由于 class Machine 可以有一个复杂的状态机,我们需要测试它在任何环境中的表现:

成功测试此类行为的最佳方法是什么?

目前,我正在创建自定义调度队列数组和已定义状态数组。然后我使用 dispatch_async 方法来改变状态并测试它的值。这引入了 XCTest 执行的新问题,因为我需要跟踪所有状态突变何时完成。该解决方案似乎相当复杂且难以维护。

我可以做哪些不同的事情来实现更好的测试?

在考虑像这样测试线程安全代码时,有两个重要的移动部分:

  1. 状态访问器仅 运行 在锁的上下文中
  2. 锁定机制实际上是线程安全的。

虽然第一个可以通过使用模拟技术进行相对测试,但后者很难测试,主要是因为验证某些代码是线程安全的涉及从多个线程访问线程安全资源的单元测试代码同一时间。即使这种技术也不是防弹的,因为我们不能完全控制我们从单元测试创​​建的线程的执行顺序,也不能完全控制每个线程的分配时间以确保我们捕获所有可能发生的竞争条件。

考虑到以上情况,我建议编写一个提供锁定机制的小 class/struct,并在 state 访问器中使用它。像这样分离职责可以更容易地通过代码审查评估锁定机制的正确性。

因此,我的建议是将线程安全代码移动到专用包装器中,并使用来自 Machine class:

的包装器
/// A struct that just wraps a value and access it in a thread safe manner
public struct ThreadSafeBox<T> {
    private var _value: T

    /// Thread safe value, uses the main thread to synchronize the accesses
    public var value: T {
        get {
            if Thread.isMainThread { return _value }
            else { return DispatchQueue.main.sync { _value } }
        }
        set {
            if Thread.isMainThread { _value = newValue }
            else { DispatchQueue.main.sync { _value = newValue } }
        }
    }

    /// Initializes the box with the given value
    init(_ value: T) {
        _value = value
    }
}

ThreadSafeBox代码相对较小,任何设计缺陷都可以在代码审查时发现,因此理论上可以通过代码分析证明其线程安全性。一旦我们证明了 ThreadSafeBox 的可靠性,那么我们就可以保证 Machine 相对于它的 state 属性.

也是线程安全的

如果你真的想测试 属性 访问器,你可以验证 get/set 操作 运行 仅在主线程上的事实,这应该足以验证线程安全。请注意,锁定机制与 class 的实现细节有关,而单元测试实现细节的缺点是将单元和单元测试紧密耦合。如果实现细节发生变化,这可能导致需要更新测试,从而降低测试的可靠性。