基于协议的 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