feat(planning): consolidate finalize+chain via TaskStateService, fix queue pickup

Slice 4 of the worker state consolidation refactor. Eliminates the
"queue never picks up planning tasks" bug structurally by routing both
the manager and MCP finalize paths through TaskStateService and
PlanningChainCoordinator.SetupChainAsync, where the auto-wake on enqueue
guarantees the queue picker claims the first child immediately.

- Delete TaskRepository.FinalizePlanningAsync; PlanningSessionManager
  now orchestrates via _state.FinalizePlanningAsync + _chain.SetupChainAsync.
- Rename QueueSubtasksSequentiallyAsync to SetupChainAsync (internal);
  layout is now Status=Queued + BlockedByTaskId, with auto-attached agent tag.
- OnChildFinishedAsync looks up the successor by BlockedByTaskId, drops
  the legacy Waiting status lookup.
- PlanningMcpService.Finalize routes through state+chain; EditableStatuses
  drops Waiting and adds Idle; gate uses PlanningPhase==Active.
- TaskStateService.FinalizePlanningAsync clears the planning session token.
- UI: TaskRowViewModel adds BlockedByTaskId; IsQueued/IsWaiting reflect
  the new layout; TasksIslandViewModel.RemoveFromQueueAsync clears
  BlockedByTaskId on dequeue.
- New regression test PlanningEndToEndTests.FinalizeAsync_FirstChildIs
  ClaimedByPicker_WithinDeadline asserts the picker claims the first
  child within 200ms with no manual WakeQueue.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Mika Kuns
2026-04-27 14:16:12 +02:00
parent 064a903076
commit 4ab906ff0b
17 changed files with 315 additions and 206 deletions

View File

@@ -358,6 +358,7 @@ public sealed class TaskRepository
.Where(t => t.Id == taskId && t.Status == TaskStatus.Manual) .Where(t => t.Id == taskId && t.Status == TaskStatus.Manual)
.ExecuteUpdateAsync(s => s .ExecuteUpdateAsync(s => s
.SetProperty(t => t.Status, TaskStatus.Planning) .SetProperty(t => t.Status, TaskStatus.Planning)
.SetProperty(t => t.PlanningPhase, PlanningPhase.Active)
.SetProperty(t => t.PlanningSessionToken, sessionToken), ct); .SetProperty(t => t.PlanningSessionToken, sessionToken), ct);
if (affected == 0) return null; if (affected == 0) return null;
@@ -396,49 +397,6 @@ public sealed class TaskRepository
.FirstOrDefaultAsync(t => t.PlanningSessionToken == token, ct); .FirstOrDefaultAsync(t => t.PlanningSessionToken == token, ct);
} }
public async Task<int> FinalizePlanningAsync(
string parentId,
bool queueAgentTasks,
CancellationToken ct = default)
{
using var tx = await _context.Database.BeginTransactionAsync(ct);
var parent = await _context.Tasks
.AsNoTracking()
.Include(t => t.List).ThenInclude(l => l.Tags)
.FirstOrDefaultAsync(t => t.Id == parentId, ct);
if (parent is null || parent.Status != TaskStatus.Planning)
throw new InvalidOperationException($"Task {parentId} is not in Planning state.");
var listHasAgentTag = parent.List.Tags.Any(t => t.Name == "agent");
var drafts = await _context.Tasks
.Include(t => t.Tags)
.Where(t => t.ParentTaskId == parentId && t.Status == TaskStatus.Draft)
.ToListAsync(ct);
int count = 0;
foreach (var draft in drafts)
{
var childHasAgentTag = draft.Tags.Any(t => t.Name == "agent");
var shouldQueue = queueAgentTasks && (childHasAgentTag || listHasAgentTag);
draft.Status = shouldQueue ? TaskStatus.Queued : TaskStatus.Manual;
count++;
}
var finalizedAt = DateTime.UtcNow;
await _context.Tasks
.Where(t => t.Id == parentId)
.ExecuteUpdateAsync(s => s
.SetProperty(t => t.Status, TaskStatus.Planned)
.SetProperty(t => t.PlanningFinalizedAt, finalizedAt)
.SetProperty(t => t.PlanningSessionToken, (string?)null), ct);
await _context.SaveChangesAsync(ct);
await tx.CommitAsync(ct);
return count;
}
public async Task<bool> DiscardPlanningAsync( public async Task<bool> DiscardPlanningAsync(
string parentId, string parentId,
CancellationToken ct = default) CancellationToken ct = default)
@@ -462,6 +420,7 @@ public sealed class TaskRepository
.Where(t => t.Id == parentId) .Where(t => t.Id == parentId)
.ExecuteUpdateAsync(s => s .ExecuteUpdateAsync(s => s
.SetProperty(t => t.Status, TaskStatus.Manual) .SetProperty(t => t.Status, TaskStatus.Manual)
.SetProperty(t => t.PlanningPhase, PlanningPhase.None)
.SetProperty(t => t.PlanningSessionId, (string?)null) .SetProperty(t => t.PlanningSessionId, (string?)null)
.SetProperty(t => t.PlanningSessionToken, (string?)null) .SetProperty(t => t.PlanningSessionToken, (string?)null)
.SetProperty(t => t.PlanningFinalizedAt, (DateTime?)null), ct); .SetProperty(t => t.PlanningFinalizedAt, (DateTime?)null), ct);

View File

