YMhut-box-C-/参考/wpfui-main/tests/Wpf.Ui.Gallery.IntegrationTests/ContentDialogAutomationTests.cs

// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file, You can obtain one at https://opensource.org/licenses/MIT.
// Copyright (C) Leszek Pomianowski and WPF UI Contributors.
// All Rights Reserved.

using FlaUI.Core.Definitions;
using FlaUI.Core.Input;
using FlaUI.Core.WindowsAPI;

#pragma warning disable IDE0008 // Use explicit type instead of 'var'
#pragma warning disable SA1512  // Single-line comments should not be followed by blank line

namespace Wpf.Ui.Gallery.IntegrationTests;

public sealed class ContentDialogAutomationTests : UiTest
{
    [Fact]
    public async Task ContentDialog_Should_Return_Correct_Text()
    {
        // Give the test app a moment to display the window before starting UI automation interactions
        await Wait(2, TestContext.Current.CancellationToken);

        // Navigate to the ContentDialog page explicitly: click parent then child nav items
        var parentNav = FindFirst(c => c.ByText("Dialogs & flyouts"));
        parentNav.Should().NotBeNull("because the Dialogs & flyouts navigation item should be present");
        parentNav.Click();

        await Wait(1, TestContext.Current.CancellationToken);

        var childNav = FindFirst(c => c.ByText("ContentDialog"));
        if (childNav == null)
        {
            // If the child item is not immediately visible, try toggling the parent to expand it and retry
            parentNav.Click();
            await Wait(1, TestContext.Current.CancellationToken);
            childNav = FindFirst(c => c.ByText("ContentDialog"));
        }

        childNav.Should().NotBeNull("because the ContentDialog navigation item should be present as a child");
        childNav.Click();

        await Wait(1, TestContext.Current.CancellationToken);

        var showButton = FindFirst(c => c.ByText("Show"));
        showButton
            .Should()
            .NotBeNull("because the ContentDialog page must contain a Show button to open the dialog");
        showButton.AsButton().Click();

        await Wait(1, TestContext.Current.CancellationToken);

        // Now exercise each dialog button (Primary, Secondary, Close) and verify
        // the page TextBlock updates according to ContentDialogViewModel.

        // Primary -> "Save" -> expect "User saved their work"
        await OpenDialog();
        await ClickButtonMatching(["Save"]);
        await WaitForText("User saved their work");

        // Secondary -> "Don't Save" -> expect "User did not save their work"
        await OpenDialog();
        await ClickButtonMatching(["Don't Save", "Do not Save", "Dont Save"]);
        await WaitForText("User did not save their work");

        // Close/Cancel -> "Cancel" -> expect "User cancelled the dialog"
        await OpenDialog();
        await ClickButtonMatching(["Cancel", "Close"]);
        await WaitForText("User cancelled the dialog");
    }

    [Fact]
    public async Task ContentDialog_CtrlF_DoesNotFocus_NavigationAutoSuggestBox()
    {
        await Wait(2, TestContext.Current.CancellationToken);

        // Open ContentDialog page and show dialog
        var parentNav = FindFirst(c => c.ByText("Dialogs & flyouts"));
        parentNav.Should().NotBeNull();
        parentNav.Click();
        await Wait(1, TestContext.Current.CancellationToken);

        var childNav = FindFirst(c => c.ByText("ContentDialog"));
        if (childNav == null)
        {
            parentNav.Click();
            await Wait(1, TestContext.Current.CancellationToken);
            childNav = FindFirst(c => c.ByText("ContentDialog"));
        }

        childNav.Should().NotBeNull();
        childNav.Click();

        await Wait(1, TestContext.Current.CancellationToken);

        var showButton = FindFirst(c => c.ByText("Show"));
        showButton.Should().NotBeNull();
        showButton.AsButton().Click();

        await Wait(1, TestContext.Current.CancellationToken);

        // Send Ctrl+F and ensure background autosuggest does not get focused
        Keyboard.Press(VirtualKeyShort.CONTROL);
        Keyboard.Type(VirtualKeyShort.KEY_F);
        Keyboard.Release(VirtualKeyShort.CONTROL);
        global::FlaUI.Core.Input.Wait.UntilInputIsProcessed();

        // "Find the element with keyboard focus within the main window."
        var focusedElement = MainWindow
            ?.FindAllDescendants()
            .FirstOrDefault(e => e.Properties.HasKeyboardFocus.ValueOrDefault);

        // Assert that a focused element is found.
        focusedElement.Should().NotBeNull("there should be a focused element after sending keys");

        // Get and assert that the AutomationId is not the background autosuggest's id.
        var focusedAutomationId = focusedElement.Properties.AutomationId.ValueOrDefault as string;
        focusedAutomationId
            .Should()
            .NotBe(
                "NavigationAutoSuggestBox",
                "because Ctrl+F should not focus the background autosuggest while the dialog is open"
            );
    }

    private Task OpenDialog()
    {
        // ensure dialog opened by clicking Show
        var showButton = FindFirst(c => c.ByText("Show"));
        if (showButton == null)
        {
            // If Show button is not found, assume dialog is already open; give UI a moment to stabilize.
            return Wait(1, TestContext.Current.CancellationToken);
        }

        showButton.AsButton().Click();
        return Wait(1, TestContext.Current.CancellationToken);
    }

    /// <summary>
    /// Polls the UI until the specified text appears or a retry limit is reached.
    /// Useful for waiting for view-model-driven UI updates after dialog interactions.
    /// </summary>
    /// <param name="text">The text to wait for.</param>
    /// <param name="retries">Number of polling attempts.</param>
    /// <param name="delaySeconds">Delay in seconds between attempts (uses test cancellation token).</param>
    /// <returns>A task that completes when the text is found or throws an assertion if not found.</returns>
    private async Task WaitForText(string text, int retries = 10, int delaySeconds = 1)
    {
        for (var i = 0; i < retries; i++)
        {
            if (FindFirst(c => c.ByText(text)) != null)
            {
                return;
            }

            await Wait(delaySeconds, TestContext.Current.CancellationToken);
        }

        // Final assertion to fail test with clear message if text never appeared
        FindFirst(c => c.ByText(text))
            .Should()
            .NotBeNull($"Expected text '{text}' to appear within timeout");
    }

    /// <summary>
    /// Finds and clicks a button matching one of the provided candidate texts.
    /// If no direct match is found, falls back to clicking the last available button in the window.
    /// </summary>
    /// <param name="candidates">Array of acceptable button texts (first match is used).</param>
    /// <returns>A task that waits a short time after clicking to allow UI to settle.</returns>
    private Task ClickButtonMatching(string[] candidates)
    {
        AutomationElement? btn = null;

        foreach (var txt in candidates)
        {
            btn = FindFirst(c => c.ByText(txt));
            if (btn != null)
            {
                break;
            }
        }

        if (btn == null)
        {
            var buttons = MainWindow?.FindAllDescendants(cf => cf.ByControlType(ControlType.Button));
            if (buttons is { Length: > 0 })
            {
                btn = buttons.Last();
            }
        }

        btn.Should().NotBeNull($"expected one of: {string.Join(',', candidates)}");
        btn.AsButton().Click();

        return Wait(1, TestContext.Current.CancellationToken);
    }
}

#pragma warning restore IDE0008 // Use explicit type instead of 'var'
#pragma warning restore SA1512  // Single-line comments should not be followed by blank line