1
# (C) 2005 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
from bzrlib.branch import Branch
19
from bzrlib.clone import copy_branch
20
from bzrlib.commit import commit
21
from bzrlib.errors import NoSuchRevision, UnlistableBranch
22
from bzrlib.selftest import TestCaseInTempDir
23
from bzrlib.trace import mutter
26
class TestBranch(TestCaseInTempDir):
28
def test_append_revisions(self):
29
"""Test appending more than one revision"""
30
br = Branch.initialize(".")
31
br.append_revision("rev1")
32
self.assertEquals(br.revision_history(), ["rev1",])
33
br.append_revision("rev2", "rev3")
34
self.assertEquals(br.revision_history(), ["rev1", "rev2", "rev3"])
37
class TestFetch(TestCaseInTempDir):
39
def test_fetch_revisions(self):
40
"""Test fetch-revision operation."""
41
from bzrlib.fetch import Fetcher
44
b1 = Branch.initialize('b1')
45
b2 = Branch.initialize('b2')
46
file(os.sep.join(['b1', 'foo']), 'w').write('hello')
47
b1.add(['foo'], ['foo-id'])
48
b1.commit('lala!', rev_id='revision-1', allow_pointless=False)
51
f = Fetcher(from_branch=b1, to_branch=b2)
52
eq = self.assertEquals
54
eq(f.last_revision, 'revision-1')
56
rev = b2.get_revision('revision-1')
57
tree = b2.revision_tree('revision-1')
58
eq(tree.get_file_text('foo-id'), 'hello')
60
def test_push_stores(self):
61
"""Copy the stores from one branch to another"""
63
br_a = Branch.initialize("a")
64
file('a/b', 'wb').write('b')
66
commit(br_a, "silly commit")
69
br_b = Branch.initialize("b")
70
self.assertRaises(NoSuchRevision, br_b.get_revision,
71
br_a.revision_history()[0])
72
br_a.push_stores(br_b)
73
rev = br_b.get_revision(br_a.revision_history()[0])
74
tree = br_b.revision_tree(br_a.revision_history()[0])
76
if tree.inventory[file_id].kind == "file":
77
tree.get_file(file_id).read()
80
def test_copy_branch(self):
81
"""Copy the stores from one branch to another"""
82
br_a, br_b = self.test_push_stores()
83
commit(br_b, "silly commit")
85
br_c = copy_branch(br_a, 'c', basis_branch=br_b)
86
self.assertEqual(br_a.revision_history(), br_c.revision_history())
87
## # basis branches currently disabled for weave format
88
## self.assertFalse(br_b.last_revision() in br_c.revision_history())
89
## br_c.get_revision(br_b.last_revision())
91
def test_copy_partial(self):
92
"""Copy only part of the history of a branch."""
93
self.build_tree(['a/', 'a/one'])
94
br_a = Branch.initialize('a')
96
br_a.commit('commit one', rev_id='u@d-1')
97
self.build_tree(['a/two'])
99
br_a.commit('commit two', rev_id='u@d-2')
100
br_b = copy_branch(br_a, 'b', revision='u@d-1')
101
self.assertEqual(br_b.last_revision(), 'u@d-1')
102
self.assertTrue(os.path.exists('b/one'))
103
self.assertFalse(os.path.exists('b/two'))
106
def test_record_initial_ghost_merge(self):
107
"""A pending merge with no revision present is still a merge."""
108
branch = Branch.initialize('.')
109
branch.add_pending_merge('non:existent@rev--ision--0--2')
110
branch.commit('pretend to merge nonexistent-revision', rev_id='first')
111
rev = branch.get_revision(branch.last_revision())
112
self.assertEqual(len(rev.parent_ids), 1)
113
# parent_sha1s is not populated now, WTF. rbc 20051003
114
self.assertEqual(len(rev.parent_sha1s), 0)
115
self.assertEqual(rev.parent_ids[0], 'non:existent@rev--ision--0--2')
118
# compare the gpg-to-sign info for a commit with a ghost and
119
# an identical tree without a ghost
120
# fetch missing should rewrite the TOC of weaves to list newly available parents.
122
# TODO: rewrite this as a regular unittest, without relying on the displayed output
123
# >>> from bzrlib.commit import commit
124
# >>> bzrlib.trace.silent = True
125
# >>> br1 = ScratchBranch(files=['foo', 'bar'])
128
# >>> commit(br1, "lala!", rev_id="REVISION-ID-1", verbose=False)
129
# >>> br2 = ScratchBranch()
130
# >>> br2.update_revisions(br1)
132
# Added 1 inventories.
134
# >>> br2.revision_history()
136
# >>> br2.update_revisions(br1)
138
# >>> br1.text_store.total_size() == br2.text_store.total_size()
added
removed
bzrlib/selftest/testbranch.py