@@ -24,6 +24,7 @@ public sealed partial class TaskRowViewModel : ViewModelBase
[ObservableProperty] private bool _dropHintAbove; [ObservableProperty] private bool _dropHintAbove;
[ObservableProperty] private bool _dropHintBelow; [ObservableProperty] private bool _dropHintBelow;
[ObservableProperty] private string? _parentTaskId; [ObservableProperty] private string? _parentTaskId;
[ObservableProperty] private string? _blockedByTaskId;
[ObservableProperty] private bool _isExpanded = true; [ObservableProperty] private bool _isExpanded = true;
[ObservableProperty] private bool _hasPlanningChildren; [ObservableProperty] private bool _hasPlanningChildren;
[ObservableProperty] private bool _hasQueuedSubtasks; [ObservableProperty] private bool _hasQueuedSubtasks;
@@ -57,8 +58,9 @@ public sealed partial class TaskRowViewModel : ViewModelBase
public bool HasSteps => StepsCount > 0; public bool HasSteps => StepsCount > 0;
public bool IsOverdue => ScheduledFor is { } d && d.Date < DateTime.Today && !Done; public bool IsOverdue => ScheduledFor is { } d && d.Date < DateTime.Today && !Done;
public bool IsRunning => Status == TaskStatus.Running; public bool IsRunning => Status == TaskStatus.Running;
public bool IsQueued => Status == TaskStatus.Queued; public bool IsQueued => Status == TaskStatus.Queued && string.IsNullOrEmpty(BlockedByTaskId);
public bool IsWaiting => Status == TaskStatus.Waiting; public bool IsWaiting => (Status == TaskStatus.Queued && !string.IsNullOrEmpty(BlockedByTaskId))
|| Status == TaskStatus.Waiting;
public bool CanRemoveFromQueue => IsQueued || HasQueuedSubtasks; public bool CanRemoveFromQueue => IsQueued || HasQueuedSubtasks;
public bool HasSchedule => ScheduledFor.HasValue; public bool HasSchedule => ScheduledFor.HasValue;
public bool HasLiveTail => IsRunning && !string.IsNullOrEmpty(LiveTail); public bool HasLiveTail => IsRunning && !string.IsNullOrEmpty(LiveTail);
@@ -67,14 +69,15 @@ public sealed partial class TaskRowViewModel : ViewModelBase
public string DiffDeletionsText => $"{DiffDeletions}"; public string DiffDeletionsText => $"{DiffDeletions}";
public string StepsText => $"{StepsCompleted}/{StepsCount} steps"; public string StepsText => $"{StepsCompleted}/{StepsCount} steps";
public string StatusChipClass => Status switch public string StatusChipClass => (Status, IsBlocked: !string.IsNullOrEmpty(BlockedByTaskId)) switch
{ {
TaskStatus.Running => "running", (TaskStatus.Running, _) => "running",
TaskStatus.Failed => "error", (TaskStatus.Failed, _) => "error",
TaskStatus.Done => "review", (TaskStatus.Done, _) => "review",
TaskStatus.Queued => "queued", (TaskStatus.Queued, true) => "waiting",
TaskStatus.Waiting => "waiting", (TaskStatus.Queued, false) => "queued",
_ => "idle", (TaskStatus.Waiting, _) => "waiting",
_ => "idle",
}; };
partial void OnStatusChanged(TaskStatus value) partial void OnStatusChanged(TaskStatus value)
@@ -95,6 +98,13 @@ public sealed partial class TaskRowViewModel : ViewModelBase
partial void OnHasQueuedSubtasksChanged(bool value) partial void OnHasQueuedSubtasksChanged(bool value)
=> OnPropertyChanged(nameof(CanRemoveFromQueue)); => OnPropertyChanged(nameof(CanRemoveFromQueue));
partial void OnBlockedByTaskIdChanged(string? value)
{
OnPropertyChanged(nameof(IsQueued));
OnPropertyChanged(nameof(IsWaiting));
OnPropertyChanged(nameof(StatusChipClass));
}
partial void OnParentTaskIdChanged(string? value) partial void OnParentTaskIdChanged(string? value)
{ {
OnPropertyChanged(nameof(IsChild)); OnPropertyChanged(nameof(IsChild));
@@ -125,18 +135,19 @@ public sealed partial class TaskRowViewModel : ViewModelBase
public void UpdateFromEntity(TaskEntity t) public void UpdateFromEntity(TaskEntity t)
{ {
var (add, del) = ParseDiffStat(t.Worktree?.DiffStat); var (add, del) = ParseDiffStat(t.Worktree?.DiffStat);
Title = t.Title; Title = t.Title;
ListName = t.List?.Name ?? ""; ListName = t.List?.Name ?? "";
Done = t.Status == TaskStatus.Done; Done = t.Status == TaskStatus.Done;
IsStarred = t.IsStarred; IsStarred = t.IsStarred;
IsMyDay = t.IsMyDay; IsMyDay = t.IsMyDay;
Status = t.Status; Status = t.Status;
Branch = t.Worktree?.BranchName; Branch = t.Worktree?.BranchName;
DiffStat = t.Worktree?.DiffStat; DiffStat = t.Worktree?.DiffStat;
ScheduledFor = t.ScheduledFor; ScheduledFor = t.ScheduledFor;
DiffAdditions = add; DiffAdditions = add;
DiffDeletions = del; DiffDeletions = del;
ParentTaskId = t.ParentTaskId; ParentTaskId = t.ParentTaskId;
BlockedByTaskId = t.BlockedByTaskId;
} }
// Best-effort parse of diff stat strings like "+12 -3" or "12 additions, 3 deletions". // Best-effort parse of diff stat strings like "+12 -3" or "12 additions, 3 deletions".

View File

@@ -495,18 +495,27 @@ public sealed partial class TasksIslandViewModel : ViewModelBase
// For a planning parent the dequeue button targets queued/waiting children, // For a planning parent the dequeue button targets queued/waiting children,
// not the parent itself (whose Status is Planning/Planned). // not the parent itself (whose Status is Planning/Planned).
if (entity.Status == TaskStatus.Planning || entity.Status == TaskStatus.Planned) if (entity.Status == TaskStatus.Planning || entity.Status == TaskStatus.Planned
|| entity.PlanningPhase != PlanningPhase.None)
{ {
var children = await db.Tasks var children = await db.Tasks
.Where(t => t.ParentTaskId == row.Id .Where(t => t.ParentTaskId == row.Id
&& (t.Status == TaskStatus.Queued || t.Status == TaskStatus.Waiting)) && (t.Status == TaskStatus.Queued || t.Status == TaskStatus.Waiting))
.ToListAsync(); .ToListAsync();
foreach (var c in children) c.Status = TaskStatus.Manual; foreach (var c in children)
{
c.Status = TaskStatus.Manual;
c.BlockedByTaskId = null;
}
await db.SaveChangesAsync(); await db.SaveChangesAsync();
foreach (var c in children) foreach (var c in children)
{ {
var childRow = Items.FirstOrDefault(r => r.Id == c.Id); var childRow = Items.FirstOrDefault(r => r.Id == c.Id);
if (childRow is not null) childRow.Status = TaskStatus.Manual; if (childRow is not null)
{
childRow.Status = TaskStatus.Manual;
childRow.BlockedByTaskId = null;
}
} }
row.HasQueuedSubtasks = false; row.HasQueuedSubtasks = false;
} }

View File

@@ -23,4 +23,8 @@
<ImplicitUsings>enable</ImplicitUsings> <ImplicitUsings>enable</ImplicitUsings>
</PropertyGroup> </PropertyGroup>
<ItemGroup>
<InternalsVisibleTo Include="ClaudeDo.Worker.Tests" />
</ItemGroup>
</Project> </Project>

View File

@@ -88,7 +88,7 @@ public sealed class WorkerHub : Microsoft.AspNetCore.SignalR.Hub
{ {
try try
{ {
await _planningChain.QueueSubtasksSequentiallyAsync(parentTaskId, Context.ConnectionAborted); await _planningChain.SetupChainAsync(parentTaskId, Context.ConnectionAborted);
} }
catch (InvalidOperationException ex) catch (InvalidOperationException ex)
{ {

View File

@@ -19,7 +19,13 @@ public sealed class PlanningChainCoordinator
_state = state; _state = state;
} }
public async Task QueueSubtasksSequentiallyAsync(string parentTaskId, CancellationToken ct = default) // Sets up a sequential queue chain over a planning parent's children.
// - First child gets Status=Queued (auto-wakes the queue picker).
// - Each subsequent child gets Status=Queued + BlockedByTaskId=<predecessor>,
// so the picker skips them until the predecessor finishes.
// The "agent" tag is auto-attached to every child so the picker can claim them.
// Returns the number of children placed in the chain.
internal async Task<int> SetupChainAsync(string parentTaskId, CancellationToken ct = default)
{ {
await using var ctx = await _dbFactory.CreateDbContextAsync(ct); await using var ctx = await _dbFactory.CreateDbContextAsync(ct);
var parent = await ctx.Tasks.FirstOrDefaultAsync(t => t.Id == parentTaskId, ct) var parent = await ctx.Tasks.FirstOrDefaultAsync(t => t.Id == parentTaskId, ct)
@@ -33,22 +39,38 @@ public sealed class PlanningChainCoordinator
if (children.Count == 0) if (children.Count == 0)
throw new InvalidOperationException("Parent has no subtasks."); throw new InvalidOperationException("Parent has no subtasks.");
// Eligibility: new layout uses Status=Idle. Tolerate legacy Manual/Planned/Draft
// values during this slice — they will be migrated away in slice 6.
var bad = children.FirstOrDefault(c => var bad = children.FirstOrDefault(c =>
c.Status != TaskStatus.Manual && c.Status != TaskStatus.Planned); c.Status != TaskStatus.Idle &&
c.Status != TaskStatus.Manual &&
c.Status != TaskStatus.Planned &&
c.Status != TaskStatus.Draft);
if (bad is not null) if (bad is not null)
throw new InvalidOperationException( throw new InvalidOperationException(
$"Child {bad.Id} is in status {bad.Status}; expected Manual or Planned."); $"Child {bad.Id} is in status {bad.Status}; expected Idle (or legacy Manual/Planned/Draft).");
// Worker queue picker requires the "agent" tag — attach it so children are pickable. // Worker queue picker requires the "agent" tag — attach it so children are pickable.
var agentTag = await ctx.Tags.FirstOrDefaultAsync(t => t.Name == "agent", ct); var agentTag = await ctx.Tags.FirstOrDefaultAsync(t => t.Name == "agent", ct);
for (int i = 0; i < children.Count; i++) if (agentTag is not null)
{ {
children[i].Status = i == 0 ? TaskStatus.Queued : TaskStatus.Waiting; foreach (var c in children)
if (agentTag is not null && !children[i].Tags.Any(t => t.Id == agentTag.Id)) {
children[i].Tags.Add(agentTag); if (!c.Tags.Any(t => t.Id == agentTag.Id))
c.Tags.Add(agentTag);
}
await ctx.SaveChangesAsync(ct);
} }
await ctx.SaveChangesAsync(ct); var state = _state();
for (int i = 0; i < children.Count; i++)
{
await state.EnqueueAsync(children[i].Id, ct);
if (i > 0)
await state.BlockOnAsync(children[i].Id, children[i - 1].Id, ct);
}
return children.Count;
} }
public async Task<string?> OnChildFinishedAsync( public async Task<string?> OnChildFinishedAsync(
@@ -57,21 +79,18 @@ public sealed class PlanningChainCoordinator
if (finalStatus != TaskStatus.Done) return null; if (finalStatus != TaskStatus.Done) return null;
await using var ctx = await _dbFactory.CreateDbContextAsync(ct); await using var ctx = await _dbFactory.CreateDbContextAsync(ct);
var child = await ctx.Tasks // The successor is whichever sibling explicitly blocks on this child.
// No status check — UnblockAsync flips legacy Waiting to Queued and is a no-op
// for already-Queued rows in the new layout.
var nextId = await ctx.Tasks
.AsNoTracking() .AsNoTracking()
.FirstOrDefaultAsync(t => t.Id == childTaskId, ct); .Where(t => t.BlockedByTaskId == childTaskId)
if (child?.ParentTaskId is null) return null; .OrderBy(t => t.SortOrder).ThenBy(t => t.CreatedAt)
.Select(t => t.Id)
var next = await ctx.Tasks
.AsNoTracking()
.Where(t => t.ParentTaskId == child.ParentTaskId
&& t.SortOrder > child.SortOrder
&& t.Status == TaskStatus.Waiting)
.OrderBy(t => t.SortOrder)
.FirstOrDefaultAsync(ct); .FirstOrDefaultAsync(ct);
if (next is null) return null; if (nextId is null) return null;
await _state().UnblockAsync(next.Id, ct); await _state().UnblockAsync(nextId, ct);
return next.Id; return nextId;
} }
} }

View File

@@ -2,6 +2,7 @@ using System.ComponentModel;
using ClaudeDo.Data.Models; using ClaudeDo.Data.Models;
using ClaudeDo.Data.Repositories; using ClaudeDo.Data.Repositories;
using ClaudeDo.Worker.Hub; using ClaudeDo.Worker.Hub;
using ClaudeDo.Worker.State;
using ModelContextProtocol.Server; using ModelContextProtocol.Server;
using TaskStatus = ClaudeDo.Data.Models.TaskStatus; using TaskStatus = ClaudeDo.Data.Models.TaskStatus;
@@ -16,15 +17,21 @@ public sealed class PlanningMcpService
private readonly TaskRepository _tasks; private readonly TaskRepository _tasks;
private readonly PlanningMcpContextAccessor _contextAccessor; private readonly PlanningMcpContextAccessor _contextAccessor;
private readonly HubBroadcaster _broadcaster; private readonly HubBroadcaster _broadcaster;
private readonly ITaskStateService _state;
private readonly PlanningChainCoordinator _chain;
public PlanningMcpService( public PlanningMcpService(
TaskRepository tasks, TaskRepository tasks,
PlanningMcpContextAccessor contextAccessor, PlanningMcpContextAccessor contextAccessor,
HubBroadcaster broadcaster) HubBroadcaster broadcaster,
ITaskStateService state,
PlanningChainCoordinator chain)
{ {
_tasks = tasks; _tasks = tasks;
_contextAccessor = contextAccessor; _contextAccessor = contextAccessor;
_broadcaster = broadcaster; _broadcaster = broadcaster;
_state = state;
_chain = chain;
} }
private Task BroadcastTaskUpdatedAsync(string taskId, CancellationToken ct) private Task BroadcastTaskUpdatedAsync(string taskId, CancellationToken ct)
@@ -61,9 +68,9 @@ public sealed class PlanningMcpService
} }
private static readonly TaskStatus[] EditableStatuses = private static readonly TaskStatus[] EditableStatuses =
{ TaskStatus.Draft, TaskStatus.Manual, TaskStatus.Queued, TaskStatus.Waiting }; { TaskStatus.Draft, TaskStatus.Idle, TaskStatus.Manual, TaskStatus.Queued };
[McpServerTool, Description("Update a child task in the active planning session. Can change title, description, tags (replaces the full set), commit type, and status. Status must be one of: Draft, Manual, Queued, Waiting.")] [McpServerTool, Description("Update a child task in the active planning session. Can change title, description, tags (replaces the full set), commit type, and status. Status must be one of: Draft, Idle, Manual, Queued.")]
public async Task<ChildTaskDto> UpdateChildTask( public async Task<ChildTaskDto> UpdateChildTask(
string taskId, string taskId,
string? title, string? title,
@@ -76,7 +83,7 @@ public sealed class PlanningMcpService
var ctx = _contextAccessor.Current; var ctx = _contextAccessor.Current;
var parent = await _tasks.GetByIdAsync(ctx.ParentTaskId, cancellationToken) var parent = await _tasks.GetByIdAsync(ctx.ParentTaskId, cancellationToken)
?? throw new InvalidOperationException("Planning parent task not found."); ?? throw new InvalidOperationException("Planning parent task not found.");
if (parent.Status != TaskStatus.Planning) if (parent.PlanningPhase != PlanningPhase.Active)
throw new InvalidOperationException("Cannot modify tasks outside an active planning session."); throw new InvalidOperationException("Cannot modify tasks outside an active planning session.");
var child = await _tasks.GetByIdAsync(taskId, cancellationToken) var child = await _tasks.GetByIdAsync(taskId, cancellationToken)
@@ -90,7 +97,7 @@ public sealed class PlanningMcpService
if (!Enum.TryParse<TaskStatus>(status, ignoreCase: true, out var parsed)) if (!Enum.TryParse<TaskStatus>(status, ignoreCase: true, out var parsed))
throw new InvalidOperationException($"Unknown status '{status}'."); throw new InvalidOperationException($"Unknown status '{status}'.");
if (!EditableStatuses.Contains(parsed)) if (!EditableStatuses.Contains(parsed))
throw new InvalidOperationException($"Status '{parsed}' cannot be set via MCP. Allowed: Draft, Manual, Queued, Waiting."); throw new InvalidOperationException($"Status '{parsed}' cannot be set via MCP. Allowed: Draft, Idle, Manual, Queued.");
newStatus = parsed; newStatus = parsed;
} }
@@ -111,7 +118,7 @@ public sealed class PlanningMcpService
var ctx = _contextAccessor.Current; var ctx = _contextAccessor.Current;
var parent = await _tasks.GetByIdAsync(ctx.ParentTaskId, cancellationToken) var parent = await _tasks.GetByIdAsync(ctx.ParentTaskId, cancellationToken)
?? throw new InvalidOperationException("Planning parent task not found."); ?? throw new InvalidOperationException("Planning parent task not found.");
if (parent.Status != TaskStatus.Planning) if (parent.PlanningPhase != PlanningPhase.Active)
throw new InvalidOperationException("Cannot delete tasks outside an active planning session."); throw new InvalidOperationException("Cannot delete tasks outside an active planning session.");
var child = await _tasks.GetByIdAsync(taskId, cancellationToken) var child = await _tasks.GetByIdAsync(taskId, cancellationToken)
@@ -141,11 +148,19 @@ public sealed class PlanningMcpService
CancellationToken cancellationToken) CancellationToken cancellationToken)
{ {
var ctx = _contextAccessor.Current; var ctx = _contextAccessor.Current;
var childIds = (await _tasks.GetChildrenAsync(ctx.ParentTaskId, cancellationToken))
.Select(c => c.Id).ToList(); var finalizeResult = await _state.FinalizePlanningAsync(ctx.ParentTaskId, cancellationToken);
var count = await _tasks.FinalizePlanningAsync(ctx.ParentTaskId, queueAgentTasks, cancellationToken); if (!finalizeResult.Ok)
foreach (var id in childIds) throw new InvalidOperationException(
await BroadcastTaskUpdatedAsync(id, cancellationToken); finalizeResult.Reason ?? $"Could not finalize planning for task {ctx.ParentTaskId}.");
var children = await _tasks.GetChildrenAsync(ctx.ParentTaskId, cancellationToken);
int count = children.Count;
if (queueAgentTasks && children.Count > 0)
count = await _chain.SetupChainAsync(ctx.ParentTaskId, cancellationToken);
foreach (var c in children)
await BroadcastTaskUpdatedAsync(c.Id, cancellationToken);
await BroadcastTaskUpdatedAsync(ctx.ParentTaskId, cancellationToken); await BroadcastTaskUpdatedAsync(ctx.ParentTaskId, cancellationToken);
return count; return count;
} }

