[#474] Create Test

Author: danielmursa-dev

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")