/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to tests/test_repository.py

  • Committer: Robert Collins
  • Date: 2010-05-06 07:48:22 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506074822-0bsgf2j4h8jx0xkk
Added ``bzrlib.tests.matchers`` as a place to put matchers, along with
our first in-tree matcher. See the module docstring for details.
(Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2007 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
 
 
17
 
"""Tests for interfacing with a Git Repository"""
18
 
 
19
 
import dulwich
20
 
from dulwich.repo import (
21
 
    Repo as GitRepo,
22
 
    )
23
 
import os
24
 
 
25
 
from bzrlib import (
26
 
    errors,
27
 
    revision,
28
 
    )
29
 
from bzrlib.repository import (
30
 
    InterRepository,
31
 
    Repository,
32
 
    )
33
 
 
34
 
from bzrlib.plugins.git import (
35
 
    dir,
36
 
    repository,
37
 
    tests,
38
 
    )
39
 
from bzrlib.plugins.git.mapping import (
40
 
    default_mapping,
41
 
    )
42
 
from bzrlib.plugins.git.object_store import (
43
 
    BazaarObjectStore,
44
 
    )
45
 
from bzrlib.plugins.git.push import (
46
 
    MissingObjectsIterator,
47
 
    )
48
 
 
49
 
class TestGitRepositoryFeatures(tests.TestCaseInTempDir):
50
 
    """Feature tests for GitRepository."""
51
 
 
52
 
    def _do_commit(self):
53
 
        builder = tests.GitBranchBuilder()
54
 
        builder.set_file('a', 'text for a\n', False)
55
 
        commit_handle = builder.commit('Joe Foo <joe@foo.com>', u'message')
56
 
        mapping = builder.finish()
57
 
        return mapping[commit_handle]
58
 
 
59
 
    def test_open_existing(self):
60
 
        GitRepo.init(self.test_dir)
61
 
 
62
 
        repo = Repository.open('.')
63
 
        self.assertIsInstance(repo, repository.GitRepository)
64
 
 
65
 
    def test_has_git_repo(self):
66
 
        GitRepo.init(self.test_dir)
67
 
 
68
 
        repo = Repository.open('.')
69
 
        self.assertIsInstance(repo._git, dulwich.repo.BaseRepo)
70
 
 
71
 
    def test_has_revision(self):
72
 
        GitRepo.init(self.test_dir)
73
 
        commit_id = self._do_commit()
74
 
        repo = Repository.open('.')
75
 
        self.assertFalse(repo.has_revision('foobar'))
76
 
        revid = default_mapping.revision_id_foreign_to_bzr(commit_id)
77
 
        self.assertTrue(repo.has_revision(revid))
78
 
 
79
 
    def test_has_revisions(self):
80
 
        GitRepo.init(self.test_dir)
81
 
        commit_id = self._do_commit()
82
 
        repo = Repository.open('.')
83
 
        self.assertEquals(set(), repo.has_revisions(['foobar']))
84
 
        revid = default_mapping.revision_id_foreign_to_bzr(commit_id)
85
 
        self.assertEquals(set([revid]), repo.has_revisions(['foobar', revid]))
86
 
 
87
 
    def test_get_revision(self):
88
 
        # GitRepository.get_revision gives a Revision object.
89
 
 
90
 
        # Create a git repository with a revision.
91
 
        GitRepo.init(self.test_dir)
92
 
        commit_id = self._do_commit()
93
 
 
94
 
        # Get the corresponding Revision object.
95
 
        revid = default_mapping.revision_id_foreign_to_bzr(commit_id)
96
 
        repo = Repository.open('.')
97
 
        rev = repo.get_revision(revid)
98
 
        self.assertIsInstance(rev, revision.Revision)
99
 
 
100
 
    def test_get_revision_unknown(self):
101
 
        GitRepo.init(self.test_dir)
102
 
 
103
 
        repo = Repository.open('.')
104
 
        self.assertRaises(errors.NoSuchRevision, repo.get_revision, "bla")
105
 
 
106
 
    def simple_commit(self):
107
 
        # Create a git repository with some interesting files in a revision.
108
 
        GitRepo.init(self.test_dir)
109
 
        builder = tests.GitBranchBuilder()
110
 
        builder.set_file('data', 'text\n', False)
111
 
        builder.set_file('executable', 'content', True)
112
 
        builder.set_link('link', 'broken')
113
 
        builder.set_file('subdir/subfile', 'subdir text\n', False)
114
 
        commit_handle = builder.commit('Joe Foo <joe@foo.com>', u'message',
115
 
            timestamp=1205433193)
116
 
        mapping = builder.finish()
117
 
        return mapping[commit_handle]
118
 
 
119
 
    def test_pack(self):
120
 
        commit_id = self.simple_commit()
121
 
        repo = Repository.open('.')
122
 
        repo.pack()
123
 
 
124
 
    def test_revision_tree(self):
125
 
        commit_id = self.simple_commit()
126
 
        revid = default_mapping.revision_id_foreign_to_bzr(commit_id)
127
 
        repo = Repository.open('.')
128
 
        tree = repo.revision_tree(revid)
129
 
        self.assertEquals(tree.get_revision_id(), revid)
130
 
        self.assertEquals("text\n", tree.get_file_text(tree.path2id("data")))
131
 
 
132
 
 
133
 
class TestGitRepository(tests.TestCaseWithTransport):
134
 
 
135
 
    def _do_commit(self):
136
 
        builder = tests.GitBranchBuilder()
137
 
        builder.set_file('a', 'text for a\n', False)
138
 
        commit_handle = builder.commit('Joe Foo <joe@foo.com>', u'message')
139
 
        mapping = builder.finish()
140
 
        return mapping[commit_handle]
141
 
 
142
 
    def setUp(self):
143
 
        tests.TestCaseWithTransport.setUp(self)
144
 
        dulwich.repo.Repo.create(self.test_dir)
145
 
        self.git_repo = Repository.open(self.test_dir)
146
 
 
147
 
    def test_supports_rich_root(self):
148
 
        repo = self.git_repo
149
 
        self.assertEqual(repo.supports_rich_root(), True)
150
 
 
151
 
    def test_get_signature_text(self):
152
 
        self.assertRaises(errors.NoSuchRevision, self.git_repo.get_signature_text, revision.NULL_REVISION)
153
 
 
154
 
    def test_has_signature_for_revision_id(self):
155
 
        self.assertEquals(False, self.git_repo.has_signature_for_revision_id(revision.NULL_REVISION))
156
 
 
157
 
    def test_all_revision_ids_none(self):
158
 
        self.assertEquals(set([]), self.git_repo.all_revision_ids())
159
 
 
160
 
    def test_get_known_graph_ancestry(self):
161
 
        cid = self._do_commit()
162
 
        revid = default_mapping.revision_id_foreign_to_bzr(cid)
163
 
        g = self.git_repo.get_known_graph_ancestry([revid])
164
 
        self.assertEquals(frozenset([revid]),
165
 
            g.heads([revid]))
166
 
        self.assertEqual([(revid, 0, (1,), True)],
167
 
            [(n.key, n.merge_depth, n.revno, n.end_of_merge)
168
 
                 for n in g.merge_sort(revid)])
169
 
 
170
 
    def test_all_revision_ids(self):
171
 
        commit_id = self._do_commit()
172
 
        self.assertEquals(
173
 
                set([default_mapping.revision_id_foreign_to_bzr(commit_id)]),
174
 
                self.git_repo.all_revision_ids())
175
 
 
176
 
    def assertIsNullInventory(self, inv):
177
 
        self.assertEqual(inv.root, None)
178
 
        self.assertEqual(inv.revision_id, revision.NULL_REVISION)
179
 
        self.assertEqual(list(inv.iter_entries()), [])
180
 
 
181
 
    def test_revision_tree_none(self):
182
 
        # GitRepository.revision_tree(None) returns the null tree.
183
 
        repo = self.git_repo
184
 
        tree = repo.revision_tree(revision.NULL_REVISION)
185
 
        self.assertEqual(tree.get_revision_id(), revision.NULL_REVISION)
186
 
 
187
 
    def test_get_parent_map_null(self):
188
 
        self.assertEquals({revision.NULL_REVISION: ()}, 
189
 
                           self.git_repo.get_parent_map([revision.NULL_REVISION]))
190
 
 
191
 
 
192
 
class GitRepositoryFormat(tests.TestCase):
193
 
 
194
 
    def setUp(self):
195
 
        super(GitRepositoryFormat, self).setUp()
196
 
        self.format = repository.GitRepositoryFormat()
197
 
 
198
 
    def test_get_format_description(self):
199
 
        self.assertEquals("Git Repository", self.format.get_format_description())
200
 
 
201
 
 
202
 
class RevisionGistImportTests(tests.TestCaseWithTransport):
203
 
 
204
 
    def setUp(self):
205
 
        tests.TestCaseWithTransport.setUp(self)
206
 
        self.git_path = os.path.join(self.test_dir, "git")
207
 
        os.mkdir(self.git_path)
208
 
        dulwich.repo.Repo.create(self.git_path)
209
 
        self.git_repo = Repository.open(self.git_path)
210
 
        self.bzr_tree = self.make_branch_and_tree("bzr")
211
 
 
212
 
    def get_inter(self):
213
 
        return InterRepository.get(self.bzr_tree.branch.repository, 
214
 
                                   self.git_repo)
215
 
 
216
 
    def object_iter(self):
217
 
        store = BazaarObjectStore(self.bzr_tree.branch.repository, default_mapping)
218
 
        store_iterator = MissingObjectsIterator(store, self.bzr_tree.branch.repository)
219
 
        return store, store_iterator
220
 
 
221
 
    def import_rev(self, revid, parent_lookup=None):
222
 
        store, store_iter = self.object_iter()
223
 
        store._cache.idmap.start_write_group()
224
 
        try:
225
 
            return store_iter.import_revision(revid, roundtrip=False)
226
 
        except:
227
 
            store._cache.idmap.abort_write_group()
228
 
            raise
229
 
        else:
230
 
            store._cache.idmap.commit_write_group()
231
 
 
232
 
    def test_pointless(self):
233
 
        revid = self.bzr_tree.commit("pointless", timestamp=1205433193,
234
 
                timezone=0,
235
 
                  committer="Jelmer Vernooij <jelmer@samba.org>")
236
 
        self.assertEquals("2caa8094a5b794961cd9bf582e3e2bb090db0b14", 
237
 
                self.import_rev(revid))
238
 
        self.assertEquals("2caa8094a5b794961cd9bf582e3e2bb090db0b14", 
239
 
                self.import_rev(revid))
240
 
 
241
 
 
242
 
class ForeignTestsRepositoryFactory(object):
243
 
 
244
 
    def make_repository(self, transport):
245
 
        return dir.LocalGitControlDirFormat().initialize_on_transport(transport).open_repository()