如何为SwiftUI App生命周期应用（含CoreData预览数据）运行UI测试？

当然可以实现！核心思路是通过给测试App传递启动参数，让App在测试模式下加载带有预设预览数据的CoreData上下文，不用再手动录入大量测试数据。下面结合你提供的代码，给出具体的实现步骤：

1. 给UI测试的App添加启动标记参数 #

在你的UI测试类的setUpWithError方法里，给XCUIApplication添加一个自定义启动参数，用来告诉App当前处于UI测试环境：

class MyUITests: XCTestCase {
    var app: XCUIApplication!
    override func setUpWithError() throws {
        app = XCUIApplication()
        // 添加测试环境标记参数
        app.launchArguments += ["-isUITesting"]
        app.launch()
        continueAfterFailure = false
    }
    
    // ... 你的测试用例
}

2. 扩展PersistenceController，添加测试专用实例 #

修改你现有的PersistenceController，新增一个测试用的初始化逻辑，直接加载你已经写好的预览数据。这里建议用内存型存储，测试结束后数据会自动清空，不会影响真实App的数据：

struct PersistenceController {
    static let shared = PersistenceController()
    // 新增测试专用的实例
    static let test = PersistenceController(inMemory: true)
    
    let container: NSPersistentContainer
    
    init(inMemory: Bool = false) {
        container = NSPersistentContainer(name: "YourModelName") // 替换成你的CoreData模型名
        if inMemory {
            // 使用内存型存储，测试数据不会持久化
            container.persistentStoreDescriptions.first?.url = URL(fileURLWithPath: "/dev/null")
        }
        container.loadPersistentStores(completionHandler: { (storeDescription, error) in
            if let error = error as NSError? {
                fatalError("Unresolved error \(error), \(error.userInfo)")
            }
        })
        container.viewContext.automaticallyMergesChangesFromParent = true
        
        // 如果是测试环境，加载你已有的预览数据
        if inMemory {
            self.loadPreviewData()
        }
    }
    
    // 这里替换成你已经实现的预览数据加载方法
    private func loadPreviewData() {
        // 示例：用你之前写好的预览数据生成逻辑填充上下文
        // 比如你之前在PreviewProvider里用到的代码，直接移到这里即可
        let sampleItem = Item(context: container.viewContext)
        sampleItem.timestamp = Date()
        sampleItem.title = "Test Sample Item"
        
        do {
            try container.viewContext.save()
        } catch {
            let nsError = error as NSError
            fatalError("Failed to load preview data: \(nsError), \(nsError.userInfo)")
        }
    }
}

3. 在App入口根据启动参数切换环境 #

修改你的MyApp结构体，在初始化时检查启动参数，决定使用普通生产环境的PersistenceController还是测试专用的：

@main struct MyApp: App {
    let persistenceController: PersistenceController
    
    init() {
        // 检查是否带有测试标记参数
        if CommandLine.arguments.contains("-isUITesting") {
            persistenceController = .test
        } else {
            persistenceController = .shared
        }
    }
    
    var body: some Scene {
        WindowGroup {
            ContentView()
                .environment(\.managedObjectContext, persistenceController.container.viewContext)
        }
    }
}

4. 验证测试效果 #

现在运行你的UI测试，App启动时会自动加载预设的预览数据，你可以直接在测试用例中验证这些数据是否存在，或者基于这些数据进行操作，不用再手动录入了。比如：

func testSampleItemDisplayed() throws {
    XCTAssert(app.staticTexts["Test Sample Item"].exists)
}

这种方式既复用了你已经写好的预览数据逻辑，又能在UI测试中快速得到预设的测试环境，完全避免了手动录入数据的繁琐。

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