1
# Copyright (C) 2007 Canonical Ltd
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.
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.
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
18
"""Tests for interfacing with a Git Branch"""
22
from dulwich.objects import (
26
from dulwich.repo import (
37
version_info as bzrlib_version,
39
from bzrlib.branch import (
43
from bzrlib.bzrdir import (
46
from bzrlib.repository import (
49
from bzrlib.symbol_versioning import (
52
from bzrlib.tests import TestSkipped
54
from bzrlib.plugins.git import (
58
from bzrlib.plugins.git.dir import (
59
LocalGitControlDirFormat,
61
from bzrlib.plugins.git.mapping import (
66
class TestGitBranch(tests.TestCaseInTempDir):
68
def test_open_by_ref(self):
71
urlutils.local_path_to_url(self.test_dir),
72
urllib.quote("refs/remotes/origin/unstable", safe='')
74
if bzrlib_version < (2, 5, 0):
75
self.assertRaises(errors.NotBranchError, BzrDir.open, url)
76
raise TestSkipped("opening by ref not supported with bzr < 2.5")
79
self.assertEquals(b.ref, "refs/remotes/origin/unstable")
81
def test_open_existing(self):
84
thebranch = d.create_branch()
85
self.assertIsInstance(thebranch, branch.GitBranch)
90
thebranch = d.create_branch()
92
"<LocalGitBranch('%s/', u'master')>" % (
93
urlutils.local_path_to_url(self.test_dir),),
96
def test_last_revision_is_null(self):
98
thedir = BzrDir.open('.')
99
thebranch = thedir.create_branch()
100
self.assertEqual(revision.NULL_REVISION, thebranch.last_revision())
101
self.assertEqual((0, revision.NULL_REVISION),
102
thebranch.last_revision_info())
104
def simple_commit_a(self):
105
r = GitRepo.init('.')
106
self.build_tree(['a'])
108
return r.do_commit("a", committer="Somebody <foo@example.com>")
110
def test_last_revision_is_valid(self):
111
head = self.simple_commit_a()
112
thebranch = Branch.open('.')
113
self.assertEqual(default_mapping.revision_id_foreign_to_bzr(head),
114
thebranch.last_revision())
116
def test_revision_history(self):
117
reva = self.simple_commit_a()
118
self.build_tree(['b'])
121
revb = r.do_commit("b", committer="Somebody <foo@example.com>")
123
thebranch = Branch.open('.')
124
(warnings, history) = self.callCatchWarnings(thebranch.revision_history)
127
(len(warnings) == 1 and isinstance(warnings[0], DeprecationWarning)),
129
self.assertEqual([default_mapping.revision_id_foreign_to_bzr(r) for r in (reva, revb)],
132
def test_tag_annotated(self):
133
reva = self.simple_commit_a()
136
o.tagger = "Jelmer <foo@example.com>"
137
o.message = "add tag"
138
o.object = (Commit, reva)
142
r.object_store.add_object(o)
143
r['refs/tags/foo'] = o.id
144
thebranch = Branch.open('.')
145
self.assertEquals({"foo": default_mapping.revision_id_foreign_to_bzr(reva)},
146
thebranch.tags.get_tag_dict())
149
reva = self.simple_commit_a()
151
r.refs["refs/tags/foo"] = reva
152
thebranch = Branch.open('.')
153
self.assertEquals({"foo": default_mapping.revision_id_foreign_to_bzr(reva)},
154
thebranch.tags.get_tag_dict())
158
class TestWithGitBranch(tests.TestCaseWithTransport):
161
tests.TestCaseWithTransport.setUp(self)
162
dulwich.repo.Repo.create(self.test_dir)
163
d = BzrDir.open(self.test_dir)
164
self.git_branch = d.create_branch()
166
def test_get_parent(self):
167
self.assertIs(None, self.git_branch.get_parent())
169
def test_get_stacked_on_url(self):
170
self.assertRaises(errors.UnstackableBranchFormat,
171
self.git_branch.get_stacked_on_url)
173
def test_get_physical_lock_status(self):
174
self.assertFalse(self.git_branch.get_physical_lock_status())
177
class TestGitBranchFormat(tests.TestCase):
180
super(TestGitBranchFormat, self).setUp()
181
self.format = branch.GitBranchFormat()
183
def test_get_format_description(self):
184
self.assertEquals("Git Branch", self.format.get_format_description())
186
def test_get_network_name(self):
187
self.assertEquals("git", self.format.network_name())
189
def test_supports_tags(self):
190
self.assertTrue(self.format.supports_tags())
193
class BranchTests(tests.TestCaseInTempDir):
195
def make_onerev_branch(self):
199
bb = tests.GitBranchBuilder()
200
bb.set_file("foobar", "foo\nbar\n", False)
201
mark = bb.commit("Somebody <somebody@someorg.org>", "mymsg")
202
gitsha = bb.finish()[mark]
204
return os.path.abspath("d"), gitsha
206
def make_tworev_branch(self):
210
bb = tests.GitBranchBuilder()
211
bb.set_file("foobar", "foo\nbar\n", False)
212
mark1 = bb.commit("Somebody <somebody@someorg.org>", "mymsg")
213
mark2 = bb.commit("Somebody <somebody@someorg.org>", "mymsg")
216
return "d", (marks[mark1], marks[mark2])
218
def clone_git_branch(self, from_url, to_url):
219
from_dir = BzrDir.open(from_url)
220
to_dir = from_dir.sprout(to_url)
221
return to_dir.open_branch()
223
def test_single_rev(self):
224
path, gitsha = self.make_onerev_branch()
225
oldrepo = Repository.open(path)
226
revid = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha)
227
self.assertEquals(gitsha, oldrepo._git.get_refs()["refs/heads/master"])
228
newbranch = self.clone_git_branch(path, "f")
229
self.assertEquals([revid], newbranch.repository.all_revision_ids())
231
def test_sprouted_tags(self):
232
path, gitsha = self.make_onerev_branch()
234
r.refs["refs/tags/lala"] = r.head()
235
oldrepo = Repository.open(path)
236
revid = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha)
237
newbranch = self.clone_git_branch(path, "f")
238
self.assertEquals({"lala": revid}, newbranch.tags.get_tag_dict())
239
self.assertEquals([revid], newbranch.repository.all_revision_ids())
241
def test_interbranch_pull(self):
242
path, (gitsha1, gitsha2) = self.make_tworev_branch()
243
oldrepo = Repository.open(path)
244
revid2 = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha2)
245
newbranch = self.make_branch('g')
246
inter_branch = InterBranch.get(Branch.open(path), newbranch)
248
self.assertEquals(revid2, newbranch.last_revision())
250
def test_interbranch_pull_noop(self):
251
path, (gitsha1, gitsha2) = self.make_tworev_branch()
252
oldrepo = Repository.open(path)
253
revid2 = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha2)
254
newbranch = self.make_branch('g')
255
inter_branch = InterBranch.get(Branch.open(path), newbranch)
257
# This is basically "assertNotRaises"
259
self.assertEquals(revid2, newbranch.last_revision())
261
def test_interbranch_pull_stop_revision(self):
262
path, (gitsha1, gitsha2) = self.make_tworev_branch()
263
oldrepo = Repository.open(path)
264
revid1 = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha1)
265
newbranch = self.make_branch('g')
266
inter_branch = InterBranch.get(Branch.open(path), newbranch)
267
inter_branch.pull(stop_revision=revid1)
268
self.assertEquals(revid1, newbranch.last_revision())
270
def test_interbranch_pull_with_tags(self):
271
path, (gitsha1, gitsha2) = self.make_tworev_branch()
272
gitrepo = GitRepo(path)
273
gitrepo.refs["refs/tags/sometag"] = gitsha2
274
oldrepo = Repository.open(path)
275
revid1 = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha1)
276
revid2 = oldrepo.get_mapping().revision_id_foreign_to_bzr(gitsha2)
277
newbranch = self.make_branch('g')
278
source_branch = Branch.open(path)
279
source_branch.get_config().set_user_option("branch.fetch_tags", True)
280
inter_branch = InterBranch.get(source_branch, newbranch)
281
inter_branch.pull(stop_revision=revid1)
282
self.assertEquals(revid1, newbranch.last_revision())
283
self.assertTrue(newbranch.repository.has_revision(revid2))
286
class ForeignTestsBranchFactory(object):
288
def make_empty_branch(self, transport):
289
d = LocalGitControlDirFormat().initialize_on_transport(transport)
290
return d.create_branch()
292
make_branch = make_empty_branch