7468401bd0
Add WriteTimeoutPolicy enum, FlushResult record struct, and GetWriteTimeoutPolicy static method as nested types in NatsClient. Models Go's client.go per-kind timeout handling: CLIENT kind closes on timeout, ROUTER/GATEWAY/LEAF use TCP-level flush recovery.
62 lines
2.5 KiB
C#
62 lines
2.5 KiB
C#
namespace NATS.Server.Tests;
|
|
|
|
// Go reference: server/client.go (write timeout handling, per-kind policies)
|
|
|
|
public class WriteTimeoutTests
|
|
{
|
|
[Fact]
|
|
public void WriteTimeoutPolicy_defaults_by_kind()
|
|
{
|
|
// Go reference: server/client.go — CLIENT closes, others use TCP flush
|
|
NatsClient.GetWriteTimeoutPolicy(ClientKind.Client).ShouldBe(NatsClient.WriteTimeoutPolicy.Close);
|
|
NatsClient.GetWriteTimeoutPolicy(ClientKind.Router).ShouldBe(NatsClient.WriteTimeoutPolicy.TcpFlush);
|
|
NatsClient.GetWriteTimeoutPolicy(ClientKind.Gateway).ShouldBe(NatsClient.WriteTimeoutPolicy.TcpFlush);
|
|
NatsClient.GetWriteTimeoutPolicy(ClientKind.Leaf).ShouldBe(NatsClient.WriteTimeoutPolicy.TcpFlush);
|
|
}
|
|
|
|
[Fact]
|
|
public void PartialFlushResult_tracks_bytes()
|
|
{
|
|
// Go reference: server/client.go — partial write tracking
|
|
var result = new NatsClient.FlushResult(BytesAttempted: 1024, BytesWritten: 512);
|
|
result.IsPartial.ShouldBeTrue();
|
|
result.BytesRemaining.ShouldBe(512L);
|
|
}
|
|
|
|
[Fact]
|
|
public void PartialFlushResult_complete_is_not_partial()
|
|
{
|
|
// Go reference: server/client.go — complete write
|
|
var result = new NatsClient.FlushResult(BytesAttempted: 1024, BytesWritten: 1024);
|
|
result.IsPartial.ShouldBeFalse();
|
|
result.BytesRemaining.ShouldBe(0L);
|
|
}
|
|
|
|
[Fact]
|
|
public void WriteTimeoutPolicy_system_kind_defaults_to_close()
|
|
{
|
|
// Go reference: server/client.go — system/internal kinds default to Close
|
|
NatsClient.GetWriteTimeoutPolicy(ClientKind.System).ShouldBe(NatsClient.WriteTimeoutPolicy.Close);
|
|
NatsClient.GetWriteTimeoutPolicy(ClientKind.JetStream).ShouldBe(NatsClient.WriteTimeoutPolicy.Close);
|
|
NatsClient.GetWriteTimeoutPolicy(ClientKind.Account).ShouldBe(NatsClient.WriteTimeoutPolicy.Close);
|
|
}
|
|
|
|
[Fact]
|
|
public void FlushResult_zero_bytes_is_partial_when_attempted_nonzero()
|
|
{
|
|
// Edge case: nothing written but something attempted
|
|
var result = new NatsClient.FlushResult(BytesAttempted: 100, BytesWritten: 0);
|
|
result.IsPartial.ShouldBeTrue();
|
|
result.BytesRemaining.ShouldBe(100L);
|
|
}
|
|
|
|
[Fact]
|
|
public void FlushResult_zero_zero_is_not_partial()
|
|
{
|
|
// Edge case: nothing attempted, nothing written
|
|
var result = new NatsClient.FlushResult(BytesAttempted: 0, BytesWritten: 0);
|
|
result.IsPartial.ShouldBeFalse();
|
|
result.BytesRemaining.ShouldBe(0L);
|
|
}
|
|
}
|