feat: add tests

revsys · Oct 21, 2024 · 54a588d · 54a588d
1 parent 2b1a92b
commit 54a588d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/health_check/contrib/db_heartbeat/__init__.py b/health_check/contrib/db_heartbeat/__init__.py
@@ -1,4 +1,4 @@
 import django
 
 if django.VERSION < (3, 2):
-    default_app_config = "health_check.contrib.db_heartbeat.apps.HealthCheckConfig"
+    default_app_config = "health_check.contrib.db_heartbeat.apps.HealthCheckConfig"
diff --git a/health_check/contrib/db_heartbeat/apps.py b/health_check/contrib/db_heartbeat/apps.py
@@ -3,8 +3,9 @@
 
 
 class HealthCheckConfig(AppConfig):
-    name = 'health_check.contrib.db_heartbeat'
+    name = "health_check.contrib.db_heartbeat"
 
     def ready(self):
         from .backends import DatabaseHeartBeatCheck
+
         plugin_dir.register(DatabaseHeartBeatCheck)
diff --git a/health_check/contrib/db_heartbeat/backends.py b/health_check/contrib/db_heartbeat/backends.py
@@ -16,6 +16,8 @@ def check_status(self):
                 result = cursor.fetchone()
 
             if result != (1,):
-                raise ServiceReturnedUnexpectedResult("Health Check query did not return the expected result.")
+                raise ServiceReturnedUnexpectedResult(
+                    "Health Check query did not return the expected result."
+                )
         except Exception as e:
             raise ServiceUnavailable(f"Database health check failed: {e}")