AtomUI/tests/AtomUI.Desktop.Controls.Tests/Input/OtpLineEditCellCustomizationTests.cs

using Avalonia.Controls;
using Avalonia.Media;
using Avalonia.Threading;
using Avalonia.VisualTree;
using AvaloniaWindow = Avalonia.Controls.Window;
using Shouldly;
using Xunit;
using AtomUIOtpLineEdit = AtomUI.Desktop.Controls.OtpLineEdit;

namespace AtomUI.Desktop.Controls.Tests.Input;

public class OtpLineEditCellCustomizationTests
{
    static OtpLineEditCellCustomizationTests()
    {
        AvaloniaTestApp.EnsureInitialized();
    }

    [Fact]
    public void Cell_Width_And_BorderBrush_Propagate_To_Every_Cell()
    {
        var brush = new SolidColorBrush(Colors.CornflowerBlue);
        var otp = new AtomUIOtpLineEdit
        {
            Length = 6,
            Separator = "*",
            CellWidth = 32,
            CellBorderBrush = brush,
            IsMotionEnabled = false
        };

        var window = Show(otp);
        try
        {
            var cells = GetCells(otp);
            cells.Count.ShouldBe(6);
            cells.ShouldAllBe(cell => cell.Bounds.Width == 32);
            cells.ShouldAllBe(cell => ReferenceEquals(cell.BorderBrush, brush));
        }
        finally
        {
            window.Close();
        }
    }

    [Fact]
    public void Unset_Cell_Overrides_Fall_Back_To_Theme_Defaults()
    {
        var otp = new AtomUIOtpLineEdit
        {
            Length = 4,
            IsMotionEnabled = false
        };

        var window = Show(otp);
        try
        {
            var cells = GetCells(otp);
            cells.Count.ShouldBe(4);
            foreach (var cell in cells)
            {
                cell.Bounds.Width.ShouldBeGreaterThan(0);
                cell.BorderBrush.ShouldNotBeNull();
            }
        }
        finally
        {
            window.Close();
        }
    }

    [Fact]
    public void Separator_Renders_By_Default_Without_Setting_Interval()
    {
        var otp = new AtomUIOtpLineEdit
        {
            Length = 6,
            Separator = "*",
            IsMotionEnabled = false
        };

        var window = Show(otp);
        try
        {
            var separators = otp.GetVisualDescendants()
                                .OfType<ContentControl>()
                                .Where(control => control.Name == "PART_SeparatorPresenter")
                                .ToArray();
            separators.Length.ShouldBe(6);
            separators.Count(static separator => separator.IsEffectivelyVisible).ShouldBe(5);

            var cell = otp.GetVisualDescendants().OfType<OtpLineEditCell>().First();
            foreach (var separator in separators.Where(static separator => separator.IsEffectivelyVisible))
            {
                ((string?)separator.Content).ShouldBe("*");

                // 结构居中锁定：wrapper 与 cell 等高、Border 垂直居中对齐；
                // ContentControl 的 token 光学补偿是渲染变换，不参与布局
                var border = separator.Parent.ShouldBeOfType<Border>();
                border.VerticalAlignment.ShouldBe(Avalonia.Layout.VerticalAlignment.Center);
                var wrapper = border.Parent.ShouldBeOfType<StackPanel>();
                wrapper.Bounds.Height.ShouldBe(cell.Bounds.Height, 0.5,
                    "separator wrapper must share the cell height so centering holds");
            }
        }
        finally
        {
            window.Close();
        }
    }

    private static IReadOnlyList<OtpLineEditCell> GetCells(AtomUIOtpLineEdit owner)
    {
        return owner.GetVisualDescendants().OfType<OtpLineEditCell>().ToArray();
    }

    private static AvaloniaWindow Show(Control content)
    {
        var window = new AvaloniaWindow
        {
            Width = 480,
            Height = 120,
            Content = content
        };
        window.Show();
        content.ApplyTemplate();
        window.UpdateLayout();
        Dispatcher.UIThread.RunJobs();
        return window;
    }
}