1
# Copyright (C) 2009 Jelmer Vernooij <jelmer@samba.org>
2
# -*- coding: utf-8 -*-
4
# This program is free software; you can redistribute it and/or modify
5
# it under the terms of the GNU General Public License as published by
6
# the Free Software Foundation; either version 2 of the License, or
7
# (at your option) any later version.
9
# This program is distributed in the hope that it will be useful,
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12
# GNU General Public License for more details.
14
# You should have received a copy of the GNU General Public License
15
# along with this program; if not, write to the Free Software
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
18
from dulwich.objects import (
22
from dulwich.repo import (
32
from bzrlib.bzrdir import (
35
from bzrlib.inventory import (
38
from bzrlib.repository import (
41
from bzrlib.tests import (
42
TestCaseWithTransport,
44
from bzrlib.transport import (
48
from bzrlib.plugins.git import (
51
from bzrlib.plugins.git.fetch import (
55
from bzrlib.plugins.git.mapping import (
59
from bzrlib.plugins.git.shamap import (
62
from bzrlib.plugins.git.tests import (
67
class RepositoryFetchTests:
69
def make_git_repo(self, path):
73
def clone_git_repo(self, from_url, to_url, revision_id=None):
74
oldrepos = self.open_git_repo(from_url)
75
dir = BzrDir.create(to_url, get_rich_root_format())
76
newrepos = dir.create_repository()
77
oldrepos.copy_content_into(newrepos, revision_id=revision_id)
81
self.make_git_repo("d")
82
newrepos = self.clone_git_repo("d", "f")
83
self.assertEquals([], newrepos.all_revision_ids())
85
def make_onerev_branch(self):
86
self.make_git_repo("d")
88
bb = GitBranchBuilder()
89
bb.set_file("foobar", "foo\nbar\n", False)
90
mark = bb.commit("Somebody <somebody@someorg.org>", "mymsg")
91
gitsha = bb.finish()[mark]
95
def test_single_rev(self):
96
path, gitsha = self.make_onerev_branch()
97
oldrepo = self.open_git_repo(path)
98
newrepo = self.clone_git_repo(path, "f")
99
self.assertEquals([oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha)], newrepo.all_revision_ids())
101
def test_single_rev_specific(self):
102
path, gitsha = self.make_onerev_branch()
103
oldrepo = self.open_git_repo(path)
104
revid = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha)
105
newrepo = self.clone_git_repo(path, "f", revision_id=revid)
106
self.assertEquals([revid], newrepo.all_revision_ids())
108
def test_incremental(self):
109
self.make_git_repo("d")
111
bb = GitBranchBuilder()
112
bb.set_file("foobar", "foo\nbar\n", False)
113
mark1 = bb.commit("Somebody <somebody@someorg.org>", "mymsg")
114
bb.set_file("foobar", "fooll\nbar\n", False)
115
mark2 = bb.commit("Somebody <somebody@someorg.org>", "nextmsg")
117
gitsha1 = marks[mark1]
118
gitsha2 = marks[mark2]
120
oldrepo = self.open_git_repo("d")
121
revid1 = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha1)
122
newrepo = self.clone_git_repo("d", "f", revision_id=revid1)
123
self.assertEquals([revid1], newrepo.all_revision_ids())
124
revid2 = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha2)
125
newrepo.fetch(oldrepo, revision_id=revid2)
126
self.assertEquals(set([revid1, revid2]), set(newrepo.all_revision_ids()))
128
def test_executable(self):
129
self.make_git_repo("d")
131
bb = GitBranchBuilder()
132
bb.set_file("foobar", "foo\nbar\n", True)
133
bb.set_file("notexec", "foo\nbar\n", False)
134
mark = bb.commit("Somebody <somebody@someorg.org>", "mymsg")
135
gitsha = bb.finish()[mark]
137
oldrepo = self.open_git_repo("d")
138
newrepo = self.clone_git_repo("d", "f")
139
revid = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha)
140
tree = newrepo.revision_tree(revid)
141
self.assertTrue(tree.has_filename("foobar"))
142
self.assertEquals(True, tree.inventory[tree.path2id("foobar")].executable)
143
self.assertTrue(tree.has_filename("notexec"))
144
self.assertEquals(False, tree.inventory[tree.path2id("notexec")].executable)
146
def test_non_ascii_characters(self):
147
self.make_git_repo("d")
149
bb = GitBranchBuilder()
150
bb.set_file(u"foőbar", "foo\nbar\n", False)
151
mark = bb.commit("Somebody <somebody@someorg.org>", "mymsg")
152
gitsha = bb.finish()[mark]
154
oldrepo = self.open_git_repo("d")
155
newrepo = self.clone_git_repo("d", "f")
156
revid = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha)
157
tree = newrepo.revision_tree(revid)
158
self.assertTrue(tree.has_filename(u"foőbar"))
161
class LocalRepositoryFetchTests(RepositoryFetchTests, TestCaseWithTransport):
163
def open_git_repo(self, path):
164
return Repository.open(path)
167
class ImportObjects(TestCaseWithTransport):
170
super(ImportObjects, self).setUp()
171
self._map = DictGitShaMap()
172
self._mapping = BzrGitMappingv1()
173
factory = knit.make_file_factory(True, versionedfile.PrefixMapper())
174
self._texts = factory(self.get_transport('texts'))
176
def test_import_blob_simple(self):
177
blob = Blob.from_string("bar")
178
base_inv = Inventory()
179
objs = { "blobname": blob}
180
ret, _= import_git_blob(self._texts, self._mapping, "bla", "blobname",
181
base_inv, None, "somerevid", [], self._map, objs.__getitem__, False,
183
self.assertEquals(set([('bla', 'somerevid')]), self._texts.keys())
184
self.assertEquals(self._texts.get_record_stream([('bla', 'somerevid')],
185
"unordered", True).next().get_bytes_as("fulltext"), "bar")
186
self.assertEquals(1, len(ret))
187
self.assertEquals(None, ret[0][0])
188
self.assertEquals("bla", ret[0][1])
190
self.assertEquals(False, ie.executable)
191
self.assertEquals("file", ie.kind)
192
self.assertEquals("somerevid", ie.revision)
193
self.assertEquals(osutils.sha_strings(["bar"]), ie.text_sha1)
195
def test_import_tree_empty_root(self):
196
base_inv = Inventory(root_id=None)
199
ret, _, _ = import_git_tree(self._texts, self._mapping, "",
201
None, "somerevid", [], self._map, {tree.id: tree}.__getitem__)
202
self.assertEquals(set([("TREE_ROOT", 'somerevid')]), self._texts.keys())
203
self.assertEquals(1, len(ret))
204
self.assertEquals(None, ret[0][0])
205
self.assertEquals("", ret[0][1])
207
self.assertEquals(False, ie.executable)
208
self.assertEquals("directory", ie.kind)
209
self.assertEquals({}, ie.children)
210
self.assertEquals("somerevid", ie.revision)
211
self.assertEquals(None, ie.text_sha1)
213
def test_import_tree_empty(self):
214
base_inv = Inventory()
217
ret, _, _ = import_git_tree(self._texts, self._mapping, "bla",
218
tree.id, base_inv, None, "somerevid", [],
219
self._map, { tree.id: tree }.__getitem__)
220
self.assertEquals(set([("bla", 'somerevid')]), self._texts.keys())
221
self.assertEquals(1, len(ret))
222
self.assertEquals(None, ret[0][0])
223
self.assertEquals("bla", ret[0][1])
225
self.assertEquals("directory", ie.kind)
226
self.assertEquals(False, ie.executable)
227
self.assertEquals({}, ie.children)
228
self.assertEquals("somerevid", ie.revision)
229
self.assertEquals(None, ie.text_sha1)
231
def test_import_tree_with_file(self):
232
base_inv = Inventory()
233
blob = Blob.from_string("bar1")
235
tree.add(0100600, "foo", blob.id)
237
objects = { blob.id: blob, tree.id: tree }
238
ret, _, _ = import_git_tree(self._texts, self._mapping, "bla", tree.id,
239
base_inv, None, "somerevid", [], self._map, objects.__getitem__)
240
self.assertEquals(2, len(ret))
241
self.assertEquals(None, ret[0][0])
242
self.assertEquals("bla", ret[0][1])
243
self.assertEquals(None, ret[1][0])
244
self.assertEquals("bla/foo", ret[1][1])
246
self.assertEquals("directory", ie.kind)
248
self.assertEquals("file", ie.kind)
249
self.assertEquals("bla/foo", ie.file_id)
250
self.assertEquals("somerevid", ie.revision)
251
self.assertEquals(osutils.sha_strings(["bar1"]), ie.text_sha1)
252
self.assertEquals(False, ie.executable)
254
def test_import_tree_with_file_exe(self):
255
base_inv = Inventory(root_id=None)
256
blob = Blob.from_string("bar")
258
tree.add(0100755, "foo", blob.id)
260
objects = { blob.id: blob, tree.id: tree }
261
ret, _, _ = import_git_tree(self._texts, self._mapping, "", tree.id,
262
base_inv, None, "somerevid", [], self._map, objects.__getitem__)
263
self.assertEquals(2, len(ret))
264
self.assertEquals(None, ret[0][0])
265
self.assertEquals("", ret[0][1])
266
self.assertEquals(None, ret[1][0])
267
self.assertEquals("foo", ret[1][1])
269
self.assertEquals("directory", ie.kind)
271
self.assertEquals("file", ie.kind)
272
self.assertEquals(True, ie.executable)
added
removed
tests/test_fetch.py
