82 lines
3.0 KiB
C#
82 lines
3.0 KiB
C#
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);
|
|
}
|
|
}
|