MQTTnet/Source/MQTTnet.Tests/Server/Server_Reference_Tests.cs

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
// See the LICENSE file in the project root for more information.

using System.Threading.Tasks;
using Microsoft.VisualStudio.TestTools.UnitTesting;
using MQTTnet.Formatter;
using MQTTnet.Internal;
using MQTTnet.Protocol;

namespace MQTTnet.Tests.Server;

// ReSharper disable InconsistentNaming
[TestClass]
public sealed class Server_Reference_Tests : BaseTestClass
{
    [TestMethod]
    public async Task Server_Reports_With_Reference_Server()
    {
        using var testEnvironment = CreateTestEnvironment();
        testEnvironment.IgnoreClientLogErrors = true;

        var server = await testEnvironment.StartServer();

        server.ValidatingConnectionAsync += e =>
        {
            e.ReasonCode = MqttConnectReasonCode.ServerMoved;
            e.ServerReference = "new_server";
            return CompletedTask.Instance;
        };

        var client = testEnvironment.CreateClient();

        var response = await client.ConnectAsync(
            new MqttClientOptionsBuilder().WithProtocolVersion(MqttProtocolVersion.V500).WithTcpServer("127.0.0.1", testEnvironment.ServerPort).Build());

        Assert.AreEqual(MqttClientConnectResultCode.ServerMoved, response.ResultCode);
        Assert.AreEqual("new_server", response.ServerReference);
    }
}