lmxopcua/tests/ZB.MOM.WW.LmxOpcUa.Client.CLI.Tests/SubscribeCommandTests.cs

using Opc.Ua;
using Shouldly;
using Xunit;
using ZB.MOM.WW.LmxOpcUa.Client.CLI.Commands;
using ZB.MOM.WW.LmxOpcUa.Client.CLI.Tests.Fakes;

namespace ZB.MOM.WW.LmxOpcUa.Client.CLI.Tests;

public class SubscribeCommandTests
{
    [Fact]
    public async Task Execute_SubscribesWithCorrectParameters()
    {
        var fakeService = new FakeOpcUaClientService();
        var factory = new FakeOpcUaClientServiceFactory(fakeService);
        var command = new SubscribeCommand(factory)
        {
            Url = "opc.tcp://localhost:4840",
            NodeId = "ns=2;s=TestVar",
            Interval = 500
        };

        using var console = TestConsoleHelper.CreateConsole();

        // The subscribe command waits for cancellation. We need to cancel it.
        // Use the console's cancellation to trigger stop.
        var task = Task.Run(async () =>
        {
            await command.ExecuteAsync(console);
        });

        // Give it a moment to subscribe, then cancel
        await Task.Delay(100);
        console.RequestCancellation();
        await task;

        fakeService.SubscribeCalls.Count.ShouldBe(1);
        fakeService.SubscribeCalls[0].IntervalMs.ShouldBe(500);
        fakeService.SubscribeCalls[0].NodeId.Identifier.ShouldBe("TestVar");
    }

    [Fact]
    public async Task Execute_UnsubscribesOnCancellation()
    {
        var fakeService = new FakeOpcUaClientService();
        var factory = new FakeOpcUaClientServiceFactory(fakeService);
        var command = new SubscribeCommand(factory)
        {
            Url = "opc.tcp://localhost:4840",
            NodeId = "ns=2;s=TestVar"
        };

        using var console = TestConsoleHelper.CreateConsole();

        var task = Task.Run(async () =>
        {
            await command.ExecuteAsync(console);
        });

        await Task.Delay(100);
        console.RequestCancellation();
        await task;

        fakeService.UnsubscribeCalls.Count.ShouldBe(1);
    }

    [Fact]
    public async Task Execute_DisconnectsInFinally()
    {
        var fakeService = new FakeOpcUaClientService();
        var factory = new FakeOpcUaClientServiceFactory(fakeService);
        var command = new SubscribeCommand(factory)
        {
            Url = "opc.tcp://localhost:4840",
            NodeId = "ns=2;s=TestVar"
        };

        using var console = TestConsoleHelper.CreateConsole();

        var task = Task.Run(async () =>
        {
            await command.ExecuteAsync(console);
        });

        await Task.Delay(100);
        console.RequestCancellation();
        await task;

        fakeService.DisconnectCalled.ShouldBeTrue();
        fakeService.DisposeCalled.ShouldBeTrue();
    }

    [Fact]
    public async Task Execute_PrintsSubscriptionMessage()
    {
        var fakeService = new FakeOpcUaClientService();
        var factory = new FakeOpcUaClientServiceFactory(fakeService);
        var command = new SubscribeCommand(factory)
        {
            Url = "opc.tcp://localhost:4840",
            NodeId = "ns=2;s=TestVar",
            Interval = 2000
        };

        using var console = TestConsoleHelper.CreateConsole();

        var task = Task.Run(async () =>
        {
            await command.ExecuteAsync(console);
        });

        await Task.Delay(100);
        console.RequestCancellation();
        await task;

        var output = TestConsoleHelper.GetOutput(console);
        output.ShouldContain("Subscribed to ns=2;s=TestVar (interval: 2000ms)");
        output.ShouldContain("Unsubscribed.");
    }
}