refactor: rename remaining tests to NATS.Server.Core.Tests
- Rename tests/NATS.Server.Tests -> tests/NATS.Server.Core.Tests - Update solution file, InternalsVisibleTo, and csproj references - Remove JETSTREAM_INTEGRATION_MATRIX and NATS.NKeys from csproj (moved to JetStream.Tests and Auth.Tests) - Update all namespaces from NATS.Server.Tests.* to NATS.Server.Core.Tests.* - Replace private GetFreePort/ReadUntilAsync helpers with TestUtilities calls - Fix stale namespace in Transport.Tests/NetworkingGoParityTests.cs
This commit is contained in:
Joseph Doherty
177
tests/NATS.Server.Core.Tests/ClientPubSubTests.cs
Normal file
177
tests/NATS.Server.Core.Tests/ClientPubSubTests.cs
Normal file
@@ -0,0 +1,177 @@
|
||||
// Go reference: golang/nats-server/server/client_test.go
|
||||
// TestClientSimplePubSub (line 666), TestClientPubSubNoEcho (line 691),
|
||||
// TestClientSimplePubSubWithReply (line 712), TestClientNoBodyPubSubWithReply (line 740),
|
||||
// TestClientPubWithQueueSub (line 768)
|
||||
|
||||
using System.Net;
|
||||
using System.Net.Sockets;
|
||||
using System.Text;
|
||||
using System.Text.RegularExpressions;
|
||||
using Microsoft.Extensions.Logging.Abstractions;
|
||||
using NATS.Server;
|
||||
using NATS.Server.TestUtilities;
|
||||
|
||||
namespace NATS.Server.Core.Tests;
|
||||
|
||||
public class ClientPubSubTests : IAsyncLifetime
|
||||
{
|
||||
private readonly NatsServer _server;
|
||||
private readonly int _port;
|
||||
private readonly CancellationTokenSource _cts = new();
|
||||
|
||||
public ClientPubSubTests()
|
||||
{
|
||||
_port = TestPortAllocator.GetFreePort();
|
||||
_server = new NatsServer(new NatsOptions { Port = _port }, NullLoggerFactory.Instance);
|
||||
}
|
||||
|
||||
public async Task InitializeAsync()
|
||||
{
|
||||
_ = _server.StartAsync(_cts.Token);
|
||||
await _server.WaitForReadyAsync();
|
||||
}
|
||||
|
||||
public async Task DisposeAsync()
|
||||
{
|
||||
await _cts.CancelAsync();
|
||||
_server.Dispose();
|
||||
}
|
||||
|
||||
|
||||
private async Task<Socket> ConnectClientAsync()
|
||||
{
|
||||
var sock = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
|
||||
await sock.ConnectAsync(IPAddress.Loopback, _port);
|
||||
return sock;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Reads from a socket until the accumulated data contains the expected substring.
|
||||
/// </summary>
|
||||
|
||||
// Go reference: TestClientSimplePubSub (client_test.go line 666)
|
||||
// SUB foo 1, PUB foo 5\r\nhello — subscriber receives MSG foo 1 5\r\nhello
|
||||
[Fact]
|
||||
public async Task Simple_pub_sub_delivers_message()
|
||||
{
|
||||
using var client = await ConnectClientAsync();
|
||||
|
||||
// Read INFO
|
||||
var buf = new byte[4096];
|
||||
await client.ReceiveAsync(buf, SocketFlags.None);
|
||||
|
||||
// CONNECT, SUB, PUB, then PING to flush delivery
|
||||
await client.SendAsync(Encoding.ASCII.GetBytes(
|
||||
"CONNECT {}\r\nSUB foo 1\r\nPUB foo 5\r\nhello\r\nPING\r\n"));
|
||||
|
||||
// Read until we see the message payload (delivered before PONG)
|
||||
var response = await SocketTestHelper.ReadUntilAsync(client, "hello\r\n");
|
||||
|
||||
// MSG line: MSG foo 1 5\r\nhello\r\n
|
||||
response.ShouldContain("MSG foo 1 5\r\nhello\r\n");
|
||||
}
|
||||
|
||||
// Go reference: TestClientPubSubNoEcho (client_test.go line 691)
|
||||
// CONNECT {"echo":false} — publishing client does NOT receive its own messages
|
||||
[Fact]
|
||||
public async Task Pub_sub_no_echo_suppresses_own_messages()
|
||||
{
|
||||
using var client = await ConnectClientAsync();
|
||||
|
||||
// Read INFO
|
||||
var buf = new byte[4096];
|
||||
await client.ReceiveAsync(buf, SocketFlags.None);
|
||||
|
||||
// Connect with echo=false, then SUB+PUB on same connection, then PING
|
||||
await client.SendAsync(Encoding.ASCII.GetBytes(
|
||||
"CONNECT {\"echo\":false}\r\nSUB foo 1\r\nPUB foo 5\r\nhello\r\nPING\r\n"));
|
||||
|
||||
// With echo=false the server must not deliver the message back to the publisher.
|
||||
// The first line we receive should be PONG, not MSG.
|
||||
var response = await SocketTestHelper.ReadUntilAsync(client, "PONG\r\n");
|
||||
|
||||
response.ShouldStartWith("PONG\r\n");
|
||||
response.ShouldNotContain("MSG");
|
||||
}
|
||||
|
||||
// Go reference: TestClientSimplePubSubWithReply (client_test.go line 712)
|
||||
// PUB foo bar 5\r\nhello — subscriber receives MSG foo 1 bar 5\r\nhello (reply subject included)
|
||||
[Fact]
|
||||
public async Task Pub_sub_with_reply_subject()
|
||||
{
|
||||
using var client = await ConnectClientAsync();
|
||||
|
||||
// Read INFO
|
||||
var buf = new byte[4096];
|
||||
await client.ReceiveAsync(buf, SocketFlags.None);
|
||||
|
||||
// PUB with reply subject "bar"
|
||||
await client.SendAsync(Encoding.ASCII.GetBytes(
|
||||
"CONNECT {}\r\nSUB foo 1\r\nPUB foo bar 5\r\nhello\r\nPING\r\n"));
|
||||
|
||||
var response = await SocketTestHelper.ReadUntilAsync(client, "hello\r\n");
|
||||
|
||||
// MSG line must include the reply subject: MSG <subject> <sid> <reply> <#bytes>
|
||||
response.ShouldContain("MSG foo 1 bar 5\r\nhello\r\n");
|
||||
}
|
||||
|
||||
// Go reference: TestClientNoBodyPubSubWithReply (client_test.go line 740)
|
||||
// PUB foo bar 0\r\n\r\n — zero-byte payload with reply subject
|
||||
[Fact]
|
||||
public async Task Empty_body_pub_sub_with_reply()
|
||||
{
|
||||
using var client = await ConnectClientAsync();
|
||||
|
||||
// Read INFO
|
||||
var buf = new byte[4096];
|
||||
await client.ReceiveAsync(buf, SocketFlags.None);
|
||||
|
||||
// PUB with reply subject and zero-length body
|
||||
await client.SendAsync(Encoding.ASCII.GetBytes(
|
||||
"CONNECT {}\r\nSUB foo 1\r\nPUB foo bar 0\r\n\r\nPING\r\n"));
|
||||
|
||||
// Read until PONG — MSG should arrive before PONG
|
||||
var response = await SocketTestHelper.ReadUntilAsync(client, "PONG\r\n");
|
||||
|
||||
// MSG line: MSG foo 1 bar 0\r\n\r\n (empty body, still CRLF terminated)
|
||||
response.ShouldContain("MSG foo 1 bar 0\r\n");
|
||||
}
|
||||
|
||||
// Go reference: TestClientPubWithQueueSub (client_test.go line 768)
|
||||
// Two queue subscribers in the same group on one connection — 100 publishes
|
||||
// distributed across both sids, each receiving at least 20 messages.
|
||||
[Fact]
|
||||
public async Task Queue_sub_distributes_messages()
|
||||
{
|
||||
const int num = 100;
|
||||
|
||||
using var client = await ConnectClientAsync();
|
||||
|
||||
// Read INFO
|
||||
var buf = new byte[4096];
|
||||
await client.ReceiveAsync(buf, SocketFlags.None);
|
||||
|
||||
// CONNECT, two queue subs with different sids, PING to confirm
|
||||
await client.SendAsync(Encoding.ASCII.GetBytes(
|
||||
"CONNECT {}\r\nSUB foo g1 1\r\nSUB foo g1 2\r\nPING\r\n"));
|
||||
await SocketTestHelper.ReadUntilAsync(client, "PONG\r\n");
|
||||
|
||||
// Publish 100 messages, then PING to flush all deliveries
|
||||
var pubSb = new StringBuilder();
|
||||
for (int i = 0; i < num; i++)
|
||||
pubSb.Append("PUB foo 5\r\nhello\r\n");
|
||||
pubSb.Append("PING\r\n");
|
||||
await client.SendAsync(Encoding.ASCII.GetBytes(pubSb.ToString()));
|
||||
|
||||
// Read until PONG — all MSGs arrive before the PONG
|
||||
var response = await SocketTestHelper.ReadUntilAsync(client, "PONG\r\n");
|
||||
|
||||
// Count deliveries per sid
|
||||
var n1 = Regex.Matches(response, @"MSG foo 1 5").Count;
|
||||
var n2 = Regex.Matches(response, @"MSG foo 2 5").Count;
|
||||
|
||||
(n1 + n2).ShouldBe(num);
|
||||
n1.ShouldBeGreaterThanOrEqualTo(20);
|
||||
n2.ShouldBeGreaterThanOrEqualTo(20);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user