feat: add TLS mixed mode tests and monitoring TLS field verification

Add TlsMixedModeTests verifying that a server with AllowNonTls=true accepts both plaintext and TLS clients on the same port. Add MonitorTlsTests verifying that /connz reports TlsVersion and TlsCipherSuite for TLS-connected clients.
2026-02-22 22:40:03 -05:00
parent 19f35e6463
commit 3b6bd08248
2 changed files with 176 additions and 0 deletions
--- a/tests/NATS.Server.Tests/TlsServerTests.cs
+++ b/tests/NATS.Server.Tests/TlsServerTests.cs
@@ -135,3 +135,91 @@ public class TlsServerTests : IAsyncLifetime
        return ((IPEndPoint)sock.LocalEndPoint!).Port;
    }
 }
+
+public class TlsMixedModeTests : IAsyncLifetime
+{
+    private readonly NatsServer _server;
+    private readonly int _port;
+    private readonly CancellationTokenSource _cts = new();
+    private readonly string _certPath;
+    private readonly string _keyPath;
+
+    public TlsMixedModeTests()
+    {
+        _port = GetFreePort();
+        (_certPath, _keyPath) = TlsHelperTests.GenerateTestCertFiles();
+        _server = new NatsServer(
+            new NatsOptions
+            {
+                Port = _port,
+                TlsCert = _certPath,
+                TlsKey = _keyPath,
+                AllowNonTls = true,
+            },
+            NullLoggerFactory.Instance);
+    }
+
+    public async Task InitializeAsync()
+    {
+        _ = _server.StartAsync(_cts.Token);
+        await _server.WaitForReadyAsync();
+    }
+
+    public async Task DisposeAsync()
+    {
+        await _cts.CancelAsync();
+        _server.Dispose();
+        File.Delete(_certPath);
+        File.Delete(_keyPath);
+    }
+
+    [Fact]
+    public async Task Mixed_mode_accepts_plain_client()
+    {
+        using var sock = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
+        await sock.ConnectAsync(new IPEndPoint(IPAddress.Loopback, _port));
+        using var stream = new NetworkStream(sock);
+
+        var buf = new byte[4096];
+        var read = await stream.ReadAsync(buf);
+        var info = Encoding.ASCII.GetString(buf, 0, read);
+        info.ShouldContain("\"tls_available\":true");
+
+        await stream.WriteAsync("CONNECT {}\r\nPING\r\n"u8.ToArray());
+        await stream.FlushAsync();
+
+        var pongBuf = new byte[64];
+        read = await stream.ReadAsync(pongBuf);
+        var pong = Encoding.ASCII.GetString(pongBuf, 0, read);
+        pong.ShouldContain("PONG");
+    }
+
+    [Fact]
+    public async Task Mixed_mode_accepts_tls_client()
+    {
+        using var tcp = new TcpClient();
+        await tcp.ConnectAsync(IPAddress.Loopback, _port);
+        using var netStream = tcp.GetStream();
+
+        var buf = new byte[4096];
+        _ = await netStream.ReadAsync(buf); // Read INFO
+
+        using var ssl = new SslStream(netStream, false, (_, _, _, _) => true);
+        await ssl.AuthenticateAsClientAsync("localhost");
+
+        await ssl.WriteAsync("CONNECT {}\r\nPING\r\n"u8.ToArray());
+        await ssl.FlushAsync();
+
+        var pongBuf = new byte[64];
+        var read = await ssl.ReadAsync(pongBuf);
+        var pong = Encoding.ASCII.GetString(pongBuf, 0, read);
+        pong.ShouldContain("PONG");
+    }
+
+    private static int GetFreePort()
+    {
+        using var sock = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
+        sock.Bind(new IPEndPoint(IPAddress.Loopback, 0));
+        return ((IPEndPoint)sock.LocalEndPoint!).Port;
+    }
+}