Swift 扩展和单元测试

Question

我有一些 UT 的问题，我正在尝试快速编写

我有一个带有“做事”扩展名的协议：

protocol MyProtocol: class
{
    var myVar: SomeClass { get }

    func doStuff(identifier: String) -> Bool
}

extension MyProtocol
{
    func doStuff(identifier: String) -> Bool {
        return true
    }
}

然后是一个实现我的协议的类

final class MyClass: MyProtocol {

}

这个类有一个扩展，它实现了另一个协议，它有一个我应该测试的方法

public protocol MyOtherProtocol: class {
    func methodToTest() -> Bool
}

extension MyClass: MyOtherProtocol {
    public func methodToTest() {
        if doStuff() {
            return doSomething()
        }

    }
}

这个设置有没有办法模拟 doStuff 方法？

Answer 1

解决协议而不是类是一个很好的做法。因此，您可以扩展协议，而不是扩展类

extension MyOtherProtocol where Self: MyProtocol {
    public func methodToTest() {
        if doStuff() {
            return doSomething()
        }

    }
}

所以你的扩展会知道 doStuff 存在，但不知道它的实现。然后让你的类同时符合两者。

extension MyClass: MyOtherProtocol {}

所以在模拟中你可以实现

class MyMockClass: MyProtocol, MyOtherProtocol {
    func doStuff() -> Bool {
        return true
    }
}