~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

[merge] Storage filename escaping

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007, 2009 Canonical Ltd
2
 
#
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.
7
 
#
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.
12
 
#
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
16
 
#
17
 
 
18
 
"""Tests of the 'bzr add' command."""
19
 
 
20
 
import os
21
 
 
22
 
from bzrlib import osutils
23
 
from bzrlib.tests import (
24
 
    condition_isinstance,
25
 
    split_suite_by_condition,
26
 
    multiply_tests,
27
 
    SymlinkFeature
28
 
    )
29
 
from bzrlib.tests.blackbox import ExternalBase
30
 
from bzrlib.tests.test_win32utils import NeedsGlobExpansionFeature
31
 
 
32
 
 
33
 
def load_tests(standard_tests, module, loader):
34
 
    """Parameterize tests for view-aware vs not."""
35
 
    to_adapt, result = split_suite_by_condition(
36
 
        standard_tests, condition_isinstance(TestAdd))
37
 
    scenarios = [
38
 
        ('pre-views', {'branch_tree_format': 'pack-0.92'}),
39
 
        ('view-aware', {'branch_tree_format': 'development6-rich-root'}),
40
 
        ]
41
 
    return multiply_tests(to_adapt, scenarios, result)
42
 
 
43
 
 
44
 
class TestAdd(ExternalBase):
45
 
 
46
 
    def make_branch_and_tree(self, dir):
47
 
        return ExternalBase.make_branch_and_tree(self, dir,
48
 
            format=self.branch_tree_format)
49
 
 
50
 
    def test_add_reports(self):
51
 
        """add command prints the names of added files."""
52
 
        tree = self.make_branch_and_tree('.')
53
 
        self.build_tree(['top.txt', 'dir/', 'dir/sub.txt', 'CVS'])
54
 
        self.build_tree_contents([('.bzrignore', 'CVS\n')])
55
 
        out = self.run_bzr('add')[0]
56
 
        # the ordering is not defined at the moment
57
 
        results = sorted(out.rstrip('\n').split('\n'))
58
 
        self.assertEquals(['adding .bzrignore',
59
 
                           'adding dir',
60
 
                           'adding dir/sub.txt',
61
 
                           'adding top.txt'],
62
 
                          results)
63
 
        out = self.run_bzr('add -v')[0]
64
 
        results = sorted(out.rstrip('\n').split('\n'))
65
 
        self.assertEquals(['ignored CVS matching "CVS"'],
66
 
                          results)
67
 
 
68
 
    def test_add_quiet_is(self):
69
 
        """add -q does not print the names of added files."""
70
 
        tree = self.make_branch_and_tree('.')
71
 
        self.build_tree(['top.txt', 'dir/', 'dir/sub.txt'])
72
 
        out = self.run_bzr('add -q')[0]
73
 
        # the ordering is not defined at the moment
74
 
        results = sorted(out.rstrip('\n').split('\n'))
75
 
        self.assertEquals([''], results)
76
 
 
77
 
    def test_add_in_unversioned(self):
78
 
        """Try to add a file in an unversioned directory.
79
 
 
80
 
        "bzr add" should add the parent(s) as necessary.
81
 
        """
82
 
        tree = self.make_branch_and_tree('.')
83
 
        self.build_tree(['inertiatic/', 'inertiatic/esp'])
84
 
        self.assertEquals(self.run_bzr('unknowns')[0], 'inertiatic\n')
85
 
        self.run_bzr('add inertiatic/esp')
86
 
        self.assertEquals(self.run_bzr('unknowns')[0], '')
87
 
 
88
 
        # Multiple unversioned parents
89
 
        self.build_tree(['veil/', 'veil/cerpin/', 'veil/cerpin/taxt'])
90
 
        self.assertEquals(self.run_bzr('unknowns')[0], 'veil\n')
91
 
        self.run_bzr('add veil/cerpin/taxt')
92
 
        self.assertEquals(self.run_bzr('unknowns')[0], '')
93
 
 
94
 
        # Check whacky paths work
95
 
        self.build_tree(['cicatriz/', 'cicatriz/esp'])
