ClaudeDo/tests/ClaudeDo.Ui.Tests/ViewModels/MissionControlViewModelTests.cs

using System.Linq;
using ClaudeDo.Data;
using ClaudeDo.Ui.Services;
using ClaudeDo.Ui.ViewModels;
using ClaudeDo.Ui.ViewModels.Islands;
using Microsoft.EntityFrameworkCore;
using Xunit;

namespace ClaudeDo.Ui.Tests.ViewModels;

public class MissionControlViewModelTests : IDisposable
{
    private readonly string _dbPath;

    public MissionControlViewModelTests()
    {
        _dbPath = Path.Combine(Path.GetTempPath(), $"claudedo_mc_test_{Guid.NewGuid():N}.db");
        using var ctx = NewContext();
        ctx.Database.EnsureCreated();
    }

    public void Dispose()
    {
        try { File.Delete(_dbPath); } catch { }
        try { File.Delete(_dbPath + "-wal"); } catch { }
        try { File.Delete(_dbPath + "-shm"); } catch { }
    }

    private ClaudeDoDbContext NewContext()
    {
        var opts = new DbContextOptionsBuilder<ClaudeDoDbContext>()
            .UseSqlite($"Data Source={_dbPath}")
            .Options;
        return new ClaudeDoDbContext(opts);
    }

    private sealed class TestDbFactory : IDbContextFactory<ClaudeDoDbContext>
    {
        private readonly Func<ClaudeDoDbContext> _create;
        public TestDbFactory(Func<ClaudeDoDbContext> create) => _create = create;
        public ClaudeDoDbContext CreateDbContext() => _create();
    }

    private sealed class FakeWorker : StubWorkerClient { }

    private MissionControlViewModel BuildVm(StubWorkerClient worker)
        => new MissionControlViewModel(new TestDbFactory(NewContext), worker);

    [Fact]
    public void TwoStarts_CreateTwoMonitors_ColumnCountTwo()
    {
        var worker = new FakeWorker();
        using var vm = BuildVm(worker);

        worker.RaiseTaskStarted("slot-1", "t1", DateTime.UtcNow);
        worker.RaiseTaskStarted("slot-2", "t2", DateTime.UtcNow);

        Assert.Equal(2, vm.Monitors.Count);
        Assert.Equal(2, vm.ColumnCount);
    }

    [Fact]
    public void DuplicateStart_DoesNotAddSecondMonitor()
    {
        var worker = new FakeWorker();
        using var vm = BuildVm(worker);

        worker.RaiseTaskStarted("slot-1", "t1", DateTime.UtcNow);
        worker.RaiseTaskStarted("slot-1", "t1", DateTime.UtcNow);

        Assert.Equal(1, vm.Monitors.Count);
    }

    [Fact]
    public void Finish_KeepsPane_AndFlipsState()
    {
        var worker = new FakeWorker();
        using var vm = BuildVm(worker);

        worker.RaiseTaskStarted("slot-1", "t1", DateTime.UtcNow);
        worker.RaiseTaskFinished("slot-1", "t1", "done", DateTime.UtcNow);

        Assert.Equal(1, vm.Monitors.Count);
        Assert.True(vm.Monitors[0].IsDone);
    }

    [Fact]
    public void ClearFinished_RemovesTerminalMonitors()
    {
        var worker = new FakeWorker();
        using var vm = BuildVm(worker);

        worker.RaiseTaskStarted("slot-1", "t1", DateTime.UtcNow);
        worker.RaiseTaskStarted("slot-2", "t2", DateTime.UtcNow);
        worker.RaiseTaskFinished("slot-1", "t1", "done", DateTime.UtcNow);

        vm.ClearFinishedCommand.Execute(null);

        Assert.Equal(1, vm.Monitors.Count);
        Assert.Equal("t2", vm.Monitors[0].SubscribedTaskId);
        Assert.Equal(1, vm.ColumnCount);
    }

    [Fact]
    public void SeedsFromActiveTasksOnConstruction()
    {
        var worker = new SeededFakeWorker();
        using var vm = BuildVm(worker);

        Assert.Equal(1, vm.Monitors.Count);
        Assert.Equal("seed1", vm.Monitors[0].SubscribedTaskId);
    }

