原文链接：Testing Swift’s ErrorType: An Exploration

译者：mmoaay

在本篇中，我们对 Swift 新错误类型的本质进行探究，观察并测试错误处理实现的可能性和限制。最后我们以一个说明样例、以及一些有用的资源结尾

如何实现 ErrorType 协议

protocol ErrorType {
}

然而，当我们试着去实现 ErrorType 时，很快就会发现为了满足这个协议至少有一些东西是必须的。比如，如果以枚举的方式实现它，一切OK。

enum MyErrorEnum : ErrorType {
}

但是如果以结构体的方式实现它，问题来了。

struct MyErrorStruct : ErrorType {
}

我们最初的想法可能是，也许 ErrorType 是一种特殊类型，编译器以特殊的方式来对它进行支持，而且只能用 Swift 原生的枚举来实现。但随后你又会想起 NSError 也满足这个协议，所以它不可能有那么特殊。所以我们下一步的尝试就是：通过一个 NSObject 的派生类实现这个协议

@objc class MyErrorClass: ErrorType {
}

不幸滴是，仍然不行。

更新：从 Xcode 7 beta 5 版本开始，我们可能不需要花费其他精力就可以为结构体和类实现 ErrorType 协议。所以下面的解决方法也不再需要了，但是仍然留作参考。

允许结构体和类实现 ErrorType 协议。（21867608）

怎么会这样？

(lldb) type lookup ErrorType
protocol ErrorType {
  var _domain: Swift.String { get }
  var _code: Swift.Int { get }
}

这样一来 NSError 满足这个定义的原因就很明白了：它有这些属性，在 ivars 的支持下，不用动态查找就可以被 Swift 访问。还有一点不明白的是为什么 Swift 的一等公民枚举可以自动满足这个协议。也许其内部仍然存在一些魔法？

如果我们用我们新获得的知识再去实现结构体和类，一切就OK了。

struct MyErrorStruct : ErrorType {
  let _domain: String
  let _code: Int
}

class MyErrorClass : ErrorType {
  let _domain: String
  let _code: Int

  init(domain: String, code: Int) {
    _domain = domain
    _code = code
  }
}

捕获其他被抛出的错误

假设我们需要反序列化一个 JSON 串，但是不确定它是不是有效的。我们将使用 Foundation 的 NSJSONSerialization 来做这件事情。当我们传给它一个异常的 JSON 串时，它会抛出一个错误码为 3840 的错误。

当然，你可以用通用的错误来捕获它，然后手动检查 _domain 和 _code 域，但是我们有更优雅的替代方案。

let json : NSString = "{"
let data = json.dataUsingEncoding(NSUTF8StringEncoding)
do {
    let object : AnyObject = try NSJSONSerialization.JSONObjectWithData(data!, options: [])
    print(object)
} catch let error {
    if   error._domain == NSCocoaErrorDomain
      && error._code   == 3840 {
        print("Invalid format")
    } else {
        throw error
    }
}

另外一个替代方案就是我们引入一个通用的错误结构体，这个结构体通过我们之前发现的方法满足 ErrorType 协议。当我们为它实现模式匹配操作符 ~= 时，我们就可以在 do … catch 分支中使用它。

struct Error : ErrorType {
    let domain: String
    let code: Int

    var _domain: String {
        return domain
    }
    var _code: Int {
        return code
    }
}

func ~=(lhs: Error, rhs: ErrorType) -> Bool {
    return lhs._domain == rhs._domain
        && rhs._code   == rhs._code
}

let json : NSString = "{"
let data = json.dataUsingEncoding(NSUTF8StringEncoding)
do {
    let object : AnyObject = try NSJSONSerialization.JSONObjectWithData(data!, options: [])
    print(object)
} catch Error(domain: NSCocoaErrorDomain, code: 3840) {
    print("Invalid format")
}

但在当前情况下，还可以用 NSCocoaError，这个辅助类包含大量定义了各种错误的静态方法。

这里所产生的叫做 NSCocoaError.PropertyListReadCorruptError 错误，虽然不是那么明显，但是它确实是有我们需要的错误码的。不管你是通过标准库还是第三方框架捕获错误，如果有像这样的东西，你就需要依赖给定的常数而不是自己再去定义一次。

let json : NSString = "{"
let data = json.dataUsingEncoding(NSUTF8StringEncoding)
do {
    let object : AnyObject = try NSJSONSerialization.JSONObjectWithData(data!, options: [])
    print(object)
} catch NSCocoaErrorDomain {
    print("Invalid format")
}

自定义错误处理的编写规范

现在我们对这个错误类型在底层的工作方式有了一些基本的认识，同时对如何在测试时让它遵循我们的意愿也有了一些想法。所以我们来展示一个小的测试用例：我们有一个银行 App，然后我们想在业务逻辑里面为现实活动建模型。我们创建了代表银行帐号的结构体 Account，它包含一个接口，这个接口暴露了一个方法用来在预算范围内进行交易。

public enum Error : ErrorType {
    case TransactionExceedsFunds
    case NonPositiveTransactionNotAllowed(amount: Int)
}

public struct Account {
    var fund: Int

