refactor: fix error and refactor connection

LLCFreedom-Space · Apr 25, 2024 · 3ccb8b5 · 3ccb8b5
1 parent ee351ff
commit 3ccb8b5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/Sources/HealthChecks/PostgresHealthChecks/PostgresHealthChecks.swift b/Sources/HealthChecks/PostgresHealthChecks/PostgresHealthChecks.swift
@@ -49,7 +49,7 @@ public struct PostgresHealthChecks: PostgresHealthChecksProtocol {
         let result = HealthCheckItem(
             componentId: app.psqlId,
             componentType: .datastore,
-            observedValue: 1,
+            observedValue: connectionDescription.contains("active") ? 1 : 0,
             status: connectionDescription.contains("active") ? .pass : .fail,
             time: app.dateTimeISOFormat.string(from: Date()),
             output: !connectionDescription.contains("active") ? connectionDescription : nil,

diff --git a/Tests/HealthChecksTests/Mocks/PostgresHealthChecksMock.swift b/Tests/HealthChecksTests/Mocks/PostgresHealthChecksMock.swift
@@ -60,4 +60,8 @@ public struct PostgresHealthChecksMock: PostgresHealthChecksProtocol {
         ]
         return result
     }
+
+    public func checkConnection() async -> String {
+        "active"
+    }
 }