Terminal.Gui/Tests/IntegrationTests/FluentTests/TreeViewFluentTests.cs

using TerminalGuiFluentTesting;
using TerminalGuiFluentTestingXunit;
using Xunit.Abstractions;

namespace IntegrationTests.FluentTests;

public class TreeViewFluentTests
{
    private readonly TextWriter _out;

    public TreeViewFluentTests (ITestOutputHelper outputHelper) { _out = new TestOutputWriter (outputHelper); }

    [Theory]
    [ClassData (typeof (TestDrivers))]
    public void TreeView_AllowReOrdering (TestDriver d)
    {
        var tv = new TreeView
        {
            Width = Dim.Fill (),
            Height = Dim.Fill ()
        };

        TreeNode car;
        TreeNode lorry;
        TreeNode bike;

        var root = new TreeNode ("Root")
        {
            Children =
            [
                car = new ("Car"),
                lorry = new ("Lorry"),
                bike = new ("Bike")
            ]
        };
        tv.AddObject (root);

        using GuiTestContext context =
            With.A<Window> (40, 10, d, _out)
                .Add (tv)
                .Focus (tv)
                .WaitIteration ()
                .ScreenShot ("Before expanding", _out)
                .AssertEqual (root, tv.GetObjectOnRow (0))
                .AssertNull (tv.GetObjectOnRow (1))
                .EnqueueKeyEvent (Key.CursorRight)
                .ScreenShot ("After expanding", _out)
                .AssertMultiple (
                                 () =>
                                 {
                                     Assert.Equal (root, tv.GetObjectOnRow (0));
                                     Assert.Equal (car, tv.GetObjectOnRow (1));
                                     Assert.Equal (lorry, tv.GetObjectOnRow (2));
                                     Assert.Equal (bike, tv.GetObjectOnRow (3));
                                 })
                .AssertIsAssignableFrom <ITreeNode>(tv.SelectedObject)
                .Then (
                       () =>
                       {
                           // Re order
                           root.Children = [bike, car, lorry];
                           tv.RefreshObject (root);
                       })
                .WaitIteration ()
                .ScreenShot ("After re-order", _out)
                .AssertMultiple (
                                 () =>
                                 {
                                     Assert.Equal (root, tv.GetObjectOnRow (0));
                                     Assert.Equal (bike, tv.GetObjectOnRow (1));
                                     Assert.Equal (car, tv.GetObjectOnRow (2));
                                     Assert.Equal (lorry, tv.GetObjectOnRow (3));
                                 })
                .WriteOutLogs (_out);

        context.Stop ();
    }

    [Theory]
    [ClassData (typeof (TestDrivers))]
    public void TreeViewReOrder_PreservesExpansion (TestDriver d)
    {
        var tv = new TreeView
        {
            Width = Dim.Fill (),
            Height = Dim.Fill ()
        };

        TreeNode car;
        TreeNode lorry;
        TreeNode bike;

        TreeNode mrA;
        TreeNode mrB;

        TreeNode mrC;

        TreeNode mrD;
        TreeNode mrE;

        var root = new TreeNode ("Root")
        {
            Children =
            [
                car = new ("Car")
                {
                    Children =
                    [
                        mrA = new ("Mr A"),
                        mrB = new ("Mr B")
                    ]
                },
                lorry = new ("Lorry")
                {
                    Children =
                    [
                        mrC = new ("Mr C")
                    ]
                },
                bike = new ("Bike")
                {
                    Children =
                    [
                        mrD = new ("Mr D"),
                        mrE = new ("Mr E")
                    ]
                }
            ]
        };

        tv.AddObject (root);
        tv.ExpandAll ();

        using GuiTestContext context =
            With.A<Window> (40, 13, d)
                .Add (tv)
                .WaitIteration ()
                .ScreenShot ("Initial State", _out)
                .AssertMultiple (
                                 () =>
                                 {
                                     Assert.Equal (root, tv.GetObjectOnRow (0));
                                     Assert.Equal (car, tv.GetObjectOnRow (1));
                                     Assert.Equal (mrA, tv.GetObjectOnRow (2));
                                     Assert.Equal (mrB, tv.GetObjectOnRow (3));
                                     Assert.Equal (lorry, tv.GetObjectOnRow (4));
                                     Assert.Equal (mrC, tv.GetObjectOnRow (5));
                                     Assert.Equal (bike, tv.GetObjectOnRow (6));
                                     Assert.Equal (mrD, tv.GetObjectOnRow (7));
                                     Assert.Equal (mrE, tv.GetObjectOnRow (8));
                                 })
                .Then (
                       () =>
                       {
                           // Re order
                           root.Children = [bike, car, lorry];
                           tv.RefreshObject (root);
                       })
                .WaitIteration ()
                .ScreenShot ("After re-order", _out)
                .AssertMultiple (
                                 () =>
                                 {
                                     Assert.Equal (root, tv.GetObjectOnRow (0));
                                     Assert.Equal (bike, tv.GetObjectOnRow (1));
                                     Assert.Equal (mrD, tv.GetObjectOnRow (2));
                                     Assert.Equal (mrE, tv.GetObjectOnRow (3));
                                     Assert.Equal (car, tv.GetObjectOnRow (4));
                                     Assert.Equal (mrA, tv.GetObjectOnRow (5));
                                     Assert.Equal (mrB, tv.GetObjectOnRow (6));
                                     Assert.Equal (lorry, tv.GetObjectOnRow (7));
                                     Assert.Equal (mrC, tv.GetObjectOnRow (8));
                                 })
                .WriteOutLogs (_out);

        context.Stop ();
    }
}