lmxopcua/tests/ZB.MOM.WW.LmxOpcUa.Tests/MxAccess/MxAccessClientMonitorTests.cs

using System;
using System.Threading.Tasks;
using Shouldly;
using Xunit;
using ZB.MOM.WW.LmxOpcUa.Host.Configuration;
using ZB.MOM.WW.LmxOpcUa.Host.Domain;
using ZB.MOM.WW.LmxOpcUa.Host.Metrics;
using ZB.MOM.WW.LmxOpcUa.Host.MxAccess;
using ZB.MOM.WW.LmxOpcUa.Tests.Helpers;

namespace ZB.MOM.WW.LmxOpcUa.Tests.MxAccess
{
    public class MxAccessClientMonitorTests : IDisposable
    {
        private readonly StaComThread _staThread;
        private readonly FakeMxProxy _proxy;
        private readonly PerformanceMetrics _metrics;

        public MxAccessClientMonitorTests()
        {
            _staThread = new StaComThread();
            _staThread.Start();
            _proxy = new FakeMxProxy();
            _metrics = new PerformanceMetrics();
        }

        public void Dispose()
        {
            _staThread.Dispose();
            _metrics.Dispose();
        }

        [Fact]
        public async Task Monitor_ReconnectsOnDisconnect()
        {
            var config = new MxAccessConfiguration
            {
                MonitorIntervalSeconds = 1,
                AutoReconnect = true
            };
            var client = new MxAccessClient(_staThread, _proxy, config, _metrics);

            await client.ConnectAsync();
            await client.DisconnectAsync();

            client.StartMonitor();

            // Wait for monitor to detect disconnect and reconnect
            await Task.Delay(2500);

            client.StopMonitor();
            client.State.ShouldBe(ConnectionState.Connected);
            client.ReconnectCount.ShouldBeGreaterThan(0);
            client.Dispose();
        }

        [Fact]
        public async Task Monitor_StopsOnCancel()
        {
            var config = new MxAccessConfiguration { MonitorIntervalSeconds = 1 };
            var client = new MxAccessClient(_staThread, _proxy, config, _metrics);

            await client.ConnectAsync();
            client.StartMonitor();
            client.StopMonitor();

            // Should not throw
            await Task.Delay(200);
            client.Dispose();
        }
    }
}