Misago/misago/acl/testutils.py

from hashlib import md5
from time import time

from misago.core import threadstore
from misago.core.cache import cache

from misago.acl.forms import get_permissions_forms
from misago.acl.models import Role


def fake_post_data(target, data_dict):
    """
    In order for form to don't fail submission, all permission fields need
    to receive values. This function populates data dict with default values
    for permissions, making form validation pass
    """
    for form in get_permissions_forms(target):
        for field in form:
            if field.value() == True:
                data_dict[field.html_name] = 1
            elif field.value() == False:
                data_dict[field.html_name] = 0
            else:
                data_dict[field.html_name] = field.value()
    return data_dict


def override_acl(user, new_acl):
    """overrides user permissions with specified ones"""
    test_role = Role(name='Fake')
    test_role.permissions = new_acl
    test_role.save()

    user.rank.roles.clear()

    user.roles.clear()
    user.roles.add(test_role)
    user.acl_key = md5(unicode(time())).hexdigest()[:8]
    user.save()