Produits Entr'ouvert » Authentic 2

import logging

from django.conf import settings

from django.contrib.contenttypes.models import ContentType

from django.core.management.base import BaseCommand

from django.utils import translation

from django.utils.translation import gettext_lazy as _

from authentic2.a2_rbac.models import Role

from authentic2.models import APIClient

logger = logging.getLogger(__name__)

class Command(BaseCommand):

    help = 'Create API clients admin role'

    def handle(self, *args, **options):

        translation.activate(settings.LANGUAGE_CODE)

        try:

            admin_role = Role.objects.get(slug='_a2-manager')

        except Role.DoesNotExist:

            logger.warning('No admin role.')

        content_type = ContentType.objects.get_for_model(APIClient)

        role = Role.objects.get_admin_role(

            instance=content_type,

            slug='_a2-manager-of-api-clients',

            name=_('Manager of API clients'),

            update_name=True,

            update_slug=True,

            create=True,

        )

        role.add_child(admin_role)

def test_create_api_clients_admin_role(db):

    from django.core.management.sql import emit_post_migrate_signal

    role = Role.objects.get(slug='_a2-manager-of-api-clients')

    # remove role created at test database setup

    Role.objects.get(slug='_a2-manager-of-api-clients').delete()

    # create role through command

    call_command('create_api_clients_admin_role')

    role = Role.objects.get(slug='_a2-manager-of-api-clients')

    assert role.permissions.count() == 1

    # check post_migrate update role

    emit_post_migrate_signal(verbosity=0, interactive=False, db='default', created_models=[])

    role_after_migrate = Role.objects.get(slug='_a2-manager-of-api-clients')

    assert role.pk == role_after_migrate.pk

    assert role.uuid == role_after_migrate.uuid

    assert role.permissions.count() == 2