View File

@@ -24,6 +24,7 @@ public sealed class PlanningSessionManager
private readonly WorkerConfig _cfg; private readonly WorkerConfig _cfg;
private readonly string _rootDirectory; private readonly string _rootDirectory;
private readonly ITaskStateService? _state; private readonly ITaskStateService? _state;
private readonly PlanningChainCoordinator? _chain;
// DI constructor. // DI constructor.
public PlanningSessionManager( public PlanningSessionManager(
@@ -31,12 +32,14 @@ public sealed class PlanningSessionManager
GitService git, GitService git,
WorkerConfig cfg, WorkerConfig cfg,
ITaskStateService state, ITaskStateService state,
PlanningChainCoordinator chain,
string rootDirectory) string rootDirectory)
{ {
_factory = factory; _factory = factory;
_git = git; _git = git;
_cfg = cfg; _cfg = cfg;
_state = state; _state = state;
_chain = chain;
_rootDirectory = rootDirectory; _rootDirectory = rootDirectory;
} }
@@ -48,7 +51,8 @@ public sealed class PlanningSessionManager
GitService git, GitService git,
WorkerConfig cfg, WorkerConfig cfg,
string rootDirectory, string rootDirectory,
ITaskStateService? state = null) ITaskStateService? state = null,
PlanningChainCoordinator? chain = null)
{ {
_tasksOverride = tasks; _tasksOverride = tasks;
_listsOverride = lists; _listsOverride = lists;
@@ -56,6 +60,7 @@ public sealed class PlanningSessionManager
_git = git; _git = git;
_cfg = cfg; _cfg = cfg;
_state = state; _state = state;
_chain = chain;
_rootDirectory = rootDirectory; _rootDirectory = rootDirectory;
} }
@@ -194,7 +199,21 @@ public sealed class PlanningSessionManager
var (tasks, lists, settings, ctx) = CreateRepos(); var (tasks, lists, settings, ctx) = CreateRepos();
await using var __ = ctx; await using var __ = ctx;
var count = await tasks.FinalizePlanningAsync(taskId, queueAgentTasks, ct); if (_state is null || _chain is null)
throw new InvalidOperationException(
"PlanningSessionManager.FinalizeAsync requires ITaskStateService and PlanningChainCoordinator.");
var finalizeResult = await _state.FinalizePlanningAsync(taskId, ct);
if (!finalizeResult.Ok)
throw new InvalidOperationException(
finalizeResult.Reason ?? $"Could not finalize planning for task {taskId}.");
int count = 0;
var children = await tasks.GetChildrenAsync(taskId, ct);
if (queueAgentTasks && children.Count > 0)
count = await _chain.SetupChainAsync(taskId, ct);
else
count = children.Count;
// Best-effort cleanup — don't block finalization on git state. // Best-effort cleanup — don't block finalization on git state.
await TryCleanupWorktreeAsync(taskId, lists, settings, ct); await TryCleanupWorktreeAsync(taskId, lists, settings, ct);

