scadaproj/ZB.MOM.WW.Overview/tests/ZB.MOM.WW.Overview.Tests/LeaderResolverTests.cs

using ZB.MOM.WW.Overview.Polling;

namespace ZB.MOM.WW.Overview.Tests;

/// <summary>
/// Leader aggregation and the split-brain flag. The disagreement rule is the one output an operator
/// would be woken by, so both directions are pinned: it must fire when two members genuinely claim
/// different leaders, and it must NOT fire for the ordinary cases that superficially resemble that
/// — a node that stopped answering, a node running a pre-0.2.0 Health package.
/// </summary>
public class LeaderResolverTests
{
    private const string LeaderA = "akka.tcp://scadabridge@site-a-a:8082";
    private const string LeaderB = "akka.tcp://scadabridge@site-a-b:8082";

    [Fact]
    public void Resolve_AgreeingPair_NamesTheLeaderInstance()
    {
        var groups = LeaderResolver.Resolve(
        [
            Snapshots.Instance("site-a-a", "site-a", "http://site-a-a:8084", leader: LeaderA),
            Snapshots.Instance("site-a-b", "site-a", "http://site-a-b:8084", leader: LeaderA),
        ]);

        var group = Assert.Single(groups);
        Assert.Equal("site-a", group.Name);
        Assert.Equal("site-a-a", group.LeaderInstance);
        Assert.Equal(LeaderA, group.LeaderAddress);
        Assert.False(group.Disagreement);
    }

    [Fact]
    public void Resolve_MembersClaimDifferentLeaders_FlagsDisagreement()
    {
        var groups = LeaderResolver.Resolve(
        [
            Snapshots.Instance("site-a-a", "site-a", "http://site-a-a:8084", leader: LeaderA),
            Snapshots.Instance("site-a-b", "site-a", "http://site-a-b:8084", leader: LeaderB),
        ]);

        var group = Assert.Single(groups);
        Assert.True(group.Disagreement);
        Assert.Equal([LeaderA, LeaderB], group.ReportedLeaders);
    }

    [Theory]
    [InlineData(InstanceStatus.Down)]
    [InlineData(InstanceStatus.Unreachable)]
    public void Resolve_FailedMemberDoesNotVote_SoAFailoverIsNotMisreadAsSplitBrain(InstanceStatus failed)
    {
        // Mid-failover the departed node's last body still names the old leader. Counting it would
        // raise a split-brain warning during the most normal event in a redundant pair's life.
        var groups = LeaderResolver.Resolve(
        [
            Snapshots.Instance("site-a-a", "site-a", "http://site-a-a:8084", status: failed, leader: LeaderA),
            Snapshots.Instance("site-a-b", "site-a", "http://site-a-b:8084", leader: LeaderB),
        ]);

        var group = Assert.Single(groups);
        Assert.False(group.Disagreement);
        Assert.Equal("site-a-b", group.LeaderInstance);
    }

    [Fact]
    public void Resolve_DegradedMemberStillVotes()
    {
        // Degraded means answering with a full cluster view — its opinion on the leader is current.
        var groups = LeaderResolver.Resolve(
        [
            Snapshots.Instance("site-a-a", "site-a", "http://site-a-a:8084", status: InstanceStatus.Degraded, leader: LeaderA),
            Snapshots.Instance("site-a-b", "site-a", "http://site-a-b:8084", leader: LeaderB),
        ]);

        Assert.True(Assert.Single(groups).Disagreement);
    }

    [Fact]
    public void Resolve_NoMemberPublishesLeaderData_YieldsNoLeaderAndNoWarning()
    {
        // The whole-fleet-on-Health-0.1.0 case: no data key anywhere. It must degrade to "leader
        // unknown", never to a false split-brain warning.
        var groups = LeaderResolver.Resolve(
        [
            Snapshots.Instance("site-a-a", "site-a", "http://site-a-a:8084"),
            Snapshots.Instance("site-a-b", "site-a", "http://site-a-b:8084"),
        ]);

        var group = Assert.Single(groups);
        Assert.Null(group.LeaderAddress);
        Assert.Null(group.LeaderInstance);
        Assert.False(group.Disagreement);
        Assert.Empty(group.ReportedLeaders);
    }

