feat(host): CentralSingletonRegistrar — canonical singleton manager+proxy+drain helper

2026-07-08 15:59:01 -04:00
parent 7138d47630
commit c255ec31c9
2 changed files with 121 additions and 0 deletions
@@ -0,0 +1,60 @@
+using Akka.Actor;
+using Akka.Configuration;
+using Microsoft.Extensions.Logging.Abstractions;
+using ZB.MOM.WW.ScadaBridge.Host.Actors;
+
+namespace ZB.MOM.WW.ScadaBridge.Host.Tests;
+
+public class CentralSingletonRegistrarTests
+{
+    private sealed class EchoActor : ReceiveActor
+    {
+        public EchoActor() => ReceiveAny(m => Sender.Tell(m));
+    }
+
+    [Fact]
+    public async Task Start_CreatesManagerAndProxy_WithCanonicalNames_AndDrainTask()
+    {
+        using var system = CreateSingleNodeClusterSystem();
+        var handle = CentralSingletonRegistrar.Start(
+            system, "test-widget", Props.Create(() => new EchoActor()),
+            NullLogger.Instance, drainTimeout: TimeSpan.FromSeconds(2));
+
+        // Canonical naming preserved: <name>-singleton / <name>-proxy.
+        Assert.EndsWith("/user/test-widget-singleton", handle.Manager.Path.ToString());
+        Assert.EndsWith("/user/test-widget-proxy", handle.Proxy.Path.ToString());
+
+        // Singleton answers through the proxy once the member is Up.
+        var echo = await handle.Proxy.Ask<string>("hi", TimeSpan.FromSeconds(20));
+        Assert.Equal("hi", echo);
+
+        // The drain task registered on PhaseClusterLeave must stop the manager
+        // during coordinated shutdown.
+        await Akka.Actor.CoordinatedShutdown.Get(system).Run(
+            Akka.Actor.CoordinatedShutdown.ClrExitReason.Instance);
+        // Manager terminated => drain ran (GracefulStop completed or PoisonPill fallback).
+        Assert.True(handle.Manager.IsNobody() || system.WhenTerminated.IsCompleted);
+    }
+
+    private static ActorSystem CreateSingleNodeClusterSystem()
+    {
+        var port = FreePort();
+        var config = ConfigurationFactory.ParseString($@"
+akka {{
+    actor.provider = cluster
+    remote.dot-netty.tcp {{ hostname = ""127.0.0.1"", port = {port} }}
+    cluster {{
+        seed-nodes = [""akka.tcp://registrar-test@127.0.0.1:{port}""]
+        roles = [""Central""]
+        min-nr-of-members = 1
+    }}
+}}");
+        return ActorSystem.Create("registrar-test", config);
+    }
+
+    private static int FreePort()
+    {
+        var l = new System.Net.Sockets.TcpListener(System.Net.IPAddress.Loopback, 0);
+        l.Start(); var p = ((System.Net.IPEndPoint)l.LocalEndpoint).Port; l.Stop(); return p;
+    }
+}