From fdb19652a219e8a5edcc08c703f9a5b3623b0e5f Mon Sep 17 00:00:00 2001
From: James Hollinger <39168456+james-hollinger@users.noreply.github.com>
Date: Mon, 24 Jun 2024 15:55:15 -0700
Subject: [PATCH] Use Correct DB Healthcheck (#549)

---
 backend/webapi/Startup.cs  | 2 +-
 backend/webapi/pidp.csproj | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/backend/webapi/Startup.cs b/backend/webapi/Startup.cs
index 15e9bd12b..043c85bfe 100644
--- a/backend/webapi/Startup.cs
+++ b/backend/webapi/Startup.cs
@@ -66,7 +66,7 @@ public void ConfigureServices(IServiceCollection services)
         services.AddHealthChecks()
             .AddApplicationStatus(tags: new[] { HealthCheckTag.Liveness.Value })
             .AddCheck<BackgroundWorkerHealthCheck>("PlrStatusUpdateSchedulingService", tags: new[] { HealthCheckTag.BackgroundServices.Value })
-            .AddNpgSql(config.ConnectionStrings.PidpDatabase, tags: new[] { HealthCheckTag.Readiness.Value })
+            .AddDbContextCheck<PidpDbContext>(tags: new[] { HealthCheckTag.Readiness.Value })
             .AddRabbitMQ(new Uri(config.RabbitMQ.HostAddress), tags: new[] { HealthCheckTag.Readiness.Value });
 
         services.AddSwaggerGen(options =>
diff --git a/backend/webapi/pidp.csproj b/backend/webapi/pidp.csproj
index b5fccc524..237ce2bf8 100644
--- a/backend/webapi/pidp.csproj
+++ b/backend/webapi/pidp.csproj
@@ -8,7 +8,6 @@
   </PropertyGroup>
   <ItemGroup>
     <PackageReference Include="AspNetCore.HealthChecks.ApplicationStatus" Version="8.0.1" />
-    <PackageReference Include="AspNetCore.HealthChecks.NpgSql" Version="8.0.1" />
     <PackageReference Include="AspNetCore.HealthChecks.RabbitMQ" Version="8.0.1" />
     <PackageReference Include="AutoMapper" Version="13.0.1" />
     <PackageReference Include="DomainResult" Version="3.2.0" />