← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/blackbox/test_cat.py

[merge] robert's knit-performance work

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/blackbox/test_cat.py
1
 
# Copyright (C) 2005-2011 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2005 by Canonical Ltd
 
2
# -*- coding: utf-8 -*-
 
3
 
3
4
# This program is free software; you can redistribute it and/or modify
4
5
# it under the terms of the GNU General Public License as published by
5
6
# the Free Software Foundation; either version 2 of the License, or
6
7
# (at your option) any later version.
7
 
#
 
8
 
8
9
# This program is distributed in the hope that it will be useful,
9
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
12
# GNU General Public License for more details.
12
 
#
 
13
 
13
14
# You should have received a copy of the GNU General Public License
14
15
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
16
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
17
 
17
18
 
18
19
"""Black-box tests for bzr cat.
20
21
 
21
22
import os
22
23
 
23
 
from bzrlib import tests
24
 
from bzrlib.transport import memory
25
 
 
26
 
 
27
 
class TestCat(tests.TestCaseWithTransport):
 
24
from bzrlib.branch import Branch
 
25
from bzrlib.tests import TestCaseInTempDir
 
26
 
 
27
class TestCat(TestCaseInTempDir):
28
28
 
29
29
    def test_cat(self):
30
 
        tree = self.make_branch_and_tree('branch')
31
 
        self.build_tree_contents([('branch/a', 'foo\n')])
32
 
        tree.add('a')
33
 
        os.chdir('branch')
 
30
 
 
31
        def bzr(*args, **kwargs):
 
32
            return self.run_bzr(*args, **kwargs)[0]
 
33
 
 
34
        bzr('init')
 
35
        open('a', 'wb').write('foo\n')
 
36
        bzr('add', 'a')
 
37
 
34
38
        # 'bzr cat' without an option should cat the last revision
35
 
        self.run_bzr(['cat', 'a'], retcode=3)
36
 
 
37
 
        tree.commit(message='1')
38
 
        self.build_tree_contents([('a', 'baz\n')])
39
 
 
40
 
        self.assertEquals(self.run_bzr(['cat', 'a'])[0], 'foo\n')
41
 
 
42
 
        # On Windows, we used to have a bug where newlines got changed into
43
 
        # crlf, whereas cat ought to write out the file exactly as it's
44
 
        # recorded (by default.)  That problem can't be reproduced in-process,
45
 
        # so we need just one test here that 
46
 
        self.assertEquals(self.run_bzr_subprocess(['cat', 'a'])[0], 'foo\n')
47
 
 
48
 
        tree.commit(message='2')
49
 
        self.assertEquals(self.run_bzr(['cat', 'a'])[0], 'baz\n')
50
 
        self.assertEquals(self.run_bzr(
51
 
            ['cat', 'a', '-r', '1'])[0],
52
 
            'foo\n')
53
 
        self.assertEquals(self.run_bzr(
54
 
            ['cat', 'a', '-r', '-1'])[0],
55
 
            'baz\n')
56
 
 
57
 
        rev_id = tree.branch.last_revision()
58
 
 
59
 
        self.assertEquals(self.run_bzr(
60
 
            ['cat', 'a', '-r', 'revid:%s' % rev_id])[0],
61
 
            'baz\n')
62
 
 
63
 
        os.chdir('..')
64
 
 
65
 
        self.assertEquals(self.run_bzr(
66
 
            ['cat', 'branch/a', '-r', 'revno:1:branch'])[0],
67
 
            'foo\n')
68
 
        self.run_bzr(['cat', 'a'], retcode=3)
69
 
        self.run_bzr(
70
 
                ['cat', 'a', '-r', 'revno:1:branch-that-does-not-exist'],
71
 
                retcode=3)
72
 
 
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', 'foo\n'),
82
 
            ('c-rev', 'baz\n'), ('d-rev', 'bar\n'), ('e-rev', 'qux\n')])
83
 
        tree.lock_write()
84
 
        try:
85
 
            tree.add(['a-rev-tree', 'c-rev', 'd-rev', 'e-rev'])
86
 
            tree.commit('add test files', rev_id='first')
87
 
            # remove currently uses self._write_inventory -
88
 
            # work around that for now.
89
 
            tree.flush()
90
 
            tree.remove(['d-rev'])
91
 
            tree.rename_one('a-rev-tree', 'b-tree')
92
 
            tree.rename_one('c-rev', 'a-rev-tree')
93
 
            tree.rename_one('e-rev', 'old-rev')
94
 
            self.build_tree_contents([('e-rev', 'new\n')])
95
 
            tree.add(['e-rev'])
96
 
        finally:
97
 
            # calling bzr as another process require free lock on win32
98
 
            tree.unlock()
99
 
 
100
 
        # 'b-tree' is not present in the old tree.
101
 
        self.run_bzr_error(["^bzr: ERROR: u?'b-tree' "
102
 
                            "is not present in revision .+$"],
103
 
                           'cat b-tree --name-from-revision')
104
 
 
105
 
        # get to the old file automatically
106
 
        out, err = self.run_bzr('cat d-rev')
107
 
        self.assertEqual('bar\n', out)
108
 
        self.assertEqual('', err)
109
 
 
110
 
        out, err = \
111
 
                self.run_bzr('cat a-rev-tree --name-from-revision')
112
 
        self.assertEqual('foo\n', out)
113
 
        self.assertEqual('', err)
114
 
 
115
 
        out, err = self.run_bzr('cat a-rev-tree')
116
 
        self.assertEqual('baz\n', out)
117
 
        self.assertEqual('', err)
118
 
 
119
 
        # the actual file-id for e-rev doesn't exist in the old tree
120
 
        out, err = self.run_bzr('cat e-rev -rrevid:first')
121
 
        self.assertEqual('qux\n', out)
122
 
        self.assertEqual('', err)
123
 
 
124
 
    def test_remote_cat(self):
125
 
        wt = self.make_branch_and_tree('.')
126
 
        self.build_tree(['README'])
127
 
        wt.add('README')
128
 
        wt.commit('Making sure there is a basis_tree available')
129
 
 
130
 
        url = self.get_readonly_url() + '/README'
131
 
        out, err = self.run_bzr(['cat', url])
132
 
        self.assertEqual('contents of README\n', out)
133
 
 
134
 
    def test_cat_branch_revspec(self):
135
 
        wt = self.make_branch_and_tree('a')
136
 
        self.build_tree(['a/README'])
137
 
        wt.add('README')
138
 
        wt.commit('Making sure there is a basis_tree available')
139
 
        wt = self.make_branch_and_tree('b')
140
 
        os.chdir('b')
141
 
 
142
 
        out, err = self.run_bzr(
143
 
            ['cat', '-r', 'branch:../a', 'README'])
144
 
        self.assertEqual('contents of a/README\n', out)
145
 
 
146
 
    def test_cat_filters(self):
147
 
        wt = self.make_branch_and_tree('.')
148
 
        self.build_tree(['README'])
149
 
        wt.add('README')
150
 
        wt.commit('Making sure there is a basis_tree available')
151
 
        url = self.get_readonly_url() + '/README'
152
 
 
153
 
        # Test unfiltered output
154
 
        out, err = self.run_bzr(['cat', url])
155
 
        self.assertEqual('contents of README\n', out)
156
 
 
157
 
        # Test --filters option is legal but has no impact if no filters
158
 
        out, err = self.run_bzr(['cat', '--filters', url])
159
 
        self.assertEqual('contents of README\n', out)
160
 
 
161
 
    def test_cat_filters_applied(self):
162
 
        # Test filtering applied to output. This is tricky to do in a
163
 
        # subprocess because we really need to patch in a plugin that
164
 
        # registers the filters. Instead, we patch in a custom
165
 
        # filter_stack and use run_bzr() ...
166
 
        from cStringIO import StringIO
167
 
        from bzrlib.commands import run_bzr
168
 
        from bzrlib.tests.test_filters import _stack_2
169
 
        from bzrlib.trace import mutter
170
 
        from bzrlib.tree import Tree
171
 
        wt = self.make_branch_and_tree('.')
172
 
        self.build_tree_contents([
173
 
            ('README', "junk\nline 1 of README\nline 2 of README\n"),
174
 
            ])
175
 
        wt.add('README')
176
 
        wt.commit('Making sure there is a basis_tree available')
177
 
        url = self.get_readonly_url() + '/README'
178
 
        real_content_filter_stack = Tree._content_filter_stack
179
 
        def _custom_content_filter_stack(tree, path=None, file_id=None):
180
 
            return _stack_2
181
 
        Tree._content_filter_stack = _custom_content_filter_stack
182
 
        try:
183
 
            out, err = self.run_bzr(['cat', url, '--filters'])
184
 
            # The filter stack will remove the first line and swapcase the rest
185
 
            self.assertEqual('LINE 1 OF readme\nLINE 2 OF readme\n', out)
186
 
            self.assertEqual('', err)
187
 
        finally:
188
 
            Tree._content_filter_stack = real_content_filter_stack
189
 
 
190
 
    def test_cat_no_working_tree(self):
191
 
        wt = self.make_branch_and_tree('.')
192
 
        self.build_tree(['README'])
193
 
        wt.add('README')
194
 
        wt.commit('Making sure there is a basis_tree available')
195
 
        wt.branch.bzrdir.destroy_workingtree()
196
 
 
197
 
        url = self.get_readonly_url() + '/README'
198
 
        out, err = self.run_bzr(['cat', url])
199
 
        self.assertEqual('contents of README\n', out)
200
 
 
201
 
    def test_cat_nonexistent_branch(self):
202
 
        self.vfs_transport_factory = memory.MemoryServer
203
 
        self.run_bzr_error(['^bzr: ERROR: Not a branch'],
204
 
                           ['cat', self.get_url()])
205
 
 
206
 
    def test_cat_directory(self):
207
 
        wt = self.make_branch_and_tree('a')
208
 
        self.build_tree(['a/README'])
209
 
        wt.add('README')
210
 
        wt.commit('Making sure there is a basis_tree available')
211
 
 
212
 
        out, err = self.run_bzr(['cat', '--directory=a', 'README'])
213
 
        self.assertEqual('contents of a/README\n', out)
214
 
 
215
 
    def test_cat_remote_directory(self):
216
 
        wt = self.make_branch_and_tree('a')
217
 
        self.build_tree(['a/README'])
218
 
        wt.add('README')
219
 
        wt.commit('Making sure there is a basis_tree available')
220
 
 
221
 
        url = self.get_readonly_url() + '/a'
222
 
        out, err = self.run_bzr(['cat', '-d', url, 'README'])
223
 
        self.assertEqual('contents of a/README\n', out)
 
39
        bzr('cat', 'a', retcode=3)
 
40
 
 
41
        bzr('commit', '-m', '1')
 
42
        open('a', 'wb').write('baz\n')
 
43
 
 
44
        self.assertEquals(bzr('cat', 'a'), 'foo\n')
 
45
 
 
46
        bzr('commit', '-m', '2')
 
47
        self.assertEquals(bzr('cat', 'a'), 'baz\n')
 
48
        self.assertEquals(bzr('cat', 'a', '-r', '1'), 'foo\n')
 
49
        self.assertEquals(bzr('cat', 'a', '-r', '-1'), 'baz\n')
 
50
 
 
51
        rev_id = bzr('revision-history').strip().split('\n')[-1]
 
52
 
 
53
        self.assertEquals(bzr('cat', 'a', '-r', 'revid:%s' % rev_id), 'baz\n')
 
54