Using factory-boy to create users with permissions in Django Tests

factories.py

""" Holds all factories used in this app's test suite """
import factory

from django.contrib.auth import get_user_model


User = get_user_model()


def get_or_create_xxx_group():
    """ Returns the XXX Group """
    try:
        group_obj = Group.objects.get(name__iexact="xxx")
    except Group.DoesNotExist:
        group_obj  = Group.objects.create(name="Xxx")
        group_obj .permissions.add(
            Permission.objects.get(codename="change_user"),
            Permission.objects.get(codename="delete_user"),
        )
    return group_obj 
  

class UserFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = User
    username = factory.Faker('email')
    email = username
    first_name = factory.Faker('first_name')
    last_name = factory.Faker('last_name')

    @factory.post_generation
    def add_to_group(obj, create, extracted, **kwargs):
        """Add mock user into group matching their role"""
        if create:
            if obj.role == User.ADMIN:
                obj.groups.add(get_or_create_xxx_group())

thanks mate!