task2: implement batch38 group A consumer lifecycle features
This commit is contained in:
@@ -127,4 +127,89 @@ public sealed class NatsConsumerTests
|
||||
q.Peek()!.Reply.ShouldBe("2a");
|
||||
q.Peek()!.N.ShouldBe(3);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void AddConsumerWithAction_CreateThenUpdate_ShouldRespectActions()
|
||||
{
|
||||
var account = new Account { Name = "A" };
|
||||
var streamCfg = new StreamConfig { Name = "S", Subjects = ["foo"], Retention = RetentionPolicy.LimitsPolicy };
|
||||
var stream = NatsStream.Create(account, streamCfg, null, null, null, null);
|
||||
stream.ShouldNotBeNull();
|
||||
|
||||
var cfg = new ConsumerConfig { Durable = "D", AckPolicy = AckPolicy.AckExplicit };
|
||||
var (created, createErr) = stream!.AddConsumerWithAction(cfg, "D", ConsumerAction.Create, pedantic: false);
|
||||
createErr.ShouldBeNull();
|
||||
created.ShouldNotBeNull();
|
||||
|
||||
var updateCfg = new ConsumerConfig { Durable = "D", AckPolicy = AckPolicy.AckAll };
|
||||
var (updated, updateErr) = stream.AddConsumerWithAction(updateCfg, "D", ConsumerAction.Update, pedantic: false);
|
||||
updateErr.ShouldBeNull();
|
||||
updated.ShouldNotBeNull();
|
||||
updated!.GetConfig().AckPolicy.ShouldBe(AckPolicy.AckAll);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void AddConsumer_WithAssignment_ShouldAttachAssignment()
|
||||
{
|
||||
var account = new Account { Name = "A" };
|
||||
var streamCfg = new StreamConfig { Name = "S", Subjects = ["foo"], Retention = RetentionPolicy.LimitsPolicy };
|
||||
var stream = NatsStream.Create(account, streamCfg, null, null, null, null);
|
||||
stream.ShouldNotBeNull();
|
||||
|
||||
var assignment = new ConsumerAssignment
|
||||
{
|
||||
Name = "D",
|
||||
Stream = "S",
|
||||
Group = new RaftGroup { Name = "RG", Peers = ["N1"] },
|
||||
};
|
||||
var cfg = new ConsumerConfig { Durable = "D", AckPolicy = AckPolicy.AckExplicit };
|
||||
|
||||
var (consumer, err) = stream!.AddConsumerWithAssignment(cfg, "D", assignment, isRecovering: false, ConsumerAction.Create, pedantic: false);
|
||||
err.ShouldBeNull();
|
||||
consumer.ShouldNotBeNull();
|
||||
consumer!.ConsumerAssignment().ShouldBeSameAs(assignment);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void UpdateInactiveThreshold_AndPauseState_ShouldTrackConfigValues()
|
||||
{
|
||||
var account = new Account { Name = "A" };
|
||||
var streamCfg = new StreamConfig { Name = "S", Subjects = ["foo"], Retention = RetentionPolicy.LimitsPolicy };
|
||||
var stream = NatsStream.Create(account, streamCfg, null, null, null, null);
|
||||
stream.ShouldNotBeNull();
|
||||
|
||||
var cfg = new ConsumerConfig { Durable = "D", AckPolicy = AckPolicy.AckExplicit };
|
||||
var consumer = NatsConsumer.Create(stream!, cfg, ConsumerAction.Create, null);
|
||||
consumer.ShouldNotBeNull();
|
||||
|
||||
consumer!.UpdateInactiveThreshold(new ConsumerConfig { InactiveThreshold = TimeSpan.FromSeconds(30) });
|
||||
consumer.GetConfig().InactiveThreshold.ShouldBe(TimeSpan.FromSeconds(30));
|
||||
|
||||
var pauseUntil = DateTime.UtcNow.AddMinutes(1);
|
||||
consumer.UpdatePauseState(new ConsumerConfig { PauseUntil = pauseUntil });
|
||||
consumer.GetConfig().PauseUntil.ShouldBe(pauseUntil);
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public void ConsumerAssignment_GetSet_ShouldRoundTrip()
|
||||
{
|
||||
var account = new Account { Name = "A" };
|
||||
var streamCfg = new StreamConfig { Name = "S", Subjects = ["foo"], Retention = RetentionPolicy.LimitsPolicy };
|
||||
var stream = NatsStream.Create(account, streamCfg, null, null, null, null);
|
||||
stream.ShouldNotBeNull();
|
||||
|
||||
var cfg = new ConsumerConfig { Durable = "D", AckPolicy = AckPolicy.AckExplicit };
|
||||
var consumer = NatsConsumer.Create(stream!, cfg, ConsumerAction.Create, null);
|
||||
consumer.ShouldNotBeNull();
|
||||
|
||||
var assignment = new ConsumerAssignment
|
||||
{
|
||||
Name = "D",
|
||||
Stream = "S",
|
||||
Group = new RaftGroup { Name = "RG", Peers = ["N1"] },
|
||||
};
|
||||
|
||||
consumer!.SetConsumerAssignment(assignment);
|
||||
consumer.ConsumerAssignment().ShouldBeSameAs(assignment);
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user