1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
# (C) 2005,2006 Canonical Ltd
# Authors: Robert Collins <robert.collins@canonical.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
from cStringIO import StringIO
import os
import bzrlib
import bzrlib.branch
from bzrlib.branch import Branch
import bzrlib.bzrdir as bzrdir
from bzrlib.bzrdir import BzrDir
import bzrlib.errors as errors
from bzrlib.errors import (NotBranchError, NotVersionedError,
UnsupportedOperation)
from bzrlib.osutils import pathjoin, getcwd, has_symlinks
from bzrlib.tests import TestSkipped, TestCase
from bzrlib.tests.workingtree_implementations import TestCaseWithWorkingTree
from bzrlib.trace import mutter
import bzrlib.ui as ui
import bzrlib.workingtree as workingtree
from bzrlib.workingtree import (TreeEntry, TreeDirectory, TreeFile, TreeLink,
WorkingTree)
class CapturingUIFactory(ui.UIFactory):
"""A UI Factory for testing - capture the updates made through it."""
def __init__(self):
super(CapturingUIFactory, self).__init__()
self._calls = []
self.depth = 0
def clear(self):
"""See progress.ProgressBar.clear()."""
def clear_term(self):
"""See progress.ProgressBar.clear_term()."""
def finished(self):
"""See progress.ProgressBar.finished()."""
self.depth -= 1
def note(self, fmt_string, *args, **kwargs):
"""See progress.ProgressBar.note()."""
def progress_bar(self):
return self
def nested_progress_bar(self):
self.depth += 1
return self
def update(self, message, count=None, total=None):
"""See progress.ProgressBar.update()."""
if self.depth == 1:
self._calls.append(("update", count, total))
class TestCapturingUI(TestCase):
def test_nested_ignore_depth_beyond_one(self):
# we only want to capture the first level out progress, not
# want sub-components might do. So we have nested bars ignored.
factory = CapturingUIFactory()
pb1 = factory.nested_progress_bar()
pb1.update('foo', 0, 1)
pb2 = factory.nested_progress_bar()
pb2.update('foo', 0, 1)
pb2.finished()
pb1.finished()
self.assertEqual([("update", 0, 1)], factory._calls)
class TestCommit(TestCaseWithWorkingTree):
def test_commit_sets_last_revision(self):
tree = self.make_branch_and_tree('tree')
tree.commit('foo', rev_id='foo', allow_pointless=True)
self.assertEqual('foo', tree.last_revision())
def test_commit_local_unbound(self):
# using the library api to do a local commit on unbound branches is
# also an error
tree = self.make_branch_and_tree('tree')
self.assertRaises(errors.LocalRequiresBoundBranch,
tree.commit,
'foo',
local=True)
def test_local_commit_ignores_master(self):
# a --local commit does not require access to the master branch
# at all, or even for it to exist.
# we test this by setting up a bound branch and then corrupting
# the master.
master = self.make_branch('master')
tree = self.make_branch_and_tree('tree')
try:
tree.branch.bind(master)
except errors.UpgradeRequired:
# older format.
return
master.bzrdir.transport.put('branch-format', StringIO('garbage'))
del master
# check its corrupted.
self.assertRaises(errors.UnknownFormatError,
bzrdir.BzrDir.open,
'master')
tree.commit('foo', rev_id='foo', local=True)
def test_local_commit_does_not_push_to_master(self):
# a --local commit does not require access to the master branch
# at all, or even for it to exist.
# we test that even when its available it does not push to it.
master = self.make_branch('master')
tree = self.make_branch_and_tree('tree')
try:
tree.branch.bind(master)
except errors.UpgradeRequired:
# older format.
return
tree.commit('foo', rev_id='foo', local=True)
self.failIf(master.repository.has_revision('foo'))
self.assertEqual(None, master.last_revision())
class TestCommmitProgress(TestCaseWithWorkingTree):
def restoreDefaults(self):
ui.ui_factory = self.old_ui_factory
def test_commit_progress_steps(self):
# during commit we one progress update for every entry in the
# inventory, and then one for the inventory, and one for the
# inventory, and one for the revision insertions.
# first we need a test commit to do. Lets setup a branch with
# 3 files, and alter one in a selected-file commit. This exercises
# a number of cases quickly. We should also test things like
# selective commits which excludes newly added files.
tree = self.make_branch_and_tree('.')
self.build_tree(['a', 'b', 'c'])
tree.add(['a', 'b', 'c'])
tree.commit('first post')
f = file('b', 'wt')
f.write('new content')
f.close()
# set a progress bar that captures the calls so we can see what is
# emitted
self.old_ui_factory = ui.ui_factory
self.addCleanup(self.restoreDefaults)
factory = CapturingUIFactory()
ui.ui_factory = factory
# TODO RBC 20060421 it would be nice to merge the reporter output
# into the factory for this test - just make the test ui factory
# pun as a reporter. Then we can check the ordering is right.
tree.commit('second post', specific_files=['b'])
# 11 steps: 1 for rev, 1 for inventory, 1 for finishing. 2 for root
# and 3 for basis files, and 3 for new inventory files.
self.assertEqual(
[("update", 0, 10),
("update", 1, 10),
("update", 2, 10),
("update", 3, 10),
("update", 4, 10),
("update", 5, 10),
("update", 6, 10),
("update", 7, 10),
("update", 8, 10),
("update", 9, 10),
("update", 10, 10)],
factory._calls
)
|