View File

@@ -83,6 +83,7 @@ builder.Services.AddSingleton(sp =>
sp.GetRequiredService<GitService>(), sp.GetRequiredService<GitService>(),
cfg, cfg,
sp.GetRequiredService<ITaskStateService>(), sp.GetRequiredService<ITaskStateService>(),
sp.GetRequiredService<PlanningChainCoordinator>(),
planningSessionsDir)); planningSessionsDir));
builder.Services.AddSingleton<IPlanningTerminalLauncher>(sp => builder.Services.AddSingleton<IPlanningTerminalLauncher>(sp =>
new WindowsTerminalPlanningLauncher("wt.exe", cfg.ClaudeBin)); new WindowsTerminalPlanningLauncher("wt.exe", cfg.ClaudeBin));

View File

@@ -164,7 +164,8 @@ public sealed class TaskStateService : ITaskStateService
.Where(t => t.Id == parentId && t.PlanningPhase == PlanningPhase.Active) .Where(t => t.Id == parentId && t.PlanningPhase == PlanningPhase.Active)
.ExecuteUpdateAsync(s => s .ExecuteUpdateAsync(s => s
.SetProperty(t => t.PlanningPhase, PlanningPhase.Finalized) .SetProperty(t => t.PlanningPhase, PlanningPhase.Finalized)
.SetProperty(t => t.PlanningFinalizedAt, DateTime.UtcNow), ct); .SetProperty(t => t.PlanningFinalizedAt, DateTime.UtcNow)
.SetProperty(t => t.PlanningSessionToken, (string?)null), ct);
if (affected == 0) if (affected == 0)
return new TransitionResult(false, "No active planning session."); return new TransitionResult(false, "No active planning session.");

