~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

  • Committer: Andrew Bennetts
  • Date: 2007-06-14 11:23:38 UTC
  • mto: (2506.2.7 container-format)
  • mto: This revision was merged to the branch mainline in revision 2572.
  • Revision ID: andrew.bennetts@canonical.com-20070614112338-6u3900u6nkag66u8
Return a callable instead of a str from read, and add more validation.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright (C) 2005, 2006 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
16
#
 
17
 
 
18
"""Tests of the 'bzr add' command."""
 
19
 
 
20
import os
 
21
 
 
22
from bzrlib.tests.blackbox import ExternalBase
 
23
 
 
24
 
 
25
class TestAdd(ExternalBase):
 
26
        
 
27
    def test_add_reports(self):
 
28
        """add command prints the names of added files."""
 
29
        self.runbzr('init')
 
30
        self.build_tree(['top.txt', 'dir/', 'dir/sub.txt', 'CVS'])
 
31
        self.build_tree_contents([('.bzrignore', 'CVS\n')])
 
32
        out = self.run_bzr_captured(['add'], retcode=0)[0]
 
33
        # the ordering is not defined at the moment
 
34
        results = sorted(out.rstrip('\n').split('\n'))
 
35
        self.assertEquals(['If you wish to add some of these files, please'\
 
36
                           ' add them by name.',
 
37
                           'added .bzrignore',
 
38
                           'added dir',
 
39
                           'added dir/sub.txt',
 
40
                           'added top.txt',
 
41
                           'ignored 1 file(s).'],
 
42
                          results)
 
43
        out = self.run_bzr_captured(['add', '-v'], retcode=0)[0]
 
44
        results = sorted(out.rstrip('\n').split('\n'))
 
45
        self.assertEquals(['If you wish to add some of these files, please'\
 
46
                           ' add them by name.',
 
47
                           'ignored CVS matching "CVS"'],
 
48
                          results)
 
49
 
 
50
    def test_add_quiet_is(self):
 
51
        """add -q does not print the names of added files."""
 
52
        self.runbzr('init')
 
53
        self.build_tree(['top.txt', 'dir/', 'dir/sub.txt'])
 
54
        out = self.run_bzr_captured(['add', '-q'], retcode=0)[0]
 
55
        # the ordering is not defined at the moment
 
56
        results = sorted(out.rstrip('\n').split('\n'))
 
57
        self.assertEquals([''], results)
 
58
 
 
59
    def test_add_in_unversioned(self):
 
60
        """Try to add a file in an unversioned directory.
 
61
 
 
62
        "bzr add" should add the parent(s) as necessary.
 
63
        """
 
64
        self.runbzr('init')
 
65
        self.build_tree(['inertiatic/', 'inertiatic/esp'])
 
66
        self.assertEquals(self.capture('unknowns'), 'inertiatic\n')
 
67
        self.run_bzr('add', 'inertiatic/esp')
 
68
        self.assertEquals(self.capture('unknowns'), '')
 
69
 
 
70
        # Multiple unversioned parents
 
71
        self.build_tree(['veil/', 'veil/cerpin/', 'veil/cerpin/taxt'])
 
72
        self.assertEquals(self.capture('unknowns'), 'veil\n')
 
73
        self.run_bzr('add', 'veil/cerpin/taxt')
 
74
        self.assertEquals(self.capture('unknowns'), '')
 
75
 
 
76
        # Check whacky paths work
 
77
        self.build_tree(['cicatriz/', 'cicatriz/esp'])
 
78
        self.assertEquals(self.capture('unknowns'), 'cicatriz\n')
 
79
        self.run_bzr('add', 'inertiatic/../cicatriz/esp')
 
80
        self.assertEquals(self.capture('unknowns'), '')
 
81
 
 
82
    def test_add_in_versioned(self):
 
83
        """Try to add a file in a versioned directory.
 
84
 
 
85
        "bzr add" should do this happily.
 
86
        """
 
87
        self.runbzr('init')
 
88
        self.build_tree(['inertiatic/', 'inertiatic/esp'])
 
89
        self.assertEquals(self.capture('unknowns'), 'inertiatic\n')
 
90
        self.run_bzr('add', '--no-recurse', 'inertiatic')
 
91
        self.assertEquals(self.capture('unknowns'), 'inertiatic/esp\n')
 
92
        self.run_bzr('add', 'inertiatic/esp')
 
93
        self.assertEquals(self.capture('unknowns'), '')
 
94
 
 
95
    def test_subdir_add(self):
 
96
        """Add in subdirectory should add only things from there down"""
 
97
        from bzrlib.workingtree import WorkingTree
 
98
 
 
99
        eq = self.assertEqual
 
100
        ass = self.assertTrue
 
101
        chdir = os.chdir
 
102
        
 
103
        t = self.make_branch_and_tree('.')
 
104
        b = t.branch
 
105
        self.build_tree(['src/', 'README'])
 
