diff --git a/tests/ZB.MOM.WW.ScadaBridge.Host.Tests/GrpcCentralTransportTests.cs b/tests/ZB.MOM.WW.ScadaBridge.Host.Tests/GrpcCentralTransportTests.cs index 1606b1a6..1be72219 100644 --- a/tests/ZB.MOM.WW.ScadaBridge.Host.Tests/GrpcCentralTransportTests.cs +++ b/tests/ZB.MOM.WW.ScadaBridge.Host.Tests/GrpcCentralTransportTests.cs @@ -10,6 +10,7 @@ using Microsoft.Extensions.Logging.Abstractions; using Microsoft.Extensions.Options; using ZB.MOM.WW.ScadaBridge.Commons.Messages.Notification; using ZB.MOM.WW.ScadaBridge.Communication; +using ZB.MOM.WW.ScadaBridge.Communication.Actors; using ZB.MOM.WW.ScadaBridge.Communication.Grpc; namespace ZB.MOM.WW.ScadaBridge.Host.Tests; @@ -542,6 +543,13 @@ public class GrpcCentralTransportTests : IAsyncLifetime Sender.Tell(new ReconcileSiteResponse([], [], "http://central/")); }); + // The gRPC service now wraps the request with the call's cancellation token. + Receive(_ => + { + Interlocked.Increment(ref counters.Reconciles); + Sender.Tell(new ReconcileSiteResponse([], [], "http://central/")); + }); + // Heartbeat lands here as a Tell (the failback probe); ignore it, no reply expected. ReceiveAny(_ => { }); }