# HG changeset patch # User Marcin Kuzminski # Date 2016-12-29 13:32:00 # Node ID 55bee39a892359da5fd6f4eb6f75960128cedad5 # Parent 59c22420943c045af68a1017d484fd71f4cc89de tests: added creation of repos for user-util fixture. diff --git a/rhodecode/tests/plugin.py b/rhodecode/tests/plugin.py --- a/rhodecode/tests/plugin.py +++ b/rhodecode/tests/plugin.py @@ -1138,6 +1138,7 @@ class UserUtility(object): self._test_name = self._sanitize_name(test_name) self.fixture = Fixture() self.repo_group_ids = [] + self.repos_ids = [] self.user_ids = [] self.user_group_ids = [] self.user_repo_permission_ids = [] @@ -1164,6 +1165,17 @@ class UserUtility(object): self.repo_group_ids.append(repo_group.group_id) return repo_group + def create_repo(self, owner=TEST_USER_ADMIN_LOGIN, parent=None, auto_cleanup=True): + repo_name = "{prefix}_repository_{count}".format( + prefix=self._test_name, + count=len(self.repos_ids)) + + repository = self.fixture.create_repo( + repo_name, cur_user=owner, repo_group=parent) + if auto_cleanup: + self.repos_ids.append(repository.repo_id) + return repository + def create_user(self, auto_cleanup=True, **kwargs): user_name = "{prefix}_user_{count}".format( prefix=self._test_name, @@ -1254,6 +1266,7 @@ class UserUtility(object): def cleanup(self): self._cleanup_permissions() + self._cleanup_repos() self._cleanup_repo_groups() self._cleanup_user_groups() self._cleanup_users() @@ -1299,6 +1312,11 @@ class UserUtility(object): for repo_group_id in sorted_repo_group_ids: self.fixture.destroy_repo_group(repo_group_id) + def _cleanup_repos(self): + sorted_repos_ids = sorted(self.repos_ids) + for repo_id in sorted_repos_ids: + self.fixture.destroy_repo(repo_id) + def _cleanup_user_groups(self): def _user_group_compare(first_group_id, second_group_id): """