    public mutating func withdraw(amount: Int) throws {
        guard amount < fund else {
            throw Error.TransactionExceedsFunds
        }
        guard amount > 0 else {
            throw Error.NonPositiveTransactionNotAllowed(amount: amount)
        }
        fund -= amount
    }
}

class AccountTests {
    func testPreventNegativeWithdrawals() {
        var account = Account(fund: 100)
        do {
            try account.withdraw(-10)
            XCTFail("Withdrawal of negative amount succeeded, but was expected to fail.")
        } catch Error.NonPositiveTransactionNotAllowed(let amount) {
            XCTAssertEqual(amount, -10)
        } catch {
            XCTFail("Catched error \"\(error)\", but not the expected: \"\(Error.NonPositiveTransactionNotAllowed)\"")
        }
    }

    func testPreventExceedingTransactions() {
        var account = Account(fund: 100)
        do {
            try account.withdraw(101)
            XCTFail("Withdrawal of amount exceeding funds succeeded, but was expected to fail.")
        } catch Error.TransactionExceedsFunds {
            // 预期结果
        } catch {
            XCTFail("Catched error \"\(error)\", but not the expected: \"\(Error.TransactionExceedsFunds)\"")
        }
    }
}

现在假想我们有更多的方法和更多的错误场景。在以测试为导向的开发方式下，我们想对它们都进行测试，从而保证所有的错误都被正确滴抛出来——我们当然不想把钱转到错误的地方去！理想情况下，我们不想在所有的测试代码中都重复这个 do-catch 。实现一个抽象，我们可以把它放到一个高阶函数中。

/// 为 ErrorType 实现模式匹配
public func ~=(lhs: ErrorType, rhs: ErrorType) -> Bool {
    return lhs._domain == rhs._domain
        && lhs._code   == rhs._code
}

func AssertThrow<R>(expectedError: ErrorType, @autoclosure _ closure: () throws -> R) -> () {
    do {
        try closure()
        XCTFail("Expected error \"\(expectedError)\", "
            + "but closure succeeded.")
    } catch expectedError {
        // 预期结果.
    } catch {
        XCTFail("Catched error \"\(error)\", "
            + "but not from the expected type "
            + "\"\(expectedError)\".")
    }
}

这段代码可以这样使用：

class AccountTests : XCTestCase {
    func testPreventExceedingTransactions() {
        var account = Account(fund: 100)
        AssertThrow(Error.TransactionExceedsFunds, try account.withdraw(101))
    }

    func testPreventNegativeWithdrawals() {
        var account = Account(fund: 100)
        AssertThrow(Error.NonPositiveTransactionNotAllowed(amount: -10), try account.withdraw(-20))
    }
}

但你可能会发现， 预期出现的参数化错误 NonPositiveTransactionNotAllowed 比这里所用到的参数要多个 amount。我们该如何对错误场景和它们相关的值做出强有力的假设呢？ 首先，我们可以为错误类型实现 Equatable 协议, 然后在相等操作符的实现中添加对相关场景的参数个数的检查。

/// 对我们的错误类型进行扩展然后实现 `Equatable`。
/// 这必须是对每一个具体的类型来做的，
/// 而不是为 `ErrorType` 统一实现。
extension Error : Equatable {}

/// 为协议 `Equatable` 以 required 的方式实现 `==` 操作符。
public func ==(lhs: Error, rhs: Error) -> Bool {
    switch (lhs, rhs) {
    case (.NonPositiveTransactionNotAllowed(let l), .NonPositiveTransactionNotAllowed(let r)):
        return l == r
    default:
        // 我们需要在默认场景，为各种组合场景返回 false。
        // 通过根据 domain 和 code 进行比较的方式，我们可以保证
        // 一旦我们添加了其他的错误场景，如果这个场景有相应的值
        // 我只需要回到并修改 Equatable 的实现即可
        return lhs._domain == rhs._domain
            && lhs._code   == rhs._code
    }
}

下一步就是让 AssertThrow 知道有合理的错误。你可能会想，我们可以扩展已存在的 AssertThrow 实现，只是简单检查一下预期的错误是否合理。但是不幸滴是根本没用：

“Equatable” 协议只能被当作泛型约束，因为它需要满足 Self 或者关联类型的必要条件

相反，我们可以通过多一个泛型参数做首参的方式重载 AssertThrow 。

func AssertThrow<R, E where E: ErrorType, E: Equatable>(expectedError: E, @autoclosure _ closure: () throws -> R) -> () {
    do {
        try closure()
        XCTFail("Expected error \"\(expectedError)\", "
            + "but closure succeeded.")
    } catch let error as E {
        XCTAssertEqual(error, expectedError,
            "Catched error is from expected type, "
                + "but not the expected case.")
    } catch {
        XCTFail("Catched error \"\(error)\", "
            + "but not the expected error "
            + "\"\(expectedError)\".")
    }
}

然后跟预期一样我们的测试最终返回了失败。

注意后者的断言实现就对错误的类型进行了强有力的假设。

不要使用“捕获其他被抛出的错误”下面的方法，因为跟目前的方法相比，它不能匹配类型。可能这只是反对意见而不是规则。

一些有用的资源

要开心滴测试哦~