如何在测试Target中配置依赖注入管理器使用测试专属图片加载器

你的思路完全没问题——把测试专用的依赖实现放在测试Target里，避免污染主Target是正确的实践。下面是几个可行的配置方案：

1. 用测试基类全局配置DI #

创建一个所有UI测试用例都继承的基类，利用XCTestCase的类级setUp()方法，在所有测试执行前一次性配置DI管理器，替换图片加载器的实现：

import XCTest
@testable import 你的主Target名称

// 测试Target专属的本地图片加载器
class TestImageLoader: ImageLoaderProtocol {
    func loadImage(url: URL, completion: @escaping (UIImage?) -> Void) {
        // 加载测试Target里的本地占位图
        let placeholder = UIImage(named: "test_placeholder", in: Bundle(for: Self.self), compatibleWith: nil)
        completion(placeholder)
    }
}

// 所有测试用例的基类
class BaseUITestCase: XCTestCase {
    override class func setUp() {
        super.setUp()
        // 替换DI管理器中的图片加载器实现
        DependencyInjector.shared.register(type: ImageLoaderProtocol.self, implementation: TestImageLoader())
    }

    override class func tearDown() {
        super.tearDown()
        // 可选：测试结束后重置DI配置，避免测试间污染
        DependencyInjector.shared.resetRegistrations()
    }
}

// 你的UI测试用例直接继承基类即可
class HomePageUITest: BaseUITestCase {
    func testBannerImagesLoad() {
        // 这里所有视图、单元格都会自动使用TestImageLoader
    }
}

2. 给主Target的DI管理器加测试专用入口 #

在主Target的DI管理器中添加一个允许覆盖注册的方法，让测试代码可以安全替换依赖实现，不影响主逻辑：

// 主Target中的DI管理器
class DependencyInjector {
    static let shared = DependencyInjector()
    private var registrations: [AnyHashable: Any] = [:]

    // 常规注册方法
    func register<T>(type: T.Type, implementation: T) {
        registrations[ObjectIdentifier(type)] = implementation
    }

    // 测试专用的覆盖注册方法
    func overrideRegistration<T>(type: T.Type, implementation: T) {
        registrations[ObjectIdentifier(type)] = implementation
    }

    func resolve<T>() -> T? {
        return registrations[ObjectIdentifier(T.self)] as? T
    }

    // 测试专用的重置方法
    func resetRegistrations() {
        registrations.removeAll()
        // 重置后可以重新注册默认依赖
        registerDefaultDependencies()
    }

    private func registerDefaultDependencies() {
        // 主Target的默认图片加载器注册逻辑
        register(type: ImageLoaderProtocol.self, implementation: NetworkImageLoader())
    }
}

之后在测试基类的setUp()中调用overrideRegistration即可替换成测试实现。

关键注意点 #

• 确保ImageLoaderProtocol定义在主Target中，测试Target通过@testable import访问协议，而TestImageLoader仅在测试Target实现。
• 类级setUp()会在所有测试用例执行前只运行一次，适合全局配置；如果需要每个测试方法前重置DI状态，改用实例级setUp()。

内容的提问来源于stack exchange，提问作者Sonastra