106
        
 
107
        eq(sorted(t.unknowns()),
 
108
           ['README', 'src'])
 
109
        
 
110
        self.run_bzr('add', 'src')
 
111
        
 
112
        self.build_tree(['src/foo.c'])
 
113
        
 
114
        # add with no arguments in a subdirectory gets only files below that
 
115
        # subdirectory
 
116
        chdir('src')
 
117
        self.run_bzr('add')
 
118
        self.assertEquals(self.capture('unknowns'), 'README\n')
 
119
        # reopen to see the new changes
 
120
        t = t.bzrdir.open_workingtree()
 
121
        versioned = [path for path, entry in t.iter_entries_by_dir()]
 
122
        self.assertEquals(versioned,
 
123
            ['', 'src', 'src/foo.c'])
 
124
                
 
125
        # add from the parent directory should pick up all file names
 
126
        chdir('..')
 
127
        self.run_bzr('add')
 
128
        self.assertEquals(self.capture('unknowns'), '')
 
129
        self.run_bzr('check')
 
130
 
 
131
    def test_add_missing(self):
 
132
        """bzr add foo where foo is missing should error."""
 
133
        self.make_branch_and_tree('.')
 
134
        self.run_bzr('add', 'missing-file', retcode=3)
 
135
 
 
136
    def test_add_from(self):
 
137
        base_tree = self.make_branch_and_tree('base')
 
138
        self.build_tree(['base/a', 'base/b/', 'base/b/c'])
 
139
        base_tree.add(['a', 'b', 'b/c'])
 
140
        base_tree.commit('foo')
 
141
 
 
142
        new_tree = self.make_branch_and_tree('new')
 
143
        self.build_tree(['new/a', 'new/b/', 'new/b/c', 'd'])
 
144
 
 
145
        os.chdir('new')
 
146
        out, err = self.run_bzr('add', '--file-ids-from', '../base')
 
147
        self.assertEqual('', err)
 
148
        self.assertEqualDiff('added a w/ file id from a\n'
 
149
                             'added b w/ file id from b\n'
 
150
                             'added b/c w/ file id from b/c\n',
 
151
                             out)
 
152
        new_tree = new_tree.bzrdir.open_workingtree()
 
153
        self.assertEqual(base_tree.path2id('a'), new_tree.path2id('a'))
 
154
        self.assertEqual(base_tree.path2id('b'), new_tree.path2id('b'))
 
155
        self.assertEqual(base_tree.path2id('b/c'), new_tree.path2id('b/c'))
 
156
 
 
157
    def test_add_from_subdir(self):
 
158
        base_tree = self.make_branch_and_tree('base')
 
159
        self.build_tree(['base/a', 'base/b/', 'base/b/c', 'base/b/d'])
 
160
        base_tree.add(['a', 'b', 'b/c', 'b/d'])
 
161
        base_tree.commit('foo')
 
162
 
 
163
        new_tree = self.make_branch_and_tree('new')
 
164
        self.build_tree(['new/c', 'new/d'])
 
165
 
 
166
        os.chdir('new')
 
167
        out, err = self.run_bzr('add', '--file-ids-from', '../base/b')
 
168
        self.assertEqual('', err)
 
169
        self.assertEqualDiff('added c w/ file id from b/c\n'
 
170
                             'added d w/ file id from b/d\n',
 
171
                             out)
 
172
 
 
173
        new_tree = new_tree.bzrdir.open_workingtree()
 
174
        self.assertEqual(base_tree.path2id('b/c'), new_tree.path2id('c'))
 
175
        self.assertEqual(base_tree.path2id('b/d'), new_tree.path2id('d'))
 
176
 
 
177
    def test_add_dry_run(self):
 
178
        # ensure that --dry-run actually don't add anything
 
179
        base_tree = self.make_branch_and_tree('.')
 
180
        self.build_tree(['spam'])
 
181
        out = self.run_bzr_captured(['add', '--dry-run'], retcode=0)[0]
 
182
        self.assertEquals('added spam\n', out)
 
183
        out = self.run_bzr_captured(['added'], retcode=0)[0]
 
184
        self.assertEquals('', out)
 
185
 
 
186
    def test_add_control_dir(self):
 
187
        """The control dir and its content should be refused."""
 
188
        self.make_branch_and_tree('.')
 
189
        err = self.run_bzr('add', '.bzr', retcode=3)[1]
 
190
        self.assertContainsRe(err, r'ERROR:.*\.bzr.*control file')
 
191
        err = self.run_bzr('add', '.bzr/README', retcode=3)[1]
 
192
        self.assertContainsRe(err, r'ERROR:.*\.bzr.*control file')
 
193
        self.build_tree(['.bzr/crescent'])
 
194
        err = self.run_bzr('add', '.bzr/crescent', retcode=3)[1]
 
195
        self.assertContainsRe(err, r'ERROR:.*\.bzr.*control file')