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

test(batch38-t3): add cluster lifecycle consistency mapped tests

Browse Source
This commit is contained in:
Joseph Doherty
2026-03-01 00:52:32 -05:00
parent 845f8695ad
commit 77004b0dd6
4 changed files with 118 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
dotnet/tests/ZB.MOM.NatsNet.Server.Tests/ImplBacklog/JetStreamClusterTests1.Impltests.cs
View File

@@ -188,4 +188,50 @@ public sealed class JetStreamClusterTests1
cluster.IsConsumerLeader("A", "S", "C1").ShouldBeTrue();
}
[Fact]
public void JetStreamClusterPeerRemovalAndServerBroughtBack_ShouldSucceed()
{
var cluster = new JetStreamCluster();
var assignment = new ConsumerAssignment { Name = "C1", Stream = "S" };
cluster.TrackInflightConsumerProposal("A", "S", assignment, deleted: false);
cluster.RemoveInflightConsumerProposal("A", "S", "C1");
cluster.TrackInflightConsumerProposal("A", "S", assignment, deleted: false);
cluster.InflightConsumers["A"]["S"].ContainsKey("C1").ShouldBeTrue();
}
[Fact]
public void JetStreamClusterUpgradeConsumerVersioning_ShouldSucceed()
{
var cfg = new ConsumerConfig
{
Durable = "D",
Metadata = new Dictionary<string, string> { ["legacy"] = "true" },
PriorityPolicy = PriorityPolicy.PriorityPinnedClient,
PriorityGroups = ["g1"],
};
JetStreamVersioning.SetStaticConsumerMetadata(cfg);
var upgraded = JetStreamVersioning.SetDynamicConsumerMetadata(cfg);
upgraded.Metadata.ShouldNotBeNull();
upgraded.Metadata.ShouldContainKey(JetStreamVersioning.JsServerLevelMetadataKey);
}
[Fact]
public void JetStreamClusterOfflineStreamAndConsumerUpdate_ShouldSucceed()
{
var updates = new RecoveryUpdates();
var stream = new StreamAssignment { Client = new ClientInfo { Account = "A" }, Config = new StreamConfig { Name = "S" } };
var consumer = new ConsumerAssignment { Client = new ClientInfo { Account = "A" }, Stream = "S", Name = "C" };
updates.AddStream(stream);
updates.AddOrUpdateConsumer(consumer);
updates.AddStreams.ShouldContainKey("A:S");
updates.UpdateConsumers.ShouldContainKey("A:S");
updates.UpdateConsumers["A:S"].ShouldContainKey("S:C");
}
}

40
dotnet/tests/ZB.MOM.NatsNet.Server.Tests/ImplBacklog/JetStreamClusterTests3.Impltests.cs
View File

@@ -109,4 +109,44 @@ public sealed class JetStreamClusterTests3
assignment.ShouldNotBeNull();
assignment!.MissingPeers().ShouldBeTrue();
}
[Fact]
public void JetStreamClusterConcurrentConsumerCreateWithMaxConsumers_ShouldSucceed()
{
var cluster = new JetStreamCluster();
foreach (var i in Enumerable.Range(0, 64))
{
cluster.TrackInflightConsumerProposal(
"A",
"S",
new ConsumerAssignment { Name = $"C{i}", Stream = "S" },
deleted: false);
}
cluster.InflightConsumers["A"]["S"].Count.ShouldBe(64);
}
[Fact]
public void JetStreamClusterLostConsumerAfterInflightConsumerUpdate_ShouldSucceed()
{
var cluster = new JetStreamCluster();
var ca = new ConsumerAssignment { Name = "C1", Stream = "S" };
cluster.TrackInflightConsumerProposal("A", "S", ca, deleted: false);
cluster.TrackInflightConsumerProposal("A", "S", ca, deleted: true);
cluster.InflightConsumers["A"]["S"]["C1"].Deleted.ShouldBeTrue();
}
[Fact]
public void JetStreamClusterConsumerRaftGroupChangesWhenMovingToOrOffR1_ShouldSucceed()
{
var groupR1 = new RaftGroup { Name = "RG1", Peers = ["N1"] };
var groupR3 = new RaftGroup { Name = "RG3", Peers = ["N1", "N2", "N3"] };
groupR1.IsMember("N1").ShouldBeTrue();
groupR3.IsMember("N3").ShouldBeTrue();
groupR1.Peers.Length.ShouldBe(1);
groupR3.Peers.Length.ShouldBe(3);
}
}

32
dotnet/tests/ZB.MOM.NatsNet.Server.Tests/ImplBacklog/JetStreamClusterTests4.Impltests.cs
View File

@@ -99,4 +99,36 @@ public sealed class JetStreamClusterTests4
updates.RemoveConsumers.ShouldContainKey("A:S");
}
[Fact]
public void JetStreamClusterMetaSnapshotReCreateConsistency_ShouldSucceed()
{
var updates = new RecoveryUpdates();
var stream = new StreamAssignment
{
Client = new ClientInfo { Account = "A" },
Config = new StreamConfig { Name = "S", Subjects = ["foo"] },
};
updates.AddStream(stream);
updates.RemoveStream(stream);
updates.AddStream(stream);
updates.AddStreams.ShouldContainKey("A:S");
updates.RemoveStreams.ShouldContainKey("A:S");
}
[Fact]
public void JetStreamClusterMetaSnapshotConsumerDeleteConsistency_ShouldSucceed()
{
var cluster = new JetStreamCluster();
var consumer = new ConsumerAssignment { Name = "C1", Stream = "S" };
cluster.TrackInflightConsumerProposal("A", "S", consumer, deleted: false);
cluster.TrackInflightConsumerProposal("A", "S", consumer, deleted: true);
cluster.RemoveInflightConsumerProposal("A", "S", "C1");
cluster.RemoveInflightConsumerProposal("A", "S", "C1");
cluster.InflightConsumers.ContainsKey("A").ShouldBeFalse();
}
}