lmxopcua/tests/Drivers/ZB.MOM.WW.OtOpcUa.Driver.Historian.SqlServer.IntegrationTests/AlarmWriterTests.cs

using Microsoft.Data.SqlClient;
using Shouldly;
using Xunit;
using ZB.MOM.WW.OtOpcUa.Core.Abstractions;
using ZB.MOM.WW.OtOpcUa.Core.AlarmHistorian;

namespace ZB.MOM.WW.OtOpcUa.Driver.Historian.SqlServer.IntegrationTests;

[Collection(SqlHistorianCollection.Name)]
public sealed class AlarmWriterTests(SqlHistorianFixture fx)
{
    private static readonly DateTime T = new DateTime(2026, 10, 9, 1, 2, 3, DateTimeKind.Utc).AddTicks(1234567);

    private (SqlServerAlarmHistorianWriter W, string S) Make(string? connection = null, bool auto = true)
    {
        var s = SqlHistorianFixture.UniqueSchema();
        var o = new SqlServerHistorianOptions { ConnectionString = connection ?? fx.ConnectionString, Schema = s, AutoCreateSchema = auto };
        var c = new SqlHistorianConnection(o);
        return (new SqlServerAlarmHistorianWriter(c, new SqlServerSchemaBootstrapper(c, o)), s);
    }

    private static AlarmHistorianEvent Ev(string kind = "Activated", string type = "LimitAlarm", string? comment = "c") =>
        new("alarm-1", "Plant/Line/Eq", "HiTemp", type, AlarmSeverity.High, kind, "msg", "user", comment, T);

    [Fact]
    public async Task Event_round_trips_and_a_repeat_is_one_row_with_both_Ack()
    {
        fx.SkipIfUnavailable();
        var (w, s) = Make();
        var first = await w.WriteBatchAsync([Ev(), Ev()], CancellationToken.None);
        first.ShouldBe([HistorianWriteOutcome.Ack, HistorianWriteOutcome.Ack]);
        (await w.WriteBatchAsync([Ev()], CancellationToken.None)).ShouldBe([HistorianWriteOutcome.Ack]);
        (await w.WriteBatchAsync([Ev(kind: "Cleared")], CancellationToken.None)).ShouldBe([HistorianWriteOutcome.Ack]);

        await using var c = new SqlConnection(fx.ConnectionString);
        await c.OpenAsync();
        await using var cmd = c.CreateCommand();
        cmd.CommandText = $"SELECT SourceName, AlarmId, AlarmName, AlarmTypeName, Severity, EventKind, Message, UserName, Comment, EventTimeUtc, ReceivedTimeUtc FROM [{s}].Event WHERE EventKind='Activated'";
        await using var r = await cmd.ExecuteReaderAsync();
        (await r.ReadAsync()).ShouldBeTrue();
        r.GetString(0).ShouldBe("Plant/Line/Eq");
        r.GetString(1).ShouldBe("alarm-1");
        r.GetString(2).ShouldBe("HiTemp");
        r.GetString(3).ShouldBe("LimitAlarm");
        r.GetInt32(4).ShouldBe(700);
        r.GetString(6).ShouldBe("msg");
        r.GetString(7).ShouldBe("user");
        r.GetString(8).ShouldBe("c");
        SqlHistorianConnection.GetUtc(r, 9).Ticks.ShouldBe(T.Ticks);
        (DateTime.UtcNow - SqlHistorianConnection.GetUtc(r, 10)).ShouldBeLessThan(TimeSpan.FromMinutes(1));
        (await r.ReadAsync()).ShouldBeFalse();
        await r.CloseAsync();
        cmd.CommandText = $"SELECT COUNT(*) FROM [{s}].Event";
        Convert.ToInt32(await cmd.ExecuteScalarAsync()).ShouldBe(2);
    }

    [Fact]
    public async Task Over_long_AlarmTypeName_is_PermanentFail_and_siblings_still_Ack()
    {
        fx.SkipIfUnavailable();
        var (w, s) = Make();
        var r = await w.WriteBatchAsync([Ev(kind: "A"), Ev(kind: "B", type: new string('t', 101)), Ev(kind: "C", comment: null)], CancellationToken.None);
        r.ShouldBe([HistorianWriteOutcome.Ack, HistorianWriteOutcome.PermanentFail, HistorianWriteOutcome.Ack]);
    }

    [Fact]
    public async Task Concurrent_writers_of_the_same_event_all_Ack_with_one_row()
    {
        fx.SkipIfUnavailable();
        var (w, s) = Make();
        await w.WriteBatchAsync([Ev(kind: "warm")], CancellationToken.None);
        var rs = await Task.WhenAll(Enumerable.Range(0, 12).Select(_ => Task.Run(() => w.WriteBatchAsync([Ev()], CancellationToken.None))));
        rs.ShouldAllBe(x => x[0] == HistorianWriteOutcome.Ack);
        await using var c = new SqlConnection(fx.ConnectionString);
        await c.OpenAsync();
        await using var cmd = c.CreateCommand();
        cmd.CommandText = $"SELECT COUNT(*) FROM [{s}].Event WHERE EventKind='Activated'";
        Convert.ToInt32(await cmd.ExecuteScalarAsync()).ShouldBe(1);
    }

    [Fact]
    public async Task Unreachable_database_is_RetryPlease_not_an_exception_and_empty_batch_is_empty()
    {
        fx.SkipIfUnavailable();
        var (w, _) = Make(fx.ConnectionString + ";Connect Timeout=1", auto: false);   // schema missing
        (await w.WriteBatchAsync([Ev()], CancellationToken.None)).ShouldBe([HistorianWriteOutcome.RetryPlease]);
        (await w.WriteBatchAsync([], CancellationToken.None)).ShouldBeEmpty();
    }
}