refactor tests and move update_institiutions to it's own file

api/draft_registrations/serializers.py

@@ -8,7 +8,6 @@
 from api.nodes.serializers import (
  DraftRegistrationLegacySerializer,
  DraftRegistrationDetailLegacySerializer,
- update_institutions,
  get_license_details,
  NodeSerializer,
  NodeLicenseSerializer,
@@ -18,6 +17,7 @@
  NodeContributorDetailSerializer,
  RegistrationSchemaRelationshipField,
 )
+from api.institutions.utils import update_institutions
 from api.taxonomies.serializers import TaxonomizableSerializerMixin
 from osf.exceptions import DraftRegistrationStateError
 from osf.models import Node

api/institutions/utils.py

@@ -0,0 +1,38 @@
+from rest_framework import exceptions
+from osf.models import Institution
+from osf.utils import permissions as osf_permissions
+from api.base.serializers import relationship_diff
+
+
+def get_institutions_to_add_remove(institutions, new_institutions):
+ diff = relationship_diff(
+ current_items={inst._id: inst for inst in institutions.all()},
+ new_items={inst['_id']: inst for inst in new_institutions},
+ )
+
+ insts_to_add = []
+ for inst_id in diff['add']:
+ inst = Institution.load(inst_id)
+ if not inst:
+ raise exceptions.NotFound(detail=f'Institution with id "{inst_id}" was not found')
+ insts_to_add.append(inst)
+
+ return insts_to_add, diff['remove'].values()
+
+
+def update_institutions(node, new_institutions, user, post=False):
+ add, remove = get_institutions_to_add_remove(
+ institutions=node.affiliated_institutions,
+ new_institutions=new_institutions,
+ )
+
+ if not post:
+ for inst in remove:
+ if not user.is_affiliated_with_institution(inst) and not node.has_permission(user, osf_permissions.ADMIN):
+ raise exceptions.PermissionDenied(detail=f'User needs to be affiliated with {inst.name}')
+ node.remove_affiliated_institution(inst, user)
+
+ for inst in add:
+ if not user.is_affiliated_with_institution(inst):
+ raise exceptions.PermissionDenied(detail=f'User needs to be affiliated with {inst.name}',)
+ node.add_affiliated_institution(inst, user)

api/nodes/serializers.py

@@ -10,7 +10,7 @@
  JSONAPISerializer, LinksField,
  NodeFileHyperLinkField, RelationshipField,
  ShowIfVersion, TargetTypeField, TypeField,
- WaterbutlerLink, relationship_diff, BaseAPISerializer,
+ WaterbutlerLink, BaseAPISerializer,
  HideIfWikiDisabled, ShowIfAdminScopeOrAnonymous,
  ValuesListField, TargetField,
 )
@@ -20,6 +20,7 @@
  get_user_auth, is_truthy,
 )
 from api.base.versioning import get_kebab_snake_case_field
+from api.institutions.utils import update_institutions
 from api.taxonomies.serializers import TaxonomizableSerializerMixin
 from django.apps import apps
 from django.conf import settings
@@ -51,43 +52,6 @@ def to_internal_value(self, data):
  return self.get_object(data)
 
 
-def get_institutions_to_add_remove(institutions, new_institutions):
- diff = relationship_diff(
- current_items={inst._id: inst for inst in institutions.all()},
- new_items={inst['_id']: inst for inst in new_institutions},
- )
-
- insts_to_add = []
- for inst_id in diff['add']:
- inst = Institution.load(inst_id)
- if not inst:
- raise exceptions.NotFound(detail='Institution with id "{}" was not found'.format(inst_id))
- insts_to_add.append(inst)
-
- return insts_to_add, diff['remove'].values()
-
-
-def update_institutions(node, new_institutions, user, post=False):
- add, remove = get_institutions_to_add_remove(
- institutions=node.affiliated_institutions,
- new_institutions=new_institutions,
- )
-
- if not post:
- for inst in remove:
- if not user.is_affiliated_with_institution(inst) and not node.has_permission(user, osf_permissions.ADMIN):
- raise exceptions.PermissionDenied(
- detail='User needs to be affiliated with {}'.format(inst.name),
- )
- node.remove_affiliated_institution(inst, user)
-
- for inst in add:
- if not user.is_affiliated_with_institution(inst):
- raise exceptions.PermissionDenied(
- detail='User needs to be affiliated with {}'.format(inst.name),
- )
- node.add_affiliated_institution(inst, user)
-
 
 class RegionRelationshipField(RelationshipField):
 

api/preprints/urls.py

@@ -20,5 +20,5 @@
  re_path(r'^(?P<preprint_id>\w+)/requests/$', views.PreprintRequestListCreate.as_view(), name=views.PreprintRequestListCreate.view_name),
  re_path(r'^(?P<preprint_id>\w+)/subjects/$', views.PreprintSubjectsList.as_view(), name=views.PreprintSubjectsList.view_name),
  re_path(r'^(?P<preprint_id>\w+)/institutions/$', views.PreprintInstitutionsList.as_view(), name=views.PreprintInstitutionsList.view_name),
- re_path(r'^(?P<preprint_id>\w+)/relationships/institutions/$', views.PreprintInstitutionsRelationshipList.as_view(), name=views.PreprintInstitutionsRelationshipList.view_name),
+ re_path(r'^(?P<preprint_id>\w+)/relationships/institutions/$', views.PreprintInstitutionsRelationship.as_view(), name=views.PreprintInstitutionsRelationship.view_name),
 ]

api/preprints/views.py

@@ -651,7 +651,7 @@ def get_queryset(self):
  return self.get_resource().affiliated_institutions.all()
 
 
-class PreprintInstitutionsRelationshipList(JSONAPIBaseView, generics.RetrieveUpdateDestroyAPIView, generics.CreateAPIView, PreprintMixin):
+class PreprintInstitutionsRelationship(JSONAPIBaseView, generics.RetrieveUpdateDestroyAPIView, generics.CreateAPIView, PreprintMixin):
  """ """
  permission_classes = (
  drf_permissions.IsAuthenticatedOrReadOnly,

api/registrations/serializers.py

@@ -19,7 +19,6 @@
  NodeStorageProviderSerializer,
  NodeLicenseRelationshipField,
  NodeLinksSerializer,
- update_institutions,
  NodeLicenseSerializer,
  NodeContributorsSerializer,
  RegistrationProviderRelationshipField,
@@ -31,13 +30,14 @@
  ShowIfVersion, VersionedDateTimeField, ValuesListField,
  HideIfWithdrawalOrWikiDisabled,
 )
+from api.institutions.utils import update_institutions
+
 from framework.auth.core import Auth
 from osf.exceptions import NodeStateError
 from osf.models import Node
 from osf.utils.registrations import strip_registered_meta_comments
 from osf.utils.workflows import ApprovalStates
 
-
 class RegistrationSerializer(NodeSerializer):
  admin_only_editable_fields = [
  'custom_citation',

api_tests/preprints/views/test_preprint_institutions_relationship.py

@@ -9,7 +9,7 @@
 
 
 @pytest.mark.django_db
-class TestPreprintInstitutionsList:
+class TestPreprintInstitutionsRelationship:
 
  @pytest.fixture()
  def user(self):