django-helpdesk/helpdesk/tests/helpers.py

# -*- coding: utf-8 -*-
import sys
try:
    from django.contrib.auth import get_user_model
except ImportError:
    from django.contrib.auth.models import User
else:
    User = get_user_model()

from helpdesk.models import Ticket, Queue


def get_staff_user(username='helpdesk.staff', password='password'):
    try:
        user = User.objects.get(username=username)
    except User.DoesNotExist:
        user = User.objects.create_user(username=username, password=password, email='staff@example.com')
        user.is_staff = True
        user.save()
    else:
        user.set_password(password)
        user.save()
    return user


def reload_urlconf(urlconf=None):
    if urlconf is None:
        from django.conf import settings

        urlconf = settings.ROOT_URLCONF
    if urlconf in sys.modules:
        from django.core.urlresolvers import clear_url_caches

        reload(sys.modules[urlconf])
        clear_url_caches()


def update_user_settings(user, **kwargs):
    usersettings = user.usersettings
    settings = usersettings.settings
    settings.update(kwargs)
    usersettings.settings = settings
    usersettings.save()


def delete_user_settings(user, *args):
    usersettings = user.usersettings
    settings = usersettings.settings
    for setting in args:
        if setting in settings:
            del settings[setting]
    usersettings.settings = settings
    usersettings.save()


def create_ticket(**kwargs):
    q = kwargs.get('queue', None)
    if q is None:
        try:
            q = Queue.objects.all()[0]
        except IndexError:
            q = Queue.objects.create(title='Test Q', slug='test', )
    data = {
        'title': "I wish to register a complaint",
        'queue': q,
    }
    data.update(kwargs)
    return Ticket.objects.create(**data)