    private sealed class SeededFakeWorker : StubWorkerClient
    {
        public override IReadOnlyList<ActiveTask> GetActiveTasks()
            => new[] { new ActiveTask("slot-1", "seed1", DateTime.UtcNow) };
    }

    [Fact]
    public void OpenInApp_PropagatesToMonitors_AndCommandInvokesHook()
    {
        var worker = new FakeWorker();
        using var vm = BuildVm(worker);

        string? revealed = null;
        vm.OpenInApp = id => revealed = id;

        worker.RaiseTaskStarted("slot-1", "t1", DateTime.UtcNow);

        vm.Monitors[0].OpenInAppCommand.Execute(null);

        Assert.Equal("t1", revealed);
    }

    [Fact]
    public void Detach_RemovesFromGrid_ThenReDockRestores()
    {
        var worker = new FakeWorker();
        using var vm = BuildVm(worker);
        TaskMonitorViewModel? detached = null;
        Action? reDock = null;
        vm.ShowDetached = (m, rd) => { detached = m; reDock = rd; };

        worker.RaiseTaskStarted("slot-1", "t1", DateTime.UtcNow);
        var monitor = vm.Monitors[0];

        monitor.DetachCommand.Execute(null);
        Assert.Empty(vm.Monitors);
        Assert.Same(monitor, detached);

        reDock!.Invoke();
        Assert.Single(vm.Monitors);
        Assert.Same(monitor, vm.Monitors[0]);
    }

    [Fact]
    public void ClearFinished_AlsoRemoves_WaitingForReview()
    {
        var worker = new FakeWorker();
        using var vm = BuildVm(worker);
        worker.RaiseTaskStarted("slot-1", "t1", DateTime.UtcNow);
        worker.RaiseTaskFinished("slot-1", "t1", "waiting_for_review", DateTime.UtcNow);

        Assert.True(vm.Monitors[0].IsWaitingForReview);
        vm.ClearFinishedCommand.Execute(null);
        Assert.Empty(vm.Monitors);
    }

    [Fact]
    public void Detach_SetsIsDetached_AndReDockClearsIt()
    {
        var worker = new FakeWorker();
        using var vm = BuildVm(worker);
        Action? reDock = null;
        vm.ShowDetached = (m, rd) => reDock = rd;
        worker.RaiseTaskStarted("slot-1", "t1", DateTime.UtcNow);
        var monitor = vm.Monitors[0];

        monitor.DetachCommand.Execute(null);
        Assert.True(monitor.IsDetached);
        Assert.Empty(vm.Monitors);

        reDock!.Invoke();
        Assert.False(monitor.IsDetached);
        Assert.Single(vm.Monitors);
    }

    [Fact]
    public void DetachCommand_WhenDetached_RequestsWindowClose()
    {
        var worker = new FakeWorker();
        using var vm = BuildVm(worker);
        vm.ShowDetached = (m, rd) => { };
        worker.RaiseTaskStarted("slot-1", "t1", DateTime.UtcNow);
        var monitor = vm.Monitors[0];

        var closeRequested = false;
        monitor.DetachCommand.Execute(null);              // detach (IsDetached = true)
        monitor.CloseWindowRequested = () => closeRequested = true;
        monitor.DetachCommand.Execute(null);              // now acts as re-dock
        Assert.True(closeRequested);
    }

    [Fact]
    public void MoveMonitor_ReordersCollection()
    {
        var worker = new FakeWorker();
        using var vm = BuildVm(worker);
        worker.RaiseTaskStarted("s1", "t1", DateTime.UtcNow);
        worker.RaiseTaskStarted("s2", "t2", DateTime.UtcNow);
        worker.RaiseTaskStarted("s3", "t3", DateTime.UtcNow);

        vm.MoveMonitor(vm.Monitors[0], vm.Monitors[2]);   // move t1 to t3's slot
        Assert.Equal(new[] { "t2", "t3", "t1" }, vm.Monitors.Select(m => m.SubscribedTaskId).ToArray());
    }
}