我想测试两个Swift enum值的相等性。例如:
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那样。
当前回答
斯威夫特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)
Swift 4.1之前
正如其他人所注意到的，Swift不会自动合成必要的相等操作符。不过，让我提出一个更清晰的实现:
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语句里面做嵌套开关。
其他回答
斯威夫特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)
Swift 4.1之前
正如其他人所注意到的，Swift不会自动合成必要的相等操作符。不过，让我提出一个更清晰的实现:
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语句里面做嵌套开关。
enum MyEnum {
case none
case simple(text: String)
case advanced(x: Int, y: Int)
}
func ==(lhs: MyEnum, rhs: MyEnum) -> Bool {
switch (lhs, rhs) {
case (.none, .none):
return true
case let (.simple(v0), .simple(v1)):
return v0 == v1
case let (.advanced(x0, y0), .advanced(x1, y1)):
return x0 == x1 && y0 == y1
default:
return false
}
}
另一种选择是比较大小写的字符串表示:
XCTAssert(String(t1) == String(t2))
例如:
let t1 = SimpleToken.Number(123) // the string representation is "Number(123)"
let t2 = SimpleToken.Number(123)
let t3 = SimpleToken.Name("bob") // the string representation is "Name(\"bob\")"
String(t1) == String(t2) //true
String(t1) == String(t3) //false
除了上面的答案之外，您还可以添加计算属性作为助手。这也是优化可读性的许多方法之一。
enum UserAccountViewStack: Hashable {
case notLoggedIn
case initialDevicePairing
case deviceMainView
case leftMenu(LeftMenuStack? = nil)
case rightMenu(RightMenuStack? = nil)
static var `default`: Self {
.deviceMainView
}
var isLeftMenu: Bool {
if case .leftMenu = self {
return true
}
return false
}
var isRightMenu: Bool {
if case .rightMenu = self {
return true
}
return false
}
}
我在单元测试代码中使用这个简单的解决方法:
extension SimpleToken: Equatable {}
func ==(lhs: SimpleToken, rhs: SimpleToken) -> Bool {
return String(stringInterpolationSegment: lhs) == String(stringInterpolationSegment: rhs)
}
它使用字符串插值来执行比较。我不建议在生产代码中使用它，但是它很简洁，并且可以用于单元测试。
