跳转至

基于协议的 Swift 依赖注入测试

通过将外部依赖(文件系统、网络、iCloud)抽象为小而聚焦的协议,使 Swift 代码更易测试。该模式支持无 I/O 的确定性测试。

启动条件

  • 编写访问文件系统、网络或外部 API 的 Swift 代码时
  • 需要测试错误处理路径,而不引发实际故障时
  • 构建需要在不同环境(应用、测试、SwiftUI 预览)中运行的模块时
  • 设计支持 Swift 并发(Actor、Sendable)的可测试架构时

核心模式

1. 定义小而聚焦的协议

每个协议只处理一个外部关注点。

// File system access
public protocol FileSystemProviding: Sendable {
    func containerURL(for purpose: Purpose) -> URL?
}

// File read/write operations
public protocol FileAccessorProviding: Sendable {
    func read(from url: URL) throws -> Data
    func write(_ data: Data, to url: URL) throws
    func fileExists(at url: URL) -> Bool
}

// Bookmark storage (e.g., for sandboxed apps)
public protocol BookmarkStorageProviding: Sendable {
    func saveBookmark(_ data: Data, for key: String) throws
    func loadBookmark(for key: String) throws -> Data?
}

2. 创建默认(生产环境)实现

public struct DefaultFileSystemProvider: FileSystemProviding {
    public init() {}

    public func containerURL(for purpose: Purpose) -> URL? {
        FileManager.default.url(forUbiquityContainerIdentifier: nil)
    }
}

public struct DefaultFileAccessor: FileAccessorProviding {
    public init() {}

    public func read(from url: URL) throws -> Data {
        try Data(contentsOf: url)
    }

    public func write(_ data: Data, to url: URL) throws {
        try data.write(to: url, options: .atomic)
    }

    public func fileExists(at url: URL) -> Bool {
        FileManager.default.fileExists(atPath: url.path)
    }
}

3. 创建用于测试的模拟实现

public final class MockFileAccessor: FileAccessorProviding, @unchecked Sendable {
    public var files: [URL: Data] = [:]
    public var readError: Error?
    public var writeError: Error?

    public init() {}

    public func read(from url: URL) throws -> Data {
        if let error = readError { throw error }
        guard let data = files[url] else {
            throw CocoaError(.fileReadNoSuchFile)
        }
        return data
    }

    public func write(_ data: Data, to url: URL) throws {
        if let error = writeError { throw error }
        files[url] = data
    }

    public func fileExists(at url: URL) -> Bool {
        files[url] != nil
    }
}

4. 使用默认参数注入依赖

生产代码使用默认值,测试注入模拟对象。

public actor SyncManager {
    private let fileSystem: FileSystemProviding
    private let fileAccessor: FileAccessorProviding

    public init(
        fileSystem: FileSystemProviding = DefaultFileSystemProvider(),
        fileAccessor: FileAccessorProviding = DefaultFileAccessor()
    ) {
        self.fileSystem = fileSystem
        self.fileAccessor = fileAccessor
    }

    public func sync() async throws {
        guard let containerURL = fileSystem.containerURL(for: .sync) else {
            throw SyncError.containerNotAvailable
        }
        let data = try fileAccessor.read(
            from: containerURL.appendingPathComponent("data.json")
        )
        // Process data...
    }
}

5. 使用 Swift Testing 编写测试

import Testing

@Test("Sync manager handles missing container")
func testMissingContainer() async {
    let mockFileSystem = MockFileSystemProvider(containerURL: nil)
    let manager = SyncManager(fileSystem: mockFileSystem)

    await #expect(throws: SyncError.containerNotAvailable) {
        try await manager.sync()
    }
}

@Test("Sync manager reads data correctly")
func testReadData() async throws {
    let mockFileAccessor = MockFileAccessor()
    mockFileAccessor.files[testURL] = testData

    let manager = SyncManager(fileAccessor: mockFileAccessor)
    let result = try await manager.loadData()

    #expect(result == expectedData)
}

@Test("Sync manager handles read errors gracefully")
func testReadError() async {
    let mockFileAccessor = MockFileAccessor()
    mockFileAccessor.readError = CocoaError(.fileReadCorruptFile)

    let manager = SyncManager(fileAccessor: mockFileAccessor)

    await #expect(throws: SyncError.self) {
        try await manager.sync()
    }
}

最佳实践

  • 单一职责:每个协议只处理一个关注点——不要创建拥有许多方法的“上帝协议”
  • Sendable 一致性:当协议跨 Actor 边界使用时是必需的
  • 默认参数:生产代码默认使用实际实现。只有测试需要指定模拟对象
  • 模拟错误:设计具有可配置错误属性的模拟对象,以测试故障路径
  • 只模拟边界:模拟外部依赖(文件系统、网络、API),不要模拟内部类型

应避免的反模式

  • 创建覆盖所有外部访问的单个大型协议
  • 模拟没有外部依赖的内部类型
  • 使用 #if DEBUG 条件语句代替适当的依赖注入
  • 与 Actor 组合使用时忘记 Sendable 一致性
  • 过度设计:如果类型没有外部依赖,则不需要协议

使用场景

  • 任何涉及文件系统、网络或外部 API 的 Swift 代码
  • 需要测试在实际环境中难以触发的错误处理路径时
  • 构建需要在应用、测试、SwiftUI 预览上下文中运行的模块时
  • 需要采用 Swift 并发(Actor、结构化并发)的可测试架构的应用

本页原文 Markdown:在 AtomGit 查看·内容源自开源项目 ev/ECC