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

feat(batch16): port outbound and parser-facing client core features

Browse Source
This commit is contained in:
Joseph Doherty
2026-02-28 18:58:52 -05:00
parent 46ea749ad0
commit a0b0b45f5e
4 changed files with 301 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

94
dotnet/tests/ZB.MOM.NatsNet.Server.Tests/ClientConnectionStubFeaturesTests.cs
View File

@@ -3,6 +3,7 @@
using System.Reflection;
using System.Text;
using System.Linq;
using Shouldly;
using ZB.MOM.NatsNet.Server;
using ZB.MOM.NatsNet.Server.Internal;
@@ -58,4 +59,97 @@ public sealed class ClientConnectionStubFeaturesTests
.GetField(field, BindingFlags.Instance | BindingFlags.NonPublic)!
.GetValue(c);
}
[Fact]
public void QueueOutbound_ChunkingAndPendingBytes_ShouldTrackState()
{
var c = new ClientConnection(ClientKind.Client)
{
OutMp = 100_000,
};
c.QueueOutbound(new byte[70_000]);
c.OutPb.ShouldBe(70_000);
c.OutNb.Count.ShouldBeGreaterThan(1);
c.OutNb.Sum(chunk => chunk.Count).ShouldBe(70_000);
}
[Fact]
public void FlushOutbound_WithoutServerOrConn_ShouldNoOpTrue()
{
var c = new ClientConnection(ClientKind.Client);
c.QueueOutbound(Encoding.ASCII.GetBytes("hello"));
c.FlushOutbound().ShouldBeTrue();
c.OutPb.ShouldBe(5);
}
[Fact]
public void HandleWriteTimeout_ClosePolicy_ShouldMarkClosed()
{
var c = new ClientConnection(ClientKind.Client)
{
OutWtp = WriteTimeoutPolicy.Close,
};
c.HandleWriteTimeout(0, 100, 1).ShouldBeTrue();
c.Flags.IsSet(ClientFlags.ConnMarkedClosed).ShouldBeTrue();
c.Flags.IsSet(ClientFlags.SkipFlushOnClose).ShouldBeTrue();
}
[Fact]
public void HandleWriteTimeout_RetryPolicy_ShouldSetSlowConsumerFlag()
{
var c = new ClientConnection(ClientKind.Client)
{
OutWtp = WriteTimeoutPolicy.Retry,
};
c.HandleWriteTimeout(1, 100, 2).ShouldBeFalse();
c.Flags.IsSet(ClientFlags.IsSlowConsumer).ShouldBeTrue();
}
[Fact]
public void ProcessPubAndHeaderPubWrappers_ShouldPopulateParseContext()
{
var c = new ClientConnection(ClientKind.Client)
{
Headers = true,
};
c.ProcessPub(Encoding.ASCII.GetBytes("foo 5")).ShouldBeNull();
Encoding.ASCII.GetString(c.ParseCtx.Pa.Subject!).ShouldBe("foo");
c.ParseCtx.Pa.Size.ShouldBe(5);
c.ProcessHeaderPub(Encoding.ASCII.GetBytes("foo 3 5"), null).ShouldBeNull();
Encoding.ASCII.GetString(c.ParseCtx.Pa.Subject!).ShouldBe("foo");
c.ParseCtx.Pa.HeaderSize.ShouldBe(3);
c.ParseCtx.Pa.Size.ShouldBe(5);
}
[Fact]
public void SplitArgParseSubAndProcessSub_ShouldCreateSubscriptions()
{
var tokens = ClientConnection.SplitArg(Encoding.ASCII.GetBytes("foo queue sid\r\n"));
tokens.Count.ShouldBe(3);
Encoding.ASCII.GetString(tokens[0]).ShouldBe("foo");
Encoding.ASCII.GetString(tokens[1]).ShouldBe("queue");
Encoding.ASCII.GetString(tokens[2]).ShouldBe("sid");
var c = new ClientConnection(ClientKind.Client);
c.ParseSub(Encoding.ASCII.GetBytes("foo queue sid"), noForward: true).ShouldBeNull();
c.Subs.Count.ShouldBe(1);
var result = c.ProcessSubEx(
Encoding.ASCII.GetBytes("bar"),
null,
Encoding.ASCII.GetBytes("sid2"),
noForward: false,
si: false,
rsi: false);
result.err.ShouldBeNull();
result.sub.ShouldNotBeNull();
c.Subs.Count.ShouldBe(2);
}
}