    [Fact]
    public void Resolve_OneMemberOnOldHealthPackage_UsesTheOneThatDoesReport()
    {
        // A partly-bumped fleet is the expected state during rollout, and one silent member is not
        // a disagreement.
        var groups = LeaderResolver.Resolve(
        [
            Snapshots.Instance("site-a-a", "site-a", "http://site-a-a:8084", leader: LeaderA),
            Snapshots.Instance("site-a-b", "site-a", "http://site-a-b:8084"),
        ]);

        var group = Assert.Single(groups);
        Assert.Equal("site-a-a", group.LeaderInstance);
        Assert.False(group.Disagreement);
    }

    [Fact]
    public void Resolve_UnresolvableAddress_KeepsTheRawAddress()
    {
        // A leader on a host that is not in the registry (a node nobody added) still tells the
        // operator something useful — showing nothing would hide the discovery.
        var groups = LeaderResolver.Resolve(
        [
            Snapshots.Instance("site-a-a", "site-a", "http://site-a-a:8084", leader: "akka.tcp://scadabridge@ghost:8082"),
        ]);

        var group = Assert.Single(groups);
        Assert.Null(group.LeaderInstance);
        Assert.Equal("akka.tcp://scadabridge@ghost:8082", group.LeaderAddress);
    }

    [Fact]
    public void Resolve_MatchesOnHostOnly_BecauseAkkaAndHttpPortsDiffer()
    {
        // The registry knows the HTTP port (8084); the leader address carries the Akka remoting
        // port (8082). Matching on authority instead of host would never resolve anything.
        var name = LeaderResolver.ResolveToInstanceName(
            LeaderA,
            [Snapshots.Instance("site-a-a", "site-a", "http://site-a-a:8084")]);

        Assert.Equal("site-a-a", name);
    }

    [Fact]
    public void Resolve_GroupWithoutActiveRole_IsNotAGroupAtAll()
    {
        // Instances grouped only for layout have no leader; a chip there would invent a concept.
        var groups = LeaderResolver.Resolve(
        [
            Snapshots.Instance("gw-1", "gateways", "http://gw-1:5120", hasActiveRole: false),
            Snapshots.Instance("gw-2", "gateways", "http://gw-2:5120", hasActiveRole: false),
        ]);

        Assert.Empty(groups);
    }

    [Fact]
    public void Resolve_UngroupedInstances_ProduceNoGroups()
    {
        Assert.Empty(LeaderResolver.Resolve([Snapshots.Instance("solo")]));
    }

    [Fact]
    public void Resolve_SeparateGroups_AreScopedIndependently()
    {
        // Two site pairs, each with its own leader: per-Cluster election is the family's actual
        // topology, so a leader from one group must never be attributed to another.
        var groups = LeaderResolver.Resolve(
        [
            Snapshots.Instance("site-a-a", "site-a", "http://site-a-a:8084", leader: LeaderA),
            Snapshots.Instance("site-b-a", "site-b", "http://site-b-a:8084", leader: "akka.tcp://scadabridge@site-b-a:8082"),
        ]);

        Assert.Equal(2, groups.Count);
        Assert.Equal("site-a-a", groups[0].LeaderInstance);
        Assert.Equal("site-b-a", groups[1].LeaderInstance);
        Assert.All(groups, g => Assert.False(g.Disagreement));
    }

    [Theory]
    [InlineData("akka.tcp://scadabridge@site-a-a:8082", "site-a-a")]
    [InlineData("akka.tcp://otopcua@10.100.0.35:4053", "10.100.0.35")]
    [InlineData("akka.tcp://sys@host", "host")]
    [InlineData("not-an-address", null)]
    [InlineData("", null)]
    [InlineData(null, null)]
    public void HostOf_ParsesAkkaAddresses(string? address, string? expected)
    {
        Assert.Equal(expected, LeaderResolver.HostOf(address));
    }
}