utils.py

from __future__ import absolute_import, unicode_literals, division

from django import test
from django.utils import timezone
from django.contrib.contenttypes.models import ContentType
from django.urls import reverse
from django.contrib.auth import get_user_model
from django.contrib.auth.models import Permission

__all__ = ('LockingClient', 'user_factory')


def user_factory(model=None):
    username = 'user%d' % timezone.now().microsecond
    email = '%s@example.com' % username
    password = 'p@ssw0rd'
    user = get_user_model().objects.create_user(username, email, password)  # user will have is_staff = False
    user.is_staff = True
    user.save()
    if model:
        content_type = ContentType.objects.get_for_model(model)
        for perm in ['add', 'change', 'delete']:
            codename = '%s_%s' % (perm, model._meta.object_name.lower())
            permission = Permission.objects.get(codename=codename, content_type=content_type)
            user.user_permissions.add(permission)
    return user, password


class LockingClient(object):

    def __init__(self, instance):
        self.instance = instance
        self.url = reverse('locking-api', kwargs={
            'app': instance._meta.app_label,
            'model': instance._meta.object_name,
            'object_id': instance.pk,
        })
        self.client = test.Client()
        self.user = None

    def login_new_user(self, has_perm=True, *args, **kwargs):
        self.client.logout()
        model_for_perm = self.instance if has_perm else None
        user, password = user_factory(model_for_perm, *args, **kwargs)
        self.client.login(username=user.username, password=password)
        self.user = user

    def get(self, *args, **kwargs):
        return self.client.get(self.url, *args, **kwargs)

    def post(self, *args, **kwargs):
        return self.client.post(self.url, *args, **kwargs)

    def put(self, *args, **kwargs):
        return self.client.put(self.url, *args, **kwargs)

    def delete(self, *args, **kwargs):
        return self.client.delete(self.url, *args, **kwargs)