View File

@@ -34,7 +34,9 @@ public sealed class PlanningHubTests : IDisposable
var cfg = new WorkerConfig { CentralWorktreeRoot = Path.Combine(_rootDir, "central") }; var cfg = new WorkerConfig { CentralWorktreeRoot = Path.Combine(_rootDir, "central") };
var settingsRepo = new AppSettingsRepository(_ctx); var settingsRepo = new AppSettingsRepository(_ctx);
settingsRepo.UpdateAsync(new AppSettingsEntity { WorktreeStrategy = "sibling" }).GetAwaiter().GetResult(); settingsRepo.UpdateAsync(new AppSettingsEntity { WorktreeStrategy = "sibling" }).GetAwaiter().GetResult();
_planning = new PlanningSessionManager(_tasks, _lists, settingsRepo, git, cfg, _rootDir); var built = TaskStateServiceBuilder.Build(_db.CreateFactory());
_planning = new PlanningSessionManager(
_tasks, _lists, settingsRepo, git, cfg, _rootDir, built.State, built.Chain);
_launcher = new FakePlanningLauncher(); _launcher = new FakePlanningLauncher();
_proxy = new RecordingClientProxy(); _proxy = new RecordingClientProxy();
} }

View File

@@ -32,7 +32,7 @@ public sealed class PlanningChainCoordinatorTests : IDisposable
public void Dispose() => _db.Dispose(); public void Dispose() => _db.Dispose();
private async Task SeedPlanningFamilyAsync(string parentId, int childCount) private async Task SeedPlanningFamilyAsync(string parentId, int childCount, TaskStatus childStatus = TaskStatus.Manual)
{ {
await using var ctx = _factory.CreateDbContext(); await using var ctx = _factory.CreateDbContext();
ctx.Tasks.Add(new TaskEntity ctx.Tasks.Add(new TaskEntity
@@ -51,7 +51,7 @@ public sealed class PlanningChainCoordinatorTests : IDisposable
ListId = _listId, ListId = _listId,
Title = $"Child {i}", Title = $"Child {i}",
CreatedAt = DateTime.UtcNow, CreatedAt = DateTime.UtcNow,
Status = TaskStatus.Manual, Status = childStatus,
ParentTaskId = parentId, ParentTaskId = parentId,
SortOrder = i, SortOrder = i,
}); });
@@ -64,31 +64,67 @@ public sealed class PlanningChainCoordinatorTests : IDisposable
await using var ctx = _factory.CreateDbContext(); await using var ctx = _factory.CreateDbContext();
return await ctx.Tasks return await ctx.Tasks
.AsNoTracking() .AsNoTracking()
.Include(t => t.Tags)
.Where(t => t.ParentTaskId == parentId) .Where(t => t.ParentTaskId == parentId)
.OrderBy(t => t.SortOrder) .OrderBy(t => t.SortOrder)
.ToListAsync(); .ToListAsync();
} }
[Fact] [Fact]
public async Task QueueSubtasksSequentially_SetsFirstQueued_RestWaiting() public async Task SetupChain_FirstChildQueuedUnblocked_RestQueuedBlockedByPredecessor()
{ {
await SeedPlanningFamilyAsync("P", 3); await SeedPlanningFamilyAsync("P", 3);
await _sut.QueueSubtasksSequentiallyAsync("P", default); var count = await _sut.SetupChainAsync("P", default);
Assert.Equal(3, count);
var kids = await GetChildrenAsync("P"); var kids = await GetChildrenAsync("P");
Assert.Equal(TaskStatus.Queued, kids[0].Status); Assert.Equal(TaskStatus.Queued, kids[0].Status);
Assert.Equal(TaskStatus.Waiting, kids[1].Status); Assert.Null(kids[0].BlockedByTaskId);
Assert.Equal(TaskStatus.Waiting, kids[2].Status); Assert.Equal(TaskStatus.Queued, kids[1].Status);
Assert.Equal(kids[0].Id, kids[1].BlockedByTaskId);
Assert.Equal(TaskStatus.Queued, kids[2].Status);
Assert.Equal(kids[1].Id, kids[2].BlockedByTaskId);
} }
[Fact] [Fact]
public async Task OnChildDone_FlipsNextWaitingToQueued() public async Task SetupChain_AttachesAgentTagToAllChildren()
{
await SeedPlanningFamilyAsync("P", 2);
await _sut.SetupChainAsync("P", default);
var kids = await GetChildrenAsync("P");
Assert.All(kids, k => Assert.Contains(k.Tags, t => t.Name == "agent"));
}
[Fact]
public async Task SetupChain_AcceptsIdleChildren()
{
await SeedPlanningFamilyAsync("P", 2, childStatus: TaskStatus.Idle);
var count = await _sut.SetupChainAsync("P", default);
Assert.Equal(2, count);
}
[Fact]
public async Task SetupChain_AcceptsDraftChildren()
{
await SeedPlanningFamilyAsync("P", 2, childStatus: TaskStatus.Draft);
var count = await _sut.SetupChainAsync("P", default);
Assert.Equal(2, count);
}
[Fact]
public async Task OnChildDone_UnblocksTheSuccessor()
{ {
await SeedPlanningFamilyAsync("P", 3); await SeedPlanningFamilyAsync("P", 3);
await _sut.QueueSubtasksSequentiallyAsync("P", default); await _sut.SetupChainAsync("P", default);
// Simulate first child finishing Done. // Mark the head child Done before announcing.
await using (var ctx = _factory.CreateDbContext()) await using (var ctx = _factory.CreateDbContext())
{ {
var first = await ctx.Tasks.FirstAsync(t => t.Id == "P-c0"); var first = await ctx.Tasks.FirstAsync(t => t.Id == "P-c0");
@@ -101,15 +137,19 @@ public sealed class PlanningChainCoordinatorTests : IDisposable
Assert.Equal("P-c1", advanced); Assert.Equal("P-c1", advanced);
var kids = await GetChildrenAsync("P"); var kids = await GetChildrenAsync("P");
Assert.Equal(TaskStatus.Done, kids[0].Status); Assert.Equal(TaskStatus.Done, kids[0].Status);
// c1 was Queued+BlockedBy=c0; UnblockAsync clears the block.
Assert.Equal(TaskStatus.Queued, kids[1].Status); Assert.Equal(TaskStatus.Queued, kids[1].Status);
Assert.Equal(TaskStatus.Waiting, kids[2].Status); Assert.Null(kids[1].BlockedByTaskId);
// c2 still blocked on c1.
Assert.Equal(TaskStatus.Queued, kids[2].Status);
Assert.Equal(kids[1].Id, kids[2].BlockedByTaskId);
} }
[Fact] [Fact]
public async Task OnChildFailed_DoesNotAdvanceChain() public async Task OnChildFailed_DoesNotAdvanceChain()
{ {
await SeedPlanningFamilyAsync("P", 3); await SeedPlanningFamilyAsync("P", 3);
await _sut.QueueSubtasksSequentiallyAsync("P", default); await _sut.SetupChainAsync("P", default);
await using (var ctx = _factory.CreateDbContext()) await using (var ctx = _factory.CreateDbContext())
{ {
@@ -123,17 +163,17 @@ public sealed class PlanningChainCoordinatorTests : IDisposable
Assert.Null(advanced); Assert.Null(advanced);
var kids = await GetChildrenAsync("P"); var kids = await GetChildrenAsync("P");
Assert.Equal(TaskStatus.Failed, kids[0].Status); Assert.Equal(TaskStatus.Failed, kids[0].Status);
Assert.Equal(TaskStatus.Waiting, kids[1].Status); // Successors remain blocked on the failed predecessor.
Assert.Equal(TaskStatus.Waiting, kids[2].Status); Assert.Equal(kids[0].Id, kids[1].BlockedByTaskId);
Assert.Equal(kids[1].Id, kids[2].BlockedByTaskId);
} }
[Fact] [Fact]
public async Task OnChildDone_LastChild_ReturnsNull() public async Task OnChildDone_LastChild_ReturnsNull()
{ {
await SeedPlanningFamilyAsync("P", 2); await SeedPlanningFamilyAsync("P", 2);
await _sut.QueueSubtasksSequentiallyAsync("P", default); await _sut.SetupChainAsync("P", default);
// Mark both done, simulating chain reaching the end.
await using (var ctx = _factory.CreateDbContext()) await using (var ctx = _factory.CreateDbContext())
{ {
foreach (var t in ctx.Tasks.Where(t => t.ParentTaskId == "P")) foreach (var t in ctx.Tasks.Where(t => t.ParentTaskId == "P"))
@@ -147,18 +187,17 @@ public sealed class PlanningChainCoordinatorTests : IDisposable
} }
[Fact] [Fact]
public async Task QueueSubtasksSequentially_RejectsNonManualChildren() public async Task SetupChain_RejectsRunningChildren()
{ {
await SeedPlanningFamilyAsync("P", 2); await SeedPlanningFamilyAsync("P", 2);
// Corrupt one child to be already Queued.
await using (var ctx = _factory.CreateDbContext()) await using (var ctx = _factory.CreateDbContext())
{ {
var first = await ctx.Tasks.FirstAsync(t => t.Id == "P-c0"); var first = await ctx.Tasks.FirstAsync(t => t.Id == "P-c0");
first.Status = TaskStatus.Queued; first.Status = TaskStatus.Running;
await ctx.SaveChangesAsync(); await ctx.SaveChangesAsync();
} }
await Assert.ThrowsAsync<InvalidOperationException>( await Assert.ThrowsAsync<InvalidOperationException>(
() => _sut.QueueSubtasksSequentiallyAsync("P", default)); () => _sut.SetupChainAsync("P", default));
} }
} }

View File

@@ -1,3 +1,4 @@
using System.Diagnostics;
using ClaudeDo.Data; using ClaudeDo.Data;
using ClaudeDo.Data.Git; using ClaudeDo.Data.Git;
using ClaudeDo.Data.Models; using ClaudeDo.Data.Models;
@@ -5,6 +6,7 @@ using ClaudeDo.Data.Repositories;
using ClaudeDo.Worker.Config; using ClaudeDo.Worker.Config;
using ClaudeDo.Worker.Hub; using ClaudeDo.Worker.Hub;
using ClaudeDo.Worker.Planning; using ClaudeDo.Worker.Planning;
using ClaudeDo.Worker.Queue;
using ClaudeDo.Worker.Tests.Infrastructure; using ClaudeDo.Worker.Tests.Infrastructure;
using Microsoft.AspNetCore.Http; using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.SignalR; using Microsoft.AspNetCore.SignalR;
@@ -50,6 +52,11 @@ public sealed class PlanningEndToEndTests : IDisposable
private readonly ClaudeDoDbContext _ctx; private readonly ClaudeDoDbContext _ctx;
private readonly TaskRepository _tasks; private readonly TaskRepository _tasks;
private readonly ListRepository _lists; private readonly ListRepository _lists;
private readonly AppSettingsRepository _settingsRepo;
private readonly GitService _git;
private readonly WorkerConfig _cfg;
private readonly string _root;
private readonly TaskStateServiceBuilder.Built _built;
private readonly PlanningSessionManager _manager; private readonly PlanningSessionManager _manager;
private readonly DefaultHttpContext _httpContext; private readonly DefaultHttpContext _httpContext;
private readonly PlanningMcpContextAccessor _accessor; private readonly PlanningMcpContextAccessor _accessor;
@@ -61,17 +68,20 @@ public sealed class PlanningEndToEndTests : IDisposable
_tasks = new TaskRepository(_ctx); _tasks = new TaskRepository(_ctx);
_lists = new ListRepository(_ctx); _lists = new ListRepository(_ctx);
var root = Path.Combine(Path.GetTempPath(), $"cd_e2e_{Guid.NewGuid():N}"); _root = Path.Combine(Path.GetTempPath(), $"cd_e2e_{Guid.NewGuid():N}");
var git = new GitService(); _git = new GitService();
var cfg = new WorkerConfig { CentralWorktreeRoot = Path.Combine(root, "central") }; _cfg = new WorkerConfig { CentralWorktreeRoot = Path.Combine(_root, "central") };
var settingsRepo = new AppSettingsRepository(_ctx); _settingsRepo = new AppSettingsRepository(_ctx);
settingsRepo.UpdateAsync(new AppSettingsEntity { WorktreeStrategy = "sibling" }).GetAwaiter().GetResult(); _settingsRepo.UpdateAsync(new AppSettingsEntity { WorktreeStrategy = "sibling" }).GetAwaiter().GetResult();
_manager = new PlanningSessionManager(_tasks, _lists, settingsRepo, git, cfg, root);
_built = TaskStateServiceBuilder.Build(_db.CreateFactory());
_manager = new PlanningSessionManager(
_tasks, _lists, _settingsRepo, _git, _cfg, _root, _built.State, _built.Chain);
_httpContext = new DefaultHttpContext(); _httpContext = new DefaultHttpContext();
_accessor = new PlanningMcpContextAccessor(new E2EFakeHttpContextAccessor { HttpContext = _httpContext }); _accessor = new PlanningMcpContextAccessor(new E2EFakeHttpContextAccessor { HttpContext = _httpContext });
var broadcaster = new HubBroadcaster(new E2EFakeHubContext()); var broadcaster = new HubBroadcaster(new E2EFakeHubContext());
_svc = new PlanningMcpService(_tasks, _accessor, broadcaster); _svc = new PlanningMcpService(_tasks, _accessor, broadcaster, _built.State, _built.Chain);
} }
public void Dispose() { _ctx.Dispose(); _db.Dispose(); } public void Dispose() { _ctx.Dispose(); _db.Dispose(); }
@@ -108,9 +118,69 @@ public sealed class PlanningEndToEndTests : IDisposable
Assert.Equal(2, count); Assert.Equal(2, count);
var reload = await _tasks.GetByIdAsync(parent.Id); var reload = await _tasks.GetByIdAsync(parent.Id);
Assert.Equal(TaskStatus.Planned, reload!.Status); Assert.Equal(PlanningPhase.Finalized, reload!.PlanningPhase);
var kids = await _tasks.GetChildrenAsync(parent.Id); var kids = await _tasks.GetChildrenAsync(parent.Id);
Assert.All(kids, k => Assert.Equal(TaskStatus.Manual, k.Status)); // SetupChainAsync auto-attaches agent tag and queues all children;
// the first one is unblocked, the rest are BlockedBy their predecessor.
Assert.Equal(TaskStatus.Queued, kids[0].Status);
Assert.Null(kids[0].BlockedByTaskId);
Assert.Equal(TaskStatus.Queued, kids[1].Status);
Assert.Equal(kids[0].Id, kids[1].BlockedByTaskId);
}
// Regression: original bug was "queue never picks up planning tasks". After Finalize
// with queueAgentTasks=true, the first child must be claimable by the queue picker
// automatically — without anyone calling WakeQueue() manually.
[Fact]
public async Task FinalizeAsync_FirstChildIsClaimedByPicker_WithinDeadline()
{
var listId = Guid.NewGuid().ToString();
var wd = Path.Combine(Path.GetTempPath(), $"cd_e2e_wd_{Guid.NewGuid():N}");
GitRepoFixture.InitRepoWithInitialCommit(wd);
await _lists.AddAsync(new ListEntity { Id = listId, Name = "L", WorkingDir = wd, CreatedAt = DateTime.UtcNow });
var parent = new TaskEntity
{
Id = Guid.NewGuid().ToString(),
ListId = listId,
Title = "Parent",
Status = TaskStatus.Manual,
CreatedAt = DateTime.UtcNow,
CommitType = "chore",
};
await _tasks.AddAsync(parent);
await _manager.StartAsync(parent.Id, CancellationToken.None);
_httpContext.Items["PlanningContext"] = new PlanningMcpContext { ParentTaskId = parent.Id };
await _svc.CreateChildTask("c1", null, null, null, CancellationToken.None);
await _svc.CreateChildTask("c2", null, null, null, CancellationToken.None);
await _svc.CreateChildTask("c3", null, null, null, CancellationToken.None);
var kidsBefore = await _tasks.GetChildrenAsync(parent.Id);
var firstChildId = kidsBefore[0].Id;
var wakesBefore = _built.WakeCount();
await _manager.FinalizeAsync(parent.Id, queueAgentTasks: true, CancellationToken.None);
// The picker should pick the first child immediately. Auto-wake fires inside
// _state.EnqueueAsync; we don't need a manual WakeQueue() for the bug to be fixed.
var picker = new QueuePicker(_db.CreateFactory());
TaskEntity? claimed = null;
var sw = Stopwatch.StartNew();
while (sw.ElapsedMilliseconds < 200)
{
claimed = await picker.ClaimNextAsync(DateTime.UtcNow, CancellationToken.None);
if (claimed is not null) break;
await Task.Delay(10);
}
Assert.NotNull(claimed);
Assert.Equal(firstChildId, claimed!.Id);
Assert.Equal(TaskStatus.Running, claimed.Status);
Assert.True(_built.WakeCount() > wakesBefore,
"TaskStateService.EnqueueAsync should auto-wake the queue.");
} }
} }

