Mercurial > kallithea

comparison rhodecode/tests/models/common.py @ 3647:8a86836fad64 beta

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
more usage of fixture tools in tests
author Marcin Kuzminski <marcin@python-works.com>
date Mon, 01 Apr 2013 23:45:25 +0200
parents b4daef4cc26d
children 5067d6e826a5
comparison
equal deleted inserted replaced
3646:63e49418a4cc 3647:8a86836fad64
1 import os 1 import os
2 import unittest 2 import unittest
3 import functools 3 import functools
4 from rhodecode.tests import * 4 from rhodecode.tests import *
5 5 from rhodecode.tests.fixture import Fixture
6 6
7 from rhodecode.model.repos_group import ReposGroupModel 7 from rhodecode.model.repos_group import ReposGroupModel
8 from rhodecode.model.repo import RepoModel 8 from rhodecode.model.repo import RepoModel
9 from rhodecode.model.db import RepoGroup, Repository, User 9 from rhodecode.model.db import RepoGroup, Repository, User
10 from rhodecode.model.user import UserModel 10 from rhodecode.model.user import UserModel
11 11
12 from rhodecode.lib.auth import AuthUser 12 from rhodecode.lib.auth import AuthUser
13 from rhodecode.model.meta import Session 13 from rhodecode.model.meta import Session
14 14
15 15
16 def _make_group(path, desc='desc', parent_id=None, 16 fixture = Fixture()
17 skip_if_exists=False):
18
19 gr = RepoGroup.get_by_group_name(path)
20 if gr and skip_if_exists:
21 return gr
22 if isinstance(parent_id, RepoGroup):
23 parent_id = parent_id.group_id
24 gr = ReposGroupModel().create(path, desc, TEST_USER_ADMIN_LOGIN, parent_id)
25 return gr
26
27
28 def _make_repo(name, repos_group=None, repo_type='hg', private=False):
29 return RepoModel().create_repo(name, repo_type, 'desc',
30 TEST_USER_ADMIN_LOGIN,
31 repos_group=repos_group,
32 private=private)
33 17
34 18
35 def _destroy_project_tree(test_u1_id): 19 def _destroy_project_tree(test_u1_id):
36 Session.remove() 20 Session.remove()
37 repos_group = RepoGroup.get_by_group_name(group_name='g0') 21 repos_group = RepoGroup.get_by_group_name(group_name='g0')
73 """ 57 """
74 test_u1 = UserModel().create_or_update( 58 test_u1 = UserModel().create_or_update(
75 username=u'test_u1', password=u'qweqwe', 59 username=u'test_u1', password=u'qweqwe',
76 email=u'test_u1@rhodecode.org', firstname=u'test_u1', lastname=u'test_u1' 60 email=u'test_u1@rhodecode.org', firstname=u'test_u1', lastname=u'test_u1'
77 ) 61 )
78 g0 = _make_group('g0') 62 g0 = fixture.create_group('g0')
79 g0_1 = _make_group('g0_1', parent_id=g0) 63 g0_1 = fixture.create_group('g0_1', group_parent_id=g0)
80 g0_1_1 = _make_group('g0_1_1', parent_id=g0_1) 64 g0_1_1 = fixture.create_group('g0_1_1', group_parent_id=g0_1)
81 g0_1_1_r1 = _make_repo('g0/g0_1/g0_1_1/g0_1_1_r1', repos_group=g0_1_1) 65 g0_1_1_r1 = fixture.create_repo('g0/g0_1/g0_1_1/g0_1_1_r1', repos_group=g0_1_1)
82 g0_1_1_r2 = _make_repo('g0/g0_1/g0_1_1/g0_1_1_r2', repos_group=g0_1_1) 66 g0_1_1_r2 = fixture.create_repo('g0/g0_1/g0_1_1/g0_1_1_r2', repos_group=g0_1_1)
83 g0_1_r1 = _make_repo('g0/g0_1/g0_1_r1', repos_group=g0_1) 67 g0_1_r1 = fixture.create_repo('g0/g0_1/g0_1_r1', repos_group=g0_1)
84 g0_2 = _make_group('g0_2', parent_id=g0) 68 g0_2 = fixture.create_group('g0_2', group_parent_id=g0)
85 g0_2_r1 = _make_repo('g0/g0_2/g0_2_r1', repos_group=g0_2) 69 g0_2_r1 = fixture.create_repo('g0/g0_2/g0_2_r1', repos_group=g0_2)
86 g0_2_r2 = _make_repo('g0/g0_2/g0_2_r2', repos_group=g0_2) 70 g0_2_r2 = fixture.create_repo('g0/g0_2/g0_2_r2', repos_group=g0_2)
87 g0_3 = _make_group('g0_3', parent_id=g0) 71 g0_3 = fixture.create_group('g0_3', group_parent_id=g0)
88 g0_3_r1 = _make_repo('g0/g0_3/g0_3_r1', repos_group=g0_3) 72 g0_3_r1 = fixture.create_repo('g0/g0_3/g0_3_r1', repos_group=g0_3)
89 g0_3_r2_private = _make_repo('g0/g0_3/g0_3_r1_private', repos_group=g0_3, 73 g0_3_r2_private = fixture.create_repo('g0/g0_3/g0_3_r1_private',
90 private=True) 74 repos_group=g0_3, repo_private=True)
91 return test_u1 75 return test_u1
92 76
93 77
94 def expected_count(group_name, objects=False): 78 def expected_count(group_name, objects=False):
95 repos_group = RepoGroup.get_by_group_name(group_name=group_name) 79 repos_group = RepoGroup.get_by_group_name(group_name=group_name)