Fix inst test (#6185)

* update institution test (#6149)

* update institution test

* update

* update inst test (#6150)

tests/api/endpoints/admin/test_institution_users.py

@@ -1,12 +1,13 @@
 import json
 import logging
+from mock import patch
 
 from django.urls import reverse
 
 from seahub.test_utils import BaseTestCase
-from tests.common.utils import randstring
+from seahub.institutions.models import Institution
-from seahub.institutions.models import Institution, InstitutionAdmin
 from seahub.profile.models import Profile
+
 logger = logging.getLogger(__name__)
 
 
@@ -25,8 +26,11 @@ class AdminInstitutionUsersTest(BaseTestCase):
         except Exception as e:
             logger.error(e)
 
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_can_get(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
 
-    def test_can_get(self):
         self.login_as(self.admin)
         inst = self._add_institution('int1')
         url = reverse('api-v2.1-admin-institution-users', args=[inst.id])
@@ -38,7 +42,11 @@ class AdminInstitutionUsersTest(BaseTestCase):
 
         inst.delete()
 
-    def test_no_permission(self):
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_no_permission(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
+
         self.logout()
         self.login_as(self.admin_no_other_permission)
         inst = self._add_institution('int1')
@@ -47,7 +55,11 @@ class AdminInstitutionUsersTest(BaseTestCase):
 
         self.assertEqual(403, resp.status_code)
 
-    def test_can_create(self):
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_can_create(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
+
         self.login_as(self.admin)
         inst = self._add_institution('int1')
         url = reverse('api-v2.1-admin-institution-users', args=[inst.id])
@@ -85,7 +97,11 @@ class AdminInstitutionUserTest(BaseTestCase):
             profile.institution = inst_name
         profile.save()
 
-    def test_can_update(self):
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_can_update(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
+
         self.login_as(self.admin)
         inst = self._add_institution('int1')
         self._add_user_in_institution(self.user.email, inst.name)
@@ -100,7 +116,11 @@ class AdminInstitutionUserTest(BaseTestCase):
 
         inst.delete()
 
-    def test_can_delete(self):
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_can_delete(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
+
         self.login_as(self.admin)
         inst = self._add_institution('int1')
         self._add_user_in_institution(self.user.email, inst.name)

tests/api/endpoints/admin/test_institutions.py

@@ -1,11 +1,13 @@
 import json
 import logging
+from mock import patch
 
 from django.urls import reverse
 
 from seahub.test_utils import BaseTestCase
 from tests.common.utils import randstring
 from seahub.institutions.models import Institution
+
 logger = logging.getLogger(__name__)
 
 
@@ -24,7 +26,11 @@ class InstitutionsTest(BaseTestCase):
         except Exception as e:
             logger.error(e)
 
-    def test_can_get(self):
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_can_get(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
+
         self.login_as(self.admin)
         inst = self._add_institution('int1')
         url = reverse('api-v2.1-admin-institutions')
@@ -37,7 +43,11 @@ class InstitutionsTest(BaseTestCase):
 
         inst.delete()
 
-    def test_no_permission(self):
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_no_permission(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
+
         self.logout()
         self.login_as(self.admin_no_other_permission)
         inst = self._add_institution('int1')
@@ -47,7 +57,11 @@ class InstitutionsTest(BaseTestCase):
         self.assertEqual(403, resp.status_code)
         inst.delete()
 
-    def test_can_create(self):
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_can_create(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
+
         self.login_as(self.admin)
         url = reverse('api-v2.1-admin-institutions')
 
@@ -79,7 +93,11 @@ class InstitutionTest(BaseTestCase):
         except Exception as e:
             logger.error(e)
 
-    def test_can_get(self):
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_can_get(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
+
         self.login_as(self.admin)
         institution_name = randstring(10)
         inst = self._add_institution(institution_name)
@@ -93,7 +111,11 @@ class InstitutionTest(BaseTestCase):
 
         inst.delete()
 
-    def test_can_update(self):
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_can_update(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
+
         self.login_as(self.admin)
         institution_name = randstring(10)
         inst = self._add_institution(institution_name)
@@ -108,8 +130,11 @@ class InstitutionTest(BaseTestCase):
 
         inst.delete()
 
+    @patch('seahub.api2.permissions.IsProVersion.has_permission')
+    def test_can_delete(self, mock_is_pro_version):
+
+        mock_is_pro_version.return_value = True
 
-    def test_can_delete(self):
         self.login_as(self.admin)
         institution_name = randstring(10)
         inst = self._add_institution(institution_name)