WPF应用弹窗致AutomationElement阻塞，UI测试无法定位MessageBox按钮

问题根源 #

核心问题在于：InsertValue触发BeginningEdit事件后，MessageBox.Show作为模态对话框会直接阻塞UI线程，同时测试代码里的操作线程也会被卡住——因为Automation的InsertValue会等待UI操作完成，而模态框不关闭就无法完成。另外原测试中查找按钮的范围（TreeScope.Children）不正确，且主线程被阻塞后，WaitForCondition的执行也会被推迟。

解决方案 #

方案1：纯测试侧修复（无需修改应用代码） #

利用UI自动化的窗口打开事件监听，提前注册回调，一旦检测到目标MessageBox弹出，立即定位并点击OK按钮。这种方式能在模态框阻塞线程前就完成监听逻辑：

// 定义MessageBox处理回调
void OnMessageBoxOpened(object sender, WindowOpenedEventArgs e)
{
    var messageBoxWindow = e.Element;
    // 验证是否是目标提示框（可通过文本内容匹配）
    var textPattern = messageBoxWindow.GetCurrentPattern(TextPattern.Pattern) as TextPattern;
    if (textPattern != null && textPattern.DocumentRange.Text.Contains("Its not allowed to edit an existing id for shifts"))
    {
        // 查找MessageBox内的OK按钮（AutomationId为2）
        var okButton = messageBoxWindow.FindFirst(TreeScope.Subtree,
            new PropertyCondition(AutomationElement.AutomationIdProperty, "2"));
        okButton?.Invoke();
    }
}

// 注册窗口打开事件监听
Automation.AddAutomationEventHandler(WindowPattern.WindowOpenedEvent,
    AutomationElement.RootElement, TreeScope.Subtree, OnMessageBoxOpened);

try
{
    // 异步执行单元格编辑操作
    await Task.Run(() =>
    {
        var cell = mainWindow.FindDescendantById("ShiftsDataGrid_Row0_Column0");
        var parentCell = cell.GetParent();
        parentCell.InsertValue("0");
    });
}
finally
{
    // 注销事件，避免内存泄漏
    Automation.RemoveAutomationEventHandler(WindowPattern.WindowOpenedEvent,
        AutomationElement.RootElement, OnMessageBoxOpened);
}

方案2：修改应用代码（更彻底的适配） #

如果允许调整WPF应用的事件逻辑，把MessageBox.Show改成异步弹出，这样BeginningEdit事件不会阻塞UI线程，测试代码就能正常执行：

private void ShiftsDataGrid_OnBeginningEdit(object? sender, DataGridBeginningEditEventArgs e)
{
    if (e.Column == ShiftIdColumn && !e.Row.IsNewItem)
    {
        CancelColorPopup();
        e.Cancel = true;
        // 用Dispatcher异步弹出提示框，避免阻塞事件线程
        Dispatcher.BeginInvoke(new Action(() =>
        {
            MessageBox.Show("Its not allowed to edit an existing id for shifts");
        }));
    }
}

修改后，原测试代码只需调整查找按钮的范围（把TreeScope.Children改成TreeScope.Subtree）即可正常工作：

var task = Task.Run(() =>
{
    var cell = mainWindow.FindDescendantById("ShiftsDataGrid_Row0_Column0");
    var parentCell = cell.GetParent();
    parentCell.InsertValue("0");
});

WaitForCondition(() =>
{
    var okButton = AutomationElement.RootElement.FindFirst(TreeScope.Subtree,
        new PropertyCondition(AutomationElement.AutomationIdProperty, "2"));

    if (okButton == null)
        return false;

    okButton.Click();
    return true;
}, TimeSpan.FromSeconds(200));

关键说明 #

• 原测试中TreeScope.Children范围错误：MessageBox是RootElement的子窗口，而OK按钮是MessageBox的后代，必须用TreeScope.Subtree才能找到。
• 模态对话框的阻塞特性：同步弹出的MessageBox会卡住UI线程和Automation操作线程，必须通过异步监听或异步弹窗的方式绕过阻塞。

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