iOS中如何用委托Mock外部框架类？Connector框架实操问题

首先，咱们先拆解你遇到的问题：你尝试用带关联类型的协议来做Mock，但Swift报错说这类协议只能用作泛型约束，不能直接当变量类型——这是Swift协议的一个核心规则：带有associatedtype或Self约束的协议，无法直接声明为属性、变量或函数参数的类型，必须作为泛型约束来使用。

为什么你的方案会报错？ #

你定义的ConnectorProtocol继承了Connector还添加了关联类型MockResult，这就触发了Swift的这个规则：当协议有associatedtype时，编译器无法确定它的具体类型（因为关联类型是留给遵守协议的类型去指定的），所以不能直接用var connector: ConnectorProtocol?这种写法。

更优雅的Mock方案（无需关联类型） #

我们可以换个思路：定义一个独立的协议来抽象Connector的行为，而不是继承原框架的Connector类，这样既能实现Mock，又能避开关联类型的限制。

步骤1：抽象Connector的行为协议

先定义一个只包含你需要的方法的协议，和原框架的Connector类解耦：

// 抽象Connector的核心行为：设置代理、发起连接
protocol ConnectorProtocol {
    func setDelegate(_ delegate: ConnectionDelegate)
    func connect()
}

然后让原框架的Connector自动遵守这个协议（因为它已经实现了这两个方法，不需要额外写代码）：

extension Connector: ConnectorProtocol {}

步骤2：创建MockConnector和MockConnectionResult

因为原ConnectionResult是框架里的复杂类，我们直接继承它来创建Mock版本，方便设置测试需要的属性：

// Mock的ConnectionResult，继承自框架的类
class MockConnectionResult: ConnectionResult {
    // 根据测试需求初始化父类的复杂属性
    init() {
        super.init()
        // 这里可以设置mock的属性值，比如：
        // self.deviceName = "Mock BLE Device"
        // self.connectionStatus = .success
    }
}

// Mock的Connector，遵守ConnectorProtocol
class MockConnector: ConnectorProtocol {
    private weak var delegate: ConnectionDelegate?
    
    func setDelegate(_ delegate: ConnectionDelegate) {
        self.delegate = delegate
    }
    
    func connect() {
        // 模拟无BLE设备时的连接结果，直接返回Mock数据
        let mockResult = MockConnectionResult()
        delegate?.onConnected(result: mockResult)
    }
}

步骤3：修改ConnectionService支持依赖注入

现在把ConnectionService改成通过初始化注入ConnectorProtocol实例，这样既可以传入真实的Connector，也可以传入MockConnector：

class ConnectionService: ConnectionDelegate {
    private let connector: ConnectorProtocol
    
    // 初始化时注入Connector实例，支持真实/Mock切换
    init(connector: ConnectorProtocol) {
        self.connector = connector
        self.connector.setDelegate(self)
    }
    
    func connect() {
        connector.connect()
    }
    
    func onConnected(result: ConnectionResult) {
        // 处理连接结果
        if let mockResult = result as? MockConnectionResult {
            // 针对Mock场景的特殊逻辑（如果需要）
            print("Mock连接成功：\(mockResult)")
        } else {
            // 真实BLE设备的处理逻辑
            print("真实设备连接成功")
        }
    }
}

使用方式

// 真实场景：使用框架的Connector
let realConnector = Connector()
let realService = ConnectionService(connector: realConnector)
realService.connect()

// 测试场景：使用MockConnector
let mockConnector = MockConnector()
let mockService = ConnectionService(connector: mockConnector)
mockService.connect()

如果你一定要用关联类型的方案（泛型版本） #

如果因为某些原因必须保留关联类型，那可以把ConnectionService改成泛型类，用协议作为泛型约束：

protocol ConnectorProtocol {
    associatedtype ResultType: ConnectionResult
    func setDelegate(_ delegate: ConnectionDelegate)
    func connect()
}

// 让原Connector遵守协议
extension Connector: ConnectorProtocol {
    typealias ResultType = ConnectionResult
}

// MockConnector指定关联类型为MockConnectionResult
class MockConnector: ConnectorProtocol {
    typealias ResultType = MockConnectionResult
    private weak var delegate: ConnectionDelegate?
    
    func setDelegate(_ delegate: ConnectionDelegate) {
        self.delegate = delegate
    }
    
    func connect() {
        let mockResult = MockConnectionResult()
        delegate?.onConnected(result: mockResult)
    }
}

// 泛型版本的ConnectionService
class ConnectionService<ConnectorType: ConnectorProtocol>: ConnectionDelegate {
    private let connector: ConnectorType
    
    init(connector: ConnectorType) {
        self.connector = connector
        self.connector.setDelegate(self)
    }
    
    func connect() {
        connector.connect()
    }
    
    func onConnected(result: ConnectionResult) {
        // 处理逻辑同上
    }
}

使用时需要指定泛型类型：

let realService = ConnectionService(connector: Connector())
let mockService = ConnectionService(connector: MockConnector())

总结 #

推荐第一种独立协议+依赖注入的方案，它更简洁、灵活，不需要引入泛型，能很好地实现Mock需求，同时符合依赖倒置原则，让你的代码和框架解耦。

内容的提问来源于stack exchange，提问作者Sazzad Hissain Khan