dohertj2/natsnet
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

task4: implement batch38 group C ack and delivery state

Browse Source
This commit is contained in:
Joseph Doherty
2026-03-01 00:30:22 -05:00
parent 804bc89246
commit bb9558e6a9
8 changed files with 741 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

145
dotnet/tests/ZB.MOM.NatsNet.Server.Tests/JetStream/NatsConsumerTests.cs
View File

@@ -296,4 +296,149 @@ public sealed class NatsConsumerTests
consumer.HandleClusterConsumerInfoRequest().ShouldNotBeNull();
consumer.ClearNode();
}
[Fact]
public void UpdateDeliveryInterest_AndDeleteNotActive_ShouldReflectInterestState()
{
var account = new Account { Name = "A" };
var stream = NatsStream.Create(account, new StreamConfig { Name = "S", Subjects = ["foo"] }, null, null, null, null);
stream.ShouldNotBeNull();
var consumer = NatsConsumer.Create(
stream!,
new ConsumerConfig { Durable = "D", DeliverSubject = "deliver.foo", InactiveThreshold = TimeSpan.FromMilliseconds(20) },
ConsumerAction.Create,
null);
consumer.ShouldNotBeNull();
consumer!.UpdateInactiveThreshold(new ConsumerConfig { InactiveThreshold = TimeSpan.FromMilliseconds(20) });
consumer.UpdateDeliveryInterest(localInterest: false).ShouldBeTrue();
Thread.Sleep(40);
consumer.IsClosed().ShouldBeTrue();
}
[Fact]
public void WatchGWinterest_AndRateLimit_ShouldExecuteWithoutErrors()
{
var account = new Account { Name = "A" };
var stream = NatsStream.Create(account, new StreamConfig { Name = "S", Subjects = ["foo"], MaxMsgSize = 4096 }, null, null, null, null);
stream.ShouldNotBeNull();
var consumer = NatsConsumer.Create(
stream!,
new ConsumerConfig { Durable = "D", DeliverSubject = "deliver.foo", RateLimit = 8_000 },
ConsumerAction.Create,
null);
consumer.ShouldNotBeNull();
consumer!.SetRateLimitNeedsLocks();
consumer.WatchGWinterest();
consumer.SubscribeInternal("deliver.foo").ShouldBeTrue();
consumer.UpdateDeliveryInterest(localInterest: true).ShouldBeFalse();
consumer.HasDeliveryInterest().ShouldBeTrue();
}
[Fact]
public void AccountCheckNewConsumerConfig_InvalidPolicyChanges_ShouldFail()
{
var account = new Account { Name = "A" };
var current = new ConsumerConfig { Durable = "D", AckPolicy = AckPolicy.AckExplicit, DeliverPolicy = DeliverPolicy.DeliverAll };
var next = new ConsumerConfig { Durable = "D", AckPolicy = AckPolicy.AckAll, DeliverPolicy = DeliverPolicy.DeliverAll };
var err = account.CheckNewConsumerConfig(current, next);
err.ShouldNotBeNull();
err.Message.ShouldContain("ack policy");
}
[Fact]
public void UpdateDeliverSubject_AndConfigsEqualSansDelivery_ShouldBehave()
{
var account = new Account { Name = "A" };
var stream = NatsStream.Create(account, new StreamConfig { Name = "S", Subjects = ["foo"] }, null, null, null, null);
stream.ShouldNotBeNull();
var cfg = new ConsumerConfig { Durable = "D", DeliverSubject = "deliver.a", AckPolicy = AckPolicy.AckExplicit };
var consumer = NatsConsumer.Create(stream!, cfg, ConsumerAction.Create, null);
consumer.ShouldNotBeNull();
consumer!.SubscribeInternal("deliver.a");
consumer.UpdateDeliverSubject("deliver.b").ShouldBeTrue();
consumer.SubscribeInternal("deliver.b");
consumer.HasDeliveryInterest().ShouldBeTrue();
var left = new ConsumerConfig { Durable = "D", DeliverSubject = "deliver.a", AckPolicy = AckPolicy.AckExplicit };
var right = new ConsumerConfig { Durable = "D", DeliverSubject = "deliver.b", AckPolicy = AckPolicy.AckExplicit };
NatsConsumer.ConfigsEqualSansDelivery(left, right).ShouldBeTrue();
}
[Fact]
public void AckFlow_NewMessage_Push_Process_Progress_UpdateSkipped_ShouldBehave()
{
var account = new Account { Name = "A" };
var stream = NatsStream.Create(account, new StreamConfig { Name = "S", Subjects = ["foo"] }, null, null, null, null);
stream.ShouldNotBeNull();
var consumer = NatsConsumer.Create(stream!, new ConsumerConfig { Durable = "D", AckPolicy = AckPolicy.AckExplicit }, ConsumerAction.Create, null);
consumer.ShouldNotBeNull();
var ack = NatsConsumer.NewJSAckMsg("a.b.10.20.1", "reply", 0, "+ACK"u8.ToArray());
ack.Subject.ShouldBe("a.b.10.20.1");
consumer!.PushAck("a.b.10.20.1", "reply", 0, "+ACK"u8.ToArray());
consumer.ProcessAck("a.b.10.20.1", "reply", 0, "+ACK"u8.ToArray());
consumer.ProgressUpdate(10);
consumer.UpdateSkipped(25);
var state = consumer.GetConsumerState();
state.AckFloor.Stream.ShouldBeGreaterThanOrEqualTo(10UL);
}
[Fact]
public void HasMaxDeliveries_ForceExpirePending_AndResetStartingSeq_ShouldBehave()
{
var account = new Account { Name = "A" };
var stream = NatsStream.Create(account, new StreamConfig { Name = "S", Subjects = ["foo"] }, null, null, null, null);
stream.ShouldNotBeNull();
var consumer = NatsConsumer.Create(
stream!,
new ConsumerConfig { Durable = "D", AckPolicy = AckPolicy.AckExplicit, MaxDeliver = 2, DeliverPolicy = DeliverPolicy.DeliverAll },
ConsumerAction.Create,
null);
consumer.ShouldNotBeNull();
consumer!.ProcessAck("a.b.5.7.1", "reply", 0, "-NAK"u8.ToArray());
consumer.HasMaxDeliveries(5).ShouldBeFalse();
consumer.HasMaxDeliveries(5).ShouldBeTrue();
consumer.ForceExpirePending();
var (seq, canRespond, err) = consumer.ResetStartingSeq(10, "reply");
err.ShouldBeNull();
seq.ShouldBe(10UL);
canRespond.ShouldBeTrue();
}
[Fact]
public void JsPubMsg_ReturnToPool_ShouldResetState()
{
var msg = new JsPubMsg
{
Subject = "foo",
Reply = "bar",
Hdr = [1, 2],
Msg = [3, 4],
Pa = new object(),
Sync = new object(),
};
msg.ReturnToPool();
msg.Subject.ShouldBe(string.Empty);
msg.Reply.ShouldBeNull();
msg.Hdr.ShouldBeNull();
msg.Msg.ShouldBeNull();
msg.Pa.ShouldBeNull();
msg.Sync.ShouldBeNull();
}
}