Added missing tests

core/common/tests.py

@@ -218,6 +218,7 @@ class OCLTestCase(TestCase, BaseTestCase):
     def setUpClass(cls):
         super().setUpClass()
         call_command("loaddata", "core/fixtures/base_entities.yaml")
+        call_command("loaddata", "core/fixtures/auth_groups.yaml")
 
     def tearDown(self):
         super().tearDown()

core/integration_tests/tests_orgs.py

@@ -175,6 +175,7 @@ def test_get_200(self):
         self.assertEqual(response.data['id'], self.org.mnemonic)
         self.assertEqual(response.data['name'], 'Stark Enterprises')
         self.assertFalse('overview' in response.data)
+        self.assertFalse('client_configs' in response.data)
 
     def test_get_200_with_overview(self):
         response = self.client.get(
@@ -188,6 +189,18 @@ def test_get_200_with_overview(self):
         self.assertEqual(response.data['id'], self.org.mnemonic)
         self.assertTrue('overview' in response.data)
 
+    def test_get_200_with_configs(self):
+        response = self.client.get(
+            f'/orgs/{self.org.mnemonic}/?includeClientConfigs=true',
+            HTTP_AUTHORIZATION='Token ' + self.token,
+            format='json'
+        )
+
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.data['uuid'], str(self.org.id))
+        self.assertEqual(response.data['id'], self.org.mnemonic)
+        self.assertTrue('client_configs' in response.data)
+
     def test_get_404(self):
         response = self.client.get(
             '/orgs/foobar/',

core/sources/tests/tests.py

@@ -1,9 +1,10 @@
 import factory
 from django.core.exceptions import ValidationError
 from django.db import transaction, IntegrityError
-from mock import patch, Mock, ANY
+from mock import patch, Mock, ANY, PropertyMock
 
-from core.common.constants import HEAD, ACCESS_TYPE_EDIT, ACCESS_TYPE_NONE, ACCESS_TYPE_VIEW
+from core.common.constants import HEAD, ACCESS_TYPE_EDIT, ACCESS_TYPE_NONE, ACCESS_TYPE_VIEW, \
+    CUSTOM_VALIDATION_SCHEMA_OPENMRS
 from core.common.tasks import seed_children
 from core.common.tests import OCLTestCase
 from core.concepts.models import Concept
@@ -576,6 +577,31 @@ def test_hierarchy_without_hierarchy_root(self):
                  name=parentless_concept.display_name, children=[parentless_concept_child.uri])
         )
 
+    def test_is_validation_necessary(self):
+        source = OrganizationSourceFactory()
+
+        self.assertFalse(source.is_validation_necessary())
+
+        source.custom_validation_schema = CUSTOM_VALIDATION_SCHEMA_OPENMRS
+
+        self.assertFalse(source.is_validation_necessary())
+
+        source.active_concepts = 1
+        self.assertTrue(source.is_validation_necessary())
+
+    @patch('core.sources.models.Source.head', new_callable=PropertyMock)
+    def test_is_hierarchy_root_belonging_to_self(self, head_mock):
+        root = Concept(id=1, parent_id=100)
+        source = Source(id=1, hierarchy_root=root, version='HEAD')
+        head_mock.return_value = source
+        self.assertFalse(source.is_hierarchy_root_belonging_to_self())
+        source_v1 = Source(id=1, hierarchy_root=root, version='v1')
+        self.assertFalse(source_v1.is_hierarchy_root_belonging_to_self())
+
+        root.parent_id = 1
+        self.assertTrue(source.is_hierarchy_root_belonging_to_self())
+        self.assertTrue(source_v1.is_hierarchy_root_belonging_to_self())
+
 
 class TasksTest(OCLTestCase):
     @patch('core.common.models.ConceptContainerModel.index_children')

core/users/tests/tests.py

@@ -1,3 +1,4 @@
+from django.contrib.auth.models import Group
 from mock import Mock, patch, ANY
 from rest_framework.authtoken.models import Token
 
@@ -7,7 +8,7 @@
 from core.common.tests import OCLTestCase, OCLAPITestCase
 from core.orgs.models import Organization
 from core.sources.tests.factories import OrganizationSourceFactory
-from core.users.constants import USER_OBJECT_TYPE
+from core.users.constants import USER_OBJECT_TYPE, OCL_SERVERS_GROUP
 from core.users.models import UserProfile
 from core.users.tests.factories import UserProfileFactory
 
@@ -204,6 +205,18 @@ def test_mark_verified(self):
         self.assertIsNone(user.verification_token)
         user.save.assert_not_called()
 
+    def test_auth_groups(self):
+        user = UserProfileFactory()
+        self.assertEqual(user.auth_groups.count(), 0)
+
+        user.groups.add(Group.objects.get(name=OCL_SERVERS_GROUP))
+
+        self.assertEqual(user.auth_groups.count(), 1)
+
+    def test_is_valid_auth_group(self):
+        self.assertFalse(UserProfile.is_valid_auth_group('foobar'))
+        self.assertTrue(UserProfile.is_valid_auth_group(OCL_SERVERS_GROUP))
+
 
 class TokenAuthenticationViewTest(OCLAPITestCase):
     def test_login(self):