96
 
        self.assertEquals(self.run_bzr('unknowns')[0], 'cicatriz\n')
97
 
        self.run_bzr('add inertiatic/../cicatriz/esp')
98
 
        self.assertEquals(self.run_bzr('unknowns')[0], '')
99
 
 
100
 
    def test_add_in_versioned(self):
101
 
        """Try to add a file in a versioned directory.
102
 
 
103
 
        "bzr add" should do this happily.
104
 
        """
105
 
        tree = self.make_branch_and_tree('.')
106
 
        self.build_tree(['inertiatic/', 'inertiatic/esp'])
107
 
        self.assertEquals(self.run_bzr('unknowns')[0], 'inertiatic\n')
108
 
        self.run_bzr('add --no-recurse inertiatic')
109
 
        self.assertEquals(self.run_bzr('unknowns')[0], 'inertiatic/esp\n')
110
 
        self.run_bzr('add inertiatic/esp')
111
 
        self.assertEquals(self.run_bzr('unknowns')[0], '')
112
 
 
113
 
    def test_subdir_add(self):
114
 
        """Add in subdirectory should add only things from there down"""
115
 
        from bzrlib.workingtree import WorkingTree
116
 
 
117
 
        eq = self.assertEqual
118
 
        ass = self.assertTrue
119
 
        chdir = os.chdir
120
 
 
121
 
        t = self.make_branch_and_tree('.')
122
 
        b = t.branch
123
 
        self.build_tree(['src/', 'README'])
124
 
 
125
 
        eq(sorted(t.unknowns()),
126
 
           ['README', 'src'])
127
 
 
128
 
        self.run_bzr('add src')
129
 
 
130
 
        self.build_tree(['src/foo.c'])
131
 
 
132
 
        # add with no arguments in a subdirectory gets only files below that
133
 
        # subdirectory
134
 
        chdir('src')
135
 
        self.run_bzr('add')
136
 
        self.assertEquals(self.run_bzr('unknowns')[0], 'README\n')
137
 
        # reopen to see the new changes
138
 
        t = t.bzrdir.open_workingtree()
139
 
        versioned = [path for path, entry in t.iter_entries_by_dir()]
140
 
        self.assertEquals(versioned,
141
 
            ['', 'src', 'src/foo.c'])
142
 
 
143
 
        # add from the parent directory should pick up all file names
144
 
        chdir('..')
145
 
        self.run_bzr('add')
146
 
        self.assertEquals(self.run_bzr('unknowns')[0], '')
147
 
        self.run_bzr('check')
148
 
 
149
 
    def test_add_missing(self):
150
 
        """bzr add foo where foo is missing should error."""
151
 
        self.make_branch_and_tree('.')
152
 
        self.run_bzr('add missing-file', retcode=3)
153
 
 
154
 
    def test_add_from(self):
155
 
        base_tree = self.make_branch_and_tree('base')
156
 
        self.build_tree(['base/a', 'base/b/', 'base/b/c'])
157
 
        base_tree.add(['a', 'b', 'b/c'])
158
 
        base_tree.commit('foo')
159
 
 
160
 
        new_tree = self.make_branch_and_tree('new')
161
 
        self.build_tree(['new/a', 'new/b/', 'new/b/c', 'd'])
162
 
 
163
 
        os.chdir('new')
164
 
        out, err = self.run_bzr('add --file-ids-from ../base')
165
 
        self.assertEqual('', err)
166
 
        self.assertEqualDiff('adding a w/ file id from a\n'
167
 
                             'adding b w/ file id from b\n'
168
 
                             'adding b/c w/ file id from b/c\n',
169
 
                             out)
170
 
        new_tree = new_tree.bzrdir.open_workingtree()
171
 
        self.assertEqual(base_tree.path2id('a'), new_tree.path2id('a'))
172
 
        self.assertEqual(base_tree.path2id('b'), new_tree.path2id('b'))
173
 
        self.assertEqual(base_tree.path2id('b/c'), new_tree.path2id('b/c'))
174
 
 
175
 
    def test_add_from_subdir(self):
