mika kuns
2700c3d817
The usage monitor polled the undocumented OAuth usage endpoint every 60s and earned 429s. It now polls every 5 min while any task is Running and every 15 min while idle (usage_poll_interval_active_seconds / _idle_seconds, both clamped to >= 60; the old single usage_poll_interval_seconds key is gone). A 429 comes back as UsageRateLimitedException carrying Retry-After and adds exponential backoff on top, capped at 30 min and never shorter than the normal cadence; the strike count resets on the first success. The schedule arithmetic is the pure static UsagePollSchedule.NextDelay. Since the idle cadence is slow on purpose, WorkerHub.RefreshUsage drives UsageMonitorService.RefreshNowAsync behind a Refresh now button in the Usage Monitor modal: an out-of-band poll that pushes the loop's next-due time out so no double poll follows, with a 10s cooldown so click-spam can't earn a 429. Staleness now measures against the slower (idle) interval so an idle worker isn't flagged stale just for not polling.
57 lines
1.6 KiB
C#
57 lines
1.6 KiB
C#
using ClaudeDo.Data.Models;
|
|
using ClaudeDo.Worker.Tests.Infrastructure;
|
|
using ClaudeDo.Worker.Usage;
|
|
using TaskStatus = ClaudeDo.Data.Models.TaskStatus;
|
|
|
|
namespace ClaudeDo.Worker.Tests.Usage;
|
|
|
|
public sealed class RunningTaskProbeTests : IDisposable
|
|
{
|
|
private readonly DbFixture _db = new();
|
|
|
|
public void Dispose() => _db.Dispose();
|
|
|
|
private async Task SeedTaskAsync(TaskStatus status)
|
|
{
|
|
await using var context = _db.CreateFactory().CreateDbContext();
|
|
var list = new ListEntity { Id = Guid.NewGuid().ToString(), Name = "L", CreatedAt = DateTime.UtcNow };
|
|
context.Lists.Add(list);
|
|
context.Tasks.Add(new TaskEntity
|
|
{
|
|
Id = Guid.NewGuid().ToString(),
|
|
ListId = list.Id,
|
|
Title = $"task-{status}",
|
|
Status = status,
|
|
CreatedAt = DateTime.UtcNow,
|
|
});
|
|
await context.SaveChangesAsync();
|
|
}
|
|
|
|
[Fact]
|
|
public async Task AnyRunningAsync_NoTasks_ReturnsFalse()
|
|
{
|
|
var probe = new RunningTaskProbe(_db.CreateFactory());
|
|
|
|
Assert.False(await probe.AnyRunningAsync());
|
|
}
|
|
|
|
[Fact]
|
|
public async Task AnyRunningAsync_QueuedTaskOnly_ReturnsFalse()
|
|
{
|
|
await SeedTaskAsync(TaskStatus.Queued);
|
|
var probe = new RunningTaskProbe(_db.CreateFactory());
|
|
|
|
Assert.False(await probe.AnyRunningAsync());
|
|
}
|
|
|
|
[Fact]
|
|
public async Task AnyRunningAsync_RunningTask_ReturnsTrue()
|
|
{
|
|
await SeedTaskAsync(TaskStatus.Idle);
|
|
await SeedTaskAsync(TaskStatus.Running);
|
|
var probe = new RunningTaskProbe(_db.CreateFactory());
|
|
|
|
Assert.True(await probe.AnyRunningAsync());
|
|
}
|
|
}
|