80 lines
1.8 KiB
Python
80 lines
1.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import pickle
|
|
|
|
from utilities import create_temporary_pub, clean_temporary_pub
|
|
|
|
from wcs.qommon.storage import StorableObject
|
|
from wcs.roles import Role, get_user_roles
|
|
|
|
from quixote import get_publisher
|
|
|
|
|
|
def setup_module(module):
|
|
pub = create_temporary_pub()
|
|
pub.cfg['language'] = {'language': 'en'}
|
|
pub.cfg['misc'] = {'charset': 'utf-8'}
|
|
|
|
|
|
def teardown_module(module):
|
|
clean_temporary_pub()
|
|
|
|
|
|
def test_slug():
|
|
Role.wipe()
|
|
role = Role(name='Hello world')
|
|
role.store()
|
|
assert role.slug == 'hello-world'
|
|
|
|
|
|
def test_duplicated_name():
|
|
Role.wipe()
|
|
role = Role(name='Hello world')
|
|
role.store()
|
|
assert role.slug == 'hello-world'
|
|
role = Role(name='Hello world')
|
|
role.store()
|
|
assert role.slug == 'hello-world-1'
|
|
|
|
|
|
def test_migrate():
|
|
Role.wipe()
|
|
role = Role(name='Hello world')
|
|
role.store()
|
|
obj = pickle.load(open(role.get_object_filename(), 'rb'))
|
|
del obj.slug
|
|
pickle.dump(obj, open(role.get_object_filename(), 'wb'))
|
|
assert pickle.load(open(role.get_object_filename(), 'rb')).slug is None
|
|
assert Role.get(role.id).slug == 'hello-world'
|
|
|
|
|
|
def test_get_user_roles():
|
|
Role.wipe()
|
|
Role(name='f1').store()
|
|
Role(name='é1').store()
|
|
Role(name='a1').store()
|
|
assert [x[1] for x in get_user_roles()] == ['a1', 'é1', 'f1']
|
|
|
|
|
|
def test_get_emails():
|
|
User = get_publisher().user_class
|
|
User.wipe()
|
|
Role.wipe()
|
|
|
|
role = Role(name='role')
|
|
role.emails_to_members = True
|
|
role.store()
|
|
|
|
users = []
|
|
for i in range(2):
|
|
user = User(name='John Doe %s' % i)
|
|
user.email = 'john.doe.%s@example.com' % i
|
|
user.add_roles([role.id])
|
|
user.store()
|
|
users.append(user)
|
|
|
|
assert len(set(role.get_emails())) == 2
|
|
users[-1].is_active = False
|
|
users[-1].store()
|
|
assert len(set(role.get_emails())) == 1
|