Ursa.Avalonia/tests/HeadlessTest.Ursa/Controls/DateTimePicker/TimePickerPresenterTests.cs

using Avalonia.Controls;
using Avalonia.Controls.Primitives;
using Avalonia.Controls.Templates;
using Avalonia.Headless.XUnit;
using Avalonia.Threading;
using TimePickerPresenter = Ursa.Controls.TimePickerPresenter;

namespace HeadlessTest.Ursa.Controls.DateTimePicker;

public class TimePickerPresenterTests
{
    [Fact]
    public void TimePickerPresenter_DefaultValues_ShouldBeCorrect()
    {
        var presenter = new TimePickerPresenter();

        Assert.False(presenter.NeedsConfirmation);
        Assert.Equal(1, presenter.MinuteIncrement);
        Assert.Equal(1, presenter.SecondIncrement);
        Assert.Null(presenter.Time);
        Assert.Equal("HH mm ss t", presenter.PanelFormat);
    }

    [Fact]
    public void TimePickerPresenter_SetTime_ShouldUpdateTimeProperty()
    {
        var presenter = new TimePickerPresenter();
        var time = new TimeSpan(10, 30, 45);

        presenter.Time = time;

        Assert.Equal(time, presenter.Time);
    }

    [AvaloniaTheory]
    [InlineData("hh mm ss", 0, 2, 4, 0, true, true, true, false)]
    [InlineData("hh mm ss t", 0, 2, 4, 6, true, true, true, true)]
    [InlineData("hh mm", 0, 2, 0, 0, true, true, false, false)]
    [InlineData("mm ss", 0, 0, 2, 0, false, true, true, false)]
    [InlineData("ss", 0, 0, 0, 0, false, false, true, false)]
    [InlineData("t", 0, 0, 0, 0, false, false, false, true)]
    public void TimePickerPresenter_SetPanelFormat_ShouldUpdatePanelLayout(
        string format, int hourColumn, int minuteColumn, int secondColumn, int amColumn,
        bool isHourPanelVisible, bool isMinutePanelVisible, bool isSecondPanelVisible, bool isAmPanelVisible
        )
    {
        var window = new Window();
        var presenter = new TimePickerPresenter();
        window.Content = presenter;
        window.Show();
        Dispatcher.UIThread.RunJobs();
        presenter.PanelFormat = format;
        Dispatcher.UIThread.RunJobs();
        
        Assert.Equal(format, presenter.PanelFormat);
        
        var hourPanel = presenter.GetTemplateChildren().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_HourScrollPanel);
        var minutePanel = presenter.GetTemplateChildren().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_MinuteScrollPanel);
        var secondPanel = presenter.GetTemplateChildren().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_SecondScrollPanel);
        var amPanel = presenter.GetTemplateChildren().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_AmPmScrollPanel);

        Assert.NotNull(hourPanel);
        Assert.NotNull(minutePanel);
        Assert.NotNull(secondPanel);
        Assert.NotNull(amPanel);
        
        Assert.Equal(hourColumn, Grid.GetColumn(hourPanel));
        Assert.Equal(minuteColumn, Grid.GetColumn(minutePanel));
        Assert.Equal(secondColumn, Grid.GetColumn(secondPanel));
        Assert.Equal(amColumn, Grid.GetColumn(amPanel));
        
        Assert.Equal(isHourPanelVisible, hourPanel.IsVisible);
        Assert.Equal(isMinutePanelVisible, minutePanel.IsVisible);
        Assert.Equal(isSecondPanelVisible, secondPanel.IsVisible);
        Assert.Equal(isAmPanelVisible, amPanel.IsVisible);
    }
    
    

    [Fact]
    public void TimePickerPresenter_Confirm_ShouldSetTimeProperty()
    {
        var presenter = new TimePickerPresenter { NeedsConfirmation = true };
        var time = new TimeSpan(10, 30, 45);
        presenter.TimeHolder = time;
        Assert.NotEqual(time, presenter.Time);

        presenter.Confirm();

        Assert.Equal(time, presenter.Time);
    }

    [AvaloniaFact]
    public void TimePickerPresenter_OnTimeChanged_ShouldRaiseEvent()
    {
        var presenter = new TimePickerPresenter();
        var oldTime = new TimeSpan(10, 30, 45);
        var newTime = new TimeSpan(11, 45, 30);
        presenter.Time = oldTime;
        var eventRaised = false;
        presenter.SelectedTimeChanged += (sender, args) =>
        {
            eventRaised = true;
            Assert.Equal(oldTime, args.OldTime);
            Assert.Equal(newTime, args.NewTime);
        };

        presenter.Time = newTime;

        Assert.True(eventRaised);
    }

    [AvaloniaTheory]
    [MemberData(nameof(GetSelectionMemberData))]
    public void TimePickerPresenter_Time_Updated_When_Panel_Selection_Changed(string format, int hourSelection, int minuteSelection, int secondSelection, int amSelection, TimeSpan expectedTime)
    {
        var window = new Window();
        var presenter = new TimePickerPresenter();
        window.Content = presenter;
        window.Show();
        Dispatcher.UIThread.RunJobs();
        presenter.PanelFormat = format;
        Dispatcher.UIThread.RunJobs();
        
        var hourPanel = presenter.GetTemplateChildren().OfType<DateTimePickerPanel>().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_HourSelector);
        var minutePanel = presenter.GetTemplateChildren().OfType<DateTimePickerPanel>().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_MinuteSelector);
        var secondPanel = presenter.GetTemplateChildren().OfType<DateTimePickerPanel>().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_SecondSelector);
        var amPanel = presenter.GetTemplateChildren().OfType<DateTimePickerPanel>().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_AmPmSelector);
        
        Assert.NotNull(hourPanel);
        Assert.NotNull(minutePanel);
        Assert.NotNull(secondPanel);
        Assert.NotNull(amPanel);
        
        hourPanel.SelectedValue = hourSelection;
        minutePanel.SelectedValue = minuteSelection;
        secondPanel.SelectedValue = secondSelection;
        amPanel.SelectedValue = amSelection;
        
        Assert.Equal(expectedTime, presenter.Time);        
    }
    
    [AvaloniaTheory]
    [MemberData(nameof(GetSelectionMemberData))]
    public void TimePickerPresenter_TimeHolder_Updated_When_Panel_Selection_Changed(string format, int hourSelection, int minuteSelection, int secondSelection, int amSelection, TimeSpan expectedTime)
    {
        var window = new Window();
        var presenter = new TimePickerPresenter(){NeedsConfirmation = true};
        window.Content = presenter;
        window.Show();
        Dispatcher.UIThread.RunJobs();
        presenter.PanelFormat = format;
        Dispatcher.UIThread.RunJobs();
        
        var hourPanel = presenter.GetTemplateChildren().OfType<DateTimePickerPanel>().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_HourSelector);
        var minutePanel = presenter.GetTemplateChildren().OfType<DateTimePickerPanel>().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_MinuteSelector);
        var secondPanel = presenter.GetTemplateChildren().OfType<DateTimePickerPanel>().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_SecondSelector);
        var amPanel = presenter.GetTemplateChildren().OfType<DateTimePickerPanel>().FirstOrDefault(a => a.Name == TimePickerPresenter.PART_AmPmSelector);
        
        Assert.NotNull(hourPanel);
        Assert.NotNull(minutePanel);
        Assert.NotNull(secondPanel);
        Assert.NotNull(amPanel);
        
        hourPanel.SelectedValue = hourSelection;
        minutePanel.SelectedValue = minuteSelection;
        secondPanel.SelectedValue = secondSelection;
        amPanel.SelectedValue = amSelection;
        
        Assert.Equal(expectedTime, presenter.TimeHolder);        
    }

    public static IEnumerable<object[]> GetSelectionMemberData()
    {
        yield return new object[] { "hh mm ss t", 1, 1, 1, 1, new TimeSpan(0, 13, 1, 1) };
        yield return new object[] { "HH mm ss t", 12, 30, 45, 0, new TimeSpan(0, 12,30,45)};
        yield return new object[] { "HH mm ss t", 12, 0, 0, 1, new TimeSpan(0, 12,0,0)};
        yield return new object[] { "HH mm ss t", 13, 0, 0, 1, new TimeSpan(0, 13,0,0)};
        yield return new object[] { "hh mm ss t", 9, 0, 0, 0, new TimeSpan(0, 9,0,0)};
        yield return new object[] { "hh mm ss t", 9, 0, 0, 1, new TimeSpan(0, 21,0,0)};
        yield return new object[] { "HH mm ss t", 0, 0, 0, 0, new TimeSpan(0, 0,0,0)};
    }
}