|
|
import os
|
|
|
from sqlalchemy.exc import IntegrityError
|
|
|
from rhodecode.model.db import User
|
|
|
|
|
|
from rhodecode.tests import *
|
|
|
from rhodecode.tests.fixture import Fixture
|
|
|
|
|
|
from rhodecode.model.user_group import UserGroupModel
|
|
|
from rhodecode.model.meta import Session
|
|
|
|
|
|
|
|
|
fixture = Fixture()
|
|
|
|
|
|
|
|
|
class TestUserGroups(BaseTestCase):
|
|
|
|
|
|
def tearDown(self):
|
|
|
# delete all groups
|
|
|
for gr in UserGroupModel.get_all():
|
|
|
fixture.destroy_user_group(gr)
|
|
|
Session().commit()
|
|
|
|
|
|
|
|
|
@parameterized.expand([
|
|
|
([], [], [], [], []),
|
|
|
([], ['regular'], [], [], ['regular']), # no changes of regular
|
|
|
(['some_other'], [], [], ['some_other'], []), # not added to regular group
|
|
|
([], ['regular'], ['container'], ['container'], ['regular', 'container']),
|
|
|
([], ['regular'], [], ['container', 'container2'], ['regular', 'container', 'container2']),
|
|
|
([], ['regular'], ['other'], [], ['regular']), # remove not used
|
|
|
(['some_other'], ['regular'], ['other', 'container'], ['container', 'container2'], ['regular', 'container', 'container2']),
|
|
|
])
|
|
|
def test_enforce_groups(self, pre_existing, regular_should_be,
|
|
|
external_should_be, groups, expected):
|
|
|
# delete all groups
|
|
|
for gr in UserGroupModel.get_all():
|
|
|
fixture.destroy_user_group(gr)
|
|
|
Session().commit()
|
|
|
|
|
|
user = User.get_by_username(TEST_USER_REGULAR_LOGIN)
|
|
|
for gr in pre_existing:
|
|
|
gr = fixture.create_user_group(gr)
|
|
|
Session().commit()
|
|
|
|
|
|
# make sure use is just in those groups
|
|
|
for gr in regular_should_be:
|
|
|
gr = fixture.create_user_group(gr)
|
|
|
Session().commit()
|
|
|
UserGroupModel().add_user_to_group(gr, user)
|
|
|
Session().commit()
|
|
|
|
|
|
# now special external groups created by auth plugins
|
|
|
for gr in external_should_be:
|
|
|
gr = fixture.create_user_group(gr, user_group_data={'extern_type': 'container'})
|
|
|
Session().commit()
|
|
|
UserGroupModel().add_user_to_group(gr, user)
|
|
|
Session().commit()
|
|
|
|
|
|
UserGroupModel().enforce_groups(user, groups, 'container')
|
|
|
Session().commit()
|
|
|
|
|
|
user = User.get_by_username(TEST_USER_REGULAR_LOGIN)
|
|
|
in_groups = user.group_member
|
|
|
self.assertEqual(expected, [x.users_group.users_group_name for x in in_groups])
|
|
|
|
