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() .UseSqlite($"Data Source={_dbPath}") .Options; return new ClaudeDoDbContext(opts); } private sealed class TestDbFactory : IDbContextFactory { private readonly Func _create; public TestDbFactory(Func 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 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()); } }