View File

@@ -64,6 +64,7 @@ public sealed class PlanningMcpServiceTests : IDisposable
public void Dispose() { _ctx.Dispose(); _db.Dispose(); } public void Dispose() { _ctx.Dispose(); _db.Dispose(); }
private List<(string Method, object?[] Args)> _hubCalls = new(); private List<(string Method, object?[] Args)> _hubCalls = new();
private TaskStateServiceBuilder.Built? _built;
private PlanningMcpService BuildSut(string parentTaskId) private PlanningMcpService BuildSut(string parentTaskId)
{ {
@@ -73,7 +74,8 @@ public sealed class PlanningMcpServiceTests : IDisposable
var hub = new FakeHubContext(); var hub = new FakeHubContext();
_hubCalls = hub.RecordingClients.Proxy.Calls; _hubCalls = hub.RecordingClients.Proxy.Calls;
var broadcaster = new HubBroadcaster(hub); var broadcaster = new HubBroadcaster(hub);
return new PlanningMcpService(_tasks, accessor, broadcaster); _built = TaskStateServiceBuilder.Build(_db.CreateFactory());
return new PlanningMcpService(_tasks, accessor, broadcaster, _built.State, _built.Chain);
} }
private IReadOnlyList<string> TaskUpdatedIds() => private IReadOnlyList<string> TaskUpdatedIds() =>
@@ -146,9 +148,12 @@ public sealed class PlanningMcpServiceTests : IDisposable
{ {
var parent = await SeedPlanningParentAsync(); var parent = await SeedPlanningParentAsync();
var c = await _tasks.CreateChildAsync(parent.Id, "c", null, null, null); var c = await _tasks.CreateChildAsync(parent.Id, "c", null, null, null);
await _tasks.FinalizePlanningAsync(parent.Id, queueAgentTasks: false); // Simulate post-finalize state directly: parent.PlanningPhase=Finalized
// is the gate the MCP service checks.
var sut = BuildSut(parent.Id); var sut = BuildSut(parent.Id);
var result = await _built!.State.FinalizePlanningAsync(parent.Id, CancellationToken.None);
Assert.True(result.Ok, result.Reason);
await Assert.ThrowsAsync<InvalidOperationException>(() => await Assert.ThrowsAsync<InvalidOperationException>(() =>
sut.UpdateChildTask(c.Id, "new", null, null, null, null, CancellationToken.None)); sut.UpdateChildTask(c.Id, "new", null, null, null, null, CancellationToken.None));
} }
@@ -258,7 +263,7 @@ public sealed class PlanningMcpServiceTests : IDisposable
Assert.Equal(2, count); Assert.Equal(2, count);
var loaded = await _tasks.GetByIdAsync(parent.Id); var loaded = await _tasks.GetByIdAsync(parent.Id);
Assert.Equal(TaskStatus.Planned, loaded!.Status); Assert.Equal(PlanningPhase.Finalized, loaded!.PlanningPhase);
Assert.Null(loaded.PlanningSessionToken); Assert.Null(loaded.PlanningSessionToken);
} }