176
 
        base_tree = self.make_branch_and_tree('base')
177
 
        self.build_tree(['base/a', 'base/b/', 'base/b/c', 'base/b/d'])
178
 
        base_tree.add(['a', 'b', 'b/c', 'b/d'])
179
 
        base_tree.commit('foo')
180
 
 
181
 
        new_tree = self.make_branch_and_tree('new')
182
 
        self.build_tree(['new/c', 'new/d'])
183
 
 
184
 
        os.chdir('new')
185
 
        out, err = self.run_bzr('add --file-ids-from ../base/b')
186
 
        self.assertEqual('', err)
187
 
        self.assertEqualDiff('adding c w/ file id from b/c\n'
188
 
                             'adding d w/ file id from b/d\n',
189
 
                             out)
190
 
 
191
 
        new_tree = new_tree.bzrdir.open_workingtree()
192
 
        self.assertEqual(base_tree.path2id('b/c'), new_tree.path2id('c'))
193
 
        self.assertEqual(base_tree.path2id('b/d'), new_tree.path2id('d'))
194
 
 
195
 
    def test_add_dry_run(self):
196
 
        """Test a dry run add, make sure nothing is added."""
197
 
        wt = self.make_branch_and_tree('.')
198
 
        self.build_tree(['inertiatic/', 'inertiatic/esp'])
199
 
        self.assertEqual(list(wt.unknowns()), ['inertiatic'])
200
 
        self.run_bzr('add --dry-run')
201
 
        self.assertEqual(list(wt.unknowns()), ['inertiatic'])
202
 
 
203
 
    def test_add_control_dir(self):
204
 
        """The control dir and its content should be refused."""
205
 
        self.make_branch_and_tree('.')
206
 
        err = self.run_bzr('add .bzr', retcode=3)[1]
207
 
        self.assertContainsRe(err, r'ERROR:.*\.bzr.*control file')
208
 
        err = self.run_bzr('add .bzr/README', retcode=3)[1]
209
 
        self.assertContainsRe(err, r'ERROR:.*\.bzr.*control file')
210
 
        self.build_tree(['.bzr/crescent'])
211
 
        err = self.run_bzr('add .bzr/crescent', retcode=3)[1]
212
 
        self.assertContainsRe(err, r'ERROR:.*\.bzr.*control file')
213
 
 
214
 
    def test_add_with_wildcards(self):
215
 
        self.requireFeature(NeedsGlobExpansionFeature)
216
 
        self.make_branch_and_tree('.')
217
 
        self.build_tree(['a1', 'a2', 'b', 'c33'])
218
 
        self.run_bzr(['add', 'a?', 'c*'])
219
 
        self.assertEquals(self.run_bzr('unknowns')[0], 'b\n')
220
 
 
221
 
    def test_add_with_wildcards_unicode(self):
222
 
        self.requireFeature(NeedsGlobExpansionFeature)
223
 
        self.make_branch_and_tree('.')
224
 
        self.build_tree([u'\u1234A', u'\u1235A', u'\u1235AA', 'cc'])
225
 
        self.run_bzr(['add', u'\u1234?', u'\u1235*'])
226
 
        self.assertEquals(self.run_bzr('unknowns')[0], 'cc\n')
227
 
 
228
 
    def test_add_via_symlink(self):
229
 
        self.requireFeature(SymlinkFeature)
230
 
        self.make_branch_and_tree('source')
231
 
        self.build_tree(['source/top.txt'])
232
 
        os.symlink('source', 'link')
233
 
        out = self.run_bzr(['add', 'link/top.txt'])[0]
234
 
        self.assertEquals(out, 'adding top.txt\n')
235
 
 
236
 
    def test_add_symlink_to_abspath(self):
237
 
        self.requireFeature(SymlinkFeature)
238
 
        self.make_branch_and_tree('tree')
239
 
        os.symlink(osutils.abspath('target'), 'tree/link')
240
 
        out = self.run_bzr(['add', 'tree/link'])[0]
241
 
        self.assertEquals(out, 'adding link\n')