Merge branch 'more-tests' into 'main'

More tests

See merge request !1

django_kubernetes_probes/tests.py

-import logging
-
-from django.test import TestCase
-from django.urls import reverse
-
-logger = logging.getLogger(__name__)
-
-
-class ProbeTestCase(TestCase):
-    def test_probe_ready(self):
-        self.assertEqual(
-            200, self.client.get(reverse("probes:probe_ready")).status_code
-        )
-
-    def test_probe_alive(self):
-        self.assertEqual(
-            200, self.client.get(reverse("probes:probe_alive")).status_code
-        )

pyproject.toml

-[build-system]
-requires = ['setuptools>=40.8.0']
-build-backend = 'setuptools.build_meta'
+[tool.black]
+line-length = 120
+skip-string-normalization = true
\ No newline at end of file
+exclude = '''(
+  migrations\/ | manage.py | wsgi.py | .venv | venv
+)
+'''

setup.cfg

 [metadata]
 name = django-kubernetes-probes
-version = 1.0.2
+version = 1.1
 description = A Django app to expose probes for kubernetes
 long_description = file: README.rst
 url = https://www.example.com/

tests/models.py

+from django.db import models
+
+
+class AModel(models.Model):
+    a_field = models.CharField("A field", max_length=50, null=True, blank=True)

tests/settings.py

@@ -6,6 +6,7 @@ INSTALLED_APPS = [
     'django.contrib.auth',
     'django.contrib.contenttypes',
     'django_kubernetes_probes',
+    'tests',
 ]
 
 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
@@ -18,3 +19,4 @@ DATABASES = {
 }
 
 ROOT_URLCONF = 'tests.urls'
+DEFAULT_AUTO_FIELD = 'django.db.models.BigAutoField'

tests/tests.py

+import logging
+
+from django.test import TestCase
+from django.urls import reverse
+from django.db import connection
+from django.contrib.auth import get_user_model
+
+from tests import models
+
+logger = logging.getLogger(__name__)
+
+
+class ProbeTestCase(TestCase):
+    def drap_table(self, model):
+        with connection.cursor() as cursor:
+            cursor.execute(f"DROP TABLE {model._meta.db_table}")
+
+    def test_probe_ready(self):
+        self.assertEqual(200, self.client.get(reverse("probes:probe_ready")).status_code)
+
+    def test_probe_ready_without_user(self):
+        self.drap_table(get_user_model())
+        self.assertEqual(500, self.client.get(reverse("probes:probe_ready")).status_code)
+
+    def test_probe_ready_without_amodel(self):
+        self.drap_table(models.AModel)
+        self.assertEqual(500, self.client.get(reverse("probes:probe_ready")).status_code)
+
+    def test_probe_alive(self):
+        self.assertEqual(200, self.client.get(reverse("probes:probe_alive")).status_code)
+
+    def test_probe_alive_without_user(self):
+        self.drap_table(get_user_model())
+        self.assertEqual(500, self.client.get(reverse("probes:probe_alive")).status_code)
+
+    def test_probe_alive_without_amodel(self):
+        """
+        for alive-ness we only look at user, other missing table in the db does not cause an error
+        """
+        self.drap_table(models.AModel)
+        self.assertEqual(200, self.client.get(reverse("probes:probe_alive")).status_code)