View File

@@ -20,6 +20,7 @@ public sealed class PlanningSessionManagerTests : IDisposable
private readonly WorkerConfig _cfg; private readonly WorkerConfig _cfg;
private readonly AppSettingsRepository _settingsRepo; private readonly AppSettingsRepository _settingsRepo;
private readonly PlanningSessionManager _sut; private readonly PlanningSessionManager _sut;
private readonly TaskStateServiceBuilder.Built _built;
public PlanningSessionManagerTests() public PlanningSessionManagerTests()
{ {
@@ -31,7 +32,9 @@ public sealed class PlanningSessionManagerTests : IDisposable
_cfg = new WorkerConfig { CentralWorktreeRoot = Path.Combine(_rootDir, "central") }; _cfg = new WorkerConfig { CentralWorktreeRoot = Path.Combine(_rootDir, "central") };
_settingsRepo = new AppSettingsRepository(_ctx); _settingsRepo = new AppSettingsRepository(_ctx);
_settingsRepo.UpdateAsync(new AppSettingsEntity { WorktreeStrategy = "sibling" }).GetAwaiter().GetResult(); _settingsRepo.UpdateAsync(new AppSettingsEntity { WorktreeStrategy = "sibling" }).GetAwaiter().GetResult();
_sut = new PlanningSessionManager(_tasks, _lists, _settingsRepo, _git, _cfg, _rootDir); _built = TaskStateServiceBuilder.Build(_db.CreateFactory());
_sut = new PlanningSessionManager(
_tasks, _lists, _settingsRepo, _git, _cfg, _rootDir, _built.State, _built.Chain);
} }
public void Dispose() public void Dispose()
@@ -173,7 +176,7 @@ public sealed class PlanningSessionManagerTests : IDisposable
} }
[Fact] [Fact]
public async Task FinalizeAsync_PromotesDraftsAndMarksPlanned() public async Task FinalizeAsync_PromotesDraftsAndMarksPlanningFinalized()
{ {
var (listId, _) = await SeedListAsync(); var (listId, _) = await SeedListAsync();
var parent = await SeedManualTaskAsync(listId); var parent = await SeedManualTaskAsync(listId);
@@ -185,7 +188,9 @@ public sealed class PlanningSessionManagerTests : IDisposable
Assert.Equal(2, count); Assert.Equal(2, count);
var loaded = await _tasks.GetByIdAsync(parent.Id); var loaded = await _tasks.GetByIdAsync(parent.Id);
Assert.Equal(TaskStatus.Planned, loaded!.Status); Assert.Equal(PlanningPhase.Finalized, loaded!.PlanningPhase);
Assert.NotNull(loaded.PlanningFinalizedAt);
Assert.Null(loaded.PlanningSessionToken);
} }
[Fact] [Fact]

