134 lines
3.9 KiB
C#
134 lines
3.9 KiB
C#
using ClaudeDo.Data;
|
|
using ClaudeDo.Ui.Services;
|
|
using ClaudeDo.Ui.ViewModels;
|
|
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);
|
|
}
|
|
}
|