Projet

Général

Profil

0001-add-role-creation-API-20706.patch

Paul Marillonnet, 16 février 2018 11:13

Télécharger (9,49 ko)

Voir les différences:

Subject: [PATCH] add role-creation API (#20706)

 src/authentic2/api_views.py |  65 ++++++++++++++++++
 tests/test_api.py           | 156 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 221 insertions(+)
src/authentic2/api_views.py
444 444
        exclude = ('date_joined', 'user_permissions', 'groups', 'last_login')
445 445

  
446 446

  
447
class RoleSerializer(serializers.ModelSerializer):
448
    ou = serializers.SlugRelatedField(
449
        many=False,
450
        required=False,
451
        default=get_default_ou,
452
        queryset=get_ou_model().objects.all(),
453
        slug_field='slug')
454

  
455
    def check_perm(self, perm, ou=None):
456
        self.context['view'].check_perm(perm, ou)
457

  
458
    def create(self, validated_data):
459
        self.check_perm('a2_rbac.add_role', validated_data.get('ou'))
460
        return super(RoleSerializer, self).create(validated_data)
461

  
462
    def update(self, instance, validated_data):
463
        self.check_perm('a2_rbac.change_role', validated_data.get('ou'))
464
        # The OU shouldn't be changed:
465
        if instance.ou != validated_data.get('ou'):
466
            validated_data['ou'] = instance.ou
467
        super(RoleSerializer, self).update(instance, validated_data)
468
        return instance
469

  
470
    def partial_update(self, instance, validated_data):
471
        self.check_perm('a2_rbac.change_role', validated_data.get('ou'))
472
        # The OU shouldn't be changed:
473
        if validated_data.get('ou') and instance.ou != validated_data['ou']:
474
            validated_data['ou'] = instance.ou
475
        super(RoleSerializer, self).partial_update(instance, validated_data)
476
        return instance
477

  
478
    class Meta:
479
        model = get_role_model()
480
        fields = ('uuid', 'name', 'slug', 'description', 'ou',)
481
        extra_kwargs = {'uuid': {'read_only': True}}
482

  
483

  
447 484
class UsersFilter(FilterSet):
448 485
    class Meta:
449 486
        model = get_user_model()
......
576 613
        return Response({'result': 1})
577 614

  
578 615

  
616
class RolesAPI(ExceptionHandlerMixin, ModelViewSet):
617
    permission_classes = (permissions.IsAuthenticated,)
618
    serializer_class = RoleSerializer
619
    lookup_field = 'slug'
620
    queryset = get_role_model().objects.all()
621

  
622
    def check_perm(self, perm, ou=None):
623
        if ou:
624
            if not self.request.user.has_ou_perm(perm, ou):
625
                raise PermissionDenied(u'User %s does not have permission %s in %s' % (user, perm, ou))
626
        else:
627
            if not self.request.user.has_perm(perm):
628
                raise PermissionDenied(u'User %s do not have permission %s' % (user, perm))
629

  
630
    def retrieve(self, request, *args, **kwargs):
631
        self.check_perm('a2_rbac.view_role', None)
632
        return super(RolesAPI, self).retrieve(request, *args, **kwargs)
633

  
634
    def list(self, request, *args, **kwargs):
635
        self.check_perm('a2_rbac.view_role', None)
636
        return super(RolesAPI, self).list(request, *args, **kwargs)
637

  
638
    def perform_destroy(self, instance):
639
        self.check_perm('a2_rbac.delete_role', instance.ou)
640
        super(RolesAPI, self).perform_destroy(instance)
641

  
642

  
579 643
class RoleMembershipsAPI(ExceptionHandlerMixin, APIView):
580 644
    permission_classes = (permissions.IsAuthenticated,)
581 645

  
......
620 684
router = SimpleRouter()
621 685
router.register(r'users', UsersAPI, base_name='a2-api-users')
622 686
router.register(r'ous', OrganizationalUnitAPI, base_name='a2-api-ous')
687
router.register(r'roles', RolesAPI, base_name='a2-api-roles')
623 688

  
624 689

  
625 690
class CheckPasswordSerializer(serializers.Serializer):
tests/test_api.py
668 668

  
669 669
    assert mail.to[0] == new_email
670 670
    assert 'http://testserver/accounts/change-email/verify/' in mail.body
671

  
672

  
673
def test_api_delete_role(app, superuser, simple_role):
674
    app.authorization = ('Basic', (superuser.username, superuser.username))
675
    Role = get_role_model()
676

  
677
    app.delete('/api/roles/simple-role/')
678
    assert not len(Role.objects.filter(slug='simple-role'))
679

  
680

  
681
def test_api_delete_role_unauthorized(app, simple_user, simple_role):
682
    app.authorization = ('Basic', (simple_user.username, simple_user.username))
683
    Role = get_role_model()
684

  
685
    app.delete('/api/roles/simple-role/', status=403)
686
    assert len(Role.objects.filter(slug='simple-role'))
687

  
688

  
689
def test_api_patch_role(app, superuser, simple_role):
690
    app.authorization = ('Basic', (superuser.username, superuser.username))
691
    Role = get_role_model()
692

  
693
    role_data = {
694
        'slug': 'updated-role',
695
    }
696
    resp = app.patch_json('/api/roles/simple-role/', params=role_data)
697
    assert not len(Role.objects.filter(slug='simple-role'))
698
    assert len(Role.objects.filter(slug='updated-role'))
699

  
700
    # The role API won't change the organizational unit attribute:
701
    role = Role.objects.get(slug='updated-role')
702
    assert role.ou.slug == get_default_ou().slug
703
    assert role.ou.slug != 'ou1'
704

  
705

  
706
def test_api_patch_role_unauthorized(app, simple_user, simple_role):
707
    app.authorization = ('Basic', (simple_user.username, simple_user.username))
708
    Role = get_role_model()
709

  
710
    role_data = {
711
        'slug': 'updated-role',
712
    }
713
    resp = app.patch_json('/api/roles/simple-role/', params=role_data, status=403)
714
    assert len(Role.objects.filter(slug='simple-role'))
715
    assert not len(Role.objects.filter(slug='updated-role'))
716

  
717

  
718
def test_api_put_role(app, superuser, simple_role, ou1):
719
    app.authorization = ('Basic', (superuser.username, superuser.username))
720
    Role = get_role_model()
721

  
722
    role_data = {
723
        'name': 'updated-role',
724
        'slug': 'updated-role',
725
        'ou': 'ou1'
726
    }
727
    resp = app.put_json('/api/roles/simple-role/', params=role_data)
728
    assert not len(Role.objects.filter(slug='simple-role'))
729
    assert len(Role.objects.filter(slug='updated-role'))
730

  
731
    # The role API won't change the organizational unit attribute:
732
    role = Role.objects.get(slug='updated-role')
733
    assert role.ou.slug == get_default_ou().slug
734
    assert role.ou.slug != 'ou1'
735

  
736

  
737
def test_api_put_role_unauthorized(app, simple_user, simple_role, ou1):
738
    app.authorization = ('Basic', (simple_user.username, simple_user.username))
739
    Role = get_role_model()
740

  
741
    role_data = {
742
        'name': 'updated-role',
743
        'slug': 'updated-role',
744
        'ou': 'ou1'
745
    }
746
    resp = app.put_json('/api/roles/simple-role/', params=role_data, status=403)
747
    assert len(Role.objects.filter(slug='simple-role'))
748
    assert not len(Role.objects.filter(slug='updated-role'))
749

  
750

  
751
def test_api_post_role(app, superuser, ou1):
752
    app.authorization = ('Basic', (superuser.username, superuser.username))
753

  
754
    role_data = {
755
        'slug': 'coffee-manager',
756
        'name': 'Coffee Manager',
757
        'ou': 'ou1'
758
        }
759
    resp = app.post_json('/api/roles/', params=role_data)
760
    assert isinstance(resp.json, dict)
761
    Role = get_role_model()
762

  
763
    # Check attribute values against the server's response:
764
    for key, value in role_data.items():
765
        assert key in resp.json.keys()
766
        assert value in resp.json.values()
767

  
768
    # Check attributes values against the DB:
769
    posted_role = Role.objects.get(slug='coffee-manager')
770
    assert posted_role.slug == role_data['slug']
771
    assert posted_role.name == role_data['name']
772
    assert posted_role.ou.slug == 'ou1'
773

  
774

  
775
def test_api_post_role_no_ou(app, superuser):
776
    app.authorization = ('Basic', (superuser.username, superuser.username))
777
    Role = get_role_model()
778

  
779
    role_data = {
780
        'slug': 'tea-manager',
781
        'name': 'Tea Manager',
782
        }
783
    resp = app.post_json('/api/roles/', params=role_data)
784
    new_role = Role.objects.get(slug='tea-manager')
785
    default_ou = get_default_ou()
786
    assert new_role.ou == default_ou
787

  
788

  
789
def test_api_post_role_unauthorized(app, simple_user, ou1):
790
    app.authorization = ('Basic', (simple_user.username, simple_user.username))
791
    Role = get_role_model()
792

  
793
    role_data = {
794
        'slug': 'mocca-manager',
795
        'name': 'Mocca Manager',
796
        'ou': 'ou1'
797
        }
798

  
799
    resp = app.post_json('/api/roles/', params=role_data, status=403)
800
    assert not len(Role.objects.filter(slug='mocca-manager'))
801

  
802

  
803
def test_api_get_role_description(app, superuser, role_random):
804
    app.authorization = ('Basic', (superuser.username, superuser.username))
805
    resp = app.get('/api/roles/rando/')
806

  
807
    assert resp.json['slug'] == 'rando'
808
    assert resp.json['ou'] == 'ou_rando'
809

  
810

  
811
def test_api_get_role_not_found(app, superuser):
812
    app.authorization = ('Basic', (superuser.username, superuser.username))
813
    resp = app.get('/api/roles/thisisnotavalidroleslug/', status=404)
814

  
815

  
816
def test_api_get_role_list(app, superuser, simple_role, role_random):
817
    app.authorization = ('Basic', (superuser.username, superuser.username))
818
    resp = app.get('/api/roles/')
819

  
820
    role_fields = ['slug', 'uuid', 'name', 'ou']
821

  
822
    assert len(resp.json['results'])
823

  
824
    for role_dict in resp.json['results']:
825
        for field in role_fields:
826
            assert field in role_dict
671
-