13
13
# You should have received a copy of the GNU General Public License
 
14
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
 
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
21
 
from bzrlib.tests import SymlinkFeature, TestSkipped
 
22
20
from bzrlib.tests.blackbox import ExternalBase
 
23
21
from bzrlib.workingtree import WorkingTree
 
24
 
from bzrlib import osutils
 
34
24
class TestRemove(ExternalBase):
 
36
 
    def _make_tree_and_add(self, paths):
 
37
 
        tree = self.make_branch_and_tree('.')
 
40
 
            self.build_tree(paths)
 
42
 
                file_id=str(path).replace('/', '_') + _id
 
43
 
                tree.add(path, file_id)
 
48
 
    def assertFilesDeleted(self, files):
 
51
 
            self.assertNotInWorkingTree(f)
 
54
 
    def assertFilesUnversioned(self, files):
 
56
 
            self.assertNotInWorkingTree(f)
 
57
 
            self.failUnlessExists(f)
 
59
 
    def changeFile(self, file_name):
 
60
 
        f = file(file_name, 'ab')
 
61
 
        f.write("\nsome other new content!")
 
64
 
    def run_bzr_remove_changed_files(self, error_regexes, files_to_remove,
 
66
 
        error_regexes.extend(["Can't safely remove modified or unknown files:",
 
67
 
            'Use --keep to not delete them,'
 
68
 
            ' or --force to delete them regardless.'
 
70
 
        self.run_bzr_error(error_regexes,
 
71
 
                           ['remove'] + list(files_to_remove),
 
72
 
                           working_dir=working_dir)
 
73
 
        #see if we can force it now
 
74
 
        self.run_bzr(['remove', '--force'] + list(files_to_remove),
 
75
 
                     working_dir=working_dir)
 
77
 
    def test_remove_new_no_files_specified(self):
 
78
 
        tree = self.make_branch_and_tree('.')
 
79
 
        self.run_bzr_error(["bzr: ERROR: No matching files."], 'remove --new')
 
80
 
        self.run_bzr_error(["bzr: ERROR: No matching files."], 'remove --new .')
 
82
 
    def test_remove_no_files_specified(self):
 
83
 
        tree = self._make_tree_and_add(['foo'])
 
84
 
        out, err = self.run_bzr(['rm'])
 
85
 
        self.assertEqual('', err)
 
86
 
        self.assertEqual('', out)
 
87
 
        self.assertInWorkingTree('foo', tree=tree)
 
88
 
        self.failUnlessExists('foo')
 
90
 
    def test_remove_no_files_specified_missing_dir_and_contents(self):
 
91
 
        tree = self._make_tree_and_add(
 
92
 
            ['foo', 'dir/', 'dir/missing/', 'dir/missing/child'])
 
93
 
        self.get_transport('.').delete_tree('dir/missing')
 
94
 
        out, err = self.run_bzr(['rm'])
 
95
 
        self.assertEqual('', out)
 
97
 
            'removed dir/missing/child\n'
 
98
 
            'removed dir/missing\n',
 
100
 
        # non-missing paths not touched:
 
101
 
        self.assertInWorkingTree('foo', tree=tree)
 
102
 
        self.failUnlessExists('foo')
 
103
 
        self.assertInWorkingTree('dir', tree=tree)
 
104
 
        self.failUnlessExists('dir')
 
105
 
        # missing files unversioned
 
106
 
        self.assertNotInWorkingTree('dir/missing', tree=tree)
 
107
 
        self.assertNotInWorkingTree('dir/missing/child', tree=tree)
 
109
 
    def test_remove_no_files_specified_already_deleted(self):
 
110
 
        tree = self._make_tree_and_add(['foo', 'bar'])
 
111
 
        tree.commit('save foo and bar')
 
114
 
        self.assertEqual(None, tree.path2id('bar'))
 
115
 
        # Running rm with a deleted file does not error.
 
116
 
        out, err = self.run_bzr(['rm'])
 
117
 
        self.assertEqual('', out)
 
118
 
        self.assertEqual('', err)
 
120
 
    def test_remove_no_files_specified_missing_file(self):
 
121
 
        tree = self._make_tree_and_add(['foo', 'bar'])
 
123
 
        out, err = self.run_bzr(['rm'])
 
124
 
        self.assertEqual('', out)
 
125
 
        self.assertEqual('removed bar\n', err)
 
126
 
        # non-missing files not touched:
 
127
 
        self.assertInWorkingTree('foo', tree=tree)
 
128
 
        self.failUnlessExists('foo')
 
129
 
        # missing files unversioned
 
130
 
        self.assertNotInWorkingTree('bar', tree=tree)
 
132
 
    def test_remove_no_files_specified_missing_link(self):
 
133
 
        self.requireFeature(SymlinkFeature)
 
134
 
        tree = self._make_tree_and_add(['foo'])
 
135
 
        os.symlink('foo', 'linkname')
 
136
 
        tree.add(['linkname'])
 
137
 
        os.unlink('linkname')
 
138
 
        out, err = self.run_bzr(['rm'])
 
139
 
        self.assertEqual('', out)
 
140
 
        self.assertEqual('removed linkname\n', err)
 
141
 
        # non-missing files not touched:
 
142
 
        self.assertInWorkingTree('foo', tree=tree)
 
143
 
        self.failUnlessExists('foo')
 
144
 
        # missing files unversioned
 
145
 
        self.assertNotInWorkingTree('linkname', tree=tree)
 
147
 
    def test_rm_one_file(self):
 
148
 
        tree = self._make_tree_and_add([a])
 
149
 
        self.run_bzr("commit -m 'added a'")
 
150
 
        self.run_bzr('rm a', error_regexes=["deleted a"])
 
151
 
        self.assertFilesDeleted([a])
 
153
 
    def test_remove_one_file(self):
 
154
 
        tree = self._make_tree_and_add([a])
 
155
 
        self.run_bzr("commit -m 'added a'")
 
156
 
        self.run_bzr('remove a', error_regexes=["deleted a"])
 
157
 
        self.assertFilesDeleted([a])
 
159
 
    def test_remove_keep_one_file(self):
 
160
 
        tree = self._make_tree_and_add([a])
 
161
 
        self.run_bzr('remove --keep a', error_regexes=["removed a"])
 
162
 
        self.assertFilesUnversioned([a])
 
164
 
    def test_remove_one_deleted_file(self):
 
165
 
        tree = self._make_tree_and_add([a])
 
166
 
        self.run_bzr("commit -m 'added a'")
 
168
 
        self.assertInWorkingTree(a)
 
169
 
        self.run_bzr('remove a')
 
170
 
        self.assertNotInWorkingTree(a)
 
172
 
    def test_remove_invalid_files(self):
 
173
 
        self.build_tree(files)
 
174
 
        tree = self.make_branch_and_tree('.')
 
175
 
        self.run_bzr(['remove', '.', 'xyz', 'abc/def'])
 
177
 
    def test_remove_unversioned_files(self):
 
178
 
        self.build_tree(files)
 
179
 
        tree = self.make_branch_and_tree('.')
 
180
 
        self.run_bzr_remove_changed_files(
 
181
 
            ['unknown:[.\s]*d/[.\s]*b/c[.\s]*b/[.\s]*a'], files)
 
183
 
    def test_remove_changed_files(self):
 
184
 
        tree = self._make_tree_and_add(files)
 
185
 
        self.run_bzr("commit -m 'added files'")
 
188
 
        self.run_bzr_remove_changed_files(['modified:[.\s]*a[.\s]*b/c'], files)
 
190
 
    def test_remove_changed_ignored_files(self):
 
191
 
        tree = self._make_tree_and_add(['a'])
 
192
 
        self.run_bzr(['ignore', 'a'])
 
193
 
        self.run_bzr_remove_changed_files(['added:[.\s]*a'], ['a'])
 
195
 
    def test_remove_changed_files_from_child_dir(self):
 
196
 
        if sys.platform == 'win32':
 
197
 
            raise TestSkipped("Windows unable to remove '.' directory")
 
198
 
        tree = self._make_tree_and_add(files)
 
199
 
        self.run_bzr("commit -m 'added files'")
 
202
 
        self.run_bzr_remove_changed_files(
 
203
 
            ['modified:[.\s]*a[.\s]*b/c'],
 
204
 
            ['../a', 'c', '.', '../d'], working_dir='b')
 
205
 
        self.assertNotInWorkingTree(files)
 
206
 
        self.failIfExists(files)
 
208
 
    def test_remove_keep_unversioned_files(self):
 
209
 
        self.build_tree(files)
 
210
 
        tree = self.make_branch_and_tree('.')
 
211
 
        self.run_bzr('remove --keep a', error_regexes=["a is not versioned."])
 
212
 
        self.assertFilesUnversioned(files)
 
214
 
    def test_remove_force_unversioned_files(self):
 
215
 
        self.build_tree(files)
 
216
 
        tree = self.make_branch_and_tree('.')
 
217
 
        self.run_bzr(['remove', '--force'] + list(files),
 
218
 
                     error_regexes=["deleted a", "deleted b",
 
219
 
                                    "deleted b/c", "deleted d"])
 
220
 
        self.assertFilesDeleted(files)
 
222
 
    def test_remove_deleted_files(self):
 
223
 
        tree = self._make_tree_and_add(files)
 
224
 
        self.run_bzr("commit -m 'added files'")
 
225
 
        my_files=[f for f in files]
 
226
 
        my_files.sort(reverse=True)
 
228
 
            osutils.delete_any(f)
 
229
 
        self.assertInWorkingTree(files)
 
230
 
        self.failIfExists(files)
 
231
 
        self.run_bzr('remove ' + ' '.join(files))
 
232
 
        self.assertNotInWorkingTree(a)
 
233
 
        self.failIfExists(files)
 
235
 
    def test_remove_non_existing_files(self):
 
236
 
        tree = self._make_tree_and_add([])
 
237
 
        self.run_bzr(['remove', 'b'])
 
239
 
    def test_remove_keep_non_existing_files(self):
 
240
 
        tree = self._make_tree_and_add([])
 
241
 
        self.run_bzr('remove --keep b', error_regexes=["b is not versioned."])
 
243
 
    def test_remove_files(self):
 
244
 
        tree = self._make_tree_and_add(files)
 
245
 
        self.run_bzr("commit -m 'added files'")
 
246
 
        self.run_bzr('remove a b b/c d',
 
247
 
                     error_regexes=["deleted a", "deleted b", "deleted b/c",
 
249
 
        self.assertFilesDeleted(files)
 
251
 
    def test_remove_keep_files(self):
 
252
 
        tree = self._make_tree_and_add(files)
 
253
 
        self.run_bzr("commit -m 'added files'")
 
254
 
        self.run_bzr('remove --keep a b b/c d',
 
255
 
                     error_regexes=["removed a", "removed b", "removed b/c",
 
257
 
        self.assertFilesUnversioned(files)
 
259
 
    def test_remove_with_new(self):
 
260
 
        tree = self._make_tree_and_add(files)
 
261
 
        self.run_bzr('remove --new --keep',
 
262
 
                     error_regexes=["removed a", "removed b", "removed b/c"])
 
263
 
        self.assertFilesUnversioned(files)
 
265
 
    def test_remove_with_new_in_dir1(self):
 
266
 
        tree = self._make_tree_and_add(files)
 
267
 
        self.run_bzr('remove --new --keep b b/c',
 
268
 
                     error_regexes=["removed b", "removed b/c"])
 
269
 
        tree = WorkingTree.open('.')
 
270
 
        self.assertInWorkingTree(a)
 
271
 
        self.assertEqual(tree.path2id(a), a + _id)
 
272
 
        self.assertFilesUnversioned([b,c])
 
274
 
    def test_remove_with_new_in_dir2(self):
 
275
 
        tree = self._make_tree_and_add(files)
 
276
 
        self.run_bzr('remove --new --keep .',
 
277
 
                     error_regexes=["removed a", "removed b", "removed b/c"])
 
278
 
        tree = WorkingTree.open('.')
 
279
 
        self.assertFilesUnversioned(files)
 
 
26
    def test_remove_deleted(self):
 
 
28
        self.build_tree(['a'])
 
 
29
        self.runbzr(['add', 'a'])
 
 
30
        self.runbzr(['commit', '-m', 'added a'])
 
 
32
        self.runbzr(['remove', 'a'])
 
 
34
    def test_remove_new(self):
 
 
35
        self.build_tree(['filefile',
 
 
38
        wt = self.make_branch_and_tree('.')
 
 
39
        wt.add(['filefile', 'dir', 'dir/filefilefile'], 
 
 
40
               ['filefile-id', 'dir-id', 'filefilefile-id'])
 
 
41
        self.assertEqual(wt.path2id('filefile'), 'filefile-id')
 
 
42
        self.assertEqual(wt.path2id('dir/filefilefile'), 'filefilefile-id')
 
 
43
        self.assertEqual(wt.path2id('dir'), 'dir-id')
 
 
44
        self.runbzr('remove --new')
 
 
45
        wt = WorkingTree.open('.')
 
 
46
        self.assertIs(wt.path2id('filefile'), None)
 
 
47
        self.assertIs(wt.path2id('dir/filefilefile'), None)
 
 
48
        self.assertIs(wt.path2id('dir'), None)
 
 
49
        wt.add(['filefile', 'dir', 'dir/filefilefile'], 
 
 
50
               ['filefile-id', 'dir-id', 'filefilefile-id'])
 
 
51
        self.assertEqual(wt.path2id('filefile'), 'filefile-id')
 
 
52
        self.assertEqual(wt.path2id('dir/filefilefile'), 'filefilefile-id')
 
 
53
        self.assertEqual(wt.path2id('dir'), 'dir-id')
 
 
54
        self.runbzr('remove --new dir')
 
 
55
        wt = WorkingTree.open('.')
 
 
56
        self.assertEqual(wt.path2id('filefile'), 'filefile-id')
 
 
57
        self.assertIs(wt.path2id('dir/filefilefile'), None)
 
 
58
        self.assertIs(wt.path2id('dir'), None)
 
 
59
        self.runbzr('remove --new .')
 
 
60
        wt = WorkingTree.open('.')
 
 
61
        self.assertIs(wt.path2id('filefile'), None)
 
 
62
        self.runbzr('remove --new .', retcode=3)