1
# Copyright (C) 2005-2012, 2016 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18
"""Black-box tests for brz cat.
22
from ..matchers import ContainsNoVfsCalls
23
from ...transport import memory
26
class TestCat(tests.TestCaseWithTransport):
29
tree = self.make_branch_and_tree('branch')
30
self.build_tree_contents([('branch/a', b'foo\n')])
32
# 'brz cat' without an option should cat the last revision
33
self.run_bzr(['cat', 'a'], retcode=3, working_dir='branch')
35
tree.commit(message='1')
36
self.build_tree_contents([('branch/a', b'baz\n')])
38
self.assertEqual('foo\n',
39
self.run_bzr(['cat', 'a'], working_dir='branch')[0])
41
# On Windows, we used to have a bug where newlines got changed into
42
# crlf, whereas cat ought to write out the file exactly as it's
43
# recorded (by default.) That problem can't be reproduced in-process,
44
# so we need just one test here that
45
self.assertEqual(b'foo\n',
46
self.run_bzr_subprocess(['cat', 'a'],
47
working_dir='branch')[0])
49
tree.commit(message='2')
51
'baz\n', self.run_bzr(['cat', 'a'], working_dir='branch')[0])
53
'foo\n', self.run_bzr(['cat', 'a', '-r', '1'],
54
working_dir='branch')[0])
56
'baz\n', self.run_bzr(['cat', 'a', '-r', '-1'],
57
working_dir='branch')[0])
59
rev_id = tree.branch.last_revision()
62
'baz\n', self.run_bzr(
63
['cat', 'a', '-r', 'revid:%s' % rev_id.decode('utf-8')],
64
working_dir='branch')[0])
66
self.assertEqual('foo\n',
67
self.run_bzr(['cat', 'branch/a',
68
'-r', 'revno:1:branch'])[0])
69
self.run_bzr(['cat', 'a'], retcode=3)
70
self.run_bzr(['cat', 'a', '-r', 'revno:1:branch-that-does-not-exist'],
73
def test_cat_different_id(self):
74
"""'cat' works with old and new files"""
75
self.disable_missing_extensions_warning()
76
tree = self.make_branch_and_tree('.')
77
# the files are named after their path in the revision and
78
# current trees later in the test case
79
# a-rev-tree is special because it appears in both the revision
80
# tree and the working tree
81
self.build_tree_contents([('a-rev-tree', b'foo\n'),
82
('c-rev', b'baz\n'), ('d-rev', b'bar\n'), ('e-rev', b'qux\n')])
83
with tree.lock_write():
84
tree.add(['a-rev-tree', 'c-rev', 'd-rev', 'e-rev'])
85
tree.commit('add test files', rev_id=b'first')
86
# remove currently uses self._write_inventory -
87
# work around that for now.
89
tree.remove(['d-rev'])
90
tree.rename_one('a-rev-tree', 'b-tree')
91
tree.rename_one('c-rev', 'a-rev-tree')
92
tree.rename_one('e-rev', 'old-rev')
93
self.build_tree_contents([('e-rev', b'new\n')])
96
# 'b-tree' is not present in the old tree.
97
self.run_bzr_error(["^brz: ERROR: u?'b-tree' "
98
"is not present in revision .+$"],
99
'cat b-tree --name-from-revision')
101
# get to the old file automatically
102
out, err = self.run_bzr('cat d-rev')
103
self.assertEqual('', err)
104
self.assertEqual('bar\n', out)
107
self.run_bzr('cat a-rev-tree --name-from-revision')
108
self.assertEqual('foo\n', out)
109
self.assertEqual('', err)
111
out, err = self.run_bzr('cat a-rev-tree')
112
self.assertEqual('baz\n', out)
113
self.assertEqual('', err)
115
# the actual file-id for e-rev doesn't exist in the old tree
116
out, err = self.run_bzr('cat e-rev -rrevid:first')
117
self.assertEqual('qux\n', out)
118
self.assertEqual('', err)
120
def test_remote_cat(self):
121
wt = self.make_branch_and_tree('.')
122
self.build_tree(['README'])
124
wt.commit('Making sure there is a basis_tree available')
126
url = self.get_readonly_url() + '/README'
127
out, err = self.run_bzr(['cat', url])
128
self.assertEqual('contents of README\n', out)
130
def test_cat_branch_revspec(self):
131
wt = self.make_branch_and_tree('a')
132
self.build_tree(['a/README'])
134
wt.commit('Making sure there is a basis_tree available')
135
wt = self.make_branch_and_tree('b')
137
out, err = self.run_bzr(['cat', '-r', 'branch:../a', 'README'],
139
self.assertEqual('contents of a/README\n', out)
141
def test_cat_filters(self):
142
wt = self.make_branch_and_tree('.')
143
self.build_tree(['README'])
145
wt.commit('Making sure there is a basis_tree available')
146
url = self.get_readonly_url() + '/README'
148
# Test unfiltered output
149
out, err = self.run_bzr(['cat', url])
150
self.assertEqual('contents of README\n', out)
152
# Test --filters option is legal but has no impact if no filters
153
out, err = self.run_bzr(['cat', '--filters', url])
154
self.assertEqual('contents of README\n', out)
156
def test_cat_filters_applied(self):
157
# Test filtering applied to output. This is tricky to do in a
158
# subprocess because we really need to patch in a plugin that
159
# registers the filters. Instead, we patch in a custom
160
# filter_stack and use run_bzr() ...
161
from ..test_filters import _stack_2
162
from ...tree import Tree
163
wt = self.make_branch_and_tree('.')
164
self.build_tree_contents([
165
('README', b"junk\nline 1 of README\nline 2 of README\n"),
168
wt.commit('Making sure there is a basis_tree available')
169
url = self.get_readonly_url() + '/README'
170
real_content_filter_stack = Tree._content_filter_stack
172
def _custom_content_filter_stack(tree, path=None, file_id=None):
174
Tree._content_filter_stack = _custom_content_filter_stack
176
out, err = self.run_bzr(['cat', url, '--filters'])
177
# The filter stack will remove the first line and swapcase the rest
178
self.assertEqual('LINE 1 OF readme\nLINE 2 OF readme\n', out)
179
self.assertEqual('', err)
181
Tree._content_filter_stack = real_content_filter_stack
183
def test_cat_no_working_tree(self):
184
wt = self.make_branch_and_tree('.')
185
self.build_tree(['README'])
187
wt.commit('Making sure there is a basis_tree available')
188
wt.branch.controldir.destroy_workingtree()
190
url = self.get_readonly_url() + '/README'
191
out, err = self.run_bzr(['cat', url])
192
self.assertEqual('contents of README\n', out)
194
def test_cat_nonexistent_branch(self):
195
self.vfs_transport_factory = memory.MemoryServer
196
self.run_bzr_error(['^brz: ERROR: Not a branch'],
197
['cat', self.get_url()])
199
def test_cat_directory(self):
200
wt = self.make_branch_and_tree('a')
201
self.build_tree(['a/README'])
203
wt.commit('Making sure there is a basis_tree available')
205
out, err = self.run_bzr(['cat', '--directory=a', 'README'])
206
self.assertEqual('contents of a/README\n', out)
208
def test_cat_remote_directory(self):
209
wt = self.make_branch_and_tree('a')
210
self.build_tree(['a/README'])
212
wt.commit('Making sure there is a basis_tree available')
214
url = self.get_readonly_url() + '/a'
215
out, err = self.run_bzr(['cat', '-d', url, 'README'])
216
self.assertEqual('contents of a/README\n', out)
219
class TestSmartServerCat(tests.TestCaseWithTransport):
221
def test_simple_branch_cat(self):
222
self.setup_smart_server_with_call_log()
223
t = self.make_branch_and_tree('branch')
224
self.build_tree_contents([('branch/foo', b'thecontents')])
227
self.reset_smart_call_log()
228
out, err = self.run_bzr(['cat', "%s/foo" % self.get_url('branch')])
229
# This figure represent the amount of work to perform this use case. It
230
# is entirely ok to reduce this number if a test fails due to rpc_count
231
# being too low. If rpc_count increases, more network roundtrips have
232
# become necessary for this use case. Please do not adjust this number
233
# upwards without agreement from bzr's network support maintainers.
234
self.assertLength(9, self.hpss_calls)
235
self.assertLength(1, self.hpss_connections)
236
self.assertThat(self.hpss_calls, ContainsNoVfsCalls)