如何为UIAlertAction设置accessibilityIdentifier？解决XCUITesting定位难题

很遗憾，UIAlertAction本身并没有直接暴露accessibilityIdentifier的公开API——这是UIKit框架设计的一个小局限。不过我们有两种正规、可维护的方案来解决这个问题，完全不需要依赖本地化字符串或者繁琐的hack逻辑：

方案一：利用KVC直接设置（简洁快速） #

UIAlertAction的底层按钮实际上支持accessibilityIdentifier属性，我们可以通过KVC的方式直接给action设置这个标识：

// 在业务代码中创建UIAlertAction时
let okAction = UIAlertAction(title: NSLocalizedString("OK", comment: ""), style: .default) { _ in
    // 你的动作逻辑
}
// 设置accessibilityIdentifier
okAction.setValue("alert_ok_button", forKey: "accessibilityIdentifier")

// 添加到alertController中
let alertController = UIAlertController(title: "...", message: "...", preferredStyle: .alert)
alertController.addAction(okAction)

在XCUITest中，你就可以直接通过标识定位按钮：

// 测试代码中定位并点击
app.alerts.buttons.matching(identifier: "alert_ok_button").firstMatch.tap()

注意：这个方案依赖UIAlertAction的私有键accessibilityIdentifier，虽然目前iOS版本都兼容，但如果苹果未来修改底层实现，可能会失效。如果追求极致稳定，可以选择方案二。

方案二：关联对象+子视图遍历（稳定兼容） #

我们可以给UIAlertAction添加一个自定义的accessibilityIdentifier属性，然后在UIAlertController显示时，自动把这个标识同步到对应的按钮上：

第一步：给UIAlertAction添加关联属性 #

import UIKit

private var ActionAccessibilityIdentifierKey = "ActionAccessibilityIdentifierKey"

extension UIAlertAction {
    var accessibilityIdentifier: String? {
        get {
            return objc_getAssociatedObject(self, &ActionAccessibilityIdentifierKey) as? String
        }
        set {
            objc_setAssociatedObject(self, &ActionAccessibilityIdentifierKey, newValue, .OBJC_ASSOCIATION_RETAIN_NONATOMIC)
        }
    }
}

第二步：给UIAlertController添加自动同步逻辑 #

extension UIAlertController {
    open override func viewWillAppear(_ animated: Bool) {
        super.viewWillAppear(animated)
        
        // 遍历所有子视图，找到alert的按钮
        for subview in view.allSubviews where subview is UIButton {
            guard let button = subview as? UIButton else { continue }
            // 根据按钮文本匹配对应的UIAlertAction，同步accessibilityIdentifier
            if let action = actions.first(where: { $0.title == button.titleLabel?.text }) {
                button.accessibilityIdentifier = action.accessibilityIdentifier
            }
        }
    }
}

// 辅助扩展：遍历视图的所有子视图
extension UIView {
    var allSubviews: [UIView] {
        return subviews + subviews.flatMap { $0.allSubviews }
    }
}

第三步：使用方式 #

// 业务代码中创建action并设置标识
let cancelAction = UIAlertAction(title: NSLocalizedString("Cancel", comment: ""), style: .cancel) { _ in }
cancelAction.accessibilityIdentifier = "alert_cancel_button"

let alertController = UIAlertController(title: "...", message: "...", preferredStyle: .alert)
alertController.addAction(cancelAction)

测试代码中的定位方式和方案一完全一致：

app.alerts.buttons["alert_cancel_button"].tap()

对比你提到的其他方案 #

• 本地化字符串方案：虽然能解决问题，但需要维护测试bundle中的本地化文件，一旦业务端的字符串变更，测试代码必须同步更新，多语言测试场景下还要切换bundle，维护成本较高。
• 繁琐的hack方案：依赖按钮顺序、索引等不稳定因素，一旦alert的按钮顺序调整，测试就会直接失败，代码冗余且稳定性差。

这两种正规方案都能让你脱离对本地化文本的依赖，用稳定的标识来定位UIAlertAction对应的按钮，更适合长期维护的项目。

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