如何在XCode的Unit Testing Bundle项目中实现iOS截图（非UI Testing Bundle）？

首先明确核心原因：XCUIScreenshot系列API属于UI Testing框架，仅能在独立的UI Testing进程中运行（该进程与宿主App进程分离，拥有系统级UI操作权限）；而Unit Testing Bundle的代码是运行在宿主App进程内部的，没有调用这类跨进程UI操作API的权限，这就是你之前两种尝试失败的根本原因。

下面提供两种实用的解决方案，按需选择：

方案一：UI Test + App内测试入口（推荐用于生成App Store截图） #

利用UI Test的截图权限，同时通过命令行参数触发App直接进入目标状态（避免纯点击操作的繁琐），兼顾白盒逻辑触发和稳定截图：

1. 在宿主App中添加测试专用逻辑（用#if DEBUG包裹，不会影响正式包）：

func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplication.LaunchOptionsKey: Any]?) -> Bool {
    #if DEBUG
    let args = ProcessInfo.processInfo.arguments
    // 跳过引导页直接进入主界面
    if args.contains("-skipOnboarding") {
        window?.rootViewController = MainViewController()
    }
    // 填充测试数据
    if args.contains("-loadTestData") {
        UserDefaults.standard.setValue("测试用户", forKey: "currentUserName")
        // 调用内部API加载页面数据
        DataManager.shared.loadMockData()
    }
    #endif
    return true
}

2. 在UI Testing Bundle中启动App并截图：

class AppStoreScreenshotTests: XCTestCase {
    func testMainScreenScreenshot() {
        let app = XCUIApplication()
        // 传入触发测试状态的参数
        app.launchArguments = ["-skipOnboarding", "-loadTestData"]
        app.launch()
        
        // 直接截取当前界面
        let screenshot = app.screenshot()
        // 保存截图到测试报告
        let attachment = XCTAttachment(screenshot: screenshot)
        attachment.name = "MainScreen_AppStore"
        attachment.lifetime = .keepAlways
        add(attachment)
    }
}

方案二：Unit Test + Core Graphics进程内截图（适合结合白盒逻辑的UI测试） #

在Unit Test中直接调用宿主App的内部API，用Core Graphics绘制当前窗口内容实现截图，完全在App进程内完成，无需依赖UI Test框架：

1. 在宿主App中添加截图工具方法（同样用#if DEBUG包裹）：

#if DEBUG
extension UIApplication {
    func captureCurrentScreen() -> UIImage? {
        guard let keyWindow = windows.first(where: { $0.isKeyWindow }) else { return nil }
        UIGraphicsBeginImageContextWithOptions(keyWindow.bounds.size, false, UIScreen.main.scale)
        keyWindow.drawHierarchy(in: keyWindow.bounds, afterScreenUpdates: true)
        let screenshot = UIGraphicsGetImageFromCurrentImageContext()
        UIGraphicsEndImageContext()
        return screenshot
    }
}
#endif

2. 在Unit Testing Bundle中调用并保存截图：

class UIScreenshotUnitTests: XCTestCase {
    func testProfileScreenScreenshot() {
        // 直接调用App内部逻辑，设置界面状态
        let profileVC = ProfileViewController()
        profileVC.user = MockUser.testUser()
        // 将VC设置为当前窗口根控制器
        UIApplication.shared.keyWindow?.rootViewController = profileVC
        
        // 触发界面更新
        profileVC.view.layoutIfNeeded()
        
        // 截图并保存
        if let screenshot = UIApplication.shared.captureCurrentScreen() {
            let attachment = XCTAttachment(image: screenshot)
            attachment.name = "ProfileScreen_UnitTest"
            attachment.lifetime = .keepAlways
            add(attachment)
        }
    }
}

两种方案对比 #

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