import os import unittest from rhodecode.tests import * from rhodecode.model.repos_group import ReposGroupModel from rhodecode.model.repo import RepoModel from rhodecode.model.db import RepoGroup, User from rhodecode.model.meta import Session from sqlalchemy.exc import IntegrityError def _make_group(path, desc='desc', parent_id=None, skip_if_exists=False): gr = RepoGroup.get_by_group_name(path) if gr and skip_if_exists: return gr if isinstance(parent_id, RepoGroup): parent_id = parent_id.group_id gr = ReposGroupModel().create(path, desc, parent_id) return gr def _update_group(id_, group_name, desc='desc', parent_id=None): form_data = _get_group_create_params(group_name=group_name, group_desc=desc, group_parent_id=parent_id) gr = ReposGroupModel().update(id_, form_data) return gr def _make_repo(name, **kwargs): form_data = _get_repo_create_params(repo_name=name, **kwargs) cur_user = User.get_by_username(TEST_USER_ADMIN_LOGIN) r = RepoModel().create(form_data, cur_user) return r def _update_repo(name, **kwargs): form_data = _get_repo_create_params(**kwargs) if not 'repo_name' in kwargs: form_data['repo_name'] = name if not 'perms_new' in kwargs: form_data['perms_new'] = [] if not 'perms_updates' in kwargs: form_data['perms_updates'] = [] r = RepoModel().update(name, **form_data) return r class TestReposGroups(unittest.TestCase): def setUp(self): self.g1 = _make_group('test1', skip_if_exists=True) Session().commit() self.g2 = _make_group('test2', skip_if_exists=True) Session().commit() self.g3 = _make_group('test3', skip_if_exists=True) Session().commit() def tearDown(self): Session.remove() def __check_path(self, *path): """ Checks the path for existance ! """ path = [TESTS_TMP_PATH] + list(path) path = os.path.join(*path) return os.path.isdir(path) def _check_folders(self): print os.listdir(TESTS_TMP_PATH) def __delete_group(self, id_): ReposGroupModel().delete(id_) def test_create_group(self): g = _make_group('newGroup') Session().commit() self.assertEqual(g.full_path, 'newGroup') self.assertTrue(self.__check_path('newGroup')) def test_create_same_name_group(self): self.assertRaises(IntegrityError, lambda: _make_group('newGroup')) Session().rollback() def test_same_subgroup(self): sg1 = _make_group('sub1', parent_id=self.g1.group_id) Session().commit() self.assertEqual(sg1.parent_group, self.g1) self.assertEqual(sg1.full_path, 'test1/sub1') self.assertTrue(self.__check_path('test1', 'sub1')) ssg1 = _make_group('subsub1', parent_id=sg1.group_id) Session().commit() self.assertEqual(ssg1.parent_group, sg1) self.assertEqual(ssg1.full_path, 'test1/sub1/subsub1') self.assertTrue(self.__check_path('test1', 'sub1', 'subsub1')) def test_remove_group(self): sg1 = _make_group('deleteme') Session().commit() self.__delete_group(sg1.group_id) self.assertEqual(RepoGroup.get(sg1.group_id), None) self.assertFalse(self.__check_path('deteteme')) sg1 = _make_group('deleteme', parent_id=self.g1.group_id) Session().commit() self.__delete_group(sg1.group_id) self.assertEqual(RepoGroup.get(sg1.group_id), None) self.assertFalse(self.__check_path('test1', 'deteteme')) def test_rename_single_group(self): sg1 = _make_group('initial') Session().commit() new_sg1 = _update_group(sg1.group_id, 'after') self.assertTrue(self.__check_path('after')) self.assertEqual(RepoGroup.get_by_group_name('initial'), None) def test_update_group_parent(self): sg1 = _make_group('initial', parent_id=self.g1.group_id) Session().commit() new_sg1 = _update_group(sg1.group_id, 'after', parent_id=self.g1.group_id) self.assertTrue(self.__check_path('test1', 'after')) self.assertEqual(RepoGroup.get_by_group_name('test1/initial'), None) new_sg1 = _update_group(sg1.group_id, 'after', parent_id=self.g3.group_id) self.assertTrue(self.__check_path('test3', 'after')) self.assertEqual(RepoGroup.get_by_group_name('test3/initial'), None) new_sg1 = _update_group(sg1.group_id, 'hello') self.assertTrue(self.__check_path('hello')) self.assertEqual(RepoGroup.get_by_group_name('hello'), new_sg1) def test_subgrouping_with_repo(self): g1 = _make_group('g1') g2 = _make_group('g2') Session().commit() # create new repo r = _make_repo('john') Session().commit() self.assertEqual(r.repo_name, 'john') # put repo into group r = _update_repo('john', repo_group=g1.group_id) Session().commit() self.assertEqual(r.repo_name, 'g1/john') _update_group(g1.group_id, 'g1', parent_id=g2.group_id) self.assertTrue(self.__check_path('g2', 'g1')) # test repo self.assertEqual(r.repo_name, RepoGroup.url_sep().join(['g2', 'g1', r.just_name])) def test_move_to_root(self): g1 = _make_group('t11') Session().commit() g2 = _make_group('t22', parent_id=g1.group_id) Session().commit() self.assertEqual(g2.full_path, 't11/t22') self.assertTrue(self.__check_path('t11', 't22')) g2 = _update_group(g2.group_id, 'g22', parent_id=None) Session().commit() self.assertEqual(g2.group_name, 'g22') # we moved out group from t1 to '' so it's full path should be 'g2' self.assertEqual(g2.full_path, 'g22') self.assertFalse(self.__check_path('t11', 't22')) self.assertTrue(self.__check_path('g22')) def test_rename_top_level_group_in_nested_setup(self): g1 = _make_group('L1') Session().commit() g2 = _make_group('L2', parent_id=g1.group_id) Session().commit() g3 = _make_group('L3', parent_id=g2.group_id) Session().commit() r = _make_repo('L1/L2/L3/L3_REPO', repo_group=g3.group_id) Session().commit() ##rename L1 all groups should be now changed _update_group(g1.group_id, 'L1_NEW') Session().commit() self.assertEqual(g1.full_path, 'L1_NEW') self.assertEqual(g2.full_path, 'L1_NEW/L2') self.assertEqual(g3.full_path, 'L1_NEW/L2/L3') self.assertEqual(r.repo_name, 'L1_NEW/L2/L3/L3_REPO') def test_change_parent_of_top_level_group_in_nested_setup(self): g1 = _make_group('R1') Session().commit() g2 = _make_group('R2', parent_id=g1.group_id) Session().commit() g3 = _make_group('R3', parent_id=g2.group_id) Session().commit() g4 = _make_group('R1_NEW') Session().commit() r = _make_repo('R1/R2/R3/R3_REPO', repo_group=g3.group_id) Session().commit() ##rename L1 all groups should be now changed _update_group(g1.group_id, 'R1', parent_id=g4.group_id) Session().commit() self.assertEqual(g1.full_path, 'R1_NEW/R1') self.assertEqual(g2.full_path, 'R1_NEW/R1/R2') self.assertEqual(g3.full_path, 'R1_NEW/R1/R2/R3') self.assertEqual(r.repo_name, 'R1_NEW/R1/R2/R3/R3_REPO') def test_change_parent_of_top_level_group_in_nested_setup_with_rename(self): g1 = _make_group('X1') Session().commit() g2 = _make_group('X2', parent_id=g1.group_id) Session().commit() g3 = _make_group('X3', parent_id=g2.group_id) Session().commit() g4 = _make_group('X1_NEW') Session().commit() r = _make_repo('X1/X2/X3/X3_REPO', repo_group=g3.group_id) Session().commit() ##rename L1 all groups should be now changed _update_group(g1.group_id, 'X1_PRIM', parent_id=g4.group_id) Session().commit() self.assertEqual(g1.full_path, 'X1_NEW/X1_PRIM') self.assertEqual(g2.full_path, 'X1_NEW/X1_PRIM/X2') self.assertEqual(g3.full_path, 'X1_NEW/X1_PRIM/X2/X3') self.assertEqual(r.repo_name, 'X1_NEW/X1_PRIM/X2/X3/X3_REPO')