如何以编程方式关闭通过runModal()显示的NSAlert？

当然可以！在Cocoa单元测试里编程关闭runModal()展示的NSAlert，有几种实用的方案，我给你详细说说：

方案1：后台线程模拟点击默认按钮 #

因为runModal()会阻塞当前线程，所以我们需要提前在后台线程触发“点击按钮”的动作，再调用弹窗展示方法。这种方法简单直接，适合大多数基础测试场景：

import XCTest

class AlertTests: XCTestCase {
    func testClosingAlertViaAction() {
        // 后台线程延迟片刻，确保弹窗已经展示
        DispatchQueue.global().asyncAfter(deadline: .now() + 0.1) {
            // 给应用发送点击默认按钮的动作
            NSApplication.shared.sendAction(Selector(("performDefaultAction:")), to: nil, from: nil)
        }
        
        // 创建并展示Alert
        let alert = NSAlert()
        alert.messageText = "测试弹窗"
        alert.addButton(withTitle: "确定")
        let modalResult = alert.runModal()
        
        // 断言结果符合预期
        XCTAssertEqual(modalResult, .alertFirstButtonReturn)
    }
}

方案2：通过辅助功能API精准定位弹窗 #

如果你的测试场景里有多个模态窗口，或者需要指定点击非默认按钮，可以用Accessibility API来精准找到Alert并触发按钮点击。不过要注意，需要在测试目标的Info.plist中添加NSAccessibilityUsageDescription权限说明：

import XCTest
import Accessibility

class AlertTests: XCTestCase {
    func testClosingAlertWithAccessibility() {
        let expectation = self.expectation(description: "弹窗应被关闭")
        
        DispatchQueue.global().async {
            // 等待弹窗加载完成
            Thread.sleep(forTimeInterval: 0.1)
            
            // 获取当前应用的Accessibility元素
            guard let appElement = AXUIElementCreateApplication(NSApplication.shared.processIdentifier) else {
                XCTFail("无法获取应用Accessibility元素")
                return
            }
            
            // 遍历所有窗口，找到Alert类型的窗口
            var windows: AnyObject?
            guard AXUIElementCopyAttributeValue(appElement, kAXWindowsAttribute as CFString, &windows) == .success,
                  let windowList = windows as? [AXUIElement] else {
                XCTFail("无法获取窗口列表")
                return
            }
            
            for window in windowList {
                var role: AnyObject?
                AXUIElementCopyAttributeValue(window, kAXRoleAttribute as CFString, &role)
                if role as? String == kAXAlertRole as String {
                    // 找到Alert的默认按钮并触发点击
                    var defaultButton: AnyObject?
                    AXUIElementCopyAttributeValue(window, kAXDefaultButtonAttribute as CFString, &defaultButton)
                    if let button = defaultButton as? AXUIElement {
                        AXUIElementPerformAction(button, kAXPressAction as CFString)
                        expectation.fulfill()
                        return
                    }
                }
            }
            
            XCTFail("未找到目标Alert")
        }
        
        // 创建并展示带多个按钮的Alert
        let alert = NSAlert()
        alert.messageText = "测试弹窗"
        alert.addButton(withTitle: "确定")
        alert.addButton(withTitle: "取消")
        let modalResult = alert.runModal()
        
        waitForExpectations(timeout: 1, handler: nil)
        XCTAssertEqual(modalResult, .alertFirstButtonReturn)
    }
}

方案3：Method Swizzling 模拟返回结果 #

如果你不需要测试弹窗的实际展示，只需要验证业务逻辑对弹窗返回值的处理，可以用Method Swizzling替换runModal()方法，直接返回预设的结果。这种方法测试速度极快，适合单元测试中隔离UI依赖的场景：

import XCTest

extension NSAlert {
    static func swizzleRunModal() {
        let originalSelector = #selector(runModal)
        let swizzledSelector = #selector(swizzled_runModal)
        
        guard let originalMethod = class_getInstanceMethod(self, originalSelector),
              let swizzledMethod = class_getInstanceMethod(self, swizzledSelector) else {
            return
        }
        
        method_exchangeImplementations(originalMethod, swizzledMethod)
    }
    
    @objc private func swizzled_runModal() -> NSApplication.ModalResponse {
        // 直接返回预设的结果，比如点击“确定”按钮的返回值
        return .alertFirstButtonReturn
    }
}

class AlertTests: XCTestCase {
    override func setUp() {
        super.setUp()
        // 在测试前替换runModal方法
        NSAlert.swizzleRunModal()
    }
    
    func testAlertLogicWithSwizzling() {
        let alert = NSAlert()
        alert.messageText = "测试弹窗"
        alert.addButton(withTitle: "确定")
        let modalResult = alert.runModal()
        
        // 断言结果符合预设值，同时可以验证后续业务逻辑
        XCTAssertEqual(modalResult, .alertFirstButtonReturn)
    }
}

选择哪种方案取决于你的测试需求：如果要覆盖完整的UI交互流程，选前两种；如果只是验证业务逻辑对弹窗结果的处理，第三种会更高效。

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