test_users.py
124 lines
| 4.3 KiB
| text/x-python
|
PythonLexer
r2527 | import unittest | |||
from rhodecode.tests import * | ||||
|
r3417 | from rhodecode.model.db import User, UserGroup, UserGroupMember, UserEmailMap,\ | ||
r2527 | Permission | |||
from rhodecode.model.user import UserModel | ||||
from rhodecode.model.meta import Session | ||||
|
r3417 | from rhodecode.model.users_group import UserGroupModel | ||
r2527 | ||||
class TestUser(unittest.TestCase): | ||||
def __init__(self, methodName='runTest'): | ||||
Session.remove() | ||||
super(TestUser, self).__init__(methodName=methodName) | ||||
def test_create_and_remove(self): | ||||
usr = UserModel().create_or_update(username=u'test_user', | ||||
password=u'qweqwe', | ||||
email=u'u232@rhodecode.org', | ||||
firstname=u'u1', lastname=u'u1') | ||||
Session().commit() | ||||
self.assertEqual(User.get_by_username(u'test_user'), usr) | ||||
r3415 | # make user group | |||
|
r3417 | users_group = UserGroupModel().create('some_example_group') | ||
r2527 | Session().commit() | |||
|
r3417 | UserGroupModel().add_user_to_group(users_group, usr) | ||
r2527 | Session().commit() | |||
|
r3417 | self.assertEqual(UserGroup.get(users_group.users_group_id), users_group) | ||
self.assertEqual(UserGroupMember.query().count(), 1) | ||||
r2527 | UserModel().delete(usr.user_id) | |||
Session().commit() | ||||
|
r3417 | self.assertEqual(UserGroupMember.query().all(), []) | ||
r2527 | ||||
def test_additonal_email_as_main(self): | ||||
usr = UserModel().create_or_update(username=u'test_user', | ||||
password=u'qweqwe', | ||||
email=u'main_email@rhodecode.org', | ||||
firstname=u'u1', lastname=u'u1') | ||||
Session().commit() | ||||
def do(): | ||||
m = UserEmailMap() | ||||
m.email = u'main_email@rhodecode.org' | ||||
m.user = usr | ||||
Session().add(m) | ||||
Session().commit() | ||||
self.assertRaises(AttributeError, do) | ||||
UserModel().delete(usr.user_id) | ||||
Session().commit() | ||||
def test_extra_email_map(self): | ||||
usr = UserModel().create_or_update(username=u'test_user', | ||||
password=u'qweqwe', | ||||
email=u'main_email@rhodecode.org', | ||||
firstname=u'u1', lastname=u'u1') | ||||
Session().commit() | ||||
m = UserEmailMap() | ||||
m.email = u'main_email2@rhodecode.org' | ||||
m.user = usr | ||||
Session().add(m) | ||||
Session().commit() | ||||
u = User.get_by_email(email='main_email@rhodecode.org') | ||||
self.assertEqual(usr.user_id, u.user_id) | ||||
self.assertEqual(usr.username, u.username) | ||||
u = User.get_by_email(email='main_email2@rhodecode.org') | ||||
self.assertEqual(usr.user_id, u.user_id) | ||||
self.assertEqual(usr.username, u.username) | ||||
u = User.get_by_email(email='main_email3@rhodecode.org') | ||||
self.assertEqual(None, u) | ||||
UserModel().delete(usr.user_id) | ||||
Session().commit() | ||||
class TestUsers(unittest.TestCase): | ||||
def __init__(self, methodName='runTest'): | ||||
super(TestUsers, self).__init__(methodName=methodName) | ||||
def setUp(self): | ||||
self.u1 = UserModel().create_or_update(username=u'u1', | ||||
password=u'qweqwe', | ||||
email=u'u1@rhodecode.org', | ||||
firstname=u'u1', lastname=u'u1') | ||||
def tearDown(self): | ||||
perm = Permission.query().all() | ||||
for p in perm: | ||||
UserModel().revoke_perm(self.u1, p) | ||||
UserModel().delete(self.u1) | ||||
Session().commit() | ||||
def test_add_perm(self): | ||||
perm = Permission.query().all()[0] | ||||
UserModel().grant_perm(self.u1, perm) | ||||
Session().commit() | ||||
self.assertEqual(UserModel().has_perm(self.u1, perm), True) | ||||
def test_has_perm(self): | ||||
perm = Permission.query().all() | ||||
for p in perm: | ||||
has_p = UserModel().has_perm(self.u1, p) | ||||
self.assertEqual(False, has_p) | ||||
def test_revoke_perm(self): | ||||
perm = Permission.query().all()[0] | ||||
UserModel().grant_perm(self.u1, perm) | ||||
Session().commit() | ||||
self.assertEqual(UserModel().has_perm(self.u1, perm), True) | ||||
#revoke | ||||
UserModel().revoke_perm(self.u1, perm) | ||||
Session().commit() | ||||
self.assertEqual(UserModel().has_perm(self.u1, perm), False) | ||||