Implement deferred WaitQueue, DiskAvailability, and NoOpCache behavior with tests

This commit is contained in:
Joseph Doherty
2026-02-27 09:58:37 -05:00
parent 8849265780
commit a660e38575
11 changed files with 508 additions and 25 deletions

View File

@@ -35,4 +35,82 @@ public sealed class NatsConsumerTests
consumer.Stop();
consumer.IsLeader().ShouldBeFalse();
}
[Fact] // T:1364
public void SortingConsumerPullRequests_ShouldSucceed()
{
var q = new WaitQueue(max: 100);
q.AddPrioritized(new WaitingRequest { Reply = "1a", PriorityGroup = new PriorityGroup { Priority = 1 }, N = 1 })
.ShouldBeTrue();
q.AddPrioritized(new WaitingRequest { Reply = "2a", PriorityGroup = new PriorityGroup { Priority = 2 }, N = 1 })
.ShouldBeTrue();
q.AddPrioritized(new WaitingRequest { Reply = "1b", PriorityGroup = new PriorityGroup { Priority = 1 }, N = 1 })
.ShouldBeTrue();
q.AddPrioritized(new WaitingRequest { Reply = "2b", PriorityGroup = new PriorityGroup { Priority = 2 }, N = 1 })
.ShouldBeTrue();
q.AddPrioritized(new WaitingRequest { Reply = "1c", PriorityGroup = new PriorityGroup { Priority = 1 }, N = 1 })
.ShouldBeTrue();
q.AddPrioritized(new WaitingRequest { Reply = "3a", PriorityGroup = new PriorityGroup { Priority = 3 }, N = 1 })
.ShouldBeTrue();
q.AddPrioritized(new WaitingRequest { Reply = "2c", PriorityGroup = new PriorityGroup { Priority = 2 }, N = 1 })
.ShouldBeTrue();
var expectedOrder = new[]
{
("1a", 1),
("1b", 1),
("1c", 1),
("2a", 2),
("2b", 2),
("2c", 2),
("3a", 3),
};
q.Len.ShouldBe(expectedOrder.Length);
foreach (var (reply, priority) in expectedOrder)
{
var current = q.Peek();
current.ShouldNotBeNull();
current!.Reply.ShouldBe(reply);
current.PriorityGroup.ShouldNotBeNull();
current.PriorityGroup!.Priority.ShouldBe(priority);
q.RemoveCurrent();
}
q.IsEmpty().ShouldBeTrue();
}
[Fact] // T:1365
public void WaitQueuePopAndRequeue_ShouldSucceed()
{
var q = new WaitQueue(max: 100);
q.AddPrioritized(new WaitingRequest { Reply = "1a", N = 2, PriorityGroup = new PriorityGroup { Priority = 1 } })
.ShouldBeTrue();
q.AddPrioritized(new WaitingRequest { Reply = "1b", N = 1, PriorityGroup = new PriorityGroup { Priority = 1 } })
.ShouldBeTrue();
q.AddPrioritized(new WaitingRequest { Reply = "2a", N = 3, PriorityGroup = new PriorityGroup { Priority = 2 } })
.ShouldBeTrue();
var wr = q.PopAndRequeue();
wr.ShouldNotBeNull();
wr!.Reply.ShouldBe("1a");
wr.N.ShouldBe(1);
q.Len.ShouldBe(3);
wr = q.PopAndRequeue();
wr.ShouldNotBeNull();
wr!.Reply.ShouldBe("1b");
wr.N.ShouldBe(0);
q.Len.ShouldBe(2);
wr = q.PopAndRequeue();
wr.ShouldNotBeNull();
wr!.Reply.ShouldBe("1a");
wr.N.ShouldBe(0);
q.Len.ShouldBe(1);
q.Peek()!.Reply.ShouldBe("2a");
q.Peek()!.N.ShouldBe(3);
}
}