101 lines
3.9 KiB
C#
101 lines
3.9 KiB
C#
using ClaudeDo.Data;
|
|
using ClaudeDo.Data.Models;
|
|
using ClaudeDo.Data.Repositories;
|
|
using ClaudeDo.Worker.External;
|
|
using ClaudeDo.Worker.Hub;
|
|
using ClaudeDo.Worker.Tests.Infrastructure;
|
|
using Xunit;
|
|
|
|
namespace ClaudeDo.Worker.Tests.Hub;
|
|
|
|
/// BaseBranch und PowerUseAllowed sitzen in der SetConfigAsync-kopiert-verbatim-Falle: jeder Writer,
|
|
/// der sie nicht durchreicht, setzt sie still zurück.
|
|
public sealed class ListConfigBaseBranchTests : IDisposable
|
|
{
|
|
private readonly DbFixture _db = new();
|
|
public void Dispose() => _db.Dispose();
|
|
|
|
private WorkerHub CreateHub()
|
|
{
|
|
var factory = _db.CreateFactory();
|
|
var broadcaster = new HubBroadcaster(new CapturingHubContext());
|
|
var hub = new WorkerHub(
|
|
null!, null!, null!, null!, broadcaster, factory,
|
|
null!, null!, null!, null!, null!, null!, null!, null!, null!, null!, null!, null!, null!,
|
|
null!, new ClaudeDo.Worker.Online.OnlineInboxConfig(), new ClaudeDo.Worker.OnlineRefreshTokenStore(new ClaudeDo.Worker.DpapiTokenStore("unused.token")),
|
|
new ClaudeDo.Worker.Runner.PendingQuestionRegistry(), null!);
|
|
hub.Clients = new FakeHubCallerClients(new RecordingClientProxy());
|
|
hub.Context = new FakeHubCallerContext();
|
|
return hub;
|
|
}
|
|
|
|
private async Task<string> SeedListAsync(string? baseBranch, bool powerUse = false)
|
|
{
|
|
var listId = Guid.NewGuid().ToString();
|
|
await using var ctx = _db.CreateContext();
|
|
var repo = new ListRepository(ctx);
|
|
await repo.AddAsync(new ListEntity { Id = listId, Name = "L", CreatedAt = DateTime.UtcNow });
|
|
await repo.SetConfigAsync(new ListConfigEntity { ListId = listId, BaseBranch = baseBranch, PowerUseAllowed = powerUse });
|
|
return listId;
|
|
}
|
|
|
|
private async Task<ListConfigEntity?> ReadConfigAsync(string listId)
|
|
{
|
|
await using var ctx = _db.CreateContext();
|
|
return await new ListRepository(ctx).GetConfigAsync(listId);
|
|
}
|
|
|
|
private ConfigMcpTools CreateTool(ClaudeDoDbContext ctx) =>
|
|
new(new ListRepository(ctx), new TaskRepository(ctx),
|
|
new HubBroadcaster(new CapturingHubContext()), _db.CreateFactory());
|
|
|
|
[Fact]
|
|
public async Task set_list_config_preserves_BaseBranch_and_PowerUseAllowed()
|
|
{
|
|
var listId = await SeedListAsync("dev", powerUse: true);
|
|
await using var ctx = _db.CreateContext();
|
|
|
|
await CreateTool(ctx).SetListConfig(listId, model: "opus", cancellationToken: CancellationToken.None);
|
|
|
|
var cfg = await ReadConfigAsync(listId);
|
|
Assert.Equal("dev", cfg!.BaseBranch);
|
|
Assert.True(cfg.PowerUseAllowed);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task get_list_config_returns_BaseBranch()
|
|
{
|
|
var listId = await SeedListAsync("dev");
|
|
await using var ctx = _db.CreateContext();
|
|
|
|
var result = await CreateTool(ctx).GetListConfig(listId, CancellationToken.None);
|
|
|
|
Assert.Equal("dev", result.Config!.BaseBranch);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task UpdateListConfig_null_keeps_BaseBranch()
|
|
{
|
|
var listId = await SeedListAsync("dev");
|
|
await CreateHub().UpdateListConfig(new UpdateListConfigDto(listId, "opus", null, null));
|
|
Assert.Equal("dev", (await ReadConfigAsync(listId))!.BaseBranch);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task UpdateListConfig_empty_clears_BaseBranch()
|
|
{
|
|
var listId = await SeedListAsync("dev");
|
|
await CreateHub().UpdateListConfig(new UpdateListConfigDto(listId, "opus", null, null, BaseBranch: ""));
|
|
Assert.Null((await ReadConfigAsync(listId))!.BaseBranch);
|
|
}
|
|
|
|
[Fact]
|
|
public async Task UpdateListConfig_sets_BaseBranch_and_GetListConfig_returns_it()
|
|
{
|
|
var listId = await SeedListAsync(null);
|
|
var hub = CreateHub();
|
|
await hub.UpdateListConfig(new UpdateListConfigDto(listId, null, null, null, BaseBranch: "dev"));
|
|
Assert.Equal("dev", (await hub.GetListConfig(listId))!.BaseBranch);
|
|
}
|
|
}
|