feat(health): gRPC dependency health check

2026-06-01 06:44:05 -04:00
parent 5b82d68ea9
commit 1ab2f32e8e
3 changed files with 178 additions and 0 deletions
@@ -0,0 +1,72 @@
+using Grpc.Core;
+using Grpc.Net.Client;
+using Microsoft.Extensions.Diagnostics.HealthChecks;
+using ZB.MOM.WW.Health;
+
+namespace ZB.MOM.WW.Health.Tests;
+
+/// <summary>
+/// Verifies <see cref="GrpcDependencyHealthCheck"/> via an injected probe (no live gRPC server):
+/// probe-true → Healthy, probe-false → Unhealthy, and an <see cref="RpcException"/> from the probe
+/// → Unhealthy. The channel is constructed but never dialled because the probe is stubbed.
+/// </summary>
+public sealed class GrpcDependencyHealthCheckTests
+{
+    private static readonly GrpcChannel Channel = GrpcChannel.ForAddress("http://localhost");
+
+    private static async Task<HealthCheckResult> RunAsync(GrpcDependencyOptions options)
+    {
+        var check = new GrpcDependencyHealthCheck(Channel, options);
+        var context = new HealthCheckContext
+        {
+            Registration = new HealthCheckRegistration("grpc-dep", check, HealthStatus.Unhealthy, tags: null),
+        };
+        return await check.CheckHealthAsync(context, CancellationToken.None);
+    }
+
+    [Fact]
+    public async Task ProbeReturnsTrue_Healthy()
+    {
+        var result = await RunAsync(new GrpcDependencyOptions
+        {
+            Probe = static (_, _) => Task.FromResult(true),
+        });
+
+        Assert.Equal(HealthStatus.Healthy, result.Status);
+    }
+
+    [Fact]
+    public async Task ProbeReturnsFalse_Unhealthy()
+    {
+        var result = await RunAsync(new GrpcDependencyOptions
+        {
+            Probe = static (_, _) => Task.FromResult(false),
+        });
+
+        Assert.Equal(HealthStatus.Unhealthy, result.Status);
+    }
+
+    [Fact]
+    public async Task ProbeThrowsRpcException_Unhealthy()
+    {
+        var result = await RunAsync(new GrpcDependencyOptions
+        {
+            Probe = static (_, _) => throw new RpcException(new Status(StatusCode.Unavailable, "down")),
+        });
+
+        Assert.Equal(HealthStatus.Unhealthy, result.Status);
+    }
+
+    [Fact]
+    public async Task DependencyName_AppearsInDescription()
+    {
+        var result = await RunAsync(new GrpcDependencyOptions
+        {
+            DependencyName = "mxaccessgw worker",
+            Probe = static (_, _) => Task.FromResult(false),
+        });
+
+        Assert.Equal(HealthStatus.Unhealthy, result.Status);
+        Assert.Contains("mxaccessgw worker", result.Description);
+    }
+}