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('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(['cat', 'a', '-r', 'revid:%s' % rev_id],
63
working_dir='branch')[0])
65
self.assertEqual('foo\n',
66
self.run_bzr(['cat', 'branch/a',
67
'-r', 'revno:1:branch'])[0])
68
self.run_bzr(['cat', 'a'], retcode=3)
69
self.run_bzr(['cat', 'a', '-r', 'revno:1:branch-that-does-not-exist'],
72
def test_cat_different_id(self):
73
"""'cat' works with old and new files"""
74
self.disable_missing_extensions_warning()
75
tree = self.make_branch_and_tree('.')
76
# the files are named after their path in the revision and
77
# current trees later in the test case
78
# a-rev-tree is special because it appears in both the revision
79
# tree and the working tree
80
self.build_tree_contents([('a-rev-tree', b'foo\n'),
81
('c-rev', b'baz\n'), ('d-rev', b'bar\n'), ('e-rev', b'qux\n')])
82
with tree.lock_write():
83
tree.add(['a-rev-tree', 'c-rev', 'd-rev', 'e-rev'])
84
tree.commit('add test files', rev_id=b'first')
85
# remove currently uses self._write_inventory -
86
# work around that for now.
88
tree.remove(['d-rev'])
89
tree.rename_one('a-rev-tree', 'b-tree')
90
tree.rename_one('c-rev', 'a-rev-tree')
91
tree.rename_one('e-rev', 'old-rev')
92
self.build_tree_contents([('e-rev', b'new\n')])
95
# 'b-tree' is not present in the old tree.
96
self.run_bzr_error(["^brz: ERROR: u?'b-tree' "
97
"is not present in revision .+$"],
98
'cat b-tree --name-from-revision')
100
# get to the old file automatically
101
out, err = self.run_bzr('cat d-rev')
102
self.assertEqual('', err)
103
self.assertEqual('bar\n', out)
106
self.run_bzr('cat a-rev-tree --name-from-revision')
107
self.assertEqual('foo\n', out)
108
self.assertEqual('', err)
110
out, err = self.run_bzr('cat a-rev-tree')
111
self.assertEqual('baz\n', out)
112
self.assertEqual('', err)
114
# the actual file-id for e-rev doesn't exist in the old tree
115
out, err = self.run_bzr('cat e-rev -rrevid:first')
116
self.assertEqual('qux\n', out)
117
self.assertEqual('', err)
119
def test_remote_cat(self):
120
wt = self.make_branch_and_tree('.')
121
self.build_tree(['README'])
123
wt.commit('Making sure there is a basis_tree available')
125
url = self.get_readonly_url() + '/README'
126
out, err = self.run_bzr(['cat', url])
127
self.assertEqual('contents of README\n', out)
129
def test_cat_branch_revspec(self):
130
wt = self.make_branch_and_tree('a')
131
self.build_tree(['a/README'])
133
wt.commit('Making sure there is a basis_tree available')
134
wt = self.make_branch_and_tree('b')
136
out, err = self.run_bzr(['cat', '-r', 'branch:../a', 'README'],
138
self.assertEqual('contents of a/README\n', out)
140
def test_cat_filters(self):
141
wt = self.make_branch_and_tree('.')
142
self.build_tree(['README'])
144
wt.commit('Making sure there is a basis_tree available')
145
url = self.get_readonly_url() + '/README'
147
# Test unfiltered output
148
out, err = self.run_bzr(['cat', url])
149
self.assertEqual('contents of README\n', out)
151
# Test --filters option is legal but has no impact if no filters
152
out, err = self.run_bzr(['cat', '--filters', url])
153
self.assertEqual('contents of README\n', out)
155
def test_cat_filters_applied(self):
156
# Test filtering applied to output. This is tricky to do in a
157
# subprocess because we really need to patch in a plugin that
158
# registers the filters. Instead, we patch in a custom
159
# filter_stack and use run_bzr() ...
160
from ..test_filters import _stack_2
161
from ...tree import Tree
162
wt = self.make_branch_and_tree('.')
163
self.build_tree_contents([
164
('README', b"junk\nline 1 of README\nline 2 of README\n"),
167
wt.commit('Making sure there is a basis_tree available')
168
url = self.get_readonly_url() + '/README'
169
real_content_filter_stack = Tree._content_filter_stack
170
def _custom_content_filter_stack(tree, path=None, file_id=None):
172
Tree._content_filter_stack = _custom_content_filter_stack
174
out, err = self.run_bzr(['cat', url, '--filters'])
175
# The filter stack will remove the first line and swapcase the rest
176
self.assertEqual('LINE 1 OF readme\nLINE 2 OF readme\n', out)
177
self.assertEqual('', err)
179
Tree._content_filter_stack = real_content_filter_stack
181
def test_cat_no_working_tree(self):
182
wt = self.make_branch_and_tree('.')
183
self.build_tree(['README'])
185
wt.commit('Making sure there is a basis_tree available')
186
wt.branch.controldir.destroy_workingtree()
188
url = self.get_readonly_url() + '/README'
189
out, err = self.run_bzr(['cat', url])
190
self.assertEqual('contents of README\n', out)
192
def test_cat_nonexistent_branch(self):
193
self.vfs_transport_factory = memory.MemoryServer
194
self.run_bzr_error(['^brz: ERROR: Not a branch'],
195
['cat', self.get_url()])
197
def test_cat_directory(self):
198
wt = self.make_branch_and_tree('a')
199
self.build_tree(['a/README'])
201
wt.commit('Making sure there is a basis_tree available')
203
out, err = self.run_bzr(['cat', '--directory=a', 'README'])
204
self.assertEqual('contents of a/README\n', out)
206
def test_cat_remote_directory(self):
207
wt = self.make_branch_and_tree('a')
208
self.build_tree(['a/README'])
210
wt.commit('Making sure there is a basis_tree available')
212
url = self.get_readonly_url() + '/a'
213
out, err = self.run_bzr(['cat', '-d', url, 'README'])
214
self.assertEqual('contents of a/README\n', out)
217
class TestSmartServerCat(tests.TestCaseWithTransport):
219
def test_simple_branch_cat(self):
220
self.setup_smart_server_with_call_log()
221
t = self.make_branch_and_tree('branch')
222
self.build_tree_contents([('branch/foo', b'thecontents')])
225
self.reset_smart_call_log()
226
out, err = self.run_bzr(['cat', "%s/foo" % self.get_url('branch')])
227
# This figure represent the amount of work to perform this use case. It
228
# is entirely ok to reduce this number if a test fails due to rpc_count
229
# being too low. If rpc_count increases, more network roundtrips have
230
# become necessary for this use case. Please do not adjust this number
231
# upwards without agreement from bzr's network support maintainers.
232
self.assertLength(9, self.hpss_calls)
233
self.assertLength(1, self.hpss_connections)
234
self.assertThat(self.hpss_calls, ContainsNoVfsCalls)