如何Mock UIImageView的Alamofire扩展以开展单元测试？

Alamofire的UIImageView扩展依赖内部的ImageDownloader处理图片下载，要Mock它的响应，有两种常用方案：

方案一：直接替换Alamofire的下载器 #

Alamofire的imageView.af属性暴露了downloader接口，我们可以自定义Mock下载器来替换它，直接控制返回结果：

1. 定义Mock下载器 #

import Alamofire
import AlamofireImage

class MockImageDownloader: ImageDownloader {
    // 预定义要返回的Mock响应
    var mockResponse: Result<Image, AFError>?
    
    override func download(
        _ urlRequest: URLRequestConvertible,
        completion: @escaping (Result<Image, AFError>) -> Void
    ) -> DownloadRequest? {
        mockResponse.map(completion)
        return nil
    }
}

2. 编写单元测试用例 #

import XCTest
import Alamofire
import AlamofireImage

class ImageLoadTests: XCTestCase {
    // 测试图片加载成功场景
    func testImageLoadSuccess() {
        let imageView = UIImageView()
        let mockDownloader = MockImageDownloader()
        let testImage = UIImage()
        
        // 设置成功响应
        mockDownloader.mockResponse = .success(testImage)
        // 替换默认下载器
        imageView.af.downloader = mockDownloader
        
        let testURL = URL(string: "https://example.com/test.jpg")!
        var receivedImage: UIImage?
        var receivedError: AFError?
        
        // 调用目标方法
        imageView.af.setImage(withURL: testURL) { response in
            receivedImage = response.value
            receivedError = response.error
        }
        
        // 断言结果符合预期
        XCTAssertEqual(receivedImage, testImage)
        XCTAssertNil(receivedError)
    }
    
    // 测试图片加载失败场景
    func testImageLoadFailure() {
        let imageView = UIImageView()
        let mockDownloader = MockImageDownloader()
        let testError = AFError.invalidURL(url: URL(string: "invalid-url")!)
        
        // 设置失败响应
        mockDownloader.mockResponse = .failure(testError)
        imageView.af.downloader = mockDownloader
        
        let testURL = URL(string: "https://example.com/test.jpg")!
        var receivedImage: UIImage?
        var receivedError: AFError?
        
        imageView.af.setImage(withURL: testURL) { response in
            receivedImage = response.value
            receivedError = response.error
        }
        
        XCTAssertNil(receivedImage)
        XCTAssertEqual(receivedError?.errorDescription, testError.errorDescription)
    }
}

方案二：抽象协议解耦（更推荐） #

通过定义图片加载协议，将业务代码与Alamofire的具体实现解耦，测试时直接使用Mock协议实现：

1. 定义图片加载协议 #

import UIKit

protocol ImageLoader {
    func loadImage(from url: URL, completion: @escaping (Result<UIImage, Error>) -> Void)
}

2. 让UIImageView遵循协议（Alamofire实现） #

import AlamofireImage

extension UIImageView: ImageLoader {
    func loadImage(from url: URL, completion: @escaping (Result<UIImage, Error>) -> Void) {
        self.af.setImage(withURL: url) { response in
            completion(response.result.mapError { $0 as Error })
        }
    }
}

3. 编写Mock协议实现 #

class MockImageLoader: ImageLoader {
    var mockResult: Result<UIImage, Error>?
    
    func loadImage(from url: URL, completion: @escaping (Result<UIImage, Error>) -> Void) {
        mockResult.map(completion)
    }
}

4. 业务代码依赖协议 #

业务中不再直接调用af.setImage，而是使用ImageLoader协议：

// 示例业务逻辑
class ImageDisplayViewModel {
    private let imageLoader: ImageLoader
    
    // 通过构造注入依赖
    init(imageLoader: ImageLoader = UIImageView()) {
        self.imageLoader = imageLoader
    }
    
    func loadImage(for imageView: UIImageView, url: URL) {
        imageLoader.loadImage(from: url) { [weak imageView] result in
            switch result {
            case .success(let image):
                imageView?.image = image
            case .failure:
                imageView?.image = UIImage(named: "placeholder")
            }
        }
    }
}

5. 测试业务逻辑 #

class ImageDisplayViewModelTests: XCTestCase {
    func testLoadImageSuccess() {
        let mockLoader = MockImageLoader()
        let testImage = UIImage()
        mockLoader.mockResult = .success(testImage)
        
        let viewModel = ImageDisplayViewModel(imageLoader: mockLoader)
        let imageView = UIImageView()
        let testURL = URL(string: "https://example.com/test.jpg")!
        
        viewModel.loadImage(for: imageView, url: testURL)
        
        XCTAssertEqual(imageView.image, testImage)
    }
    
    func testLoadImageFailure() {
        let mockLoader = MockImageLoader()
        mockLoader.mockResult = .failure(NSError(domain: "Test", code: -1))
        
        let viewModel = ImageDisplayViewModel(imageLoader: mockLoader)
        let imageView = UIImageView()
        let testURL = URL(string: "https://example.com/test.jpg")!
        
        viewModel.loadImage(for: imageView, url: testURL)
        
        XCTAssertEqual(imageView.image, UIImage(named: "placeholder"))
    }
}

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