View File

@@ -188,65 +188,6 @@ public sealed class TaskRepositoryPlanningTests : IDisposable
Assert.Null(found); Assert.Null(found);
} }
[Fact]
public async Task FinalizePlanningAsync_TransitionsDraftsAndParent()
{
var listId = await CreateListAsync();
var parent = MakeTask(listId, TaskStatus.Manual);
await _tasks.AddAsync(parent);
await _tasks.SetPlanningStartedAsync(parent.Id, "tok");
var c1 = await _tasks.CreateChildAsync(parent.Id, "c1", null, tagNames: new[] { "agent" }, commitType: null);
var c2 = await _tasks.CreateChildAsync(parent.Id, "c2", null, tagNames: null, commitType: null);
var count = await _tasks.FinalizePlanningAsync(parent.Id, queueAgentTasks: true);
Assert.Equal(2, count);
var c1Loaded = await _tasks.GetByIdAsync(c1.Id);
var c2Loaded = await _tasks.GetByIdAsync(c2.Id);
var parentLoaded = await _tasks.GetByIdAsync(parent.Id);
Assert.Equal(TaskStatus.Queued, c1Loaded!.Status);
Assert.Equal(TaskStatus.Manual, c2Loaded!.Status);
Assert.Equal(TaskStatus.Planned, parentLoaded!.Status);
Assert.NotNull(parentLoaded.PlanningFinalizedAt);
Assert.Null(parentLoaded.PlanningSessionToken);
}
[Fact]
public async Task FinalizePlanningAsync_QueueAgentTasksFalse_AllToManual()
{
var listId = await CreateListAsync();
var parent = MakeTask(listId, TaskStatus.Manual);
await _tasks.AddAsync(parent);
await _tasks.SetPlanningStartedAsync(parent.Id, "tok");
var c = await _tasks.CreateChildAsync(parent.Id, "c", null, tagNames: new[] { "agent" }, commitType: null);
await _tasks.FinalizePlanningAsync(parent.Id, queueAgentTasks: false);
var cLoaded = await _tasks.GetByIdAsync(c.Id);
Assert.Equal(TaskStatus.Manual, cLoaded!.Status);
}
[Fact]
public async Task FinalizePlanningAsync_ParentWithAgentListTag_ChildIsQueued()
{
var listId = await CreateListAsync();
var agentTagId = await _tags.GetOrCreateAsync("agent");
await _lists.AddTagAsync(listId, agentTagId);
var parent = MakeTask(listId, TaskStatus.Manual);
await _tasks.AddAsync(parent);
await _tasks.SetPlanningStartedAsync(parent.Id, "tok");
var c = await _tasks.CreateChildAsync(parent.Id, "c", null, tagNames: null, commitType: null);
await _tasks.FinalizePlanningAsync(parent.Id, queueAgentTasks: true);
var cLoaded = await _tasks.GetByIdAsync(c.Id);
Assert.Equal(TaskStatus.Queued, cLoaded!.Status);
}
[Fact] [Fact]
public async Task DiscardPlanningAsync_DeletesDraftsAndResetsParent() public async Task DiscardPlanningAsync_DeletesDraftsAndResetsParent()
{ {

View File

@@ -36,7 +36,11 @@ public sealed class TaskStateServiceTests : IDisposable
public void Dispose() => _db.Dispose(); public void Dispose() => _db.Dispose();
private async Task<string> SeedTaskAsync(TaskStatus status, string? parentId = null, int sortOrder = 0) private async Task<string> SeedTaskAsync(
TaskStatus status,
string? parentId = null,
int sortOrder = 0,
string? blockedBy = null)
{ {
var id = Guid.NewGuid().ToString(); var id = Guid.NewGuid().ToString();
await using var ctx = _factory.CreateDbContext(); await using var ctx = _factory.CreateDbContext();
@@ -49,6 +53,7 @@ public sealed class TaskStateServiceTests : IDisposable
CreatedAt = DateTime.UtcNow, CreatedAt = DateTime.UtcNow,
ParentTaskId = parentId, ParentTaskId = parentId,
SortOrder = sortOrder, SortOrder = sortOrder,
BlockedByTaskId = blockedBy,
}); });
await ctx.SaveChangesAsync(); await ctx.SaveChangesAsync();
return id; return id;
@@ -364,20 +369,24 @@ public sealed class TaskStateServiceTests : IDisposable
// ─── Child terminal → chain advance ─────────────────────────────────── // ─── Child terminal → chain advance ───────────────────────────────────
[Fact] [Fact]
public async Task CompleteAsync_OnChild_AdvancesNextWaitingSibling() public async Task CompleteAsync_OnChild_AdvancesNextBlockedSibling()
{ {
var parent = await SeedTaskAsync(TaskStatus.Planned); var parent = await SeedTaskAsync(TaskStatus.Planned);
var c0 = await SeedTaskAsync(TaskStatus.Running, parentId: parent, sortOrder: 0); var c0 = await SeedTaskAsync(TaskStatus.Running, parentId: parent, sortOrder: 0);
var c1 = await SeedTaskAsync(TaskStatus.Waiting, parentId: parent, sortOrder: 1); var c1 = await SeedTaskAsync(TaskStatus.Queued, parentId: parent, sortOrder: 1, blockedBy: c0);
var c2 = await SeedTaskAsync(TaskStatus.Waiting, parentId: parent, sortOrder: 2); var c2 = await SeedTaskAsync(TaskStatus.Queued, parentId: parent, sortOrder: 2, blockedBy: c1);
var result = await _sut.CompleteAsync(c0, DateTime.UtcNow, "ok", default); var result = await _sut.CompleteAsync(c0, DateTime.UtcNow, "ok", default);
Assert.True(result.Ok); Assert.True(result.Ok);
Assert.Equal(TaskStatus.Done, await GetStatusAsync(c0)); Assert.Equal(TaskStatus.Done, await GetStatusAsync(c0));
// Next sibling was Waiting → chain coordinator unblocks → Queued. // c1 was BlockedBy=c0 → chain coordinator unblocks → BlockedByTaskId cleared, still Queued.
Assert.Equal(TaskStatus.Queued, await GetStatusAsync(c1)); var t1 = await GetTaskAsync(c1);
// Subsequent sibling untouched. Assert.Equal(TaskStatus.Queued, t1.Status);
Assert.Equal(TaskStatus.Waiting, await GetStatusAsync(c2)); Assert.Null(t1.BlockedByTaskId);
// c2 still blocked on c1.
var t2 = await GetTaskAsync(c2);
Assert.Equal(TaskStatus.Queued, t2.Status);
Assert.Equal(c1, t2.BlockedByTaskId);
} }
} }