[#474] Create Test

maykinmedia · Jan 2, 2025 · 407b61f · 407b61f
1 parent 737bdd4
commit 407b61f
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/objects/tests/admin/test_token_permissions.py b/src/objects/tests/admin/test_token_permissions.py
@@ -1,3 +1,5 @@
+import os
+
 from django.urls import reverse_lazy
 
 import requests_mock
@@ -63,3 +65,21 @@ def test_get_permission_with_unavailable_objecttypes(self, m):
         response = self.app.get(self.url)
 
         self.assertEqual(response.status_code, 200)
+
+
+@disable_admin_mfa()
+@requests_mock.Mocker()
+class AdminVersionTests(WebTest):
+    url = reverse_lazy("admin:index")
+
+    def setUp(self):
+        user = UserFactory(is_superuser=True, is_staff=True)
+        self.app.set_user(user)
+
+    def test_version(self, m):
+        response = self.app.get(self.url)
+        self.assertEqual(response.status_code, 200)
+        if os.environ["ENVIRONMENT"] == "development":
+            self.assertEqual(os.environ["RELEASE"], "dev")
+        else:
+            self.assertEqual(os.environ["RELEASE"], "prod")