django-helpdeskmig/helpdesk/tests/helpers.py
2019-03-20 16:12:19 +01:00

51 lines
1.2 KiB
Python

# -*- coding: utf-8 -*-
import sys
from django.contrib.auth import get_user_model
User = get_user_model()
def get_user(username='helpdesk.staff',
password='password',
is_staff=False,
is_superuser=False):
try:
user = User.objects.get(username=username)
except User.DoesNotExist:
user = User.objects.create_user(username=username,
password=password,
email='%s@example.com' % username)
user.is_staff = is_staff
user.is_superuser = is_superuser
user.save()
else:
user.set_password(password)
user.save()
return user
def reload_urlconf(urlconf=None):
from django.utils import six
if six.PY2:
from imp import reload
else:
from importlib import reload
if urlconf is None:
from django.conf import settings
urlconf = settings.ROOT_URLCONF
if HELPDESK_URLCONF in sys.modules:
reload(sys.modules[HELPDESK_URLCONF])
if urlconf in sys.modules:
reload(sys.modules[urlconf])
from django.urls import clear_url_caches
clear_url_caches()
HELPDESK_URLCONF = 'helpdesk.urls'