如何测试 Swift 枚举与关联值的相等性

发布于11月03日

我想测试两个Swift枚举值的相等性.例如:

enum SimpleToken {
    case Name(String)
    case Number(Int)
}
let t1 = SimpleToken.Number(123)
let t2 = SimpleToken.Number(123)
XCTAssert(t1 == t2)

但是，编译器不会编译等式表达式:

error: could not find an overload for '==' that accepts the supplied arguments
    XCTAssert(t1 == t2)
    ^~~~~~~~~~~~~~~~~~~

我必须定义自己的相等运算符重载吗？我希望Swift编译器能像Scala和Ocaml一样自动处理它.

Swift 4.1+

正如@jedwidz所指出的，从Swift 4.1(由于SE-0185)，Swift还支持为带有关联值的枚举合成Equatable和Hashable.

因此，如果您使用的是Swift 4.1或更新版本，以下内容将自动合成必要的方法，以便XCTAssert(t1 == t2)种方法都能正常工作.关键是将Equatable协议添加到枚举中.

enum SimpleToken: Equatable {
    case Name(String)
    case Number(Int)
}
let t1 = SimpleToken.Number(123)
let t2 = SimpleToken.Number(123)

Before Swift 4.1

正如其他人所指出的，Swift不会自动合成必要的等式运算符.不过，让我提出一个更清洁(IMHO)的实施方案:

enum SimpleToken: Equatable {
    case Name(String)
    case Number(Int)
}

public func ==(lhs: SimpleToken, rhs: SimpleToken) -> Bool {
    switch (lhs, rhs) {
    case let (.Name(a),   .Name(b)),
         let (.Number(a), .Number(b)):
      return a == b
    default:
      return false
    }
}

这远非理想——有很多重复——但至少你不需要在内部使用if语句进行嵌套切换.