task5: implement batch38 group D proposal and ack flow

2026-03-01 00:35:46 -05:00
parent bb9558e6a9
commit 9fbcafb2a3
5 changed files with 391 additions and 0 deletions
--- a/dotnet/tests/ZB.MOM.NatsNet.Server.Tests/JetStream/ConsumerStateTests.cs
+++ b/dotnet/tests/ZB.MOM.NatsNet.Server.Tests/JetStream/ConsumerStateTests.cs
@@ -0,0 +1,81 @@
+using Shouldly;
+using ZB.MOM.NatsNet.Server;
+
+namespace ZB.MOM.NatsNet.Server.Tests.JetStream;
+
+public sealed class ConsumerStateTests
+{
+    private static NatsConsumer CreateConsumer()
+    {
+        var account = new Account { Name = "A" };
+        var stream = NatsStream.Create(account, new StreamConfig { Name = "S", Subjects = ["foo"] }, null, null, null, null)!;
+        return NatsConsumer.Create(stream, new ConsumerConfig { Durable = "D", AckPolicy = AckPolicy.AckExplicit }, ConsumerAction.Create, null)!;
+    }
+
+    [Fact]
+    public void ProposalAndPendingRequestFlow_ShouldBehave()
+    {
+        var consumer = CreateConsumer();
+
+        consumer.Propose([1, 2, 3]);
+        consumer.Propose([4]);
+        consumer.LoopAndForwardProposals().ShouldBe(2);
+
+        consumer.AddClusterPendingRequest("r1");
+        consumer.AddClusterPendingRequest("r2");
+        consumer.CheckPendingRequests(TimeSpan.FromMinutes(1)).ShouldBe(2);
+        consumer.RemoveClusterPendingRequest("r2");
+        consumer.CheckPendingRequests(TimeSpan.FromMinutes(1)).ShouldBe(1);
+
+        consumer.SetPendingRequestsOk(false);
+        consumer.PendingRequestsOk().ShouldBeFalse();
+        consumer.CheckAndSetPendingRequestsOk(true).ShouldBeFalse();
+        consumer.PendingRequestsOk().ShouldBeTrue();
+        consumer.ReleaseAnyPendingRequests().ShouldBe(1);
+    }
+
+    [Fact]
+    public void DeliveredAckReplyAndAcks_ShouldBehave()
+    {
+        var consumer = CreateConsumer();
+
+        consumer.UpdateDelivered(10, 20, 2, DateTimeOffset.UtcNow.ToUnixTimeMilliseconds());
+        consumer.AddAckReply(20, "reply");
+        consumer.UpdateAcks().ShouldBe(1);
+
+        var state = consumer.ReadStoredState();
+        state.Delivered.Consumer.ShouldBeGreaterThanOrEqualTo(10UL);
+        state.Delivered.Stream.ShouldBeGreaterThanOrEqualTo(20UL);
+        state.Redelivered.ShouldNotBeNull();
+        state.Redelivered!.ShouldContainKey(20UL);
+    }
+
+    [Fact]
+    public void ReplicatedQueueAndNakTermFlow_ShouldBehave()
+    {
+        var consumer = CreateConsumer();
+
+        consumer.AddReplicatedQueuedMsg(33, new JsPubMsg { Subject = "foo" });
+        consumer.ProcessNak(33, 2, 1, "-NAK"u8.ToArray()).ShouldBeTrue();
+        consumer.CheckRedelivered(33).ShouldBeTrue();
+        consumer.ProcessNak(33, 2, 2, "-NAK"u8.ToArray()).ShouldBeTrue();
+        consumer.CheckRedelivered(33).ShouldBeTrue();
+
+        consumer.ProcessTerm(33, 2, 2, "done", "reply").ShouldBeTrue();
+        consumer.AckWait(TimeSpan.Zero).ShouldBe(TimeSpan.FromSeconds(30));
+        consumer.AckWait(TimeSpan.FromSeconds(5)).ShouldBe(TimeSpan.FromSeconds(5));
+    }
+
+    [Fact]
+    public void ResetLocalStartingSeq_ShouldResetState()
+    {
+        var consumer = CreateConsumer();
+
+        consumer.UpdateDelivered(1, 1, 1, DateTimeOffset.UtcNow.ToUnixTimeMilliseconds());
+        consumer.ResetLocalStartingSeq(100);
+
+        var state = consumer.GetConsumerState();
+        state.Delivered.Stream.ShouldBe(100UL);
+        state.AckFloor.Stream.ShouldBe(99UL);
+    }
+}