~bzr-pqm/bzr/bzr.dev

4634.125.4 by John Arbash Meinel
Merge bzr/2.0@4725, resolve NEWS
1
# Copyright (C) 2006-2010 Canonical Ltd
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
2
#
1534.7.106 by Aaron Bentley
Cleaned up imports, added copyright statements
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.
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
7
#
1534.7.106 by Aaron Bentley
Cleaned up imports, added copyright statements
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.
1887.1.1 by Adeodato Simó
Do not separate paragraphs in the copyright statement with blank lines,
12
#
1534.7.106 by Aaron Bentley
Cleaned up imports, added copyright statements
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
4183.7.1 by Sabin Iacob
update FSF mailing address
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
1534.7.106 by Aaron Bentley
Cleaned up imports, added copyright statements
16
17
import os
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
18
from StringIO import StringIO
2027.1.1 by John Arbash Meinel
Fix bug #56549, and write a direct test that the right path is being statted
19
import sys
4934.1.1 by John Arbash Meinel
Basic implementation for windows and bug #488724.
20
import time
1558.1.3 by Aaron Bentley
Fixed deprecated op use in test suite
21
2027.1.1 by John Arbash Meinel
Fix bug #56549, and write a direct test that the right path is being statted
22
from bzrlib import (
2694.5.4 by Jelmer Vernooij
Move bzrlib.util.bencode to bzrlib._bencode_py.
23
    bencode,
2090.2.1 by Martin Pool
Fix some code which relies on assertions and breaks under python -O
24
    errors,
4826.1.5 by Andrew Bennetts
Add test that content filtered files are not hardlinked by build_tree.
25
    filters,
2116.4.1 by John Arbash Meinel
Update file and revision id generators.
26
    generate_ids,
3199.1.4 by Vincent Ladeuil
Fix 16 leaked tmp dirs. Probably indicates a lock handling problem with TransformPreview
27
    osutils,
3008.1.18 by Aaron Bentley
Get supported PreviewTree functionality under test
28
    revision as _mod_revision,
4826.1.5 by Andrew Bennetts
Add test that content filtered files are not hardlinked by build_tree.
29
    rules,
2027.1.1 by John Arbash Meinel
Fix bug #56549, and write a direct test that the right path is being statted
30
    tests,
31
    urlutils,
32
    )
1558.1.3 by Aaron Bentley
Fixed deprecated op use in test suite
33
from bzrlib.bzrdir import BzrDir
5162.4.1 by Aaron Bentley
TreeTransform supports normal commit parameters and includes branch nick.
34
from bzrlib.conflicts import (
35
    DeletingParent,
36
    DuplicateEntry,
37
    DuplicateID,
38
    MissingParent,
39
    NonDirectoryParent,
40
    ParentLoop,
41
    UnversionedParent,
42
)
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
43
from bzrlib.diff import show_diff_trees
5162.4.1 by Aaron Bentley
TreeTransform supports normal commit parameters and includes branch nick.
44
from bzrlib.errors import (
45
    DuplicateKey,
46
    ExistingLimbo,
47
    ExistingPendingDeletion,
48
    ImmortalLimbo,
49
    ImmortalPendingDeletion,
50
    LockError,
51
    MalformedTransform,
52
    NoSuchFile,
53
    ReusingTransform,
54
)
55
from bzrlib.osutils import (
56
    file_kind,
57
    pathjoin,
58
)
3363.17.1 by Aaron Bentley
Avoid inventory for merge and transform code
59
from bzrlib.merge import Merge3Merger, Merger
2949.5.1 by Alexander Belchenko
selftest: use SymlinkFeature instead of TestSkipped where appropriate
60
from bzrlib.tests import (
3136.1.1 by Aaron Bentley
Add support for hardlinks to TreeTransform
61
    HardlinkFeature,
2949.5.1 by Alexander Belchenko
selftest: use SymlinkFeature instead of TestSkipped where appropriate
62
    SymlinkFeature,
63
    TestCase,
64
    TestCaseInTempDir,
65
    TestSkipped,
5162.4.1 by Aaron Bentley
TreeTransform supports normal commit parameters and includes branch nick.
66
)
67
from bzrlib.transform import (
68
    build_tree,
69
    create_from_tree,
70
    cook_conflicts,
71
    _FileMover,
72
    FinalPaths,
73
    get_backup_name,
74
    resolve_conflicts,
75
    resolve_checkout,
76
    ROOT_PARENT,
77
    TransformPreview,
78
    TreeTransform,
79
)
0.13.13 by Aaron Bentley
Add direct test of serialization records
80
1534.7.1 by Aaron Bentley
Got creation of a versioned file working
81
2100.3.21 by Aaron Bentley
Work on checking out by-reference trees
82
class TestTreeTransform(tests.TestCaseWithTransport):
1740.2.4 by Aaron Bentley
Update transform tests and docs
83
1534.7.59 by Aaron Bentley
Simplified tests
84
    def setUp(self):
85
        super(TestTreeTransform, self).setUp()
2255.2.194 by Robert Collins
[BROKEN] Many updates to stop using experimental formats in tests.
86
        self.wt = self.make_branch_and_tree('.', format='dirstate-with-subtree')
1534.7.161 by Aaron Bentley
Used appropriate control_files
87
        os.chdir('..')
1534.7.59 by Aaron Bentley
Simplified tests
88
89
    def get_transform(self):
90
        transform = TreeTransform(self.wt)
3453.2.7 by Aaron Bentley
Remove test kipple
91
        self.addCleanup(transform.finalize)
1731.1.33 by Aaron Bentley
Revert no-special-root changes
92
        return transform, transform.root
1534.7.59 by Aaron Bentley
Simplified tests
93
1534.7.162 by Aaron Bentley
Handle failures creating/deleting the Limbo directory
94
    def test_existing_limbo(self):
95
        transform, root = self.get_transform()
2733.2.6 by Aaron Bentley
Make TreeTransform commits rollbackable
96
        limbo_name = transform._limbodir
97
        deletion_path = transform._deletiondir
1534.7.176 by abentley
Fixed up tests for Windows
98
        os.mkdir(pathjoin(limbo_name, 'hehe'))
1534.7.162 by Aaron Bentley
Handle failures creating/deleting the Limbo directory
99
        self.assertRaises(ImmortalLimbo, transform.apply)
100
        self.assertRaises(LockError, self.wt.unlock)
101
        self.assertRaises(ExistingLimbo, self.get_transform)
102
        self.assertRaises(LockError, self.wt.unlock)
1534.7.176 by abentley
Fixed up tests for Windows
103
        os.rmdir(pathjoin(limbo_name, 'hehe'))
1534.7.162 by Aaron Bentley
Handle failures creating/deleting the Limbo directory
104
        os.rmdir(limbo_name)
2733.2.6 by Aaron Bentley
Make TreeTransform commits rollbackable
105
        os.rmdir(deletion_path)
1534.7.162 by Aaron Bentley
Handle failures creating/deleting the Limbo directory
106
        transform, root = self.get_transform()
107
        transform.apply()
1534.7.59 by Aaron Bentley
Simplified tests
108
2733.2.11 by Aaron Bentley
Detect irregularities with the pending-deletion directory
109
    def test_existing_pending_deletion(self):
110
        transform, root = self.get_transform()
111
        deletion_path = self._limbodir = urlutils.local_path_from_url(
3407.2.8 by Martin Pool
Deprecate LockableFiles.controlfilename
112
            transform._tree._transport.abspath('pending-deletion'))
2733.2.11 by Aaron Bentley
Detect irregularities with the pending-deletion directory
113
        os.mkdir(pathjoin(deletion_path, 'blocking-directory'))
114
        self.assertRaises(ImmortalPendingDeletion, transform.apply)
115
        self.assertRaises(LockError, self.wt.unlock)
116
        self.assertRaises(ExistingPendingDeletion, self.get_transform)
117
1534.7.1 by Aaron Bentley
Got creation of a versioned file working
118
    def test_build(self):
3034.2.1 by Aaron Bentley
Fix is_executable tests for win32
119
        transform, root = self.get_transform()
120
        self.wt.lock_tree_write()
121
        self.addCleanup(self.wt.unlock)
1534.7.59 by Aaron Bentley
Simplified tests
122
        self.assertIs(transform.get_tree_parent(root), ROOT_PARENT)
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
123
        imaginary_id = transform.trans_id_tree_path('imaginary')
1534.10.32 by Aaron Bentley
Test and fix case where name has trailing slash
124
        imaginary_id2 = transform.trans_id_tree_path('imaginary/')
125
        self.assertEqual(imaginary_id, imaginary_id2)
1534.7.59 by Aaron Bentley
Simplified tests
126
        self.assertEqual(transform.get_tree_parent(imaginary_id), root)
127
        self.assertEqual(transform.final_kind(root), 'directory')
128
        self.assertEqual(transform.final_file_id(root), self.wt.get_root_id())
129
        trans_id = transform.create_path('name', root)
130
        self.assertIs(transform.final_file_id(trans_id), None)
131
        self.assertRaises(NoSuchFile, transform.final_kind, trans_id)
132
        transform.create_file('contents', trans_id)
133
        transform.set_executability(True, trans_id)
134
        transform.version_file('my_pretties', trans_id)
135
        self.assertRaises(DuplicateKey, transform.version_file,
136
                          'my_pretties', trans_id)
137
        self.assertEqual(transform.final_file_id(trans_id), 'my_pretties')
138
        self.assertEqual(transform.final_parent(trans_id), root)
139
        self.assertIs(transform.final_parent(root), ROOT_PARENT)
140
        self.assertIs(transform.get_tree_parent(root), ROOT_PARENT)
141
        oz_id = transform.create_path('oz', root)
142
        transform.create_directory(oz_id)
143
        transform.version_file('ozzie', oz_id)
144
        trans_id2 = transform.create_path('name2', root)
145
        transform.create_file('contents', trans_id2)
146
        transform.set_executability(False, trans_id2)
147
        transform.version_file('my_pretties2', trans_id2)
1534.7.191 by Aaron Bentley
Got transform.apply to list modified paths
148
        modified_paths = transform.apply().modified_paths
1534.7.100 by Aaron Bentley
Fixed path-relative test cases
149
        self.assertEqual('contents', self.wt.get_file_byname('name').read())
1534.7.59 by Aaron Bentley
Simplified tests
150
        self.assertEqual(self.wt.path2id('name'), 'my_pretties')
151
        self.assertIs(self.wt.is_executable('my_pretties'), True)
152
        self.assertIs(self.wt.is_executable('my_pretties2'), False)
1534.7.100 by Aaron Bentley
Fixed path-relative test cases
153
        self.assertEqual('directory', file_kind(self.wt.abspath('oz')))
1534.7.191 by Aaron Bentley
Got transform.apply to list modified paths
154
        self.assertEqual(len(modified_paths), 3)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
155
        tree_mod_paths = [self.wt.id2abspath(f) for f in
1534.7.191 by Aaron Bentley
Got transform.apply to list modified paths
156
                          ('ozzie', 'my_pretties', 'my_pretties2')]
157
        self.assertSubset(tree_mod_paths, modified_paths)
1534.7.1 by Aaron Bentley
Got creation of a versioned file working
158
        # is it safe to finalize repeatedly?
159
        transform.finalize()
1534.7.59 by Aaron Bentley
Simplified tests
160
        transform.finalize()
1534.7.2 by Aaron Bentley
Added convenience function
161
4934.1.1 by John Arbash Meinel
Basic implementation for windows and bug #488724.
162
    def test_create_files_same_timestamp(self):
163
        transform, root = self.get_transform()
164
        self.wt.lock_tree_write()
165
        self.addCleanup(self.wt.unlock)
166
        # Roll back the clock, so that we know everything is being set to the
167
        # exact time
4934.1.10 by John Arbash Meinel
Rework the tests a bit.
168
        transform._creation_mtime = creation_mtime = time.time() - 20.0
4934.1.1 by John Arbash Meinel
Basic implementation for windows and bug #488724.
169
        transform.create_file('content-one',
170
                              transform.create_path('one', root))
171
        time.sleep(1) # *ugly*
172
        transform.create_file('content-two',
173
                              transform.create_path('two', root))
174
        transform.apply()
175
        fo, st1 = self.wt.get_file_with_stat(None, path='one', filtered=False)
176
        fo.close()
177
        fo, st2 = self.wt.get_file_with_stat(None, path='two', filtered=False)
178
        fo.close()
4934.2.2 by Martin
Rearrange osutils.fset_mtime tests a little and make them robust on FAT filesystems
179
        # We only guarantee 2s resolution
4934.1.11 by John Arbash Meinel
2 => 2.0 because it looks better.
180
        self.assertTrue(abs(creation_mtime - st1.st_mtime) < 2.0,
4934.1.10 by John Arbash Meinel
Rework the tests a bit.
181
            "%s != %s within 2 seconds" % (creation_mtime, st1.st_mtime))
4934.1.6 by John Arbash Meinel
It turns out that with os/python/C buffering, we need to flush
182
        # But if we have more than that, all files should get the same result
4934.1.1 by John Arbash Meinel
Basic implementation for windows and bug #488724.
183
        self.assertEqual(st1.st_mtime, st2.st_mtime)
184
4634.122.2 by Aaron Bentley, John Arbash Meinel
Bring the fixup_new_roots code from the nested-trees code.
185
    def test_change_root_id(self):
186
        transform, root = self.get_transform()
187
        self.assertNotEqual('new-root-id', self.wt.get_root_id())
188
        transform.new_directory('', ROOT_PARENT, 'new-root-id')
189
        transform.delete_contents(root)
190
        transform.unversion_file(root)
191
        transform.fixup_new_roots()
192
        transform.apply()
193
        self.assertEqual('new-root-id', self.wt.get_root_id())
194
195
    def test_change_root_id_add_files(self):
196
        transform, root = self.get_transform()
197
        self.assertNotEqual('new-root-id', self.wt.get_root_id())
198
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'new-root-id')
199
        transform.new_file('file', new_trans_id, ['new-contents\n'],
200
                           'new-file-id')
201
        transform.delete_contents(root)
202
        transform.unversion_file(root)
203
        transform.fixup_new_roots()
204
        transform.apply()
205
        self.assertEqual('new-root-id', self.wt.get_root_id())
206
        self.assertEqual('new-file-id', self.wt.path2id('file'))
207
        self.assertFileEqual('new-contents\n', self.wt.abspath('file'))
208
209
    def test_add_two_roots(self):
210
        transform, root = self.get_transform()
211
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'new-root-id')
212
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'alt-root-id')
213
        self.assertRaises(ValueError, transform.fixup_new_roots)
214
3136.1.1 by Aaron Bentley
Add support for hardlinks to TreeTransform
215
    def test_hardlink(self):
216
        self.requireFeature(HardlinkFeature)
217
        transform, root = self.get_transform()
218
        transform.new_file('file1', root, 'contents')
219
        transform.apply()
220
        target = self.make_branch_and_tree('target')
221
        target_transform = TreeTransform(target)
222
        trans_id = target_transform.create_path('file1', target_transform.root)
223
        target_transform.create_hardlink(self.wt.abspath('file1'), trans_id)
224
        target_transform.apply()
225
        self.failUnlessExists('target/file1')
226
        source_stat = os.stat(self.wt.abspath('file1'))
227
        target_stat = os.stat('target/file1')
228
        self.assertEqual(source_stat, target_stat)
229
1534.7.2 by Aaron Bentley
Added convenience function
230
    def test_convenience(self):
1534.7.59 by Aaron Bentley
Simplified tests
231
        transform, root = self.get_transform()
3034.2.1 by Aaron Bentley
Fix is_executable tests for win32
232
        self.wt.lock_tree_write()
233
        self.addCleanup(self.wt.unlock)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
234
        trans_id = transform.new_file('name', root, 'contents',
1534.7.59 by Aaron Bentley
Simplified tests
235
                                      'my_pretties', True)
236
        oz = transform.new_directory('oz', root, 'oz-id')
237
        dorothy = transform.new_directory('dorothy', oz, 'dorothy-id')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
238
        toto = transform.new_file('toto', dorothy, 'toto-contents',
1534.7.59 by Aaron Bentley
Simplified tests
239
                                  'toto-id', False)
240
241
        self.assertEqual(len(transform.find_conflicts()), 0)
242
        transform.apply()
243
        self.assertRaises(ReusingTransform, transform.find_conflicts)
1534.7.100 by Aaron Bentley
Fixed path-relative test cases
244
        self.assertEqual('contents', file(self.wt.abspath('name')).read())
1534.7.59 by Aaron Bentley
Simplified tests
245
        self.assertEqual(self.wt.path2id('name'), 'my_pretties')
246
        self.assertIs(self.wt.is_executable('my_pretties'), True)
247
        self.assertEqual(self.wt.path2id('oz'), 'oz-id')
248
        self.assertEqual(self.wt.path2id('oz/dorothy'), 'dorothy-id')
249
        self.assertEqual(self.wt.path2id('oz/dorothy/toto'), 'toto-id')
250
2255.2.194 by Robert Collins
[BROKEN] Many updates to stop using experimental formats in tests.
251
        self.assertEqual('toto-contents',
1534.7.100 by Aaron Bentley
Fixed path-relative test cases
252
                         self.wt.get_file_byname('oz/dorothy/toto').read())
1534.7.59 by Aaron Bentley
Simplified tests
253
        self.assertIs(self.wt.is_executable('toto-id'), False)
1534.7.6 by Aaron Bentley
Added conflict handling
254
2100.3.21 by Aaron Bentley
Work on checking out by-reference trees
255
    def test_tree_reference(self):
256
        transform, root = self.get_transform()
257
        tree = transform._tree
258
        trans_id = transform.new_directory('reference', root, 'subtree-id')
259
        transform.set_tree_reference('subtree-revision', trans_id)
260
        transform.apply()
2255.2.194 by Robert Collins
[BROKEN] Many updates to stop using experimental formats in tests.
261
        tree.lock_read()
262
        self.addCleanup(tree.unlock)
263
        self.assertEqual('subtree-revision',
2100.3.21 by Aaron Bentley
Work on checking out by-reference trees
264
                         tree.inventory['subtree-id'].reference_revision)
265
1534.7.6 by Aaron Bentley
Added conflict handling
266
    def test_conflicts(self):
1534.7.59 by Aaron Bentley
Simplified tests
267
        transform, root = self.get_transform()
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
268
        trans_id = transform.new_file('name', root, 'contents',
1534.7.59 by Aaron Bentley
Simplified tests
269
                                      'my_pretties')
270
        self.assertEqual(len(transform.find_conflicts()), 0)
271
        trans_id2 = transform.new_file('name', root, 'Crontents', 'toto')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
272
        self.assertEqual(transform.find_conflicts(),
1534.7.59 by Aaron Bentley
Simplified tests
273
                         [('duplicate', trans_id, trans_id2, 'name')])
274
        self.assertRaises(MalformedTransform, transform.apply)
275
        transform.adjust_path('name', trans_id, trans_id2)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
276
        self.assertEqual(transform.find_conflicts(),
1534.7.59 by Aaron Bentley
Simplified tests
277
                         [('non-directory parent', trans_id)])
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
278
        tinman_id = transform.trans_id_tree_path('tinman')
1534.7.59 by Aaron Bentley
Simplified tests
279
        transform.adjust_path('name', tinman_id, trans_id2)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
280
        self.assertEqual(transform.find_conflicts(),
281
                         [('unversioned parent', tinman_id),
1534.7.59 by Aaron Bentley
Simplified tests
282
                          ('missing parent', tinman_id)])
283
        lion_id = transform.create_path('lion', root)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
284
        self.assertEqual(transform.find_conflicts(),
285
                         [('unversioned parent', tinman_id),
1534.7.59 by Aaron Bentley
Simplified tests
286
                          ('missing parent', tinman_id)])
287
        transform.adjust_path('name', lion_id, trans_id2)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
288
        self.assertEqual(transform.find_conflicts(),
1534.7.59 by Aaron Bentley
Simplified tests
289
                         [('unversioned parent', lion_id),
290
                          ('missing parent', lion_id)])
291
        transform.version_file("Courage", lion_id)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
292
        self.assertEqual(transform.find_conflicts(),
293
                         [('missing parent', lion_id),
1534.7.59 by Aaron Bentley
Simplified tests
294
                          ('versioning no contents', lion_id)])
295
        transform.adjust_path('name2', root, trans_id2)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
296
        self.assertEqual(transform.find_conflicts(),
1534.7.59 by Aaron Bentley
Simplified tests
297
                         [('versioning no contents', lion_id)])
298
        transform.create_file('Contents, okay?', lion_id)
299
        transform.adjust_path('name2', trans_id2, trans_id2)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
300
        self.assertEqual(transform.find_conflicts(),
301
                         [('parent loop', trans_id2),
1534.7.59 by Aaron Bentley
Simplified tests
302
                          ('non-directory parent', trans_id2)])
303
        transform.adjust_path('name2', root, trans_id2)
304
        oz_id = transform.new_directory('oz', root)
305
        transform.set_executability(True, oz_id)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
306
        self.assertEqual(transform.find_conflicts(),
1534.7.59 by Aaron Bentley
Simplified tests
307
                         [('unversioned executability', oz_id)])
308
        transform.version_file('oz-id', oz_id)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
309
        self.assertEqual(transform.find_conflicts(),
1534.7.59 by Aaron Bentley
Simplified tests
310
                         [('non-file executability', oz_id)])
311
        transform.set_executability(None, oz_id)
1534.7.71 by abentley
All tests pass under Windows
312
        tip_id = transform.new_file('tip', oz_id, 'ozma', 'tip-id')
1534.7.59 by Aaron Bentley
Simplified tests
313
        transform.apply()
314
        self.assertEqual(self.wt.path2id('name'), 'my_pretties')
315
        self.assertEqual('contents', file(self.wt.abspath('name')).read())
316
        transform2, root = self.get_transform()
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
317
        oz_id = transform2.trans_id_tree_file_id('oz-id')
1534.7.59 by Aaron Bentley
Simplified tests
318
        newtip = transform2.new_file('tip', oz_id, 'other', 'tip-id')
319
        result = transform2.find_conflicts()
1534.7.135 by Aaron Bentley
Fixed deletion handling
320
        fp = FinalPaths(transform2)
1534.7.59 by Aaron Bentley
Simplified tests
321
        self.assert_('oz/tip' in transform2._tree_path_ids)
1534.7.176 by abentley
Fixed up tests for Windows
322
        self.assertEqual(fp.get_path(newtip), pathjoin('oz', 'tip'))
1534.7.59 by Aaron Bentley
Simplified tests
323
        self.assertEqual(len(result), 2)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
324
        self.assertEqual((result[0][0], result[0][1]),
1534.7.59 by Aaron Bentley
Simplified tests
325
                         ('duplicate', newtip))
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
326
        self.assertEqual((result[1][0], result[1][2]),
1534.7.59 by Aaron Bentley
Simplified tests
327
                         ('duplicate id', newtip))
1534.7.73 by Aaron Bentley
Changed model again. Now iterator is used immediately.
328
        transform2.finalize()
1534.7.59 by Aaron Bentley
Simplified tests
329
        transform3 = TreeTransform(self.wt)
330
        self.addCleanup(transform3.finalize)
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
331
        oz_id = transform3.trans_id_tree_file_id('oz-id')
1534.7.59 by Aaron Bentley
Simplified tests
332
        transform3.delete_contents(oz_id)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
333
        self.assertEqual(transform3.find_conflicts(),
1534.7.59 by Aaron Bentley
Simplified tests
334
                         [('missing parent', oz_id)])
1731.1.33 by Aaron Bentley
Revert no-special-root changes
335
        root_id = transform3.root
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
336
        tip_id = transform3.trans_id_tree_file_id('tip-id')
1534.7.59 by Aaron Bentley
Simplified tests
337
        transform3.adjust_path('tip', root_id, tip_id)
338
        transform3.apply()
1534.7.36 by Aaron Bentley
Added rename tests
339
3034.4.1 by Aaron Bentley
Start handling case-insensitivity
340
    def test_conflict_on_case_insensitive(self):
341
        tree = self.make_branch_and_tree('tree')
342
        # Don't try this at home, kids!
343
        # Force the tree to report that it is case sensitive, for conflict
344
        # resolution tests
345
        tree.case_sensitive = True
346
        transform = TreeTransform(tree)
347
        self.addCleanup(transform.finalize)
348
        transform.new_file('file', transform.root, 'content')
349
        transform.new_file('FiLe', transform.root, 'content')
350
        result = transform.find_conflicts()
351
        self.assertEqual([], result)
3008.1.15 by Aaron Bentley
Make case_sensitive an aspect of the transform, not the source tree
352
        transform.finalize()
3034.4.1 by Aaron Bentley
Start handling case-insensitivity
353
        # Force the tree to report that it is case insensitive, for conflict
354
        # generation tests
355
        tree.case_sensitive = False
3008.1.15 by Aaron Bentley
Make case_sensitive an aspect of the transform, not the source tree
356
        transform = TreeTransform(tree)
357
        self.addCleanup(transform.finalize)
358
        transform.new_file('file', transform.root, 'content')
359
        transform.new_file('FiLe', transform.root, 'content')
3034.4.1 by Aaron Bentley
Start handling case-insensitivity
360
        result = transform.find_conflicts()
361
        self.assertEqual([('duplicate', 'new-1', 'new-2', 'file')], result)
362
3034.4.7 by Aaron Bentley
Add test for filename conflicts with existing files
363
    def test_conflict_on_case_insensitive_existing(self):
364
        tree = self.make_branch_and_tree('tree')
365
        self.build_tree(['tree/FiLe'])
366
        # Don't try this at home, kids!
367
        # Force the tree to report that it is case sensitive, for conflict
368
        # resolution tests
369
        tree.case_sensitive = True
370
        transform = TreeTransform(tree)
371
        self.addCleanup(transform.finalize)
372
        transform.new_file('file', transform.root, 'content')
373
        result = transform.find_conflicts()
374
        self.assertEqual([], result)
3008.1.15 by Aaron Bentley
Make case_sensitive an aspect of the transform, not the source tree
375
        transform.finalize()
3034.4.7 by Aaron Bentley
Add test for filename conflicts with existing files
376
        # Force the tree to report that it is case insensitive, for conflict
377
        # generation tests
378
        tree.case_sensitive = False
3008.1.15 by Aaron Bentley
Make case_sensitive an aspect of the transform, not the source tree
379
        transform = TreeTransform(tree)
380
        self.addCleanup(transform.finalize)
381
        transform.new_file('file', transform.root, 'content')
3034.4.7 by Aaron Bentley
Add test for filename conflicts with existing files
382
        result = transform.find_conflicts()
383
        self.assertEqual([('duplicate', 'new-1', 'new-2', 'file')], result)
384
3034.4.1 by Aaron Bentley
Start handling case-insensitivity
385
    def test_resolve_case_insensitive_conflict(self):
386
        tree = self.make_branch_and_tree('tree')
387
        # Don't try this at home, kids!
388
        # Force the tree to report that it is case insensitive, for conflict
389
        # resolution tests
390
        tree.case_sensitive = False
391
        transform = TreeTransform(tree)
392
        self.addCleanup(transform.finalize)
393
        transform.new_file('file', transform.root, 'content')
394
        transform.new_file('FiLe', transform.root, 'content')
395
        resolve_conflicts(transform)
396
        transform.apply()
397
        self.failUnlessExists('tree/file')
398
        self.failUnlessExists('tree/FiLe.moved')
399
3034.4.2 by Aaron Bentley
Get conflict handling and case-insensitive tree creation under test
400
    def test_resolve_checkout_case_conflict(self):
3034.4.1 by Aaron Bentley
Start handling case-insensitivity
401
        tree = self.make_branch_and_tree('tree')
402
        # Don't try this at home, kids!
403
        # Force the tree to report that it is case insensitive, for conflict
404
        # resolution tests
405
        tree.case_sensitive = False
406
        transform = TreeTransform(tree)
407
        self.addCleanup(transform.finalize)
408
        transform.new_file('file', transform.root, 'content')
409
        transform.new_file('FiLe', transform.root, 'content')
3034.4.2 by Aaron Bentley
Get conflict handling and case-insensitive tree creation under test
410
        resolve_conflicts(transform,
411
                          pass_func=lambda t, c: resolve_checkout(t, c, []))
3034.4.1 by Aaron Bentley
Start handling case-insensitivity
412
        transform.apply()
413
        self.failUnlessExists('tree/file')
414
        self.failUnlessExists('tree/FiLe.moved')
415
3034.4.2 by Aaron Bentley
Get conflict handling and case-insensitive tree creation under test
416
    def test_apply_case_conflict(self):
3034.4.6 by Aaron Bentley
Update apply_case_conflict tests
417
        """Ensure that a transform with case conflicts can always be applied"""
3034.4.2 by Aaron Bentley
Get conflict handling and case-insensitive tree creation under test
418
        tree = self.make_branch_and_tree('tree')
419
        transform = TreeTransform(tree)
420
        self.addCleanup(transform.finalize)
421
        transform.new_file('file', transform.root, 'content')
422
        transform.new_file('FiLe', transform.root, 'content')
3034.4.6 by Aaron Bentley
Update apply_case_conflict tests
423
        dir = transform.new_directory('dir', transform.root)
424
        transform.new_file('dirfile', dir, 'content')
425
        transform.new_file('dirFiLe', dir, 'content')
426
        resolve_conflicts(transform)
3034.4.2 by Aaron Bentley
Get conflict handling and case-insensitive tree creation under test
427
        transform.apply()
3034.4.3 by Aaron Bentley
Add case-sensitivity handling to WorkingTree
428
        self.failUnlessExists('tree/file')
429
        if not os.path.exists('tree/FiLe.moved'):
430
            self.failUnlessExists('tree/FiLe')
3034.4.6 by Aaron Bentley
Update apply_case_conflict tests
431
        self.failUnlessExists('tree/dir/dirfile')
432
        if not os.path.exists('tree/dir/dirFiLe.moved'):
433
            self.failUnlessExists('tree/dir/dirFiLe')
3034.4.2 by Aaron Bentley
Get conflict handling and case-insensitive tree creation under test
434
3034.4.1 by Aaron Bentley
Start handling case-insensitivity
435
    def test_case_insensitive_limbo(self):
436
        tree = self.make_branch_and_tree('tree')
437
        # Don't try this at home, kids!
438
        # Force the tree to report that it is case insensitive
439
        tree.case_sensitive = False
440
        transform = TreeTransform(tree)
441
        self.addCleanup(transform.finalize)
442
        dir = transform.new_directory('dir', transform.root)
443
        first = transform.new_file('file', dir, 'content')
444
        second = transform.new_file('FiLe', dir, 'content')
445
        self.assertContainsRe(transform._limbo_name(first), 'new-1/file')
446
        self.assertNotContainsRe(transform._limbo_name(second), 'new-1/FiLe')
447
4634.78.1 by Aaron Bentley
adjust_path updatest limbo paths.
448
    def test_adjust_path_updates_child_limbo_names(self):
449
        tree = self.make_branch_and_tree('tree')
450
        transform = TreeTransform(tree)
451
        self.addCleanup(transform.finalize)
452
        foo_id = transform.new_directory('foo', transform.root)
453
        bar_id = transform.new_directory('bar', foo_id)
454
        baz_id = transform.new_directory('baz', bar_id)
455
        qux_id = transform.new_directory('qux', baz_id)
456
        transform.adjust_path('quxx', foo_id, bar_id)
457
        self.assertStartsWith(transform._limbo_name(qux_id),
458
                              transform._limbo_name(bar_id))
459
1558.7.11 by Aaron Bentley
Avoid spurious conflict on add/delete
460
    def test_add_del(self):
461
        start, root = self.get_transform()
462
        start.new_directory('a', root, 'a')
463
        start.apply()
464
        transform, root = self.get_transform()
465
        transform.delete_versioned(transform.trans_id_tree_file_id('a'))
466
        transform.new_directory('a', root, 'a')
467
        transform.apply()
468
1534.7.46 by Aaron Bentley
Ensured a conflict when parents of versioned files are unversioned
469
    def test_unversioning(self):
1534.7.59 by Aaron Bentley
Simplified tests
470
        create_tree, root = self.get_transform()
471
        parent_id = create_tree.new_directory('parent', root, 'parent-id')
472
        create_tree.new_file('child', parent_id, 'child', 'child-id')
473
        create_tree.apply()
474
        unversion = TreeTransform(self.wt)
475
        self.addCleanup(unversion.finalize)
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
476
        parent = unversion.trans_id_tree_path('parent')
1534.7.59 by Aaron Bentley
Simplified tests
477
        unversion.unversion_file(parent)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
478
        self.assertEqual(unversion.find_conflicts(),
1534.7.59 by Aaron Bentley
Simplified tests
479
                         [('unversioned parent', parent_id)])
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
480
        file_id = unversion.trans_id_tree_file_id('child-id')
1534.7.59 by Aaron Bentley
Simplified tests
481
        unversion.unversion_file(file_id)
482
        unversion.apply()
1534.7.46 by Aaron Bentley
Ensured a conflict when parents of versioned files are unversioned
483
1534.7.36 by Aaron Bentley
Added rename tests
484
    def test_name_invariants(self):
1534.7.59 by Aaron Bentley
Simplified tests
485
        create_tree, root = self.get_transform()
486
        # prepare tree
1731.1.33 by Aaron Bentley
Revert no-special-root changes
487
        root = create_tree.root
1534.7.59 by Aaron Bentley
Simplified tests
488
        create_tree.new_file('name1', root, 'hello1', 'name1')
489
        create_tree.new_file('name2', root, 'hello2', 'name2')
490
        ddir = create_tree.new_directory('dying_directory', root, 'ddir')
491
        create_tree.new_file('dying_file', ddir, 'goodbye1', 'dfile')
492
        create_tree.new_file('moving_file', ddir, 'later1', 'mfile')
493
        create_tree.new_file('moving_file2', root, 'later2', 'mfile2')
494
        create_tree.apply()
495
496
        mangle_tree,root = self.get_transform()
1731.1.33 by Aaron Bentley
Revert no-special-root changes
497
        root = mangle_tree.root
1534.7.59 by Aaron Bentley
Simplified tests
498
        #swap names
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
499
        name1 = mangle_tree.trans_id_tree_file_id('name1')
500
        name2 = mangle_tree.trans_id_tree_file_id('name2')
1534.7.59 by Aaron Bentley
Simplified tests
501
        mangle_tree.adjust_path('name2', root, name1)
502
        mangle_tree.adjust_path('name1', root, name2)
503
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
504
        #tests for deleting parent directories
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
505
        ddir = mangle_tree.trans_id_tree_file_id('ddir')
1534.7.59 by Aaron Bentley
Simplified tests
506
        mangle_tree.delete_contents(ddir)
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
507
        dfile = mangle_tree.trans_id_tree_file_id('dfile')
1534.7.59 by Aaron Bentley
Simplified tests
508
        mangle_tree.delete_versioned(dfile)
509
        mangle_tree.unversion_file(dfile)
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
510
        mfile = mangle_tree.trans_id_tree_file_id('mfile')
1534.7.59 by Aaron Bentley
Simplified tests
511
        mangle_tree.adjust_path('mfile', root, mfile)
512
513
        #tests for adding parent directories
514
        newdir = mangle_tree.new_directory('new_directory', root, 'newdir')
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
515
        mfile2 = mangle_tree.trans_id_tree_file_id('mfile2')
1534.7.59 by Aaron Bentley
Simplified tests
516
        mangle_tree.adjust_path('mfile2', newdir, mfile2)
517
        mangle_tree.new_file('newfile', newdir, 'hello3', 'dfile')
518
        self.assertEqual(mangle_tree.final_file_id(mfile2), 'mfile2')
519
        self.assertEqual(mangle_tree.final_parent(mfile2), newdir)
520
        self.assertEqual(mangle_tree.final_file_id(mfile2), 'mfile2')
521
        mangle_tree.apply()
522
        self.assertEqual(file(self.wt.abspath('name1')).read(), 'hello2')
523
        self.assertEqual(file(self.wt.abspath('name2')).read(), 'hello1')
1534.7.176 by abentley
Fixed up tests for Windows
524
        mfile2_path = self.wt.abspath(pathjoin('new_directory','mfile2'))
1534.7.41 by Aaron Bentley
Got inventory ID movement working
525
        self.assertEqual(mangle_tree.final_parent(mfile2), newdir)
1534.7.38 by Aaron Bentley
Tested adding paths
526
        self.assertEqual(file(mfile2_path).read(), 'later2')
1534.7.59 by Aaron Bentley
Simplified tests
527
        self.assertEqual(self.wt.id2path('mfile2'), 'new_directory/mfile2')
528
        self.assertEqual(self.wt.path2id('new_directory/mfile2'), 'mfile2')
1534.7.176 by abentley
Fixed up tests for Windows
529
        newfile_path = self.wt.abspath(pathjoin('new_directory','newfile'))
1534.7.38 by Aaron Bentley
Tested adding paths
530
        self.assertEqual(file(newfile_path).read(), 'hello3')
1534.7.59 by Aaron Bentley
Simplified tests
531
        self.assertEqual(self.wt.path2id('dying_directory'), 'ddir')
532
        self.assertIs(self.wt.path2id('dying_directory/dying_file'), None)
1534.7.176 by abentley
Fixed up tests for Windows
533
        mfile2_path = self.wt.abspath(pathjoin('new_directory','mfile2'))
1534.7.43 by abentley
Fixed some Windows bugs, introduced a conflicts bug
534
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
535
    def test_both_rename(self):
536
        create_tree,root = self.get_transform()
537
        newdir = create_tree.new_directory('selftest', root, 'selftest-id')
538
        create_tree.new_file('blackbox.py', newdir, 'hello1', 'blackbox-id')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
539
        create_tree.apply()
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
540
        mangle_tree,root = self.get_transform()
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
541
        selftest = mangle_tree.trans_id_tree_file_id('selftest-id')
542
        blackbox = mangle_tree.trans_id_tree_file_id('blackbox-id')
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
543
        mangle_tree.adjust_path('test', root, selftest)
544
        mangle_tree.adjust_path('test_too_much', root, selftest)
545
        mangle_tree.set_executability(True, blackbox)
546
        mangle_tree.apply()
547
548
    def test_both_rename2(self):
549
        create_tree,root = self.get_transform()
550
        bzrlib = create_tree.new_directory('bzrlib', root, 'bzrlib-id')
551
        tests = create_tree.new_directory('tests', bzrlib, 'tests-id')
552
        blackbox = create_tree.new_directory('blackbox', tests, 'blackbox-id')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
553
        create_tree.new_file('test_too_much.py', blackbox, 'hello1',
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
554
                             'test_too_much-id')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
555
        create_tree.apply()
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
556
        mangle_tree,root = self.get_transform()
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
557
        bzrlib = mangle_tree.trans_id_tree_file_id('bzrlib-id')
558
        tests = mangle_tree.trans_id_tree_file_id('tests-id')
559
        test_too_much = mangle_tree.trans_id_tree_file_id('test_too_much-id')
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
560
        mangle_tree.adjust_path('selftest', bzrlib, tests)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
561
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much)
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
562
        mangle_tree.set_executability(True, test_too_much)
563
        mangle_tree.apply()
564
565
    def test_both_rename3(self):
566
        create_tree,root = self.get_transform()
567
        tests = create_tree.new_directory('tests', root, 'tests-id')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
568
        create_tree.new_file('test_too_much.py', tests, 'hello1',
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
569
                             'test_too_much-id')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
570
        create_tree.apply()
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
571
        mangle_tree,root = self.get_transform()
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
572
        tests = mangle_tree.trans_id_tree_file_id('tests-id')
573
        test_too_much = mangle_tree.trans_id_tree_file_id('test_too_much-id')
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
574
        mangle_tree.adjust_path('selftest', root, tests)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
575
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much)
1534.7.150 by Aaron Bentley
Handled simultaneous renames of parent and child better
576
        mangle_tree.set_executability(True, test_too_much)
577
        mangle_tree.apply()
578
1534.7.48 by Aaron Bentley
Ensured we can move/rename dangling inventory entries
579
    def test_move_dangling_ie(self):
1534.7.59 by Aaron Bentley
Simplified tests
580
        create_tree, root = self.get_transform()
581
        # prepare tree
1731.1.33 by Aaron Bentley
Revert no-special-root changes
582
        root = create_tree.root
1534.7.59 by Aaron Bentley
Simplified tests
583
        create_tree.new_file('name1', root, 'hello1', 'name1')
584
        create_tree.apply()
585
        delete_contents, root = self.get_transform()
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
586
        file = delete_contents.trans_id_tree_file_id('name1')
1534.7.59 by Aaron Bentley
Simplified tests
587
        delete_contents.delete_contents(file)
588
        delete_contents.apply()
589
        move_id, root = self.get_transform()
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
590
        name1 = move_id.trans_id_tree_file_id('name1')
1534.7.59 by Aaron Bentley
Simplified tests
591
        newdir = move_id.new_directory('dir', root, 'newdir')
592
        move_id.adjust_path('name2', newdir, name1)
593
        move_id.apply()
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
594
1534.7.50 by Aaron Bentley
Detect duplicate inventory ids
595
    def test_replace_dangling_ie(self):
1534.7.59 by Aaron Bentley
Simplified tests
596
        create_tree, root = self.get_transform()
597
        # prepare tree
1731.1.33 by Aaron Bentley
Revert no-special-root changes
598
        root = create_tree.root
1534.7.59 by Aaron Bentley
Simplified tests
599
        create_tree.new_file('name1', root, 'hello1', 'name1')
600
        create_tree.apply()
601
        delete_contents = TreeTransform(self.wt)
602
        self.addCleanup(delete_contents.finalize)
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
603
        file = delete_contents.trans_id_tree_file_id('name1')
1534.7.59 by Aaron Bentley
Simplified tests
604
        delete_contents.delete_contents(file)
605
        delete_contents.apply()
606
        delete_contents.finalize()
607
        replace = TreeTransform(self.wt)
608
        self.addCleanup(replace.finalize)
609
        name2 = replace.new_file('name2', root, 'hello2', 'name1')
610
        conflicts = replace.find_conflicts()
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
611
        name1 = replace.trans_id_tree_file_id('name1')
1534.7.59 by Aaron Bentley
Simplified tests
612
        self.assertEqual(conflicts, [('duplicate id', name1, name2)])
613
        resolve_conflicts(replace)
614
        replace.apply()
1534.7.48 by Aaron Bentley
Ensured we can move/rename dangling inventory entries
615
4241.14.17 by Vincent Ladeuil
Add more tests for unicode symlinks to test_transform.
616
    def _test_symlinks(self, link_name1,link_target1,
617
                       link_name2, link_target2):
618
619
        def ozpath(p): return 'oz/' + p
620
2949.5.1 by Alexander Belchenko
selftest: use SymlinkFeature instead of TestSkipped where appropriate
621
        self.requireFeature(SymlinkFeature)
4241.14.12 by Vincent Ladeuil
Far too many modifications for a single commit, need to restart.
622
        transform, root = self.get_transform()
1534.7.59 by Aaron Bentley
Simplified tests
623
        oz_id = transform.new_directory('oz', root, 'oz-id')
4241.14.17 by Vincent Ladeuil
Add more tests for unicode symlinks to test_transform.
624
        wizard = transform.new_symlink(link_name1, oz_id, link_target1,
1534.7.59 by Aaron Bentley
Simplified tests
625
                                       'wizard-id')
4241.14.17 by Vincent Ladeuil
Add more tests for unicode symlinks to test_transform.
626
        wiz_id = transform.create_path(link_name2, oz_id)
627
        transform.create_symlink(link_target2, wiz_id)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
628
        transform.version_file('wiz-id2', wiz_id)
1534.7.71 by abentley
All tests pass under Windows
629
        transform.set_executability(True, wiz_id)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
630
        self.assertEqual(transform.find_conflicts(),
1534.7.71 by abentley
All tests pass under Windows
631
                         [('non-file executability', wiz_id)])
632
        transform.set_executability(None, wiz_id)
1534.7.59 by Aaron Bentley
Simplified tests
633
        transform.apply()
4241.14.17 by Vincent Ladeuil
Add more tests for unicode symlinks to test_transform.
634
        self.assertEqual(self.wt.path2id(ozpath(link_name1)), 'wizard-id')
635
        self.assertEqual('symlink',
636
                         file_kind(self.wt.abspath(ozpath(link_name1))))
637
        self.assertEqual(link_target2,
638
                         osutils.readlink(self.wt.abspath(ozpath(link_name2))))
639
        self.assertEqual(link_target1,
640
                         osutils.readlink(self.wt.abspath(ozpath(link_name1))))
641
642
    def test_symlinks(self):
643
        self._test_symlinks('wizard', 'wizard-target',
644
                            'wizard2', 'behind_curtain')
645
646
    def test_symlinks_unicode(self):
647
        self.requireFeature(tests.UnicodeFilenameFeature)
648
        self._test_symlinks(u'\N{Euro Sign}wizard',
649
                            u'wizard-targ\N{Euro Sign}t',
650
                            u'\N{Euro Sign}wizard2',
651
                            u'b\N{Euro Sign}hind_curtain')
1534.7.60 by Aaron Bentley
Tested existing conflict resolution functionality
652
3006.2.2 by Alexander Belchenko
tests added.
653
    def test_unable_create_symlink(self):
654
        def tt_helper():
655
            wt = self.make_branch_and_tree('.')
656
            tt = TreeTransform(wt)  # TreeTransform obtains write lock
657
            try:
658
                tt.new_symlink('foo', tt.root, 'bar')
659
                tt.apply()
660
            finally:
661
                wt.unlock()
662
        os_symlink = getattr(os, 'symlink', None)
663
        os.symlink = None
664
        try:
665
            err = self.assertRaises(errors.UnableCreateSymlink, tt_helper)
666
            self.assertEquals(
667
                "Unable to create symlink 'foo' on this platform",
668
                str(err))
669
        finally:
670
            if os_symlink:
671
                os.symlink = os_symlink
672
1534.7.170 by Aaron Bentley
Cleaned up filesystem conflict handling
673
    def get_conflicted(self):
1534.7.60 by Aaron Bentley
Tested existing conflict resolution functionality
674
        create,root = self.get_transform()
675
        create.new_file('dorothy', root, 'dorothy', 'dorothy-id')
1534.7.61 by Aaron Bentley
Handled parent loops, missing parents, unversioned parents
676
        oz = create.new_directory('oz', root, 'oz-id')
677
        create.new_directory('emeraldcity', oz, 'emerald-id')
1534.7.60 by Aaron Bentley
Tested existing conflict resolution functionality
678
        create.apply()
679
        conflicts,root = self.get_transform()
1534.7.65 by Aaron Bentley
Text cleaup/docs
680
        # set up duplicate entry, duplicate id
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
681
        new_dorothy = conflicts.new_file('dorothy', root, 'dorothy',
1534.7.60 by Aaron Bentley
Tested existing conflict resolution functionality
682
                                         'dorothy-id')
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
683
        old_dorothy = conflicts.trans_id_tree_file_id('dorothy-id')
684
        oz = conflicts.trans_id_tree_file_id('oz-id')
1551.8.22 by Aaron Bentley
Improve message when OTHER deletes an in-use tree
685
        # set up DeletedParent parent conflict
1534.7.65 by Aaron Bentley
Text cleaup/docs
686
        conflicts.delete_versioned(oz)
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
687
        emerald = conflicts.trans_id_tree_file_id('emerald-id')
1551.8.22 by Aaron Bentley
Improve message when OTHER deletes an in-use tree
688
        # set up MissingParent conflict
689
        munchkincity = conflicts.trans_id_file_id('munchkincity-id')
690
        conflicts.adjust_path('munchkincity', root, munchkincity)
691
        conflicts.new_directory('auntem', munchkincity, 'auntem-id')
1534.7.65 by Aaron Bentley
Text cleaup/docs
692
        # set up parent loop
1534.7.61 by Aaron Bentley
Handled parent loops, missing parents, unversioned parents
693
        conflicts.adjust_path('emeraldcity', emerald, emerald)
1534.7.170 by Aaron Bentley
Cleaned up filesystem conflict handling
694
        return conflicts, emerald, oz, old_dorothy, new_dorothy
695
696
    def test_conflict_resolution(self):
697
        conflicts, emerald, oz, old_dorothy, new_dorothy =\
698
            self.get_conflicted()
1534.7.60 by Aaron Bentley
Tested existing conflict resolution functionality
699
        resolve_conflicts(conflicts)
700
        self.assertEqual(conflicts.final_name(old_dorothy), 'dorothy.moved')
701
        self.assertIs(conflicts.final_file_id(old_dorothy), None)
702
        self.assertEqual(conflicts.final_name(new_dorothy), 'dorothy')
1534.7.170 by Aaron Bentley
Cleaned up filesystem conflict handling
703
        self.assertEqual(conflicts.final_file_id(new_dorothy), 'dorothy-id')
1534.7.64 by Aaron Bentley
Extra testing
704
        self.assertEqual(conflicts.final_parent(emerald), oz)
1534.7.63 by Aaron Bentley
Ensure transform can be applied after resolution
705
        conflicts.apply()
1534.7.62 by Aaron Bentley
Fixed moving versioned directories
706
1534.7.170 by Aaron Bentley
Cleaned up filesystem conflict handling
707
    def test_cook_conflicts(self):
708
        tt, emerald, oz, old_dorothy, new_dorothy = self.get_conflicted()
709
        raw_conflicts = resolve_conflicts(tt)
710
        cooked_conflicts = cook_conflicts(raw_conflicts, tt)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
711
        duplicate = DuplicateEntry('Moved existing file to', 'dorothy.moved',
1534.10.20 by Aaron Bentley
Got all tests passing
712
                                   'dorothy', None, 'dorothy-id')
1534.7.170 by Aaron Bentley
Cleaned up filesystem conflict handling
713
        self.assertEqual(cooked_conflicts[0], duplicate)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
714
        duplicate_id = DuplicateID('Unversioned existing file',
1534.10.20 by Aaron Bentley
Got all tests passing
715
                                   'dorothy.moved', 'dorothy', None,
716
                                   'dorothy-id')
1534.7.170 by Aaron Bentley
Cleaned up filesystem conflict handling
717
        self.assertEqual(cooked_conflicts[1], duplicate_id)
1551.8.22 by Aaron Bentley
Improve message when OTHER deletes an in-use tree
718
        missing_parent = MissingParent('Created directory', 'munchkincity',
719
                                       'munchkincity-id')
720
        deleted_parent = DeletingParent('Not deleting', 'oz', 'oz-id')
1534.7.170 by Aaron Bentley
Cleaned up filesystem conflict handling
721
        self.assertEqual(cooked_conflicts[2], missing_parent)
1551.8.22 by Aaron Bentley
Improve message when OTHER deletes an in-use tree
722
        unversioned_parent = UnversionedParent('Versioned directory',
723
                                               'munchkincity',
724
                                               'munchkincity-id')
725
        unversioned_parent2 = UnversionedParent('Versioned directory', 'oz',
1534.10.20 by Aaron Bentley
Got all tests passing
726
                                               'oz-id')
1534.7.170 by Aaron Bentley
Cleaned up filesystem conflict handling
727
        self.assertEqual(cooked_conflicts[3], unversioned_parent)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
728
        parent_loop = ParentLoop('Cancelled move', 'oz/emeraldcity',
1534.10.20 by Aaron Bentley
Got all tests passing
729
                                 'oz/emeraldcity', 'emerald-id', 'emerald-id')
1551.8.22 by Aaron Bentley
Improve message when OTHER deletes an in-use tree
730
        self.assertEqual(cooked_conflicts[4], deleted_parent)
731
        self.assertEqual(cooked_conflicts[5], unversioned_parent2)
732
        self.assertEqual(cooked_conflicts[6], parent_loop)
733
        self.assertEqual(len(cooked_conflicts), 7)
1534.7.170 by Aaron Bentley
Cleaned up filesystem conflict handling
734
        tt.finalize()
735
736
    def test_string_conflicts(self):
737
        tt, emerald, oz, old_dorothy, new_dorothy = self.get_conflicted()
738
        raw_conflicts = resolve_conflicts(tt)
739
        cooked_conflicts = cook_conflicts(raw_conflicts, tt)
740
        tt.finalize()
1534.10.24 by Aaron Bentley
Eliminated conflicts_to_strings, made remove_files a ConflictList member
741
        conflicts_s = [str(c) for c in cooked_conflicts]
1534.7.171 by Aaron Bentley
Implemented stringifying filesystem conflicts
742
        self.assertEqual(len(cooked_conflicts), len(conflicts_s))
743
        self.assertEqual(conflicts_s[0], 'Conflict adding file dorothy.  '
744
                                         'Moved existing file to '
745
                                         'dorothy.moved.')
746
        self.assertEqual(conflicts_s[1], 'Conflict adding id to dorothy.  '
747
                                         'Unversioned existing file '
748
                                         'dorothy.moved.')
1551.8.22 by Aaron Bentley
Improve message when OTHER deletes an in-use tree
749
        self.assertEqual(conflicts_s[2], 'Conflict adding files to'
750
                                         ' munchkincity.  Created directory.')
751
        self.assertEqual(conflicts_s[3], 'Conflict because munchkincity is not'
752
                                         ' versioned, but has versioned'
753
                                         ' children.  Versioned directory.')
1551.8.23 by Aaron Bentley
Tweaked conflict message to be more understandable
754
        self.assertEqualDiff(conflicts_s[4], "Conflict: can't delete oz because it"
755
                                         " is not empty.  Not deleting.")
1551.8.22 by Aaron Bentley
Improve message when OTHER deletes an in-use tree
756
        self.assertEqual(conflicts_s[5], 'Conflict because oz is not'
757
                                         ' versioned, but has versioned'
758
                                         ' children.  Versioned directory.')
759
        self.assertEqual(conflicts_s[6], 'Conflict moving oz/emeraldcity into'
4597.8.11 by Vincent Ladeuil
The ParentLoop.format has been updated, fix fallouts.
760
                                         ' oz/emeraldcity. Cancelled move.')
1534.7.170 by Aaron Bentley
Cleaned up filesystem conflict handling
761
3144.4.2 by Aaron Bentley
Handle non-directory parent conflicts (abentley, #177390)
762
    def prepare_wrong_parent_kind(self):
763
        tt, root = self.get_transform()
764
        tt.new_file('parent', root, 'contents', 'parent-id')
765
        tt.apply()
766
        tt, root = self.get_transform()
767
        parent_id = tt.trans_id_file_id('parent-id')
768
        tt.new_file('child,', parent_id, 'contents2', 'file-id')
769
        return tt
770
3144.4.1 by Aaron Bentley
Handle trying to list parents of a non-directory
771
    def test_find_conflicts_wrong_parent_kind(self):
3144.4.2 by Aaron Bentley
Handle non-directory parent conflicts (abentley, #177390)
772
        tt = self.prepare_wrong_parent_kind()
3144.4.1 by Aaron Bentley
Handle trying to list parents of a non-directory
773
        tt.find_conflicts()
774
3144.4.2 by Aaron Bentley
Handle non-directory parent conflicts (abentley, #177390)
775
    def test_resolve_conflicts_wrong_existing_parent_kind(self):
776
        tt = self.prepare_wrong_parent_kind()
777
        raw_conflicts = resolve_conflicts(tt)
778
        self.assertEqual(set([('non-directory parent', 'Created directory',
779
                         'new-3')]), raw_conflicts)
780
        cooked_conflicts = cook_conflicts(raw_conflicts, tt)
781
        self.assertEqual([NonDirectoryParent('Created directory', 'parent.new',
782
        'parent-id')], cooked_conflicts)
783
        tt.apply()
784
        self.assertEqual(None, self.wt.path2id('parent'))
785
        self.assertEqual('parent-id', self.wt.path2id('parent.new'))
786
787
    def test_resolve_conflicts_wrong_new_parent_kind(self):
788
        tt, root = self.get_transform()
789
        parent_id = tt.new_directory('parent', root, 'parent-id')
790
        tt.new_file('child,', parent_id, 'contents2', 'file-id')
791
        tt.apply()
792
        tt, root = self.get_transform()
793
        parent_id = tt.trans_id_file_id('parent-id')
794
        tt.delete_contents(parent_id)
795
        tt.create_file('contents', parent_id)
796
        raw_conflicts = resolve_conflicts(tt)
797
        self.assertEqual(set([('non-directory parent', 'Created directory',
798
                         'new-3')]), raw_conflicts)
799
        tt.apply()
800
        self.assertEqual(None, self.wt.path2id('parent'))
801
        self.assertEqual('parent-id', self.wt.path2id('parent.new'))
802
803
    def test_resolve_conflicts_wrong_parent_kind_unversioned(self):
804
        tt, root = self.get_transform()
805
        parent_id = tt.new_directory('parent', root)
806
        tt.new_file('child,', parent_id, 'contents2')
807
        tt.apply()
808
        tt, root = self.get_transform()
809
        parent_id = tt.trans_id_tree_path('parent')
810
        tt.delete_contents(parent_id)
811
        tt.create_file('contents', parent_id)
812
        resolve_conflicts(tt)
813
        tt.apply()
814
        self.assertIs(None, self.wt.path2id('parent'))
815
        self.assertIs(None, self.wt.path2id('parent.new'))
816
1534.7.62 by Aaron Bentley
Fixed moving versioned directories
817
    def test_moving_versioned_directories(self):
818
        create, root = self.get_transform()
819
        kansas = create.new_directory('kansas', root, 'kansas-id')
820
        create.new_directory('house', kansas, 'house-id')
821
        create.new_directory('oz', root, 'oz-id')
822
        create.apply()
823
        cyclone, root = self.get_transform()
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
824
        oz = cyclone.trans_id_tree_file_id('oz-id')
825
        house = cyclone.trans_id_tree_file_id('house-id')
1534.7.62 by Aaron Bentley
Fixed moving versioned directories
826
        cyclone.adjust_path('house', oz, house)
827
        cyclone.apply()
1534.7.66 by Aaron Bentley
Ensured we don't accidentally move the root directory
828
829
    def test_moving_root(self):
1534.7.68 by Aaron Bentley
Got semi-reasonable root directory renaming working
830
        create, root = self.get_transform()
831
        fun = create.new_directory('fun', root, 'fun-id')
832
        create.new_directory('sun', root, 'sun-id')
833
        create.new_directory('moon', root, 'moon')
834
        create.apply()
1534.7.66 by Aaron Bentley
Ensured we don't accidentally move the root directory
835
        transform, root = self.get_transform()
1534.7.68 by Aaron Bentley
Got semi-reasonable root directory renaming working
836
        transform.adjust_root_path('oldroot', fun)
4634.122.2 by Aaron Bentley, John Arbash Meinel
Bring the fixup_new_roots code from the nested-trees code.
837
        new_root = transform.trans_id_tree_path('')
1534.7.69 by Aaron Bentley
Got real root moves working
838
        transform.version_file('new-root', new_root)
1534.7.68 by Aaron Bentley
Got semi-reasonable root directory renaming working
839
        transform.apply()
1534.7.93 by Aaron Bentley
Added text merge test
840
1534.7.114 by Aaron Bentley
Added file renaming test case
841
    def test_renames(self):
842
        create, root = self.get_transform()
843
        old = create.new_directory('old-parent', root, 'old-id')
844
        intermediate = create.new_directory('intermediate', old, 'im-id')
845
        myfile = create.new_file('myfile', intermediate, 'myfile-text',
846
                                 'myfile-id')
847
        create.apply()
848
        rename, root = self.get_transform()
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
849
        old = rename.trans_id_file_id('old-id')
1534.7.114 by Aaron Bentley
Added file renaming test case
850
        rename.adjust_path('new', root, old)
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
851
        myfile = rename.trans_id_file_id('myfile-id')
1534.7.114 by Aaron Bentley
Added file renaming test case
852
        rename.set_executability(True, myfile)
853
        rename.apply()
854
5186.2.4 by Martin Pool
Add failing specific test for 491763
855
    def test_rename_fails(self):
856
        # see https://bugs.launchpad.net/bzr/+bug/491763
857
        create, root_id = self.get_transform()
858
        first_dir = create.new_directory('first-dir', root_id, 'first-id')
859
        myfile = create.new_file('myfile', root_id, 'myfile-text',
860
                                 'myfile-id')
861
        create.apply()
5050.15.2 by Martin
Correct bt.test_transform.TestTreeTransform.test_rename_fails on non-posix platforms
862
        if os.name == "posix" and sys.platform != "cygwin":
863
            # posix filesystems fail on renaming if the readonly bit is set
864
            osutils.make_readonly(self.wt.abspath('first-dir'))
865
        elif os.name == "nt":
866
            # windows filesystems fail on renaming open files
867
            self.addCleanup(file(self.wt.abspath('myfile')).close)
868
        else:
869
            self.skip("Don't know how to force a permissions error on rename")
5186.2.4 by Martin Pool
Add failing specific test for 491763
870
        # now transform to rename
871
        rename_transform, root_id = self.get_transform()
872
        file_trans_id = rename_transform.trans_id_file_id('myfile-id')
873
        dir_id = rename_transform.trans_id_file_id('first-id')
874
        rename_transform.adjust_path('newname', dir_id, file_trans_id)
5186.2.5 by Martin Pool
Raise a specific clearer error when a rename fails inside transform
875
        e = self.assertRaises(errors.TransformRenameFailed,
876
            rename_transform.apply)
5050.15.2 by Martin
Correct bt.test_transform.TestTreeTransform.test_rename_fails on non-posix platforms
877
        # On nix looks like: 
5186.2.5 by Martin Pool
Raise a specific clearer error when a rename fails inside transform
878
        # "Failed to rename .../work/.bzr/checkout/limbo/new-1
879
        # to .../first-dir/newname: [Errno 13] Permission denied"
5050.15.2 by Martin
Correct bt.test_transform.TestTreeTransform.test_rename_fails on non-posix platforms
880
        # On windows looks like:
881
        # "Failed to rename .../work/myfile to 
882
        # .../work/.bzr/checkout/limbo/new-1: [Errno 13] Permission denied"
883
        # The strerror will vary per OS and language so it's not checked here
5186.2.5 by Martin Pool
Raise a specific clearer error when a rename fails inside transform
884
        self.assertContainsRe(str(e),
5050.15.2 by Martin
Correct bt.test_transform.TestTreeTransform.test_rename_fails on non-posix platforms
885
            "Failed to rename .*(first-dir.newname:|myfile)")
5186.2.4 by Martin Pool
Add failing specific test for 491763
886
1740.2.4 by Aaron Bentley
Update transform tests and docs
887
    def test_set_executability_order(self):
888
        """Ensure that executability behaves the same, no matter what order.
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
889
1740.2.4 by Aaron Bentley
Update transform tests and docs
890
        - create file and set executability simultaneously
891
        - create file and set executability afterward
892
        - unsetting the executability of a file whose executability has not been
893
        declared should throw an exception (this may happen when a
894
        merge attempts to create a file with a duplicate ID)
895
        """
896
        transform, root = self.get_transform()
897
        wt = transform._tree
3034.2.1 by Aaron Bentley
Fix is_executable tests for win32
898
        wt.lock_read()
899
        self.addCleanup(wt.unlock)
1740.2.4 by Aaron Bentley
Update transform tests and docs
900
        transform.new_file('set_on_creation', root, 'Set on creation', 'soc',
901
                           True)
2502.1.1 by Aaron Bentley
Ensure renames only root children are renamed when building trees
902
        sac = transform.new_file('set_after_creation', root,
903
                                 'Set after creation', 'sac')
1740.2.4 by Aaron Bentley
Update transform tests and docs
904
        transform.set_executability(True, sac)
2502.1.1 by Aaron Bentley
Ensure renames only root children are renamed when building trees
905
        uws = transform.new_file('unset_without_set', root, 'Unset badly',
906
                                 'uws')
1740.2.4 by Aaron Bentley
Update transform tests and docs
907
        self.assertRaises(KeyError, transform.set_executability, None, uws)
908
        transform.apply()
909
        self.assertTrue(wt.is_executable('soc'))
910
        self.assertTrue(wt.is_executable('sac'))
911
1534.12.2 by Aaron Bentley
Added test for preserving file mode
912
    def test_preserve_mode(self):
913
        """File mode is preserved when replacing content"""
914
        if sys.platform == 'win32':
915
            raise TestSkipped('chmod has no effect on win32')
916
        transform, root = self.get_transform()
917
        transform.new_file('file1', root, 'contents', 'file1-id', True)
918
        transform.apply()
3146.4.12 by Aaron Bentley
Add needed write lock to test
919
        self.wt.lock_write()
920
        self.addCleanup(self.wt.unlock)
1534.12.2 by Aaron Bentley
Added test for preserving file mode
921
        self.assertTrue(self.wt.is_executable('file1-id'))
922
        transform, root = self.get_transform()
923
        file1_id = transform.trans_id_tree_file_id('file1-id')
924
        transform.delete_contents(file1_id)
925
        transform.create_file('contents2', file1_id)
926
        transform.apply()
927
        self.assertTrue(self.wt.is_executable('file1-id'))
928
2027.1.1 by John Arbash Meinel
Fix bug #56549, and write a direct test that the right path is being statted
929
    def test__set_mode_stats_correctly(self):
930
        """_set_mode stats to determine file mode."""
931
        if sys.platform == 'win32':
932
            raise TestSkipped('chmod has no effect on win32')
933
934
        stat_paths = []
935
        real_stat = os.stat
936
        def instrumented_stat(path):
937
            stat_paths.append(path)
938
            return real_stat(path)
939
940
        transform, root = self.get_transform()
941
942
        bar1_id = transform.new_file('bar', root, 'bar contents 1\n',
943
                                     file_id='bar-id-1', executable=False)
944
        transform.apply()
945
946
        transform, root = self.get_transform()
947
        bar1_id = transform.trans_id_tree_path('bar')
948
        bar2_id = transform.trans_id_tree_path('bar2')
949
        try:
950
            os.stat = instrumented_stat
951
            transform.create_file('bar2 contents\n', bar2_id, mode_id=bar1_id)
952
        finally:
953
            os.stat = real_stat
954
            transform.finalize()
955
956
        bar1_abspath = self.wt.abspath('bar')
957
        self.assertEqual([bar1_abspath], stat_paths)
958
1551.11.1 by Aaron Bentley
Initial work on converting TreeTransform to iter_changes format
959
    def test_iter_changes(self):
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
960
        self.wt.set_root_id('eert_toor')
1551.11.1 by Aaron Bentley
Initial work on converting TreeTransform to iter_changes format
961
        transform, root = self.get_transform()
962
        transform.new_file('old', root, 'blah', 'id-1', True)
963
        transform.apply()
964
        transform, root = self.get_transform()
965
        try:
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
966
            self.assertEqual([], list(transform.iter_changes()))
1551.11.1 by Aaron Bentley
Initial work on converting TreeTransform to iter_changes format
967
            old = transform.trans_id_tree_file_id('id-1')
968
            transform.unversion_file(old)
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
969
            self.assertEqual([('id-1', ('old', None), False, (True, False),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
970
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
971
                (True, True))], list(transform.iter_changes()))
1551.11.1 by Aaron Bentley
Initial work on converting TreeTransform to iter_changes format
972
            transform.new_directory('new', root, 'id-1')
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
973
            self.assertEqual([('id-1', ('old', 'new'), True, (True, True),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
974
                ('eert_toor', 'eert_toor'), ('old', 'new'),
1551.11.1 by Aaron Bentley
Initial work on converting TreeTransform to iter_changes format
975
                ('file', 'directory'),
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
976
                (True, False))], list(transform.iter_changes()))
1551.11.1 by Aaron Bentley
Initial work on converting TreeTransform to iter_changes format
977
        finally:
978
            transform.finalize()
979
980
    def test_iter_changes_new(self):
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
981
        self.wt.set_root_id('eert_toor')
1551.11.1 by Aaron Bentley
Initial work on converting TreeTransform to iter_changes format
982
        transform, root = self.get_transform()
983
        transform.new_file('old', root, 'blah')
984
        transform.apply()
985
        transform, root = self.get_transform()
986
        try:
987
            old = transform.trans_id_tree_path('old')
988
            transform.version_file('id-1', old)
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
989
            self.assertEqual([('id-1', (None, 'old'), False, (False, True),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
990
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
991
                (False, False))], list(transform.iter_changes()))
1551.11.1 by Aaron Bentley
Initial work on converting TreeTransform to iter_changes format
992
        finally:
993
            transform.finalize()
994
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
995
    def test_iter_changes_modifications(self):
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
996
        self.wt.set_root_id('eert_toor')
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
997
        transform, root = self.get_transform()
998
        transform.new_file('old', root, 'blah', 'id-1')
999
        transform.new_file('new', root, 'blah')
1000
        transform.new_directory('subdir', root, 'subdir-id')
1001
        transform.apply()
1002
        transform, root = self.get_transform()
1003
        try:
1004
            old = transform.trans_id_tree_path('old')
1005
            subdir = transform.trans_id_tree_file_id('subdir-id')
1006
            new = transform.trans_id_tree_path('new')
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1007
            self.assertEqual([], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1008
1009
            #content deletion
1010
            transform.delete_contents(old)
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
1011
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
1012
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', None),
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1013
                (False, False))], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1014
1015
            #content change
1016
            transform.create_file('blah', old)
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
1017
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
1018
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1019
                (False, False))], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1020
            transform.cancel_deletion(old)
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
1021
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
1022
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1023
                (False, False))], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1024
            transform.cancel_creation(old)
1025
1026
            # move file_id to a different file
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1027
            self.assertEqual([], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1028
            transform.unversion_file(old)
1029
            transform.version_file('id-1', new)
1030
            transform.adjust_path('old', root, new)
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
1031
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
1032
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1033
                (False, False))], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1034
            transform.cancel_versioning(new)
1035
            transform._removed_id = set()
1036
1037
            #execute bit
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1038
            self.assertEqual([], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1039
            transform.set_executability(True, old)
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
1040
            self.assertEqual([('id-1', ('old', 'old'), False, (True, True),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
1041
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1042
                (False, True))], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1043
            transform.set_executability(None, old)
1044
1045
            # filename
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1046
            self.assertEqual([], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1047
            transform.adjust_path('new', root, old)
1048
            transform._new_parent = {}
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
1049
            self.assertEqual([('id-1', ('old', 'new'), False, (True, True),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
1050
                ('eert_toor', 'eert_toor'), ('old', 'new'), ('file', 'file'),
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1051
                (False, False))], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1052
            transform._new_name = {}
1053
1054
            # parent directory
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1055
            self.assertEqual([], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1056
            transform.adjust_path('new', subdir, old)
1057
            transform._new_name = {}
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
1058
            self.assertEqual([('id-1', ('old', 'subdir/old'), False,
2255.2.180 by Martin Pool
merge dirstate
1059
                (True, True), ('eert_toor', 'subdir-id'), ('old', 'old'),
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
1060
                ('file', 'file'), (False, False))],
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1061
                list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1062
            transform._new_path = {}
1063
1064
        finally:
1065
            transform.finalize()
1066
1551.11.7 by Aaron Bentley
Stop modified flag bleeding into later changes
1067
    def test_iter_changes_modified_bleed(self):
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
1068
        self.wt.set_root_id('eert_toor')
1551.11.7 by Aaron Bentley
Stop modified flag bleeding into later changes
1069
        """Modified flag should not bleed from one change to another"""
1070
        # unfortunately, we have no guarantee that file1 (which is modified)
1071
        # will be applied before file2.  And if it's applied after file2, it
1072
        # obviously can't bleed into file2's change output.  But for now, it
1073
        # works.
1074
        transform, root = self.get_transform()
1075
        transform.new_file('file1', root, 'blah', 'id-1')
1076
        transform.new_file('file2', root, 'blah', 'id-2')
1077
        transform.apply()
1078
        transform, root = self.get_transform()
1079
        try:
1080
            transform.delete_contents(transform.trans_id_file_id('id-1'))
1081
            transform.set_executability(True,
1082
            transform.trans_id_file_id('id-2'))
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
1083
            self.assertEqual([('id-1', (u'file1', u'file1'), True, (True, True),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
1084
                ('eert_toor', 'eert_toor'), ('file1', u'file1'),
1551.11.7 by Aaron Bentley
Stop modified flag bleeding into later changes
1085
                ('file', None), (False, False)),
2255.7.96 by Robert Collins
Change _iter_changes interface to yield both old and new paths.
1086
                ('id-2', (u'file2', u'file2'), False, (True, True),
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
1087
                ('eert_toor', 'eert_toor'), ('file2', u'file2'),
1551.11.7 by Aaron Bentley
Stop modified flag bleeding into later changes
1088
                ('file', 'file'), (False, True))],
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1089
                list(transform.iter_changes()))
1551.11.7 by Aaron Bentley
Stop modified flag bleeding into later changes
1090
        finally:
1091
            transform.finalize()
1092
1551.10.37 by Aaron Bentley
recommit of TreeTransform._iter_changes fix with missing files
1093
    def test_iter_changes_move_missing(self):
1094
        """Test moving ids with no files around"""
1095
        self.wt.set_root_id('toor_eert')
1096
        # Need two steps because versioning a non-existant file is a conflict.
1097
        transform, root = self.get_transform()
1098
        transform.new_directory('floater', root, 'floater-id')
1099
        transform.apply()
1100
        transform, root = self.get_transform()
1101
        transform.delete_contents(transform.trans_id_tree_path('floater'))
1102
        transform.apply()
1103
        transform, root = self.get_transform()
1104
        floater = transform.trans_id_tree_path('floater')
1105
        try:
1106
            transform.adjust_path('flitter', root, floater)
1107
            self.assertEqual([('floater-id', ('floater', 'flitter'), False,
1108
            (True, True), ('toor_eert', 'toor_eert'), ('floater', 'flitter'),
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1109
            (None, None), (False, False))], list(transform.iter_changes()))
1551.10.37 by Aaron Bentley
recommit of TreeTransform._iter_changes fix with missing files
1110
        finally:
1111
            transform.finalize()
1112
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1113
    def test_iter_changes_pointless(self):
1114
        """Ensure that no-ops are not treated as modifications"""
2100.3.33 by Aaron Bentley
Handle unique roots in tests for TreeTransform.iter_changes
1115
        self.wt.set_root_id('eert_toor')
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1116
        transform, root = self.get_transform()
1117
        transform.new_file('old', root, 'blah', 'id-1')
1118
        transform.new_directory('subdir', root, 'subdir-id')
1119
        transform.apply()
1120
        transform, root = self.get_transform()
1121
        try:
1122
            old = transform.trans_id_tree_path('old')
1123
            subdir = transform.trans_id_tree_file_id('subdir-id')
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1124
            self.assertEqual([], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1125
            transform.delete_contents(subdir)
1126
            transform.create_directory(subdir)
1127
            transform.set_executability(False, old)
1128
            transform.unversion_file(old)
1129
            transform.version_file('id-1', old)
1130
            transform.adjust_path('old', root, old)
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1131
            self.assertEqual([], list(transform.iter_changes()))
1551.11.2 by Aaron Bentley
Get kind change detection working for iter_changes
1132
        finally:
1133
            transform.finalize()
1534.7.93 by Aaron Bentley
Added text merge test
1134
2502.1.1 by Aaron Bentley
Ensure renames only root children are renamed when building trees
1135
    def test_rename_count(self):
1136
        transform, root = self.get_transform()
2502.1.8 by Aaron Bentley
Updates from review comments
1137
        transform.new_file('name1', root, 'contents')
2502.1.1 by Aaron Bentley
Ensure renames only root children are renamed when building trees
1138
        self.assertEqual(transform.rename_count, 0)
1139
        transform.apply()
1140
        self.assertEqual(transform.rename_count, 1)
1141
        transform2, root = self.get_transform()
2502.1.8 by Aaron Bentley
Updates from review comments
1142
        transform2.adjust_path('name2', root,
1143
                               transform2.trans_id_tree_path('name1'))
2502.1.1 by Aaron Bentley
Ensure renames only root children are renamed when building trees
1144
        self.assertEqual(transform2.rename_count, 0)
1145
        transform2.apply()
1146
        self.assertEqual(transform2.rename_count, 2)
1147
2502.1.2 by Aaron Bentley
Make the limited-renames functionality safer in the general case
1148
    def test_change_parent(self):
2502.1.8 by Aaron Bentley
Updates from review comments
1149
        """Ensure that after we change a parent, the results are still right.
1150
1151
        Renames and parent changes on pending transforms can happen as part
1152
        of conflict resolution, and are explicitly permitted by the
1153
        TreeTransform API.
1154
1155
        This test ensures they work correctly with the rename-avoidance
1156
        optimization.
1157
        """
2502.1.2 by Aaron Bentley
Make the limited-renames functionality safer in the general case
1158
        transform, root = self.get_transform()
2502.1.8 by Aaron Bentley
Updates from review comments
1159
        parent1 = transform.new_directory('parent1', root)
1160
        child1 = transform.new_file('child1', parent1, 'contents')
1161
        parent2 = transform.new_directory('parent2', root)
1162
        transform.adjust_path('child1', parent2, child1)
2502.1.2 by Aaron Bentley
Make the limited-renames functionality safer in the general case
1163
        transform.apply()
2502.1.8 by Aaron Bentley
Updates from review comments
1164
        self.failIfExists(self.wt.abspath('parent1/child1'))
1165
        self.failUnlessExists(self.wt.abspath('parent2/child1'))
1166
        # rename limbo/new-1 => parent1, rename limbo/new-3 => parent2
1167
        # no rename for child1 (counting only renames during apply)
2502.1.2 by Aaron Bentley
Make the limited-renames functionality safer in the general case
1168
        self.failUnlessEqual(2, transform.rename_count)
1169
1170
    def test_cancel_parent(self):
1171
        """Cancelling a parent doesn't cause deletion of a non-empty directory
1172
1173
        This is like the test_change_parent, except that we cancel the parent
1174
        before adjusting the path.  The transform must detect that the
1175
        directory is non-empty, and move children to safe locations.
1176
        """
1177
        transform, root = self.get_transform()
2502.1.8 by Aaron Bentley
Updates from review comments
1178
        parent1 = transform.new_directory('parent1', root)
1179
        child1 = transform.new_file('child1', parent1, 'contents')
1180
        child2 = transform.new_file('child2', parent1, 'contents')
1181
        try:
1182
            transform.cancel_creation(parent1)
1183
        except OSError:
1184
            self.fail('Failed to move child1 before deleting parent1')
1185
        transform.cancel_creation(child2)
1186
        transform.create_directory(parent1)
1187
        try:
1188
            transform.cancel_creation(parent1)
1189
        # If the transform incorrectly believes that child2 is still in
1190
        # parent1's limbo directory, it will try to rename it and fail
1191
        # because was already moved by the first cancel_creation.
1192
        except OSError:
1193
            self.fail('Transform still thinks child2 is a child of parent1')
1194
        parent2 = transform.new_directory('parent2', root)
1195
        transform.adjust_path('child1', parent2, child1)
2502.1.2 by Aaron Bentley
Make the limited-renames functionality safer in the general case
1196
        transform.apply()
2502.1.8 by Aaron Bentley
Updates from review comments
1197
        self.failIfExists(self.wt.abspath('parent1'))
1198
        self.failUnlessExists(self.wt.abspath('parent2/child1'))
1199
        # rename limbo/new-3 => parent2, rename limbo/new-2 => child1
2502.1.2 by Aaron Bentley
Make the limited-renames functionality safer in the general case
1200
        self.failUnlessEqual(2, transform.rename_count)
1201
1202
    def test_adjust_and_cancel(self):
2502.1.8 by Aaron Bentley
Updates from review comments
1203
        """Make sure adjust_path keeps track of limbo children properly"""
2502.1.2 by Aaron Bentley
Make the limited-renames functionality safer in the general case
1204
        transform, root = self.get_transform()
2502.1.8 by Aaron Bentley
Updates from review comments
1205
        parent1 = transform.new_directory('parent1', root)
1206
        child1 = transform.new_file('child1', parent1, 'contents')
1207
        parent2 = transform.new_directory('parent2', root)
1208
        transform.adjust_path('child1', parent2, child1)
1209
        transform.cancel_creation(child1)
2502.1.2 by Aaron Bentley
Make the limited-renames functionality safer in the general case
1210
        try:
2502.1.8 by Aaron Bentley
Updates from review comments
1211
            transform.cancel_creation(parent1)
1212
        # if the transform thinks child1 is still in parent1's limbo
1213
        # directory, it will attempt to move it and fail.
2502.1.2 by Aaron Bentley
Make the limited-renames functionality safer in the general case
1214
        except OSError:
2502.1.8 by Aaron Bentley
Updates from review comments
1215
            self.fail('Transform still thinks child1 is a child of parent1')
2502.1.2 by Aaron Bentley
Make the limited-renames functionality safer in the general case
1216
        transform.finalize()
1217
2502.1.3 by Aaron Bentley
Don't cause errors when creating contents for trans_ids with no parent/name
1218
    def test_noname_contents(self):
2502.1.8 by Aaron Bentley
Updates from review comments
1219
        """TreeTransform should permit deferring naming files."""
2502.1.3 by Aaron Bentley
Don't cause errors when creating contents for trans_ids with no parent/name
1220
        transform, root = self.get_transform()
2502.1.8 by Aaron Bentley
Updates from review comments
1221
        parent = transform.trans_id_file_id('parent-id')
2502.1.3 by Aaron Bentley
Don't cause errors when creating contents for trans_ids with no parent/name
1222
        try:
2502.1.8 by Aaron Bentley
Updates from review comments
1223
            transform.create_directory(parent)
2502.1.3 by Aaron Bentley
Don't cause errors when creating contents for trans_ids with no parent/name
1224
        except KeyError:
1225
            self.fail("Can't handle contents with no name")
1226
        transform.finalize()
1227
2502.1.9 by Aaron Bentley
Add additional test for no-name contents
1228
    def test_noname_contents_nested(self):
1229
        """TreeTransform should permit deferring naming files."""
1230
        transform, root = self.get_transform()
1231
        parent = transform.trans_id_file_id('parent-id')
1232
        try:
1233
            transform.create_directory(parent)
1234
        except KeyError:
1235
            self.fail("Can't handle contents with no name")
1236
        child = transform.new_directory('child', parent)
1237
        transform.adjust_path('parent', root, parent)
1238
        transform.apply()
1239
        self.failUnlessExists(self.wt.abspath('parent/child'))
1240
        self.assertEqual(1, transform.rename_count)
1241
2502.1.4 by Aaron Bentley
Ensure we only reuse limbo names appropriately
1242
    def test_reuse_name(self):
1243
        """Avoid reusing the same limbo name for different files"""
1244
        transform, root = self.get_transform()
2502.1.8 by Aaron Bentley
Updates from review comments
1245
        parent = transform.new_directory('parent', root)
1246
        child1 = transform.new_directory('child', parent)
2502.1.4 by Aaron Bentley
Ensure we only reuse limbo names appropriately
1247
        try:
2502.1.8 by Aaron Bentley
Updates from review comments
1248
            child2 = transform.new_directory('child', parent)
2502.1.4 by Aaron Bentley
Ensure we only reuse limbo names appropriately
1249
        except OSError:
1250
            self.fail('Tranform tried to use the same limbo name twice')
2502.1.8 by Aaron Bentley
Updates from review comments
1251
        transform.adjust_path('child2', parent, child2)
2502.1.4 by Aaron Bentley
Ensure we only reuse limbo names appropriately
1252
        transform.apply()
2502.1.8 by Aaron Bentley
Updates from review comments
1253
        # limbo/new-1 => parent, limbo/new-3 => parent/child2
1254
        # child2 is put into top-level limbo because child1 has already
1255
        # claimed the direct limbo path when child2 is created.  There is no
1256
        # advantage in renaming files once they're in top-level limbo, except
1257
        # as part of apply.
2502.1.4 by Aaron Bentley
Ensure we only reuse limbo names appropriately
1258
        self.assertEqual(2, transform.rename_count)
1259
1260
    def test_reuse_when_first_moved(self):
1261
        """Don't avoid direct paths when it is safe to use them"""
1262
        transform, root = self.get_transform()
2502.1.8 by Aaron Bentley
Updates from review comments
1263
        parent = transform.new_directory('parent', root)
1264
        child1 = transform.new_directory('child', parent)
1265
        transform.adjust_path('child1', parent, child1)
1266
        child2 = transform.new_directory('child', parent)
2502.1.4 by Aaron Bentley
Ensure we only reuse limbo names appropriately
1267
        transform.apply()
2502.1.8 by Aaron Bentley
Updates from review comments
1268
        # limbo/new-1 => parent
2502.1.4 by Aaron Bentley
Ensure we only reuse limbo names appropriately
1269
        self.assertEqual(1, transform.rename_count)
1270
1271
    def test_reuse_after_cancel(self):
1272
        """Don't avoid direct paths when it is safe to use them"""
1273
        transform, root = self.get_transform()
2502.1.8 by Aaron Bentley
Updates from review comments
1274
        parent2 = transform.new_directory('parent2', root)
1275
        child1 = transform.new_directory('child1', parent2)
1276
        transform.cancel_creation(parent2)
1277
        transform.create_directory(parent2)
1278
        child2 = transform.new_directory('child1', parent2)
1279
        transform.adjust_path('child2', parent2, child1)
2502.1.4 by Aaron Bentley
Ensure we only reuse limbo names appropriately
1280
        transform.apply()
2502.1.8 by Aaron Bentley
Updates from review comments
1281
        # limbo/new-1 => parent2, limbo/new-2 => parent2/child1
2502.1.4 by Aaron Bentley
Ensure we only reuse limbo names appropriately
1282
        self.assertEqual(2, transform.rename_count)
1283
2502.1.7 by Aaron Bentley
Fix finalize deletion ordering
1284
    def test_finalize_order(self):
2502.1.8 by Aaron Bentley
Updates from review comments
1285
        """Finalize must be done in child-to-parent order"""
2502.1.7 by Aaron Bentley
Fix finalize deletion ordering
1286
        transform, root = self.get_transform()
2502.1.8 by Aaron Bentley
Updates from review comments
1287
        parent = transform.new_directory('parent', root)
1288
        child = transform.new_directory('child', parent)
2502.1.7 by Aaron Bentley
Fix finalize deletion ordering
1289
        try:
1290
            transform.finalize()
1291
        except OSError:
2502.1.8 by Aaron Bentley
Updates from review comments
1292
            self.fail('Tried to remove parent before child1')
2502.1.7 by Aaron Bentley
Fix finalize deletion ordering
1293
2502.1.13 by Aaron Bentley
Updates from review
1294
    def test_cancel_with_cancelled_child_should_succeed(self):
2502.1.12 by Aaron Bentley
Avoid renaming children with no content
1295
        transform, root = self.get_transform()
1296
        parent = transform.new_directory('parent', root)
1297
        child = transform.new_directory('child', parent)
1298
        transform.cancel_creation(child)
2502.1.13 by Aaron Bentley
Updates from review
1299
        transform.cancel_creation(parent)
2502.1.12 by Aaron Bentley
Avoid renaming children with no content
1300
        transform.finalize()
1301
3638.3.15 by Vincent Ladeuil
Fix test_case_insensitive_clash to pass on all platforms (renamed too).
1302
    def test_rollback_on_directory_clash(self):
3063.1.1 by Alexander Belchenko
Catch OSError 17 (file exists) in final phase of tree transform and show filename to user (#111758).
1303
        def tt_helper():
3638.3.17 by Vincent Ladeuil
Fixed as per Aaron's review.
1304
            wt = self.make_branch_and_tree('.')
3063.1.1 by Alexander Belchenko
Catch OSError 17 (file exists) in final phase of tree transform and show filename to user (#111758).
1305
            tt = TreeTransform(wt)  # TreeTransform obtains write lock
1306
            try:
3638.3.15 by Vincent Ladeuil
Fix test_case_insensitive_clash to pass on all platforms (renamed too).
1307
                foo = tt.new_directory('foo', tt.root)
1308
                tt.new_file('bar', foo, 'foobar')
1309
                baz = tt.new_directory('baz', tt.root)
1310
                tt.new_file('qux', baz, 'quux')
1311
                # Ask for a rename 'foo' -> 'baz'
1312
                tt.adjust_path('baz', tt.root, foo)
3063.1.3 by Aaron Bentley
Update for Linux
1313
                # Lie to tt that we've already resolved all conflicts.
3063.1.1 by Alexander Belchenko
Catch OSError 17 (file exists) in final phase of tree transform and show filename to user (#111758).
1314
                tt.apply(no_conflicts=True)
3063.1.3 by Aaron Bentley
Update for Linux
1315
            except:
3063.1.1 by Alexander Belchenko
Catch OSError 17 (file exists) in final phase of tree transform and show filename to user (#111758).
1316
                wt.unlock()
3063.1.3 by Aaron Bentley
Update for Linux
1317
                raise
3638.3.17 by Vincent Ladeuil
Fixed as per Aaron's review.
1318
        # The rename will fail because the target directory is not empty (but
1319
        # raises FileExists anyway).
3063.1.1 by Alexander Belchenko
Catch OSError 17 (file exists) in final phase of tree transform and show filename to user (#111758).
1320
        err = self.assertRaises(errors.FileExists, tt_helper)
1321
        self.assertContainsRe(str(err),
3638.3.15 by Vincent Ladeuil
Fix test_case_insensitive_clash to pass on all platforms (renamed too).
1322
            "^File exists: .+/baz")
3063.1.1 by Alexander Belchenko
Catch OSError 17 (file exists) in final phase of tree transform and show filename to user (#111758).
1323
3063.1.2 by Alexander Belchenko
test for two directories clash
1324
    def test_two_directories_clash(self):
1325
        def tt_helper():
1326
            wt = self.make_branch_and_tree('.')
1327
            tt = TreeTransform(wt)  # TreeTransform obtains write lock
1328
            try:
3063.1.3 by Aaron Bentley
Update for Linux
1329
                foo_1 = tt.new_directory('foo', tt.root)
1330
                tt.new_directory('bar', foo_1)
3638.3.15 by Vincent Ladeuil
Fix test_case_insensitive_clash to pass on all platforms (renamed too).
1331
                # Adding the same directory with a different content
3063.1.3 by Aaron Bentley
Update for Linux
1332
                foo_2 = tt.new_directory('foo', tt.root)
1333
                tt.new_directory('baz', foo_2)
1334
                # Lie to tt that we've already resolved all conflicts.
3063.1.2 by Alexander Belchenko
test for two directories clash
1335
                tt.apply(no_conflicts=True)
3063.1.3 by Aaron Bentley
Update for Linux
1336
            except:
3063.1.2 by Alexander Belchenko
test for two directories clash
1337
                wt.unlock()
3063.1.3 by Aaron Bentley
Update for Linux
1338
                raise
3063.1.2 by Alexander Belchenko
test for two directories clash
1339
        err = self.assertRaises(errors.FileExists, tt_helper)
1340
        self.assertContainsRe(str(err),
1341
            "^File exists: .+/foo")
1342
3100.1.1 by Aaron Bentley
Fix ImmortalLimbo errors when transforms fail
1343
    def test_two_directories_clash_finalize(self):
1344
        def tt_helper():
1345
            wt = self.make_branch_and_tree('.')
1346
            tt = TreeTransform(wt)  # TreeTransform obtains write lock
1347
            try:
1348
                foo_1 = tt.new_directory('foo', tt.root)
1349
                tt.new_directory('bar', foo_1)
3638.3.15 by Vincent Ladeuil
Fix test_case_insensitive_clash to pass on all platforms (renamed too).
1350
                # Adding the same directory with a different content
3100.1.1 by Aaron Bentley
Fix ImmortalLimbo errors when transforms fail
1351
                foo_2 = tt.new_directory('foo', tt.root)
1352
                tt.new_directory('baz', foo_2)
1353
                # Lie to tt that we've already resolved all conflicts.
1354
                tt.apply(no_conflicts=True)
1355
            except:
1356
                tt.finalize()
1357
                raise
1358
        err = self.assertRaises(errors.FileExists, tt_helper)
1359
        self.assertContainsRe(str(err),
1360
            "^File exists: .+/foo")
1361
3535.6.1 by James Westby
Handle a file turning in to a directory in TreeTransform.
1362
    def test_file_to_directory(self):
1363
        wt = self.make_branch_and_tree('.')
3535.6.2 by James Westby
Fixes from review. Thanks Aaron and John.
1364
        self.build_tree(['foo'])
3535.6.1 by James Westby
Handle a file turning in to a directory in TreeTransform.
1365
        wt.add(['foo'])
3590.3.1 by James Westby
Make TreeTransform update the inventory with new kind information.
1366
        wt.commit("one")
3535.6.1 by James Westby
Handle a file turning in to a directory in TreeTransform.
1367
        tt = TreeTransform(wt)
3535.6.2 by James Westby
Fixes from review. Thanks Aaron and John.
1368
        self.addCleanup(tt.finalize)
3535.6.3 by James Westby
Fix the test to not create transform conflicts.
1369
        foo_trans_id = tt.trans_id_tree_path("foo")
1370
        tt.delete_contents(foo_trans_id)
1371
        tt.create_directory(foo_trans_id)
1372
        bar_trans_id = tt.trans_id_tree_path("foo/bar")
1373
        tt.create_file(["aa\n"], bar_trans_id)
1374
        tt.version_file("bar-1", bar_trans_id)
3535.6.2 by James Westby
Fixes from review. Thanks Aaron and John.
1375
        tt.apply()
3535.6.1 by James Westby
Handle a file turning in to a directory in TreeTransform.
1376
        self.failUnlessExists("foo/bar")
3590.3.2 by James Westby
Handle ->symlink changes as well.
1377
        wt.lock_read()
1378
        try:
1379
            self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
1380
                    "directory")
1381
        finally:
1382
            wt.unlock()
3590.3.1 by James Westby
Make TreeTransform update the inventory with new kind information.
1383
        wt.commit("two")
1384
        changes = wt.changes_from(wt.basis_tree())
1385
        self.assertFalse(changes.has_changed(), changes)
3535.6.1 by James Westby
Handle a file turning in to a directory in TreeTransform.
1386
3590.3.2 by James Westby
Handle ->symlink changes as well.
1387
    def test_file_to_symlink(self):
3590.3.3 by James Westby
Make ->file changes work as well.
1388
        self.requireFeature(SymlinkFeature)
3590.3.2 by James Westby
Handle ->symlink changes as well.
1389
        wt = self.make_branch_and_tree('.')
1390
        self.build_tree(['foo'])
1391
        wt.add(['foo'])
1392
        wt.commit("one")
1393
        tt = TreeTransform(wt)
1394
        self.addCleanup(tt.finalize)
1395
        foo_trans_id = tt.trans_id_tree_path("foo")
1396
        tt.delete_contents(foo_trans_id)
1397
        tt.create_symlink("bar", foo_trans_id)
1398
        tt.apply()
1399
        self.failUnlessExists("foo")
1400
        wt.lock_read()
1401
        self.addCleanup(wt.unlock)
1402
        self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
1403
                "symlink")
1404
3590.3.3 by James Westby
Make ->file changes work as well.
1405
    def test_dir_to_file(self):
1406
        wt = self.make_branch_and_tree('.')
1407
        self.build_tree(['foo/', 'foo/bar'])
1408
        wt.add(['foo', 'foo/bar'])
1409
        wt.commit("one")
1410
        tt = TreeTransform(wt)
1411
        self.addCleanup(tt.finalize)
1412
        foo_trans_id = tt.trans_id_tree_path("foo")
1413
        bar_trans_id = tt.trans_id_tree_path("foo/bar")
1414
        tt.delete_contents(foo_trans_id)
1415
        tt.delete_versioned(bar_trans_id)
1416
        tt.create_file(["aa\n"], foo_trans_id)
1417
        tt.apply()
1418
        self.failUnlessExists("foo")
1419
        wt.lock_read()
1420
        self.addCleanup(wt.unlock)
1421
        self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
1422
                "file")
1423
3590.3.4 by James Westby
Add a test for creating hardlinks as well.
1424
    def test_dir_to_hardlink(self):
3590.3.5 by James Westby
Use HardlinkFeature for the hardlink test.
1425
        self.requireFeature(HardlinkFeature)
3590.3.4 by James Westby
Add a test for creating hardlinks as well.
1426
        wt = self.make_branch_and_tree('.')
1427
        self.build_tree(['foo/', 'foo/bar'])
1428
        wt.add(['foo', 'foo/bar'])
1429
        wt.commit("one")
1430
        tt = TreeTransform(wt)
1431
        self.addCleanup(tt.finalize)
1432
        foo_trans_id = tt.trans_id_tree_path("foo")
1433
        bar_trans_id = tt.trans_id_tree_path("foo/bar")
1434
        tt.delete_contents(foo_trans_id)
1435
        tt.delete_versioned(bar_trans_id)
1436
        self.build_tree(['baz'])
1437
        tt.create_hardlink("baz", foo_trans_id)
1438
        tt.apply()
1439
        self.failUnlessExists("foo")
1440
        self.failUnlessExists("baz")
1441
        wt.lock_read()
1442
        self.addCleanup(wt.unlock)
1443
        self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
1444
                "file")
1445
3619.2.10 by Aaron Bentley
Compensate for stale entries in TT._needs_rename
1446
    def test_no_final_path(self):
1447
        transform, root = self.get_transform()
1448
        trans_id = transform.trans_id_file_id('foo')
1449
        transform.create_file('bar', trans_id)
1450
        transform.cancel_creation(trans_id)
1451
        transform.apply()
1452
3363.17.24 by Aaron Bentley
Implement create_by_tree
1453
    def test_create_from_tree(self):
1454
        tree1 = self.make_branch_and_tree('tree1')
1455
        self.build_tree_contents([('tree1/foo/',), ('tree1/bar', 'baz')])
1456
        tree1.add(['foo', 'bar'], ['foo-id', 'bar-id'])
1457
        tree2 = self.make_branch_and_tree('tree2')
1458
        tt = TreeTransform(tree2)
1459
        foo_trans_id = tt.create_path('foo', tt.root)
1460
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id')
1461
        bar_trans_id = tt.create_path('bar', tt.root)
1462
        create_from_tree(tt, bar_trans_id, tree1, 'bar-id')
1463
        tt.apply()
1464
        self.assertEqual('directory', osutils.file_kind('tree2/foo'))
1465
        self.assertFileEqual('baz', 'tree2/bar')
1466
3363.17.25 by Aaron Bentley
remove get_inventory_entry, replace with create_from_tree
1467
    def test_create_from_tree_bytes(self):
1468
        """Provided lines are used instead of tree content."""
1469
        tree1 = self.make_branch_and_tree('tree1')
1470
        self.build_tree_contents([('tree1/foo', 'bar'),])
1471
        tree1.add('foo', 'foo-id')
1472
        tree2 = self.make_branch_and_tree('tree2')
1473
        tt = TreeTransform(tree2)
1474
        foo_trans_id = tt.create_path('foo', tt.root)
1475
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id', bytes='qux')
1476
        tt.apply()
1477
        self.assertFileEqual('qux', 'tree2/foo')
1478
1479
    def test_create_from_tree_symlink(self):
3363.17.24 by Aaron Bentley
Implement create_by_tree
1480
        self.requireFeature(SymlinkFeature)
1481
        tree1 = self.make_branch_and_tree('tree1')
1482
        os.symlink('bar', 'tree1/foo')
1483
        tree1.add('foo', 'foo-id')
1484
        tt = TreeTransform(self.make_branch_and_tree('tree2'))
1485
        foo_trans_id = tt.create_path('foo', tt.root)
1486
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id')
1487
        tt.apply()
1488
        self.assertEqual('bar', os.readlink('tree2/foo'))
1489
2502.1.3 by Aaron Bentley
Don't cause errors when creating contents for trans_ids with no parent/name
1490
1534.7.93 by Aaron Bentley
Added text merge test
1491
class TransformGroup(object):
3063.1.1 by Alexander Belchenko
Catch OSError 17 (file exists) in final phase of tree transform and show filename to user (#111758).
1492
1731.1.33 by Aaron Bentley
Revert no-special-root changes
1493
    def __init__(self, dirname, root_id):
1534.7.101 by Aaron Bentley
Got conflicts on symlinks working properly
1494
        self.name = dirname
1534.7.93 by Aaron Bentley
Added text merge test
1495
        os.mkdir(dirname)
1558.1.3 by Aaron Bentley
Fixed deprecated op use in test suite
1496
        self.wt = BzrDir.create_standalone_workingtree(dirname)
1731.1.33 by Aaron Bentley
Revert no-special-root changes
1497
        self.wt.set_root_id(root_id)
1558.1.3 by Aaron Bentley
Fixed deprecated op use in test suite
1498
        self.b = self.wt.branch
1534.7.93 by Aaron Bentley
Added text merge test
1499
        self.tt = TreeTransform(self.wt)
1534.7.181 by Aaron Bentley
Renamed a bunch of functions
1500
        self.root = self.tt.trans_id_tree_file_id(self.wt.get_root_id())
1534.7.93 by Aaron Bentley
Added text merge test
1501
1551.11.1 by Aaron Bentley
Initial work on converting TreeTransform to iter_changes format
1502
1534.7.95 by Aaron Bentley
Added more text merge tests
1503
def conflict_text(tree, merge):
1504
    template = '%s TREE\n%s%s\n%s%s MERGE-SOURCE\n'
1505
    return template % ('<' * 7, tree, '=' * 7, merge, '>' * 7)
1506
1534.7.93 by Aaron Bentley
Added text merge test
1507
1508
class TestTransformMerge(TestCaseInTempDir):
3199.1.5 by Vincent Ladeuil
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.
1509
1534.7.93 by Aaron Bentley
Added text merge test
1510
    def test_text_merge(self):
2116.4.1 by John Arbash Meinel
Update file and revision id generators.
1511
        root_id = generate_ids.gen_root_id()
1731.1.33 by Aaron Bentley
Revert no-special-root changes
1512
        base = TransformGroup("base", root_id)
1534.7.93 by Aaron Bentley
Added text merge test
1513
        base.tt.new_file('a', base.root, 'a\nb\nc\nd\be\n', 'a')
1534.7.95 by Aaron Bentley
Added more text merge tests
1514
        base.tt.new_file('b', base.root, 'b1', 'b')
1515
        base.tt.new_file('c', base.root, 'c', 'c')
1516
        base.tt.new_file('d', base.root, 'd', 'd')
1517
        base.tt.new_file('e', base.root, 'e', 'e')
1518
        base.tt.new_file('f', base.root, 'f', 'f')
1534.7.96 by Aaron Bentley
Tested with BASE as directory
1519
        base.tt.new_directory('g', base.root, 'g')
1534.7.97 by Aaron Bentley
Ensured foo.BASE is a directory if there's a conflict
1520
        base.tt.new_directory('h', base.root, 'h')
1534.7.93 by Aaron Bentley
Added text merge test
1521
        base.tt.apply()
1731.1.33 by Aaron Bentley
Revert no-special-root changes
1522
        other = TransformGroup("other", root_id)
1534.7.93 by Aaron Bentley
Added text merge test
1523
        other.tt.new_file('a', other.root, 'y\nb\nc\nd\be\n', 'a')
1534.7.95 by Aaron Bentley
Added more text merge tests
1524
        other.tt.new_file('b', other.root, 'b2', 'b')
1525
        other.tt.new_file('c', other.root, 'c2', 'c')
1526
        other.tt.new_file('d', other.root, 'd', 'd')
1527
        other.tt.new_file('e', other.root, 'e2', 'e')
1528
        other.tt.new_file('f', other.root, 'f', 'f')
1534.7.96 by Aaron Bentley
Tested with BASE as directory
1529
        other.tt.new_file('g', other.root, 'g', 'g')
1534.7.97 by Aaron Bentley
Ensured foo.BASE is a directory if there's a conflict
1530
        other.tt.new_file('h', other.root, 'h\ni\nj\nk\n', 'h')
1534.7.99 by Aaron Bentley
Handle non-existent BASE properly
1531
        other.tt.new_file('i', other.root, 'h\ni\nj\nk\n', 'i')
1534.7.93 by Aaron Bentley
Added text merge test
1532
        other.tt.apply()
1731.1.33 by Aaron Bentley
Revert no-special-root changes
1533
        this = TransformGroup("this", root_id)
1534.7.93 by Aaron Bentley
Added text merge test
1534
        this.tt.new_file('a', this.root, 'a\nb\nc\nd\bz\n', 'a')
1534.7.95 by Aaron Bentley
Added more text merge tests
1535
        this.tt.new_file('b', this.root, 'b', 'b')
1536
        this.tt.new_file('c', this.root, 'c', 'c')
1537
        this.tt.new_file('d', this.root, 'd2', 'd')
1538
        this.tt.new_file('e', this.root, 'e2', 'e')
1539
        this.tt.new_file('f', this.root, 'f', 'f')
1534.7.96 by Aaron Bentley
Tested with BASE as directory
1540
        this.tt.new_file('g', this.root, 'g', 'g')
1534.7.97 by Aaron Bentley
Ensured foo.BASE is a directory if there's a conflict
1541
        this.tt.new_file('h', this.root, '1\n2\n3\n4\n', 'h')
1534.7.99 by Aaron Bentley
Handle non-existent BASE properly
1542
        this.tt.new_file('i', this.root, '1\n2\n3\n4\n', 'i')
1534.7.93 by Aaron Bentley
Added text merge test
1543
        this.tt.apply()
3008.1.11 by Michael Hudson
restore the default behaviour of Merge3Merger.__init__().
1544
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
3008.1.6 by Michael Hudson
chop up Merge3Merger.__init__ into pieces
1545
1534.7.95 by Aaron Bentley
Added more text merge tests
1546
        # textual merge
1534.7.93 by Aaron Bentley
Added text merge test
1547
        self.assertEqual(this.wt.get_file('a').read(), 'y\nb\nc\nd\bz\n')
1534.7.95 by Aaron Bentley
Added more text merge tests
1548
        # three-way text conflict
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1549
        self.assertEqual(this.wt.get_file('b').read(),
1534.7.95 by Aaron Bentley
Added more text merge tests
1550
                         conflict_text('b', 'b2'))
1551
        # OTHER wins
1552
        self.assertEqual(this.wt.get_file('c').read(), 'c2')
1553
        # THIS wins
1554
        self.assertEqual(this.wt.get_file('d').read(), 'd2')
1555
        # Ambigious clean merge
1556
        self.assertEqual(this.wt.get_file('e').read(), 'e2')
1557
        # No change
1558
        self.assertEqual(this.wt.get_file('f').read(), 'f')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1559
        # Correct correct results when THIS == OTHER
1534.7.96 by Aaron Bentley
Tested with BASE as directory
1560
        self.assertEqual(this.wt.get_file('g').read(), 'g')
1534.7.97 by Aaron Bentley
Ensured foo.BASE is a directory if there's a conflict
1561
        # Text conflict when THIS & OTHER are text and BASE is dir
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1562
        self.assertEqual(this.wt.get_file('h').read(),
1534.7.97 by Aaron Bentley
Ensured foo.BASE is a directory if there's a conflict
1563
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
1564
        self.assertEqual(this.wt.get_file_byname('h.THIS').read(),
1565
                         '1\n2\n3\n4\n')
1566
        self.assertEqual(this.wt.get_file_byname('h.OTHER').read(),
1567
                         'h\ni\nj\nk\n')
1568
        self.assertEqual(file_kind(this.wt.abspath('h.BASE')), 'directory')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1569
        self.assertEqual(this.wt.get_file('i').read(),
1534.7.99 by Aaron Bentley
Handle non-existent BASE properly
1570
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
1571
        self.assertEqual(this.wt.get_file_byname('i.THIS').read(),
1572
                         '1\n2\n3\n4\n')
1573
        self.assertEqual(this.wt.get_file_byname('i.OTHER').read(),
1574
                         'h\ni\nj\nk\n')
1575
        self.assertEqual(os.path.exists(this.wt.abspath('i.BASE')), False)
1534.7.192 by Aaron Bentley
Record hashes produced by merges
1576
        modified = ['a', 'b', 'c', 'h', 'i']
1577
        merge_modified = this.wt.merge_modified()
1578
        self.assertSubset(merge_modified, modified)
1579
        self.assertEqual(len(merge_modified), len(modified))
1580
        file(this.wt.id2abspath('a'), 'wb').write('booga')
1581
        modified.pop(0)
1582
        merge_modified = this.wt.merge_modified()
1583
        self.assertSubset(merge_modified, modified)
1584
        self.assertEqual(len(merge_modified), len(modified))
1558.12.10 by Aaron Bentley
Be robust when merge_hash file_id not in inventory
1585
        this.wt.remove('b')
2796.1.4 by Aaron Bentley
Fix up various test cases
1586
        this.wt.revert()
1534.7.101 by Aaron Bentley
Got conflicts on symlinks working properly
1587
1588
    def test_file_merge(self):
2949.5.1 by Alexander Belchenko
selftest: use SymlinkFeature instead of TestSkipped where appropriate
1589
        self.requireFeature(SymlinkFeature)
2116.4.1 by John Arbash Meinel
Update file and revision id generators.
1590
        root_id = generate_ids.gen_root_id()
1731.1.33 by Aaron Bentley
Revert no-special-root changes
1591
        base = TransformGroup("BASE", root_id)
1592
        this = TransformGroup("THIS", root_id)
1593
        other = TransformGroup("OTHER", root_id)
1534.7.101 by Aaron Bentley
Got conflicts on symlinks working properly
1594
        for tg in this, base, other:
1595
            tg.tt.new_directory('a', tg.root, 'a')
1596
            tg.tt.new_symlink('b', tg.root, 'b', 'b')
1597
            tg.tt.new_file('c', tg.root, 'c', 'c')
1598
            tg.tt.new_symlink('d', tg.root, tg.name, 'd')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1599
        targets = ((base, 'base-e', 'base-f', None, None),
1600
                   (this, 'other-e', 'this-f', 'other-g', 'this-h'),
1534.7.104 by Aaron Bentley
Fixed set_versioned, enhanced conflict testing
1601
                   (other, 'other-e', None, 'other-g', 'other-h'))
1602
        for tg, e_target, f_target, g_target, h_target in targets:
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1603
            for link, target in (('e', e_target), ('f', f_target),
1534.7.104 by Aaron Bentley
Fixed set_versioned, enhanced conflict testing
1604
                                 ('g', g_target), ('h', h_target)):
1605
                if target is not None:
1606
                    tg.tt.new_symlink(link, tg.root, target, link)
1534.7.102 by Aaron Bentley
Deleted old pre-conflict contents
1607
1608
        for tg in this, base, other:
1534.7.101 by Aaron Bentley
Got conflicts on symlinks working properly
1609
            tg.tt.apply()
3008.1.11 by Michael Hudson
restore the default behaviour of Merge3Merger.__init__().
1610
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
1534.7.101 by Aaron Bentley
Got conflicts on symlinks working properly
1611
        self.assertIs(os.path.isdir(this.wt.abspath('a')), True)
1612
        self.assertIs(os.path.islink(this.wt.abspath('b')), True)
1613
        self.assertIs(os.path.isfile(this.wt.abspath('c')), True)
1614
        for suffix in ('THIS', 'BASE', 'OTHER'):
1615
            self.assertEqual(os.readlink(this.wt.abspath('d.'+suffix)), suffix)
1534.7.102 by Aaron Bentley
Deleted old pre-conflict contents
1616
        self.assertIs(os.path.lexists(this.wt.abspath('d')), False)
1534.7.104 by Aaron Bentley
Fixed set_versioned, enhanced conflict testing
1617
        self.assertEqual(this.wt.id2path('d'), 'd.OTHER')
1618
        self.assertEqual(this.wt.id2path('f'), 'f.THIS')
1534.7.102 by Aaron Bentley
Deleted old pre-conflict contents
1619
        self.assertEqual(os.readlink(this.wt.abspath('e')), 'other-e')
1620
        self.assertIs(os.path.lexists(this.wt.abspath('e.THIS')), False)
1621
        self.assertIs(os.path.lexists(this.wt.abspath('e.OTHER')), False)
1622
        self.assertIs(os.path.lexists(this.wt.abspath('e.BASE')), False)
1534.7.104 by Aaron Bentley
Fixed set_versioned, enhanced conflict testing
1623
        self.assertIs(os.path.lexists(this.wt.abspath('g')), True)
1624
        self.assertIs(os.path.lexists(this.wt.abspath('g.BASE')), False)
1625
        self.assertIs(os.path.lexists(this.wt.abspath('h')), False)
1626
        self.assertIs(os.path.lexists(this.wt.abspath('h.BASE')), False)
1627
        self.assertIs(os.path.lexists(this.wt.abspath('h.THIS')), True)
1628
        self.assertIs(os.path.lexists(this.wt.abspath('h.OTHER')), True)
1534.7.105 by Aaron Bentley
Got merge with rename working
1629
1630
    def test_filename_merge(self):
2116.4.1 by John Arbash Meinel
Update file and revision id generators.
1631
        root_id = generate_ids.gen_root_id()
1731.1.33 by Aaron Bentley
Revert no-special-root changes
1632
        base = TransformGroup("BASE", root_id)
1633
        this = TransformGroup("THIS", root_id)
1634
        other = TransformGroup("OTHER", root_id)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1635
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
1534.7.105 by Aaron Bentley
Got merge with rename working
1636
                                   for t in [base, this, other]]
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1637
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
1534.7.105 by Aaron Bentley
Got merge with rename working
1638
                                   for t in [base, this, other]]
1639
        base.tt.new_directory('c', base_a, 'c')
1640
        this.tt.new_directory('c1', this_a, 'c')
1641
        other.tt.new_directory('c', other_b, 'c')
1642
1643
        base.tt.new_directory('d', base_a, 'd')
1644
        this.tt.new_directory('d1', this_b, 'd')
1645
        other.tt.new_directory('d', other_a, 'd')
1646
1647
        base.tt.new_directory('e', base_a, 'e')
1648
        this.tt.new_directory('e', this_a, 'e')
1649
        other.tt.new_directory('e1', other_b, 'e')
1650
1651
        base.tt.new_directory('f', base_a, 'f')
1652
        this.tt.new_directory('f1', this_b, 'f')
1653
        other.tt.new_directory('f1', other_b, 'f')
1654
1655
        for tg in [this, base, other]:
1656
            tg.tt.apply()
3008.1.11 by Michael Hudson
restore the default behaviour of Merge3Merger.__init__().
1657
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
1534.7.176 by abentley
Fixed up tests for Windows
1658
        self.assertEqual(this.wt.id2path('c'), pathjoin('b/c1'))
1659
        self.assertEqual(this.wt.id2path('d'), pathjoin('b/d1'))
1660
        self.assertEqual(this.wt.id2path('e'), pathjoin('b/e1'))
1661
        self.assertEqual(this.wt.id2path('f'), pathjoin('b/f1'))
1534.7.105 by Aaron Bentley
Got merge with rename working
1662
1663
    def test_filename_merge_conflicts(self):
2116.4.1 by John Arbash Meinel
Update file and revision id generators.
1664
        root_id = generate_ids.gen_root_id()
1731.1.33 by Aaron Bentley
Revert no-special-root changes
1665
        base = TransformGroup("BASE", root_id)
1666
        this = TransformGroup("THIS", root_id)
1667
        other = TransformGroup("OTHER", root_id)
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1668
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
1534.7.105 by Aaron Bentley
Got merge with rename working
1669
                                   for t in [base, this, other]]
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1670
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
1534.7.105 by Aaron Bentley
Got merge with rename working
1671
                                   for t in [base, this, other]]
1672
1673
        base.tt.new_file('g', base_a, 'g', 'g')
1674
        other.tt.new_file('g1', other_b, 'g1', 'g')
1675
1676
        base.tt.new_file('h', base_a, 'h', 'h')
1677
        this.tt.new_file('h1', this_b, 'h1', 'h')
1678
1679
        base.tt.new_file('i', base.root, 'i', 'i')
1534.7.153 by Aaron Bentley
Handled test cases involving symlinks
1680
        other.tt.new_directory('i1', this_b, 'i')
1534.7.105 by Aaron Bentley
Got merge with rename working
1681
1682
        for tg in [this, base, other]:
1683
            tg.tt.apply()
3008.1.11 by Michael Hudson
restore the default behaviour of Merge3Merger.__init__().
1684
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
1534.7.105 by Aaron Bentley
Got merge with rename working
1685
1534.7.176 by abentley
Fixed up tests for Windows
1686
        self.assertEqual(this.wt.id2path('g'), pathjoin('b/g1.OTHER'))
1534.7.105 by Aaron Bentley
Got merge with rename working
1687
        self.assertIs(os.path.lexists(this.wt.abspath('b/g1.BASE')), True)
1688
        self.assertIs(os.path.lexists(this.wt.abspath('b/g1.THIS')), False)
1534.7.176 by abentley
Fixed up tests for Windows
1689
        self.assertEqual(this.wt.id2path('h'), pathjoin('b/h1.THIS'))
1534.7.105 by Aaron Bentley
Got merge with rename working
1690
        self.assertIs(os.path.lexists(this.wt.abspath('b/h1.BASE')), True)
1691
        self.assertIs(os.path.lexists(this.wt.abspath('b/h1.OTHER')), False)
1534.7.176 by abentley
Fixed up tests for Windows
1692
        self.assertEqual(this.wt.id2path('i'), pathjoin('b/i1.OTHER'))
1534.7.183 by Aaron Bentley
Fixed build_tree with symlinks
1693
2027.1.1 by John Arbash Meinel
Fix bug #56549, and write a direct test that the right path is being statted
1694
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1695
class TestBuildTree(tests.TestCaseWithTransport):
1696
3006.2.2 by Alexander Belchenko
tests added.
1697
    def test_build_tree_with_symlinks(self):
2949.5.1 by Alexander Belchenko
selftest: use SymlinkFeature instead of TestSkipped where appropriate
1698
        self.requireFeature(SymlinkFeature)
1534.7.183 by Aaron Bentley
Fixed build_tree with symlinks
1699
        os.mkdir('a')
1700
        a = BzrDir.create_standalone_workingtree('a')
1701
        os.mkdir('a/foo')
1702
        file('a/foo/bar', 'wb').write('contents')
1703
        os.symlink('a/foo/bar', 'a/foo/baz')
1704
        a.add(['foo', 'foo/bar', 'foo/baz'])
1705
        a.commit('initial commit')
1706
        b = BzrDir.create_standalone_workingtree('b')
2255.2.194 by Robert Collins
[BROKEN] Many updates to stop using experimental formats in tests.
1707
        basis = a.basis_tree()
1708
        basis.lock_read()
1709
        self.addCleanup(basis.unlock)
1710
        build_tree(basis, b)
1534.7.183 by Aaron Bentley
Fixed build_tree with symlinks
1711
        self.assertIs(os.path.isdir('b/foo'), True)
1712
        self.assertEqual(file('b/foo/bar', 'rb').read(), "contents")
1713
        self.assertEqual(os.readlink('b/foo/baz'), 'a/foo/bar')
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1714
2100.3.21 by Aaron Bentley
Work on checking out by-reference trees
1715
    def test_build_with_references(self):
2255.2.194 by Robert Collins
[BROKEN] Many updates to stop using experimental formats in tests.
1716
        tree = self.make_branch_and_tree('source',
1717
            format='dirstate-with-subtree')
1718
        subtree = self.make_branch_and_tree('source/subtree',
1719
            format='dirstate-with-subtree')
2100.3.21 by Aaron Bentley
Work on checking out by-reference trees
1720
        tree.add_reference(subtree)
1721
        tree.commit('a revision')
1722
        tree.branch.create_checkout('target')
1723
        self.failUnlessExists('target')
1724
        self.failUnlessExists('target/subtree')
1725
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1726
    def test_file_conflict_handling(self):
1727
        """Ensure that when building trees, conflict handling is done"""
1728
        source = self.make_branch_and_tree('source')
1729
        target = self.make_branch_and_tree('target')
1730
        self.build_tree(['source/file', 'target/file'])
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1731
        source.add('file', 'new-file')
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1732
        source.commit('added file')
1733
        build_tree(source.basis_tree(), target)
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1734
        self.assertEqual([DuplicateEntry('Moved existing file to',
1735
                          'file.moved', 'file', None, 'new-file')],
1736
                         target.conflicts())
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1737
        target2 = self.make_branch_and_tree('target2')
1738
        target_file = file('target2/file', 'wb')
1739
        try:
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1740
            source_file = file('source/file', 'rb')
1741
            try:
1742
                target_file.write(source_file.read())
1743
            finally:
1744
                source_file.close()
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1745
        finally:
1746
            target_file.close()
1747
        build_tree(source.basis_tree(), target2)
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1748
        self.assertEqual([], target2.conflicts())
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1749
1750
    def test_symlink_conflict_handling(self):
1751
        """Ensure that when building trees, conflict handling is done"""
2949.5.1 by Alexander Belchenko
selftest: use SymlinkFeature instead of TestSkipped where appropriate
1752
        self.requireFeature(SymlinkFeature)
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1753
        source = self.make_branch_and_tree('source')
1754
        os.symlink('foo', 'source/symlink')
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1755
        source.add('symlink', 'new-symlink')
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1756
        source.commit('added file')
1757
        target = self.make_branch_and_tree('target')
1758
        os.symlink('bar', 'target/symlink')
1759
        build_tree(source.basis_tree(), target)
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1760
        self.assertEqual([DuplicateEntry('Moved existing file to',
1761
            'symlink.moved', 'symlink', None, 'new-symlink')],
1762
            target.conflicts())
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1763
        target = self.make_branch_and_tree('target2')
1764
        os.symlink('foo', 'target2/symlink')
1765
        build_tree(source.basis_tree(), target)
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1766
        self.assertEqual([], target.conflicts())
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1767
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1768
    def test_directory_conflict_handling(self):
1769
        """Ensure that when building trees, conflict handling is done"""
1770
        source = self.make_branch_and_tree('source')
1771
        target = self.make_branch_and_tree('target')
1772
        self.build_tree(['source/dir1/', 'source/dir1/file', 'target/dir1/'])
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1773
        source.add(['dir1', 'dir1/file'], ['new-dir1', 'new-file'])
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1774
        source.commit('added file')
1775
        build_tree(source.basis_tree(), target)
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1776
        self.assertEqual([], target.conflicts())
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1777
        self.failUnlessExists('target/dir1/file')
1778
1779
        # Ensure contents are merged
1780
        target = self.make_branch_and_tree('target2')
1781
        self.build_tree(['target2/dir1/', 'target2/dir1/file2'])
1782
        build_tree(source.basis_tree(), target)
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1783
        self.assertEqual([], target.conflicts())
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1784
        self.failUnlessExists('target2/dir1/file2')
1785
        self.failUnlessExists('target2/dir1/file')
1786
1787
        # Ensure new contents are suppressed for existing branches
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1788
        target = self.make_branch_and_tree('target3')
1789
        self.make_branch('target3/dir1')
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1790
        self.build_tree(['target3/dir1/file2'])
1791
        build_tree(source.basis_tree(), target)
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1792
        self.failIfExists('target3/dir1/file')
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1793
        self.failUnlessExists('target3/dir1/file2')
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1794
        self.failUnlessExists('target3/dir1.diverted/file')
1795
        self.assertEqual([DuplicateEntry('Diverted to',
1796
            'dir1.diverted', 'dir1', 'new-dir1', None)],
1797
            target.conflicts())
1798
1799
        target = self.make_branch_and_tree('target4')
1800
        self.build_tree(['target4/dir1/'])
1801
        self.make_branch('target4/dir1/file')
1802
        build_tree(source.basis_tree(), target)
1803
        self.failUnlessExists('target4/dir1/file')
1804
        self.assertEqual('directory', file_kind('target4/dir1/file'))
1805
        self.failUnlessExists('target4/dir1/file.diverted')
1806
        self.assertEqual([DuplicateEntry('Diverted to',
1807
            'dir1/file.diverted', 'dir1/file', 'new-file', None)],
1808
            target.conflicts())
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1809
1810
    def test_mixed_conflict_handling(self):
1811
        """Ensure that when building trees, conflict handling is done"""
1812
        source = self.make_branch_and_tree('source')
1813
        target = self.make_branch_and_tree('target')
1814
        self.build_tree(['source/name', 'target/name/'])
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1815
        source.add('name', 'new-name')
1816
        source.commit('added file')
1817
        build_tree(source.basis_tree(), target)
1818
        self.assertEqual([DuplicateEntry('Moved existing file to',
1819
            'name.moved', 'name', None, 'new-name')], target.conflicts())
1820
1821
    def test_raises_in_populated(self):
1822
        source = self.make_branch_and_tree('source')
1823
        self.build_tree(['source/name'])
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1824
        source.add('name')
1966.1.2 by Aaron Bentley
Divert files instead of failing to create them, update from review
1825
        source.commit('added name')
1826
        target = self.make_branch_and_tree('target')
1827
        self.build_tree(['target/name'])
1828
        target.add('name')
3943.8.1 by Marius Kruger
remove all trailing whitespace from bzr source
1829
        self.assertRaises(errors.WorkingTreeAlreadyPopulated,
2090.2.1 by Martin Pool
Fix some code which relies on assertions and breaks under python -O
1830
            build_tree, source.basis_tree(), target)
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
1831
2502.1.1 by Aaron Bentley
Ensure renames only root children are renamed when building trees
1832
    def test_build_tree_rename_count(self):
1833
        source = self.make_branch_and_tree('source')
1834
        self.build_tree(['source/file1', 'source/dir1/'])
1835
        source.add(['file1', 'dir1'])
1836
        source.commit('add1')
1837
        target1 = self.make_branch_and_tree('target1')
2502.1.6 by Aaron Bentley
Update from review comments
1838
        transform_result = build_tree(source.basis_tree(), target1)
1839
        self.assertEqual(2, transform_result.rename_count)
2502.1.1 by Aaron Bentley
Ensure renames only root children are renamed when building trees
1840
1841
        self.build_tree(['source/dir1/file2'])
1842
        source.add(['dir1/file2'])
1843
        source.commit('add3')
1844
        target2 = self.make_branch_and_tree('target2')
2502.1.6 by Aaron Bentley
Update from review comments
1845
        transform_result = build_tree(source.basis_tree(), target2)
2502.1.1 by Aaron Bentley
Ensure renames only root children are renamed when building trees
1846
        # children of non-root directories should not be renamed
2502.1.6 by Aaron Bentley
Update from review comments
1847
        self.assertEqual(2, transform_result.rename_count)
2502.1.1 by Aaron Bentley
Ensure renames only root children are renamed when building trees
1848
3136.1.2 by Aaron Bentley
Implement hard-linking for build_tree
1849
    def create_ab_tree(self):
1850
        """Create a committed test tree with two files"""
1851
        source = self.make_branch_and_tree('source')
1852
        self.build_tree_contents([('source/file1', 'A')])
1853
        self.build_tree_contents([('source/file2', 'B')])
1854
        source.add(['file1', 'file2'], ['file1-id', 'file2-id'])
1855
        source.commit('commit files')
1856
        source.lock_write()
1857
        self.addCleanup(source.unlock)
1858
        return source
1859
3123.5.1 by Aaron Bentley
Make build-tree able to use an additional 'accelerator' tree
1860
    def test_build_tree_accelerator_tree(self):
3136.1.2 by Aaron Bentley
Implement hard-linking for build_tree
1861
        source = self.create_ab_tree()
3123.5.1 by Aaron Bentley
Make build-tree able to use an additional 'accelerator' tree
1862
        self.build_tree_contents([('source/file2', 'C')])
1863
        calls = []
1864
        real_source_get_file = source.get_file
1865
        def get_file(file_id, path=None):
1866
            calls.append(file_id)
1867
            return real_source_get_file(file_id, path)
1868
        source.get_file = get_file
1869
        target = self.make_branch_and_tree('target')
3123.5.19 by Aaron Bentley
Ensure content is exactly the same, when accelerator used
1870
        revision_tree = source.basis_tree()
1871
        revision_tree.lock_read()
1872
        self.addCleanup(revision_tree.unlock)
1873
        build_tree(revision_tree, target, source)
3123.5.1 by Aaron Bentley
Make build-tree able to use an additional 'accelerator' tree
1874
        self.assertEqual(['file1-id'], calls)
3123.5.19 by Aaron Bentley
Ensure content is exactly the same, when accelerator used
1875
        target.lock_read()
1876
        self.addCleanup(target.unlock)
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1877
        self.assertEqual([], list(target.iter_changes(revision_tree)))
3123.5.1 by Aaron Bentley
Make build-tree able to use an additional 'accelerator' tree
1878
3123.5.4 by Aaron Bentley
Use an accelerator tree when branching, handle no-such-id correctly
1879
    def test_build_tree_accelerator_tree_missing_file(self):
3136.1.2 by Aaron Bentley
Implement hard-linking for build_tree
1880
        source = self.create_ab_tree()
3123.5.4 by Aaron Bentley
Use an accelerator tree when branching, handle no-such-id correctly
1881
        os.unlink('source/file1')
1882
        source.remove(['file2'])
1883
        target = self.make_branch_and_tree('target')
3123.5.19 by Aaron Bentley
Ensure content is exactly the same, when accelerator used
1884
        revision_tree = source.basis_tree()
1885
        revision_tree.lock_read()
1886
        self.addCleanup(revision_tree.unlock)
1887
        build_tree(revision_tree, target, source)
1888
        target.lock_read()
1889
        self.addCleanup(target.unlock)
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1890
        self.assertEqual([], list(target.iter_changes(revision_tree)))
3123.5.4 by Aaron Bentley
Use an accelerator tree when branching, handle no-such-id correctly
1891
3123.5.16 by Aaron Bentley
Test handling of conversion to non-file
1892
    def test_build_tree_accelerator_wrong_kind(self):
3146.4.8 by Aaron Bentley
Add missing symlink requirement
1893
        self.requireFeature(SymlinkFeature)
3123.5.16 by Aaron Bentley
Test handling of conversion to non-file
1894
        source = self.make_branch_and_tree('source')
1895
        self.build_tree_contents([('source/file1', '')])
1896
        self.build_tree_contents([('source/file2', '')])
1897
        source.add(['file1', 'file2'], ['file1-id', 'file2-id'])
1898
        source.commit('commit files')
1899
        os.unlink('source/file2')
1900
        self.build_tree_contents([('source/file2/', 'C')])
1901
        os.unlink('source/file1')
1902
        os.symlink('file2', 'source/file1')
1903
        calls = []
1904
        real_source_get_file = source.get_file
1905
        def get_file(file_id, path=None):
1906
            calls.append(file_id)
1907
            return real_source_get_file(file_id, path)
1908
        source.get_file = get_file
1909
        target = self.make_branch_and_tree('target')
3123.5.19 by Aaron Bentley
Ensure content is exactly the same, when accelerator used
1910
        revision_tree = source.basis_tree()
1911
        revision_tree.lock_read()
1912
        self.addCleanup(revision_tree.unlock)
1913
        build_tree(revision_tree, target, source)
3123.5.16 by Aaron Bentley
Test handling of conversion to non-file
1914
        self.assertEqual([], calls)
3123.5.19 by Aaron Bentley
Ensure content is exactly the same, when accelerator used
1915
        target.lock_read()
1916
        self.addCleanup(target.unlock)
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1917
        self.assertEqual([], list(target.iter_changes(revision_tree)))
3123.5.16 by Aaron Bentley
Test handling of conversion to non-file
1918
3136.1.2 by Aaron Bentley
Implement hard-linking for build_tree
1919
    def test_build_tree_hardlink(self):
1920
        self.requireFeature(HardlinkFeature)
1921
        source = self.create_ab_tree()
1922
        target = self.make_branch_and_tree('target')
1923
        revision_tree = source.basis_tree()
1924
        revision_tree.lock_read()
1925
        self.addCleanup(revision_tree.unlock)
1926
        build_tree(revision_tree, target, source, hardlink=True)
1927
        target.lock_read()
1928
        self.addCleanup(target.unlock)
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1929
        self.assertEqual([], list(target.iter_changes(revision_tree)))
3136.1.2 by Aaron Bentley
Implement hard-linking for build_tree
1930
        source_stat = os.stat('source/file1')
1931
        target_stat = os.stat('target/file1')
1932
        self.assertEqual(source_stat, target_stat)
1933
1934
        # Explicitly disallowing hardlinks should prevent them.
1935
        target2 = self.make_branch_and_tree('target2')
1936
        build_tree(revision_tree, target2, source, hardlink=False)
1937
        target2.lock_read()
1938
        self.addCleanup(target2.unlock)
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1939
        self.assertEqual([], list(target2.iter_changes(revision_tree)))
3136.1.2 by Aaron Bentley
Implement hard-linking for build_tree
1940
        source_stat = os.stat('source/file1')
1941
        target2_stat = os.stat('target2/file1')
1942
        self.assertNotEqual(source_stat, target2_stat)
1943
3137.1.1 by Aaron Bentley
Fix build_tree acceleration when file is moved in accelerator_tree
1944
    def test_build_tree_accelerator_tree_moved(self):
1945
        source = self.make_branch_and_tree('source')
1946
        self.build_tree_contents([('source/file1', 'A')])
1947
        source.add(['file1'], ['file1-id'])
1948
        source.commit('commit files')
1949
        source.rename_one('file1', 'file2')
1950
        source.lock_read()
1951
        self.addCleanup(source.unlock)
1952
        target = self.make_branch_and_tree('target')
1953
        revision_tree = source.basis_tree()
1954
        revision_tree.lock_read()
1955
        self.addCleanup(revision_tree.unlock)
1956
        build_tree(revision_tree, target, source)
1957
        target.lock_read()
1958
        self.addCleanup(target.unlock)
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1959
        self.assertEqual([], list(target.iter_changes(revision_tree)))
3137.1.1 by Aaron Bentley
Fix build_tree acceleration when file is moved in accelerator_tree
1960
3136.1.2 by Aaron Bentley
Implement hard-linking for build_tree
1961
    def test_build_tree_hardlinks_preserve_execute(self):
1962
        self.requireFeature(HardlinkFeature)
1963
        source = self.create_ab_tree()
1964
        tt = TreeTransform(source)
1965
        trans_id = tt.trans_id_tree_file_id('file1-id')
1966
        tt.set_executability(True, trans_id)
1967
        tt.apply()
1968
        self.assertTrue(source.is_executable('file1-id'))
1969
        target = self.make_branch_and_tree('target')
1970
        revision_tree = source.basis_tree()
1971
        revision_tree.lock_read()
1972
        self.addCleanup(revision_tree.unlock)
1973
        build_tree(revision_tree, target, source, hardlink=True)
1974
        target.lock_read()
1975
        self.addCleanup(target.unlock)
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
1976
        self.assertEqual([], list(target.iter_changes(revision_tree)))
3136.1.2 by Aaron Bentley
Implement hard-linking for build_tree
1977
        self.assertTrue(source.is_executable('file1-id'))
1978
4826.1.5 by Andrew Bennetts
Add test that content filtered files are not hardlinked by build_tree.
1979
    def install_rot13_content_filter(self, pattern):
4985.2.1 by Vincent Ladeuil
Deploy addAttrCleanup on the whole test suite.
1980
        # We could use
1981
        # self.addCleanup(filters._reset_registry, filters._reset_registry())
1982
        # below, but that looks a bit... hard to read even if it's exactly
1983
        # the same thing.
4826.1.5 by Andrew Bennetts
Add test that content filtered files are not hardlinked by build_tree.
1984
        original_registry = filters._reset_registry()
1985
        def restore_registry():
1986
            filters._reset_registry(original_registry)
1987
        self.addCleanup(restore_registry)
1988
        def rot13(chunks, context=None):
1989
            return [''.join(chunks).encode('rot13')]
1990
        rot13filter = filters.ContentFilter(rot13, rot13)
1991
        filters.register_filter_stack_map('rot13', {'yes': [rot13filter]}.get)
1992
        os.mkdir(self.test_home_dir + '/.bazaar')
1993
        rules_filename = self.test_home_dir + '/.bazaar/rules'
4826.1.8 by Andrew Bennetts
Tweaks suggested by John.
1994
        f = open(rules_filename, 'wb')
4826.1.5 by Andrew Bennetts
Add test that content filtered files are not hardlinked by build_tree.
1995
        f.write('[name %s]\nrot13=yes\n' % (pattern,))
1996
        f.close()
1997
        def uninstall_rules():
1998
            os.remove(rules_filename)
1999
            rules.reset_rules()
2000
        self.addCleanup(uninstall_rules)
2001
        rules.reset_rules()
2002
2003
    def test_build_tree_content_filtered_files_are_not_hardlinked(self):
2004
        """build_tree will not hardlink files that have content filtering rules
2005
        applied to them (but will still hardlink other files from the same tree
2006
        if it can).
2007
        """
2008
        self.requireFeature(HardlinkFeature)
2009
        self.install_rot13_content_filter('file1')
2010
        source = self.create_ab_tree()
2011
        target = self.make_branch_and_tree('target')
2012
        revision_tree = source.basis_tree()
2013
        revision_tree.lock_read()
2014
        self.addCleanup(revision_tree.unlock)
2015
        build_tree(revision_tree, target, source, hardlink=True)
2016
        target.lock_read()
2017
        self.addCleanup(target.unlock)
2018
        self.assertEqual([], list(target.iter_changes(revision_tree)))
2019
        source_stat = os.stat('source/file1')
2020
        target_stat = os.stat('target/file1')
2021
        self.assertNotEqual(source_stat, target_stat)
2022
        source_stat = os.stat('source/file2')
2023
        target_stat = os.stat('target/file2')
4826.1.8 by Andrew Bennetts
Tweaks suggested by John.
2024
        self.assertEqualStat(source_stat, target_stat)
4826.1.5 by Andrew Bennetts
Add test that content filtered files are not hardlinked by build_tree.
2025
3453.2.4 by Aaron Bentley
Disable fast-path when conflicts are encountered
2026
    def test_case_insensitive_build_tree_inventory(self):
4241.14.12 by Vincent Ladeuil
Far too many modifications for a single commit, need to restart.
2027
        if (tests.CaseInsensitiveFilesystemFeature.available()
2028
            or tests.CaseInsCasePresFilenameFeature.available()):
4241.9.4 by Vincent Ladeuil
Fix test_case_insensitive_build_tree_inventory failure on OSX.
2029
            raise tests.UnavailableFeature('Fully case sensitive filesystem')
3453.2.4 by Aaron Bentley
Disable fast-path when conflicts are encountered
2030
        source = self.make_branch_and_tree('source')
2031
        self.build_tree(['source/file', 'source/FILE'])
2032
        source.add(['file', 'FILE'], ['lower-id', 'upper-id'])
2033
        source.commit('added files')
2034
        # Don't try this at home, kids!
2035
        # Force the tree to report that it is case insensitive
2036
        target = self.make_branch_and_tree('target')
2037
        target.case_sensitive = False
3453.2.6 by Aaron Bentley
Rename mutate_tree to delta_from_tree, add comment
2038
        build_tree(source.basis_tree(), target, source, delta_from_tree=True)
3453.2.4 by Aaron Bentley
Disable fast-path when conflicts are encountered
2039
        self.assertEqual('file.moved', target.id2path('lower-id'))
2040
        self.assertEqual('FILE', target.id2path('upper-id'))
2041
1966.1.1 by Aaron Bentley
Implement disk-content merge and conflict resolution for build_tree
2042
4526.8.1 by Aaron Bentley
Support committing a TreeTransform to a branch.
2043
class TestCommitTransform(tests.TestCaseWithTransport):
2044
2045
    def get_branch(self):
2046
        tree = self.make_branch_and_tree('tree')
2047
        tree.lock_write()
2048
        self.addCleanup(tree.unlock)
2049
        tree.commit('empty commit')
2050
        return tree.branch
2051
2052
    def get_branch_and_transform(self):
2053
        branch = self.get_branch()
2054
        tt = TransformPreview(branch.basis_tree())
2055
        self.addCleanup(tt.finalize)
2056
        return branch, tt
2057
2058
    def test_commit_wrong_basis(self):
2059
        branch = self.get_branch()
2060
        basis = branch.repository.revision_tree(
2061
            _mod_revision.NULL_REVISION)
2062
        tt = TransformPreview(basis)
2063
        self.addCleanup(tt.finalize)
4526.8.5 by Aaron Bentley
Updates from review.
2064
        e = self.assertRaises(ValueError, tt.commit, branch, '')
4526.8.1 by Aaron Bentley
Support committing a TreeTransform to a branch.
2065
        self.assertEqual('TreeTransform not based on branch basis: null:',
2066
                         str(e))
2067
2068
    def test_empy_commit(self):
2069
        branch, tt = self.get_branch_and_transform()
2070
        rev = tt.commit(branch, 'my message')
2071
        self.assertEqual(2, branch.revno())
2072
        repo = branch.repository
2073
        self.assertEqual('my message', repo.get_revision(rev).message)
2074
2075
    def test_merge_parents(self):
2076
        branch, tt = self.get_branch_and_transform()
2077
        rev = tt.commit(branch, 'my message', ['rev1b', 'rev1c'])
2078
        self.assertEqual(['rev1b', 'rev1c'],
2079
                         branch.basis_tree().get_parent_ids()[1:])
2080
4526.8.3 by Aaron Bentley
Clean up merge parent handling.
2081
    def test_first_commit(self):
2082
        branch = self.make_branch('branch')
2083
        branch.lock_write()
2084
        self.addCleanup(branch.unlock)
2085
        tt = TransformPreview(branch.basis_tree())
4659.2.4 by Vincent Ladeuil
Cleanup remaining bzr-limbo-XXXXXX leaks in /tmp during selftest.
2086
        self.addCleanup(tt.finalize)
4526.9.23 by Robert Collins
Change the tree transform test_first_commit test to set a root id in the new tree, and workaround an apparent bug in TreeTransform._determine_path.
2087
        tt.new_directory('', ROOT_PARENT, 'TREE_ROOT')
4526.8.3 by Aaron Bentley
Clean up merge parent handling.
2088
        rev = tt.commit(branch, 'my message')
2089
        self.assertEqual([], branch.basis_tree().get_parent_ids())
4526.8.5 by Aaron Bentley
Updates from review.
2090
        self.assertNotEqual(_mod_revision.NULL_REVISION,
2091
                            branch.last_revision())
4526.8.3 by Aaron Bentley
Clean up merge parent handling.
2092
4526.8.5 by Aaron Bentley
Updates from review.
2093
    def test_first_commit_with_merge_parents(self):
4526.8.3 by Aaron Bentley
Clean up merge parent handling.
2094
        branch = self.make_branch('branch')
2095
        branch.lock_write()
2096
        self.addCleanup(branch.unlock)
2097
        tt = TransformPreview(branch.basis_tree())
4659.2.4 by Vincent Ladeuil
Cleanup remaining bzr-limbo-XXXXXX leaks in /tmp during selftest.
2098
        self.addCleanup(tt.finalize)
4526.8.5 by Aaron Bentley
Updates from review.
2099
        e = self.assertRaises(ValueError, tt.commit, branch,
4526.8.3 by Aaron Bentley
Clean up merge parent handling.
2100
                          'my message', ['rev1b-id'])
4526.8.5 by Aaron Bentley
Updates from review.
2101
        self.assertEqual('Cannot supply merge parents for first commit.',
2102
                         str(e))
2103
        self.assertEqual(_mod_revision.NULL_REVISION, branch.last_revision())
4526.8.3 by Aaron Bentley
Clean up merge parent handling.
2104
4526.8.1 by Aaron Bentley
Support committing a TreeTransform to a branch.
2105
    def test_add_files(self):
2106
        branch, tt = self.get_branch_and_transform()
2107
        tt.new_file('file', tt.root, 'contents', 'file-id')
2108
        trans_id = tt.new_directory('dir', tt.root, 'dir-id')
4789.25.2 by John Arbash Meinel
Fix 3 tests that meant to skip exec or symlink testing on win32
2109
        if SymlinkFeature.available():
2110
            tt.new_symlink('symlink', trans_id, 'target', 'symlink-id')
4526.8.1 by Aaron Bentley
Support committing a TreeTransform to a branch.
2111
        rev = tt.commit(branch, 'message')
2112
        tree = branch.basis_tree()
2113
        self.assertEqual('file', tree.id2path('file-id'))
2114
        self.assertEqual('contents', tree.get_file_text('file-id'))
2115
        self.assertEqual('dir', tree.id2path('dir-id'))
4789.25.2 by John Arbash Meinel
Fix 3 tests that meant to skip exec or symlink testing on win32
2116
        if SymlinkFeature.available():
2117
            self.assertEqual('dir/symlink', tree.id2path('symlink-id'))
2118
            self.assertEqual('target', tree.get_symlink_target('symlink-id'))
4526.8.1 by Aaron Bentley
Support committing a TreeTransform to a branch.
2119
4526.8.2 by Aaron Bentley
Proved strict commit handling.
2120
    def test_add_unversioned(self):
2121
        branch, tt = self.get_branch_and_transform()
2122
        tt.new_file('file', tt.root, 'contents')
2123
        self.assertRaises(errors.StrictCommitFailed, tt.commit, branch,
2124
                          'message', strict=True)
2125
2126
    def test_modify_strict(self):
2127
        branch, tt = self.get_branch_and_transform()
2128
        tt.new_file('file', tt.root, 'contents', 'file-id')
2129
        tt.commit(branch, 'message', strict=True)
2130
        tt = TransformPreview(branch.basis_tree())
4659.2.4 by Vincent Ladeuil
Cleanup remaining bzr-limbo-XXXXXX leaks in /tmp during selftest.
2131
        self.addCleanup(tt.finalize)
4526.8.2 by Aaron Bentley
Proved strict commit handling.
2132
        trans_id = tt.trans_id_file_id('file-id')
2133
        tt.delete_contents(trans_id)
2134
        tt.create_file('contents', trans_id)
2135
        tt.commit(branch, 'message', strict=True)
2136
4526.8.6 by Aaron Bentley
Check for malformed transforms before committing.
2137
    def test_commit_malformed(self):
2138
        """Committing a malformed transform should raise an exception.
2139
2140
        In this case, we are adding a file without adding its parent.
2141
        """
2142
        branch, tt = self.get_branch_and_transform()
2143
        parent_id = tt.trans_id_file_id('parent-id')
2144
        tt.new_file('file', parent_id, 'contents', 'file-id')
2145
        self.assertRaises(errors.MalformedTransform, tt.commit, branch,
2146
                          'message')
2147
5162.4.1 by Aaron Bentley
TreeTransform supports normal commit parameters and includes branch nick.
2148
    def test_commit_rich_revision_data(self):
2149
        branch, tt = self.get_branch_and_transform()
5162.4.3 by Aaron Bentley
Fix failing test.
2150
        rev_id = tt.commit(branch, 'message', timestamp=1, timezone=43201,
5162.4.1 by Aaron Bentley
TreeTransform supports normal commit parameters and includes branch nick.
2151
                           committer='me <me@example.com>',
2152
                           revprops={'foo': 'bar'}, revision_id='revid-1',
2153
                           authors=['Author1 <author1@example.com>',
2154
                              'Author2 <author2@example.com>',
2155
                               ])
2156
        self.assertEqual('revid-1', rev_id)
2157
        revision = branch.repository.get_revision(rev_id)
2158
        self.assertEqual(1, revision.timestamp)
5162.4.3 by Aaron Bentley
Fix failing test.
2159
        self.assertEqual(43201, revision.timezone)
5162.4.1 by Aaron Bentley
TreeTransform supports normal commit parameters and includes branch nick.
2160
        self.assertEqual('me <me@example.com>', revision.committer)
2161
        self.assertEqual(['Author1 <author1@example.com>',
2162
                          'Author2 <author2@example.com>'],
2163
                         revision.get_apparent_authors())
2164
        del revision.properties['authors']
2165
        self.assertEqual({'foo': 'bar',
2166
                          'branch-nick': 'tree'},
2167
                         revision.properties)
2168
2169
    def test_no_explicit_revprops(self):
2170
        branch, tt = self.get_branch_and_transform()
2171
        rev_id = tt.commit(branch, 'message', authors=[
2172
            'Author1 <author1@example.com>',
2173
            'Author2 <author2@example.com>', ])
2174
        revision = branch.repository.get_revision(rev_id)
2175
        self.assertEqual(['Author1 <author1@example.com>',
2176
                          'Author2 <author2@example.com>'],
2177
                         revision.get_apparent_authors())
2178
        self.assertEqual('tree', revision.properties['branch-nick'])
2179
4526.8.1 by Aaron Bentley
Support committing a TreeTransform to a branch.
2180
1534.10.28 by Aaron Bentley
Use numbered backup files
2181
class MockTransform(object):
2182
2183
    def has_named_child(self, by_parent, parent_id, name):
2184
        for child_id in by_parent[parent_id]:
2185
            if child_id == '0':
2186
                if name == "name~":
2187
                    return True
1534.10.29 by Aaron Bentley
Fixed backup numbering to match GNU standard better
2188
            elif name == "name.~%s~" % child_id:
1534.10.28 by Aaron Bentley
Use numbered backup files
2189
                return True
2190
        return False
2191
2502.1.1 by Aaron Bentley
Ensure renames only root children are renamed when building trees
2192
1534.10.28 by Aaron Bentley
Use numbered backup files
2193
class MockEntry(object):
2194
    def __init__(self):
2195
        object.__init__(self)
2196
        self.name = "name"
2197
3063.1.1 by Alexander Belchenko
Catch OSError 17 (file exists) in final phase of tree transform and show filename to user (#111758).
2198
1534.10.28 by Aaron Bentley
Use numbered backup files
2199
class TestGetBackupName(TestCase):
2200
    def test_get_backup_name(self):
2201
        tt = MockTransform()
2202
        name = get_backup_name(MockEntry(), {'a':[]}, 'a', tt)
1534.10.29 by Aaron Bentley
Fixed backup numbering to match GNU standard better
2203
        self.assertEqual(name, 'name.~1~')
2204
        name = get_backup_name(MockEntry(), {'a':['1']}, 'a', tt)
2205
        self.assertEqual(name, 'name.~2~')
1534.10.28 by Aaron Bentley
Use numbered backup files
2206
        name = get_backup_name(MockEntry(), {'a':['2']}, 'a', tt)
1534.10.29 by Aaron Bentley
Fixed backup numbering to match GNU standard better
2207
        self.assertEqual(name, 'name.~1~')
1534.10.28 by Aaron Bentley
Use numbered backup files
2208
        name = get_backup_name(MockEntry(), {'a':['2'], 'b':[]}, 'b', tt)
1534.10.29 by Aaron Bentley
Fixed backup numbering to match GNU standard better
2209
        self.assertEqual(name, 'name.~1~')
2210
        name = get_backup_name(MockEntry(), {'a':['1', '2', '3']}, 'a', tt)
2211
        self.assertEqual(name, 'name.~4~')
2733.2.1 by Aaron Bentley
Implement FileMover, to support TreeTransform rollback
2212
2213
2214
class TestFileMover(tests.TestCaseWithTransport):
2215
2216
    def test_file_mover(self):
2217
        self.build_tree(['a/', 'a/b', 'c/', 'c/d'])
2218
        mover = _FileMover()
2219
        mover.rename('a', 'q')
2220
        self.failUnlessExists('q')
2221
        self.failIfExists('a')
2733.2.12 by Aaron Bentley
Updates from review
2222
        self.failUnlessExists('q/b')
2223
        self.failUnlessExists('c')
2224
        self.failUnlessExists('c/d')
2733.2.1 by Aaron Bentley
Implement FileMover, to support TreeTransform rollback
2225
2733.2.5 by Aaron Bentley
Implement FileMover.pre_delete and FileMover.apply_deletions
2226
    def test_pre_delete_rollback(self):
2227
        self.build_tree(['a/'])
2228
        mover = _FileMover()
2229
        mover.pre_delete('a', 'q')
2230
        self.failUnlessExists('q')
2231
        self.failIfExists('a')
2232
        mover.rollback()
2233
        self.failIfExists('q')
2234
        self.failUnlessExists('a')
2235
2236
    def test_apply_deletions(self):
2733.2.12 by Aaron Bentley
Updates from review
2237
        self.build_tree(['a/', 'b/'])
2733.2.5 by Aaron Bentley
Implement FileMover.pre_delete and FileMover.apply_deletions
2238
        mover = _FileMover()
2239
        mover.pre_delete('a', 'q')
2733.2.12 by Aaron Bentley
Updates from review
2240
        mover.pre_delete('b', 'r')
2733.2.5 by Aaron Bentley
Implement FileMover.pre_delete and FileMover.apply_deletions
2241
        self.failUnlessExists('q')
2733.2.12 by Aaron Bentley
Updates from review
2242
        self.failUnlessExists('r')
2733.2.5 by Aaron Bentley
Implement FileMover.pre_delete and FileMover.apply_deletions
2243
        self.failIfExists('a')
2733.2.12 by Aaron Bentley
Updates from review
2244
        self.failIfExists('b')
2733.2.5 by Aaron Bentley
Implement FileMover.pre_delete and FileMover.apply_deletions
2245
        mover.apply_deletions()
2246
        self.failIfExists('q')
2733.2.12 by Aaron Bentley
Updates from review
2247
        self.failIfExists('r')
2733.2.5 by Aaron Bentley
Implement FileMover.pre_delete and FileMover.apply_deletions
2248
        self.failIfExists('a')
2733.2.12 by Aaron Bentley
Updates from review
2249
        self.failIfExists('b')
2733.2.5 by Aaron Bentley
Implement FileMover.pre_delete and FileMover.apply_deletions
2250
2733.2.1 by Aaron Bentley
Implement FileMover, to support TreeTransform rollback
2251
    def test_file_mover_rollback(self):
2252
        self.build_tree(['a/', 'a/b', 'c/', 'c/d/', 'c/e/'])
2253
        mover = _FileMover()
2254
        mover.rename('c/d', 'c/f')
2255
        mover.rename('c/e', 'c/d')
2256
        try:
2257
            mover.rename('a', 'c')
3063.1.3 by Aaron Bentley
Update for Linux
2258
        except errors.FileExists, e:
2733.2.1 by Aaron Bentley
Implement FileMover, to support TreeTransform rollback
2259
            mover.rollback()
2260
        self.failUnlessExists('a')
2261
        self.failUnlessExists('c/d')
2733.2.3 by Aaron Bentley
Test tranform rollback
2262
2263
2264
class Bogus(Exception):
2265
    pass
2266
2267
2268
class TestTransformRollback(tests.TestCaseWithTransport):
2269
2270
    class ExceptionFileMover(_FileMover):
2271
2733.2.4 by Aaron Bentley
Test transform rollback when renaming into place
2272
        def __init__(self, bad_source=None, bad_target=None):
2273
            _FileMover.__init__(self)
2274
            self.bad_source = bad_source
2275
            self.bad_target = bad_target
2276
2733.2.3 by Aaron Bentley
Test tranform rollback
2277
        def rename(self, source, target):
2733.2.4 by Aaron Bentley
Test transform rollback when renaming into place
2278
            if (self.bad_source is not None and
2279
                source.endswith(self.bad_source)):
2280
                raise Bogus
2281
            elif (self.bad_target is not None and
2282
                target.endswith(self.bad_target)):
2733.2.3 by Aaron Bentley
Test tranform rollback
2283
                raise Bogus
2284
            else:
2285
                _FileMover.rename(self, source, target)
2286
2287
    def test_rollback_rename(self):
2288
        tree = self.make_branch_and_tree('.')
2289
        self.build_tree(['a/', 'a/b'])
2290
        tt = TreeTransform(tree)
2291
        self.addCleanup(tt.finalize)
2292
        a_id = tt.trans_id_tree_path('a')
2293
        tt.adjust_path('c', tt.root, a_id)
2294
        tt.adjust_path('d', a_id, tt.trans_id_tree_path('a/b'))
2733.2.4 by Aaron Bentley
Test transform rollback when renaming into place
2295
        self.assertRaises(Bogus, tt.apply,
2296
                          _mover=self.ExceptionFileMover(bad_source='a'))
2297
        self.failUnlessExists('a')
2298
        self.failUnlessExists('a/b')
2299
        tt.apply()
2300
        self.failUnlessExists('c')
2301
        self.failUnlessExists('c/d')
2302
2303
    def test_rollback_rename_into_place(self):
2304
        tree = self.make_branch_and_tree('.')
2305
        self.build_tree(['a/', 'a/b'])
2306
        tt = TreeTransform(tree)
2307
        self.addCleanup(tt.finalize)
2308
        a_id = tt.trans_id_tree_path('a')
2309
        tt.adjust_path('c', tt.root, a_id)
2310
        tt.adjust_path('d', a_id, tt.trans_id_tree_path('a/b'))
2311
        self.assertRaises(Bogus, tt.apply,
2312
                          _mover=self.ExceptionFileMover(bad_target='c/d'))
2313
        self.failUnlessExists('a')
2314
        self.failUnlessExists('a/b')
2315
        tt.apply()
2316
        self.failUnlessExists('c')
2317
        self.failUnlessExists('c/d')
2733.2.6 by Aaron Bentley
Make TreeTransform commits rollbackable
2318
2319
    def test_rollback_deletion(self):
2320
        tree = self.make_branch_and_tree('.')
2321
        self.build_tree(['a/', 'a/b'])
2322
        tt = TreeTransform(tree)
2323
        self.addCleanup(tt.finalize)
2324
        a_id = tt.trans_id_tree_path('a')
2325
        tt.delete_contents(a_id)
2326
        tt.adjust_path('d', tt.root, tt.trans_id_tree_path('a/b'))
2327
        self.assertRaises(Bogus, tt.apply,
2328
                          _mover=self.ExceptionFileMover(bad_target='d'))
2329
        self.failUnlessExists('a')
2330
        self.failUnlessExists('a/b')
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
2331
1551.19.6 by Aaron Bentley
Revert doesn't crash restoring a file from a deleted directory
2332
    def test_resolve_no_parent(self):
2333
        wt = self.make_branch_and_tree('.')
2334
        tt = TreeTransform(wt)
2335
        self.addCleanup(tt.finalize)
2336
        parent = tt.trans_id_file_id('parent-id')
2337
        tt.new_file('file', parent, 'Contents')
2338
        resolve_conflicts(tt)
3008.1.13 by Michael Hudson
merge bzr.dev
2339
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
2340
3363.19.1 by Aaron Bentley
Make PreviewTree.iter_changes accept all options.
2341
A_ENTRY = ('a-id', ('a', 'a'), True, (True, True),
2342
                  ('TREE_ROOT', 'TREE_ROOT'), ('a', 'a'), ('file', 'file'),
2343
                  (False, False))
2344
ROOT_ENTRY = ('TREE_ROOT', ('', ''), False, (True, True), (None, None),
2345
              ('', ''), ('directory', 'directory'), (False, None))
2346
2347
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
2348
class TestTransformPreview(tests.TestCaseWithTransport):
2349
2350
    def create_tree(self):
2351
        tree = self.make_branch_and_tree('.')
2352
        self.build_tree_contents([('a', 'content 1')])
4600.3.1 by Robert Collins
Set tree root ID in tree transform tests that don't care about the root id.
2353
        tree.set_root_id('TREE_ROOT')
3008.2.1 by Aaron Bentley
Ensure conflict resolution works
2354
        tree.add('a', 'a-id')
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
2355
        tree.commit('rev1', rev_id='rev1')
2356
        return tree.branch.repository.revision_tree('rev1')
2357
3008.1.18 by Aaron Bentley
Get supported PreviewTree functionality under test
2358
    def get_empty_preview(self):
2359
        repository = self.make_repository('repo')
2360
        tree = repository.revision_tree(_mod_revision.NULL_REVISION)
3199.1.4 by Vincent Ladeuil
Fix 16 leaked tmp dirs. Probably indicates a lock handling problem with TransformPreview
2361
        preview = TransformPreview(tree)
3199.1.5 by Vincent Ladeuil
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.
2362
        self.addCleanup(preview.finalize)
3199.1.4 by Vincent Ladeuil
Fix 16 leaked tmp dirs. Probably indicates a lock handling problem with TransformPreview
2363
        return preview
2364
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
2365
    def test_transform_preview(self):
2366
        revision_tree = self.create_tree()
2367
        preview = TransformPreview(revision_tree)
3199.1.5 by Vincent Ladeuil
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.
2368
        self.addCleanup(preview.finalize)
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
2369
2370
    def test_transform_preview_tree(self):
2371
        revision_tree = self.create_tree()
2372
        preview = TransformPreview(revision_tree)
3199.1.5 by Vincent Ladeuil
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.
2373
        self.addCleanup(preview.finalize)
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
2374
        preview.get_preview_tree()
2375
3008.1.5 by Michael Hudson
a more precise test
2376
    def test_transform_new_file(self):
2377
        revision_tree = self.create_tree()
2378
        preview = TransformPreview(revision_tree)
3199.1.5 by Vincent Ladeuil
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.
2379
        self.addCleanup(preview.finalize)
3008.1.5 by Michael Hudson
a more precise test
2380
        preview.new_file('file2', preview.root, 'content B\n', 'file2-id')
2381
        preview_tree = preview.get_preview_tree()
2382
        self.assertEqual(preview_tree.kind('file2-id'), 'file')
2383
        self.assertEqual(
2384
            preview_tree.get_file('file2-id').read(), 'content B\n')
2385
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
2386
    def test_diff_preview_tree(self):
2387
        revision_tree = self.create_tree()
2388
        preview = TransformPreview(revision_tree)
3199.1.5 by Vincent Ladeuil
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.
2389
        self.addCleanup(preview.finalize)
3008.1.4 by Michael Hudson
Merge test enhancements
2390
        preview.new_file('file2', preview.root, 'content B\n', 'file2-id')
3008.1.1 by Aaron Bentley
Start work allowing previews of transforms
2391
        preview_tree = preview.get_preview_tree()
2392
        out = StringIO()
2393
        show_diff_trees(revision_tree, preview_tree, out)
3008.1.4 by Michael Hudson
Merge test enhancements
2394
        lines = out.getvalue().splitlines()
2395
        self.assertEqual(lines[0], "=== added file 'file2'")
2396
        # 3 lines of diff administrivia
2397
        self.assertEqual(lines[4], "+content B")
3008.2.1 by Aaron Bentley
Ensure conflict resolution works
2398
2399
    def test_transform_conflicts(self):
2400
        revision_tree = self.create_tree()
2401
        preview = TransformPreview(revision_tree)
3199.1.5 by Vincent Ladeuil
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.
2402
        self.addCleanup(preview.finalize)
3008.2.1 by Aaron Bentley
Ensure conflict resolution works
2403
        preview.new_file('a', preview.root, 'content 2')
2404
        resolve_conflicts(preview)
2405
        trans_id = preview.trans_id_file_id('a-id')
2406
        self.assertEqual('a.moved', preview.final_name(trans_id))
3008.1.17 by Aaron Bentley
Test unused parameters of preview_tree._iter_changes
2407
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2408
    def get_tree_and_preview_tree(self):
3008.1.17 by Aaron Bentley
Test unused parameters of preview_tree._iter_changes
2409
        revision_tree = self.create_tree()
2410
        preview = TransformPreview(revision_tree)
3199.1.5 by Vincent Ladeuil
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.
2411
        self.addCleanup(preview.finalize)
3008.1.17 by Aaron Bentley
Test unused parameters of preview_tree._iter_changes
2412
        a_trans_id = preview.trans_id_file_id('a-id')
2413
        preview.delete_contents(a_trans_id)
2414
        preview.create_file('b content', a_trans_id)
2415
        preview_tree = preview.get_preview_tree()
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2416
        return revision_tree, preview_tree
2417
2418
    def test_iter_changes(self):
2419
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
2420
        root = revision_tree.inventory.root.file_id
3008.1.17 by Aaron Bentley
Test unused parameters of preview_tree._iter_changes
2421
        self.assertEqual([('a-id', ('a', 'a'), True, (True, True),
2422
                          (root, root), ('a', 'a'), ('file', 'file'),
2423
                          (False, False))],
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
2424
                          list(preview_tree.iter_changes(revision_tree)))
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2425
3363.19.1 by Aaron Bentley
Make PreviewTree.iter_changes accept all options.
2426
    def test_include_unchanged_succeeds(self):
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2427
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
3363.19.1 by Aaron Bentley
Make PreviewTree.iter_changes accept all options.
2428
        changes = preview_tree.iter_changes(revision_tree,
2429
                                            include_unchanged=True)
2430
        root = revision_tree.inventory.root.file_id
2431
2432
        self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2433
2434
    def test_specific_files(self):
2435
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
3363.19.1 by Aaron Bentley
Make PreviewTree.iter_changes accept all options.
2436
        changes = preview_tree.iter_changes(revision_tree,
2437
                                            specific_files=[''])
2438
        self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2439
3363.19.1 by Aaron Bentley
Make PreviewTree.iter_changes accept all options.
2440
    def test_want_unversioned(self):
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2441
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
3363.19.1 by Aaron Bentley
Make PreviewTree.iter_changes accept all options.
2442
        changes = preview_tree.iter_changes(revision_tree,
2443
                                            want_unversioned=True)
2444
        self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
3008.1.17 by Aaron Bentley
Test unused parameters of preview_tree._iter_changes
2445
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2446
    def test_ignore_extra_trees_no_specific_files(self):
3008.1.17 by Aaron Bentley
Test unused parameters of preview_tree._iter_changes
2447
        # extra_trees is harmless without specific_files, so we'll silently
2448
        # accept it, even though we won't use it.
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2449
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
2450
        preview_tree.iter_changes(revision_tree, extra_trees=[preview_tree])
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2451
2452
    def test_ignore_require_versioned_no_specific_files(self):
3008.1.17 by Aaron Bentley
Test unused parameters of preview_tree._iter_changes
2453
        # require_versioned is meaningless without specific_files.
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2454
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
3254.1.1 by Aaron Bentley
Make Tree.iter_changes a public method
2455
        preview_tree.iter_changes(revision_tree, require_versioned=False)
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2456
2457
    def test_ignore_pb(self):
3008.1.17 by Aaron Bentley
Test unused parameters of preview_tree._iter_changes
2458
        # pb could be supported, but TT.iter_changes doesn't support it.
3008.1.31 by Aaron Bentley
Split PreviewTree._iter_changes parameter tests into smaller tests
2459
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
4961.2.9 by Martin Pool
Rip out most remaining uses of DummyProgressBar
2460
        preview_tree.iter_changes(revision_tree)
3008.1.18 by Aaron Bentley
Get supported PreviewTree functionality under test
2461
2462
    def test_kind(self):
2463
        revision_tree = self.create_tree()
2464
        preview = TransformPreview(revision_tree)
3199.1.5 by Vincent Ladeuil
Fix two more leaking tmp dirs, by reworking TransformPreview lock handling.
2465
        self.addCleanup(preview.finalize)
3008.1.18 by Aaron Bentley
Get supported PreviewTree functionality under test
2466
        preview.new_file('file', preview.root, 'contents', 'file-id')
2467
        preview.new_directory('directory', preview.root, 'dir-id')
2468
        preview_tree = preview.get_preview_tree()
2469
        self.assertEqual('file', preview_tree.kind('file-id'))
2470
        self.assertEqual('directory', preview_tree.kind('dir-id'))
2471
2472
    def test_get_file_mtime(self):
2473
        preview = self.get_empty_preview()
2474
        file_trans_id = preview.new_file('file', preview.root, 'contents',
2475
                                         'file-id')
2476
        limbo_path = preview._limbo_name(file_trans_id)
2477
        preview_tree = preview.get_preview_tree()
2478
        self.assertEqual(os.stat(limbo_path).st_mtime,
2479
                         preview_tree.get_file_mtime('file-id'))
2480
4635.1.1 by Aaron Bentley
Fix OSError with renamed files in PreviewTree.
2481
    def test_get_file_mtime_renamed(self):
2482
        work_tree = self.make_branch_and_tree('tree')
2483
        self.build_tree(['tree/file'])
2484
        work_tree.add('file', 'file-id')
2485
        preview = TransformPreview(work_tree)
2486
        self.addCleanup(preview.finalize)
2487
        file_trans_id = preview.trans_id_tree_file_id('file-id')
2488
        preview.adjust_path('renamed', preview.root, file_trans_id)
2489
        preview_tree = preview.get_preview_tree()
2490
        preview_mtime = preview_tree.get_file_mtime('file-id', 'renamed')
2491
        work_mtime = work_tree.get_file_mtime('file-id', 'file')
2492
3008.1.18 by Aaron Bentley
Get supported PreviewTree functionality under test
2493
    def test_get_file(self):
2494
        preview = self.get_empty_preview()
2495
        preview.new_file('file', preview.root, 'contents', 'file-id')
2496
        preview_tree = preview.get_preview_tree()
2497
        tree_file = preview_tree.get_file('file-id')
2498
        try:
2499
            self.assertEqual('contents', tree_file.read())
2500
        finally:
2501
            tree_file.close()
3228.1.2 by James Henstridge
Simplify test, and move it down to be next to the other _PreviewTree tests.
2502
2503
    def test_get_symlink_target(self):
2504
        self.requireFeature(SymlinkFeature)
2505
        preview = self.get_empty_preview()
2506
        preview.new_symlink('symlink', preview.root, 'target', 'symlink-id')
2507
        preview_tree = preview.get_preview_tree()
2508
        self.assertEqual('target',
2509
                         preview_tree.get_symlink_target('symlink-id'))
3363.2.18 by Aaron Bentley
Implement correct all_file_ids for PreviewTree
2510
2511
    def test_all_file_ids(self):
2512
        tree = self.make_branch_and_tree('tree')
2513
        self.build_tree(['tree/a', 'tree/b', 'tree/c'])
2514
        tree.add(['a', 'b', 'c'], ['a-id', 'b-id', 'c-id'])
2515
        preview = TransformPreview(tree)
2516
        self.addCleanup(preview.finalize)
2517
        preview.unversion_file(preview.trans_id_file_id('b-id'))
2518
        c_trans_id = preview.trans_id_file_id('c-id')
2519
        preview.unversion_file(c_trans_id)
2520
        preview.version_file('c-id', c_trans_id)
2521
        preview_tree = preview.get_preview_tree()
2522
        self.assertEqual(set(['a-id', 'c-id', tree.get_root_id()]),
2523
                         preview_tree.all_file_ids())
3363.2.19 by Aaron Bentley
Make PreviewTree.path2id correct
2524
2525
    def test_path2id_deleted_unchanged(self):
2526
        tree = self.make_branch_and_tree('tree')
2527
        self.build_tree(['tree/unchanged', 'tree/deleted'])
2528
        tree.add(['unchanged', 'deleted'], ['unchanged-id', 'deleted-id'])
2529
        preview = TransformPreview(tree)
2530
        self.addCleanup(preview.finalize)
2531
        preview.unversion_file(preview.trans_id_file_id('deleted-id'))
2532
        preview_tree = preview.get_preview_tree()
2533
        self.assertEqual('unchanged-id', preview_tree.path2id('unchanged'))
2534
        self.assertIs(None, preview_tree.path2id('deleted'))
2535
2536
    def test_path2id_created(self):
2537
        tree = self.make_branch_and_tree('tree')
2538
        self.build_tree(['tree/unchanged'])
2539
        tree.add(['unchanged'], ['unchanged-id'])
2540
        preview = TransformPreview(tree)
2541
        self.addCleanup(preview.finalize)
2542
        preview.new_file('new', preview.trans_id_file_id('unchanged-id'),
2543
            'contents', 'new-id')
2544
        preview_tree = preview.get_preview_tree()
2545
        self.assertEqual('new-id', preview_tree.path2id('unchanged/new'))
2546
2547
    def test_path2id_moved(self):
2548
        tree = self.make_branch_and_tree('tree')
2549
        self.build_tree(['tree/old_parent/', 'tree/old_parent/child'])
2550
        tree.add(['old_parent', 'old_parent/child'],
2551
                 ['old_parent-id', 'child-id'])
2552
        preview = TransformPreview(tree)
2553
        self.addCleanup(preview.finalize)
2554
        new_parent = preview.new_directory('new_parent', preview.root,
2555
                                           'new_parent-id')
2556
        preview.adjust_path('child', new_parent,
2557
                            preview.trans_id_file_id('child-id'))
2558
        preview_tree = preview.get_preview_tree()
2559
        self.assertIs(None, preview_tree.path2id('old_parent/child'))
2560
        self.assertEqual('child-id', preview_tree.path2id('new_parent/child'))
2561
2562
    def test_path2id_renamed_parent(self):
2563
        tree = self.make_branch_and_tree('tree')
2564
        self.build_tree(['tree/old_name/', 'tree/old_name/child'])
2565
        tree.add(['old_name', 'old_name/child'],
2566
                 ['parent-id', 'child-id'])
2567
        preview = TransformPreview(tree)
2568
        self.addCleanup(preview.finalize)
2569
        preview.adjust_path('new_name', preview.root,
2570
                            preview.trans_id_file_id('parent-id'))
2571
        preview_tree = preview.get_preview_tree()
2572
        self.assertIs(None, preview_tree.path2id('old_name/child'))
2573
        self.assertEqual('child-id', preview_tree.path2id('new_name/child'))
3363.2.21 by Aaron Bentley
Implement iter_entries_by_dir
2574
2575
    def assertMatchingIterEntries(self, tt, specific_file_ids=None):
2576
        preview_tree = tt.get_preview_tree()
2577
        preview_result = list(preview_tree.iter_entries_by_dir(
2578
                              specific_file_ids))
2579
        tree = tt._tree
2580
        tt.apply()
2581
        actual_result = list(tree.iter_entries_by_dir(specific_file_ids))
2582
        self.assertEqual(actual_result, preview_result)
2583
2584
    def test_iter_entries_by_dir_new(self):
2585
        tree = self.make_branch_and_tree('tree')
2586
        tt = TreeTransform(tree)
2587
        tt.new_file('new', tt.root, 'contents', 'new-id')
2588
        self.assertMatchingIterEntries(tt)
2589
2590
    def test_iter_entries_by_dir_deleted(self):
2591
        tree = self.make_branch_and_tree('tree')
2592
        self.build_tree(['tree/deleted'])
2593
        tree.add('deleted', 'deleted-id')
2594
        tt = TreeTransform(tree)
2595
        tt.delete_contents(tt.trans_id_file_id('deleted-id'))
2596
        self.assertMatchingIterEntries(tt)
2597
2598
    def test_iter_entries_by_dir_unversioned(self):
2599
        tree = self.make_branch_and_tree('tree')
2600
        self.build_tree(['tree/removed'])
2601
        tree.add('removed', 'removed-id')
2602
        tt = TreeTransform(tree)
2603
        tt.unversion_file(tt.trans_id_file_id('removed-id'))
2604
        self.assertMatchingIterEntries(tt)
2605
2606
    def test_iter_entries_by_dir_moved(self):
2607
        tree = self.make_branch_and_tree('tree')
2608
        self.build_tree(['tree/moved', 'tree/new_parent/'])
2609
        tree.add(['moved', 'new_parent'], ['moved-id', 'new_parent-id'])
2610
        tt = TreeTransform(tree)
2611
        tt.adjust_path('moved', tt.trans_id_file_id('new_parent-id'),
2612
                       tt.trans_id_file_id('moved-id'))
2613
        self.assertMatchingIterEntries(tt)
2614
2615
    def test_iter_entries_by_dir_specific_file_ids(self):
2616
        tree = self.make_branch_and_tree('tree')
2617
        tree.set_root_id('tree-root-id')
2618
        self.build_tree(['tree/parent/', 'tree/parent/child'])
2619
        tree.add(['parent', 'parent/child'], ['parent-id', 'child-id'])
2620
        tt = TreeTransform(tree)
2621
        self.assertMatchingIterEntries(tt, ['tree-root-id', 'child-id'])
3363.2.26 by Aaron Bentley
Get symlinks working
2622
2623
    def test_symlink_content_summary(self):
2624
        self.requireFeature(SymlinkFeature)
2625
        preview = self.get_empty_preview()
2626
        preview.new_symlink('path', preview.root, 'target', 'path-id')
2627
        summary = preview.get_preview_tree().path_content_summary('path')
2628
        self.assertEqual(('symlink', None, None, 'target'), summary)
3363.2.27 by Aaron Bentley
Make path_content_summary a core API
2629
2630
    def test_missing_content_summary(self):
2631
        preview = self.get_empty_preview()
2632
        summary = preview.get_preview_tree().path_content_summary('path')
2633
        self.assertEqual(('missing', None, None, None), summary)
2634
2635
    def test_deleted_content_summary(self):
2636
        tree = self.make_branch_and_tree('tree')
2637
        self.build_tree(['tree/path/'])
2638
        tree.add('path')
2639
        preview = TransformPreview(tree)
2640
        self.addCleanup(preview.finalize)
2641
        preview.delete_contents(preview.trans_id_tree_path('path'))
2642
        summary = preview.get_preview_tree().path_content_summary('path')
2643
        self.assertEqual(('missing', None, None, None), summary)
2644
3363.2.30 by Aaron Bentley
Improve execute bit testing
2645
    def test_file_content_summary_executable(self):
2646
        preview = self.get_empty_preview()
2647
        path_id = preview.new_file('path', preview.root, 'contents', 'path-id')
2648
        preview.set_executability(True, path_id)
2649
        summary = preview.get_preview_tree().path_content_summary('path')
2650
        self.assertEqual(4, len(summary))
2651
        self.assertEqual('file', summary[0])
2652
        # size must be known
2653
        self.assertEqual(len('contents'), summary[1])
2654
        # executable
2655
        self.assertEqual(True, summary[2])
3363.2.31 by Aaron Bentley
Tweak tests
2656
        # will not have hash (not cheap to determine)
2657
        self.assertIs(None, summary[3])
3363.2.30 by Aaron Bentley
Improve execute bit testing
2658
2659
    def test_change_executability(self):
2660
        tree = self.make_branch_and_tree('tree')
2661
        self.build_tree(['tree/path'])
2662
        tree.add('path')
2663
        preview = TransformPreview(tree)
2664
        self.addCleanup(preview.finalize)
2665
        path_id = preview.trans_id_tree_path('path')
2666
        preview.set_executability(True, path_id)
2667
        summary = preview.get_preview_tree().path_content_summary('path')
2668
        self.assertEqual(True, summary[2])
2669
3363.2.27 by Aaron Bentley
Make path_content_summary a core API
2670
    def test_file_content_summary_non_exec(self):
2671
        preview = self.get_empty_preview()
2672
        preview.new_file('path', preview.root, 'contents', 'path-id')
2673
        summary = preview.get_preview_tree().path_content_summary('path')
2674
        self.assertEqual(4, len(summary))
2675
        self.assertEqual('file', summary[0])
2676
        # size must be known
3363.2.30 by Aaron Bentley
Improve execute bit testing
2677
        self.assertEqual(len('contents'), summary[1])
3363.2.27 by Aaron Bentley
Make path_content_summary a core API
2678
        # not executable
4789.15.3 by John Arbash Meinel
PreviewTree now returns False for exec bit properly.
2679
        self.assertEqual(False, summary[2])
3363.2.31 by Aaron Bentley
Tweak tests
2680
        # will not have hash (not cheap to determine)
2681
        self.assertIs(None, summary[3])
3363.2.27 by Aaron Bentley
Make path_content_summary a core API
2682
2683
    def test_dir_content_summary(self):
2684
        preview = self.get_empty_preview()
2685
        preview.new_directory('path', preview.root, 'path-id')
2686
        summary = preview.get_preview_tree().path_content_summary('path')
2687
        self.assertEqual(('directory', None, None, None), summary)
2688
2689
    def test_tree_content_summary(self):
2690
        preview = self.get_empty_preview()
2691
        path = preview.new_directory('path', preview.root, 'path-id')
2692
        preview.set_tree_reference('rev-1', path)
2693
        summary = preview.get_preview_tree().path_content_summary('path')
2694
        self.assertEqual(4, len(summary))
2695
        self.assertEqual('tree-reference', summary[0])
3363.2.33 by Aaron Bentley
Implement PreviewTree.annotate_iter
2696
2697
    def test_annotate(self):
2698
        tree = self.make_branch_and_tree('tree')
2699
        self.build_tree_contents([('tree/file', 'a\n')])
2700
        tree.add('file', 'file-id')
2701
        tree.commit('a', rev_id='one')
2702
        self.build_tree_contents([('tree/file', 'a\nb\n')])
2703
        preview = TransformPreview(tree)
2704
        self.addCleanup(preview.finalize)
2705
        file_trans_id = preview.trans_id_file_id('file-id')
2706
        preview.delete_contents(file_trans_id)
2707
        preview.create_file('a\nb\nc\n', file_trans_id)
2708
        preview_tree = preview.get_preview_tree()
2709
        expected = [
2710
            ('one', 'a\n'),
2711
            ('me:', 'b\n'),
2712
            ('me:', 'c\n'),
2713
        ]
2714
        annotation = preview_tree.annotate_iter('file-id', 'me:')
2715
        self.assertEqual(expected, annotation)
2716
2717
    def test_annotate_missing(self):
2718
        preview = self.get_empty_preview()
2719
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
2720
        preview_tree = preview.get_preview_tree()
2721
        expected = [
2722
            ('me:', 'a\n'),
2723
            ('me:', 'b\n'),
2724
            ('me:', 'c\n'),
2725
         ]
2726
        annotation = preview_tree.annotate_iter('file-id', 'me:')
2727
        self.assertEqual(expected, annotation)
2728
3363.7.3 by Aaron Bentley
Add test that annotate correctly handles renames
2729
    def test_annotate_rename(self):
2730
        tree = self.make_branch_and_tree('tree')
2731
        self.build_tree_contents([('tree/file', 'a\n')])
2732
        tree.add('file', 'file-id')
2733
        tree.commit('a', rev_id='one')
2734
        preview = TransformPreview(tree)
2735
        self.addCleanup(preview.finalize)
2736
        file_trans_id = preview.trans_id_file_id('file-id')
2737
        preview.adjust_path('newname', preview.root, file_trans_id)
2738
        preview_tree = preview.get_preview_tree()
2739
        expected = [
2740
            ('one', 'a\n'),
2741
        ]
2742
        annotation = preview_tree.annotate_iter('file-id', 'me:')
2743
        self.assertEqual(expected, annotation)
2744
3363.2.33 by Aaron Bentley
Implement PreviewTree.annotate_iter
2745
    def test_annotate_deleted(self):
2746
        tree = self.make_branch_and_tree('tree')
2747
        self.build_tree_contents([('tree/file', 'a\n')])
2748
        tree.add('file', 'file-id')
2749
        tree.commit('a', rev_id='one')
2750
        self.build_tree_contents([('tree/file', 'a\nb\n')])
2751
        preview = TransformPreview(tree)
2752
        self.addCleanup(preview.finalize)
2753
        file_trans_id = preview.trans_id_file_id('file-id')
2754
        preview.delete_contents(file_trans_id)
2755
        preview_tree = preview.get_preview_tree()
2756
        annotation = preview_tree.annotate_iter('file-id', 'me:')
2757
        self.assertIs(None, annotation)
2758
3363.2.36 by Aaron Bentley
Fix PreviewTree.stored_kind
2759
    def test_stored_kind(self):
2760
        preview = self.get_empty_preview()
2761
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
2762
        preview_tree = preview.get_preview_tree()
2763
        self.assertEqual('file', preview_tree.stored_kind('file-id'))
3363.2.37 by Aaron Bentley
Fix is_executable
2764
2765
    def test_is_executable(self):
2766
        preview = self.get_empty_preview()
2767
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
2768
        preview.set_executability(True, preview.trans_id_file_id('file-id'))
2769
        preview_tree = preview.get_preview_tree()
2770
        self.assertEqual(True, preview_tree.is_executable('file-id'))
3363.9.1 by Aaron Bentley
Implement plan_merge, refactoring various bits
2771
3571.1.1 by Aaron Bentley
Allow set/get of parent_ids in PreviewTree
2772
    def test_get_set_parent_ids(self):
2773
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
2774
        self.assertEqual([], preview_tree.get_parent_ids())
2775
        preview_tree.set_parent_ids(['rev-1'])
2776
        self.assertEqual(['rev-1'], preview_tree.get_parent_ids())
2777
3363.9.1 by Aaron Bentley
Implement plan_merge, refactoring various bits
2778
    def test_plan_file_merge(self):
2779
        work_a = self.make_branch_and_tree('wta')
2780
        self.build_tree_contents([('wta/file', 'a\nb\nc\nd\n')])
2781
        work_a.add('file', 'file-id')
3363.9.7 by Aaron Bentley
Fix up to use set_parent_ids
2782
        base_id = work_a.commit('base version')
3363.9.1 by Aaron Bentley
Implement plan_merge, refactoring various bits
2783
        tree_b = work_a.bzrdir.sprout('wtb').open_workingtree()
2784
        preview = TransformPreview(work_a)
2785
        self.addCleanup(preview.finalize)
2786
        trans_id = preview.trans_id_file_id('file-id')
2787
        preview.delete_contents(trans_id)
2788
        preview.create_file('b\nc\nd\ne\n', trans_id)
2789
        self.build_tree_contents([('wtb/file', 'a\nc\nd\nf\n')])
2790
        tree_a = preview.get_preview_tree()
3363.9.7 by Aaron Bentley
Fix up to use set_parent_ids
2791
        tree_a.set_parent_ids([base_id])
3363.9.1 by Aaron Bentley
Implement plan_merge, refactoring various bits
2792
        self.assertEqual([
3363.9.5 by Aaron Bentley
Move killed-a from top to bottom
2793
            ('killed-a', 'a\n'),
3363.9.1 by Aaron Bentley
Implement plan_merge, refactoring various bits
2794
            ('killed-b', 'b\n'),
2795
            ('unchanged', 'c\n'),
2796
            ('unchanged', 'd\n'),
2797
            ('new-a', 'e\n'),
2798
            ('new-b', 'f\n'),
2799
        ], list(tree_a.plan_file_merge('file-id', tree_b)))
3363.9.8 by Aaron Bentley
Ensure plan_file_merge works with a RevisionTree as the basis
2800
2801
    def test_plan_file_merge_revision_tree(self):
2802
        work_a = self.make_branch_and_tree('wta')
2803
        self.build_tree_contents([('wta/file', 'a\nb\nc\nd\n')])
2804
        work_a.add('file', 'file-id')
2805
        base_id = work_a.commit('base version')
2806
        tree_b = work_a.bzrdir.sprout('wtb').open_workingtree()
2807
        preview = TransformPreview(work_a.basis_tree())
2808
        self.addCleanup(preview.finalize)
2809
        trans_id = preview.trans_id_file_id('file-id')
2810
        preview.delete_contents(trans_id)
2811
        preview.create_file('b\nc\nd\ne\n', trans_id)
2812
        self.build_tree_contents([('wtb/file', 'a\nc\nd\nf\n')])
2813
        tree_a = preview.get_preview_tree()
2814
        tree_a.set_parent_ids([base_id])
2815
        self.assertEqual([
2816
            ('killed-a', 'a\n'),
2817
            ('killed-b', 'b\n'),
2818
            ('unchanged', 'c\n'),
2819
            ('unchanged', 'd\n'),
2820
            ('new-a', 'e\n'),
2821
            ('new-b', 'f\n'),
2822
        ], list(tree_a.plan_file_merge('file-id', tree_b)))
3363.9.9 by Aaron Bentley
Implement walkdirs in terms of TreeTransform
2823
2824
    def test_walkdirs(self):
2825
        preview = self.get_empty_preview()
4634.57.3 by Aaron Bentley
Fix failing test.
2826
        root = preview.new_directory('', ROOT_PARENT, 'tree-root')
2827
        # FIXME: new_directory should mark root.
4634.122.6 by John Arbash Meinel
Fix a test that used 'adjust_path' to set the root.
2828
        preview.fixup_new_roots()
3363.9.9 by Aaron Bentley
Implement walkdirs in terms of TreeTransform
2829
        preview_tree = preview.get_preview_tree()
2830
        file_trans_id = preview.new_file('a', preview.root, 'contents',
2831
                                         'a-id')
2832
        expected = [(('', 'tree-root'),
2833
                    [('a', 'a', 'file', None, 'a-id', 'file')])]
2834
        self.assertEqual(expected, list(preview_tree.walkdirs()))
3363.13.2 by Aaron Bentley
Test specific cases for PreviewTree.extras
2835
2836
    def test_extras(self):
2837
        work_tree = self.make_branch_and_tree('tree')
2838
        self.build_tree(['tree/removed-file', 'tree/existing-file',
2839
                         'tree/not-removed-file'])
2840
        work_tree.add(['removed-file', 'not-removed-file'])
2841
        preview = TransformPreview(work_tree)
3363.13.3 by Aaron Bentley
Add cleanup
2842
        self.addCleanup(preview.finalize)
3363.13.2 by Aaron Bentley
Test specific cases for PreviewTree.extras
2843
        preview.new_file('new-file', preview.root, 'contents')
2844
        preview.new_file('new-versioned-file', preview.root, 'contents',
2845
                         'new-versioned-id')
2846
        tree = preview.get_preview_tree()
2847
        preview.unversion_file(preview.trans_id_tree_path('removed-file'))
2848
        self.assertEqual(set(['new-file', 'removed-file', 'existing-file']),
2849
                         set(tree.extras()))
3363.17.1 by Aaron Bentley
Avoid inventory for merge and transform code
2850
3363.17.2 by Aaron Bentley
Add text checking
2851
    def test_merge_into_preview(self):
3363.17.1 by Aaron Bentley
Avoid inventory for merge and transform code
2852
        work_tree = self.make_branch_and_tree('tree')
3363.17.2 by Aaron Bentley
Add text checking
2853
        self.build_tree_contents([('tree/file','b\n')])
2854
        work_tree.add('file', 'file-id')
3363.17.1 by Aaron Bentley
Avoid inventory for merge and transform code
2855
        work_tree.commit('first commit')
2856
        child_tree = work_tree.bzrdir.sprout('child').open_workingtree()
3363.17.2 by Aaron Bentley
Add text checking
2857
        self.build_tree_contents([('child/file','b\nc\n')])
3363.17.1 by Aaron Bentley
Avoid inventory for merge and transform code
2858
        child_tree.commit('child commit')
2859
        child_tree.lock_write()
2860
        self.addCleanup(child_tree.unlock)
2861
        work_tree.lock_write()
2862
        self.addCleanup(work_tree.unlock)
2863
        preview = TransformPreview(work_tree)
2864
        self.addCleanup(preview.finalize)
3363.17.6 by Aaron Bentley
Improve test scenario
2865
        file_trans_id = preview.trans_id_file_id('file-id')
2866
        preview.delete_contents(file_trans_id)
2867
        preview.create_file('a\nb\n', file_trans_id)
4634.57.2 by Aaron Bentley
Fix failing test.
2868
        preview_tree = preview.get_preview_tree()
4961.2.9 by Martin Pool
Rip out most remaining uses of DummyProgressBar
2869
        merger = Merger.from_revision_ids(None, preview_tree,
3363.17.1 by Aaron Bentley
Avoid inventory for merge and transform code
2870
                                          child_tree.branch.last_revision(),
2871
                                          other_branch=child_tree.branch,
2872
                                          tree_branch=work_tree.branch)
2873
        merger.merge_type = Merge3Merger
2874
        tt = merger.make_merger().make_preview_transform()
3363.17.2 by Aaron Bentley
Add text checking
2875
        self.addCleanup(tt.finalize)
2876
        final_tree = tt.get_preview_tree()
2877
        self.assertEqual('a\nb\nc\n', final_tree.get_file_text('file-id'))
3363.17.17 by Aaron Bentley
Start testing merging PreviewTree as OTHER
2878
2879
    def test_merge_preview_into_workingtree(self):
2880
        tree = self.make_branch_and_tree('tree')
4600.3.1 by Robert Collins
Set tree root ID in tree transform tests that don't care about the root id.
2881
        tree.set_root_id('TREE_ROOT')
3363.17.17 by Aaron Bentley
Start testing merging PreviewTree as OTHER
2882
        tt = TransformPreview(tree)
2883
        self.addCleanup(tt.finalize)
2884
        tt.new_file('name', tt.root, 'content', 'file-id')
2885
        tree2 = self.make_branch_and_tree('tree2')
4600.3.1 by Robert Collins
Set tree root ID in tree transform tests that don't care about the root id.
2886
        tree2.set_root_id('TREE_ROOT')
3363.17.17 by Aaron Bentley
Start testing merging PreviewTree as OTHER
2887
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
4961.2.9 by Martin Pool
Rip out most remaining uses of DummyProgressBar
2888
                                         None, tree.basis_tree())
3363.17.17 by Aaron Bentley
Start testing merging PreviewTree as OTHER
2889
        merger.merge_type = Merge3Merger
2890
        merger.do_merge()
3363.17.18 by Aaron Bentley
Fix is_executable for PreviewTree
2891
3363.17.21 by Aaron Bentley
Conflicts are handled when merging from preview trees
2892
    def test_merge_preview_into_workingtree_handles_conflicts(self):
2893
        tree = self.make_branch_and_tree('tree')
2894
        self.build_tree_contents([('tree/foo', 'bar')])
2895
        tree.add('foo', 'foo-id')
2896
        tree.commit('foo')
2897
        tt = TransformPreview(tree)
2898
        self.addCleanup(tt.finalize)
2899
        trans_id = tt.trans_id_file_id('foo-id')
2900
        tt.delete_contents(trans_id)
2901
        tt.create_file('baz', trans_id)
2902
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
2903
        self.build_tree_contents([('tree2/foo', 'qux')])
4961.2.9 by Martin Pool
Rip out most remaining uses of DummyProgressBar
2904
        pb = None
3363.17.21 by Aaron Bentley
Conflicts are handled when merging from preview trees
2905
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
2906
                                         pb, tree.basis_tree())
2907
        merger.merge_type = Merge3Merger
2908
        merger.do_merge()
2909
3363.17.18 by Aaron Bentley
Fix is_executable for PreviewTree
2910
    def test_is_executable(self):
2911
        tree = self.make_branch_and_tree('tree')
2912
        preview = TransformPreview(tree)
2913
        self.addCleanup(preview.finalize)
2914
        preview.new_file('foo', preview.root, 'bar', 'baz-id')
2915
        preview_tree = preview.get_preview_tree()
2916
        self.assertEqual(False, preview_tree.is_executable('baz-id',
2917
                                                           'tree/foo'))
2918
        self.assertEqual(False, preview_tree.is_executable('baz-id'))
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
2919
4354.4.4 by Aaron Bentley
Simplify by using CommitBuilder directly
2920
    def test_commit_preview_tree(self):
2921
        tree = self.make_branch_and_tree('tree')
2922
        rev_id = tree.commit('rev1')
2923
        tree.branch.lock_write()
2924
        self.addCleanup(tree.branch.unlock)
2925
        tt = TransformPreview(tree)
2926
        tt.new_file('file', tt.root, 'contents', 'file_id')
2927
        self.addCleanup(tt.finalize)
2928
        preview = tt.get_preview_tree()
2929
        preview.set_parent_ids([rev_id])
2930
        builder = tree.branch.get_commit_builder([rev_id])
2931
        list(builder.record_iter_changes(preview, rev_id, tt.iter_changes()))
2932
        builder.finish_inventory()
2933
        rev2_id = builder.commit('rev2')
2934
        rev2_tree = tree.branch.repository.revision_tree(rev2_id)
2935
        self.assertEqual('contents', rev2_tree.get_file_text('file_id'))
2936
4634.79.1 by Aaron Bentley
TransformPreview uses ascii-only filenames.
2937
    def test_ascii_limbo_paths(self):
4634.79.2 by Aaron Bentley
Avoid runing test on non-unicode filesystems.
2938
        self.requireFeature(tests.UnicodeFilenameFeature)
4634.79.1 by Aaron Bentley
TransformPreview uses ascii-only filenames.
2939
        branch = self.make_branch('any')
2940
        tree = branch.repository.revision_tree(_mod_revision.NULL_REVISION)
2941
        tt = TransformPreview(tree)
4789.25.7 by John Arbash Meinel
We can run the executable tests.
2942
        self.addCleanup(tt.finalize)
4634.79.1 by Aaron Bentley
TransformPreview uses ascii-only filenames.
2943
        foo_id = tt.new_directory('', ROOT_PARENT)
2944
        bar_id = tt.new_file(u'\u1234bar', foo_id, 'contents')
2945
        limbo_path = tt._limbo_name(bar_id)
2946
        self.assertEqual(limbo_path.encode('ascii', 'replace'), limbo_path)
2947
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
2948
0.13.13 by Aaron Bentley
Add direct test of serialization records
2949
class FakeSerializer(object):
2950
    """Serializer implementation that simply returns the input.
2951
2952
    The input is returned in the order used by pack.ContainerPushParser.
2953
    """
2954
    @staticmethod
2955
    def bytes_record(bytes, names):
2956
        return names, bytes
2957
2958
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
2959
class TestSerializeTransform(tests.TestCaseWithTransport):
2960
0.13.22 by Aaron Bentley
More unicodeness for Shelf tests
2961
    _test_needs_features = [tests.UnicodeFilenameFeature]
2962
0.13.17 by Aaron Bentley
Convert roundtrip destruction test to serialization/deserialization pair
2963
    def get_preview(self, tree=None):
2964
        if tree is None:
2965
            tree = self.make_branch_and_tree('tree')
0.13.14 by Aaron Bentley
Add deserialization test, remove roundtrip test.
2966
        tt = TransformPreview(tree)
2967
        self.addCleanup(tt.finalize)
2968
        return tt
2969
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
2970
    def assertSerializesTo(self, expected, tt):
2971
        records = list(tt.serialize(FakeSerializer()))
2972
        self.assertEqual(expected, records)
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
2973
0.13.13 by Aaron Bentley
Add direct test of serialization records
2974
    @staticmethod
2975
    def default_attribs():
2976
        return {
0.13.15 by Aaron Bentley
Convert symlink tests to avoid roundtripping
2977
            '_id_number': 1,
0.13.13 by Aaron Bentley
Add direct test of serialization records
2978
            '_new_name': {},
2979
            '_new_parent': {},
2980
            '_new_executability': {},
2981
            '_new_id': {},
0.13.15 by Aaron Bentley
Convert symlink tests to avoid roundtripping
2982
            '_tree_path_ids': {'': 'new-0'},
0.13.13 by Aaron Bentley
Add direct test of serialization records
2983
            '_removed_id': [],
2984
            '_removed_contents': [],
2985
            '_non_present_ids': {},
2986
            }
2987
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
2988
    def make_records(self, attribs, contents):
2989
        records = [
2990
            (((('attribs'),),), bencode.bencode(attribs))]
2991
        records.extend([(((n, k),), c) for n, k, c in contents])
2992
        return records
2993
0.13.13 by Aaron Bentley
Add direct test of serialization records
2994
    def creation_records(self):
2995
        attribs = self.default_attribs()
2996
        attribs['_id_number'] = 3
2997
        attribs['_new_name'] = {
2998
            'new-1': u'foo\u1234'.encode('utf-8'), 'new-2': 'qux'}
2999
        attribs['_new_id'] = {'new-1': 'baz', 'new-2': 'quxx'}
3000
        attribs['_new_parent'] = {'new-1': 'new-0', 'new-2': 'new-0'}
3001
        attribs['_new_executability'] = {'new-1': 1}
3002
        contents = [
3003
            ('new-1', 'file', 'i 1\nbar\n'),
3004
            ('new-2', 'directory', ''),
3005
            ]
3006
        return self.make_records(attribs, contents)
3007
3008
    def test_serialize_creation(self):
0.13.14 by Aaron Bentley
Add deserialization test, remove roundtrip test.
3009
        tt = self.get_preview()
0.13.13 by Aaron Bentley
Add direct test of serialization records
3010
        tt.new_file(u'foo\u1234', tt.root, 'bar', 'baz', True)
3011
        tt.new_directory('qux', tt.root, 'quxx')
0.13.21 by Aaron Bentley
Use assertSerializesTo in more places
3012
        self.assertSerializesTo(self.creation_records(), tt)
0.13.13 by Aaron Bentley
Add direct test of serialization records
3013
0.13.14 by Aaron Bentley
Add deserialization test, remove roundtrip test.
3014
    def test_deserialize_creation(self):
3015
        tt = self.get_preview()
3016
        tt.deserialize(iter(self.creation_records()))
3017
        self.assertEqual(3, tt._id_number)
3018
        self.assertEqual({'new-1': u'foo\u1234',
3019
                          'new-2': 'qux'}, tt._new_name)
3020
        self.assertEqual({'new-1': 'baz', 'new-2': 'quxx'}, tt._new_id)
3021
        self.assertEqual({'new-1': tt.root, 'new-2': tt.root}, tt._new_parent)
3022
        self.assertEqual({'baz': 'new-1', 'quxx': 'new-2'}, tt._r_new_id)
3023
        self.assertEqual({'new-1': True}, tt._new_executability)
3024
        self.assertEqual({'new-1': 'file',
3025
                          'new-2': 'directory'}, tt._new_contents)
3026
        foo_limbo = open(tt._limbo_name('new-1'), 'rb')
3027
        try:
3028
            foo_content = foo_limbo.read()
3029
        finally:
3030
            foo_limbo.close()
3031
        self.assertEqual('bar', foo_content)
3032
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3033
    def symlink_creation_records(self):
3034
        attribs = self.default_attribs()
3035
        attribs['_id_number'] = 2
3036
        attribs['_new_name'] = {'new-1': u'foo\u1234'.encode('utf-8')}
3037
        attribs['_new_parent'] = {'new-1': 'new-0'}
3038
        contents = [('new-1', 'symlink', u'bar\u1234'.encode('utf-8'))]
3039
        return self.make_records(attribs, contents)
3040
0.13.15 by Aaron Bentley
Convert symlink tests to avoid roundtripping
3041
    def test_serialize_symlink_creation(self):
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3042
        self.requireFeature(tests.SymlinkFeature)
0.13.15 by Aaron Bentley
Convert symlink tests to avoid roundtripping
3043
        tt = self.get_preview()
0.13.16 by Aaron Bentley
Add unicode symlink targets to tests
3044
        tt.new_symlink(u'foo\u1234', tt.root, u'bar\u1234')
0.13.21 by Aaron Bentley
Use assertSerializesTo in more places
3045
        self.assertSerializesTo(self.symlink_creation_records(), tt)
0.13.15 by Aaron Bentley
Convert symlink tests to avoid roundtripping
3046
3047
    def test_deserialize_symlink_creation(self):
4241.14.12 by Vincent Ladeuil
Far too many modifications for a single commit, need to restart.
3048
        self.requireFeature(tests.SymlinkFeature)
0.13.15 by Aaron Bentley
Convert symlink tests to avoid roundtripping
3049
        tt = self.get_preview()
3050
        tt.deserialize(iter(self.symlink_creation_records()))
4241.14.12 by Vincent Ladeuil
Far too many modifications for a single commit, need to restart.
3051
        abspath = tt._limbo_name('new-1')
4241.14.17 by Vincent Ladeuil
Add more tests for unicode symlinks to test_transform.
3052
        foo_content = osutils.readlink(abspath)
0.13.22 by Aaron Bentley
More unicodeness for Shelf tests
3053
        self.assertEqual(u'bar\u1234', foo_content)
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3054
0.13.19 by Aaron Bentley
Clean up serialization tests
3055
    def make_destruction_preview(self):
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3056
        tree = self.make_branch_and_tree('.')
3057
        self.build_tree([u'foo\u1234', 'bar'])
3058
        tree.add([u'foo\u1234', 'bar'], ['foo-id', 'bar-id'])
0.13.19 by Aaron Bentley
Clean up serialization tests
3059
        return self.get_preview(tree)
0.13.17 by Aaron Bentley
Convert roundtrip destruction test to serialization/deserialization pair
3060
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3061
    def destruction_records(self):
3062
        attribs = self.default_attribs()
3063
        attribs['_id_number'] = 3
3064
        attribs['_removed_id'] = ['new-1']
3065
        attribs['_removed_contents'] = ['new-2']
3066
        attribs['_tree_path_ids'] = {
3067
            '': 'new-0',
3068
            u'foo\u1234'.encode('utf-8'): 'new-1',
3069
            'bar': 'new-2',
3070
            }
3071
        return self.make_records(attribs, [])
3072
0.13.17 by Aaron Bentley
Convert roundtrip destruction test to serialization/deserialization pair
3073
    def test_serialize_destruction(self):
0.13.19 by Aaron Bentley
Clean up serialization tests
3074
        tt = self.make_destruction_preview()
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3075
        foo_trans_id = tt.trans_id_tree_file_id('foo-id')
3076
        tt.unversion_file(foo_trans_id)
3077
        bar_trans_id = tt.trans_id_tree_file_id('bar-id')
3078
        tt.delete_contents(bar_trans_id)
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3079
        self.assertSerializesTo(self.destruction_records(), tt)
0.13.17 by Aaron Bentley
Convert roundtrip destruction test to serialization/deserialization pair
3080
3081
    def test_deserialize_destruction(self):
0.13.19 by Aaron Bentley
Clean up serialization tests
3082
        tt = self.make_destruction_preview()
0.13.17 by Aaron Bentley
Convert roundtrip destruction test to serialization/deserialization pair
3083
        tt.deserialize(iter(self.destruction_records()))
3084
        self.assertEqual({u'foo\u1234': 'new-1',
3085
                          'bar': 'new-2',
3086
                          '': tt.root}, tt._tree_path_ids)
3087
        self.assertEqual({'new-1': u'foo\u1234',
3088
                          'new-2': 'bar',
3089
                          tt.root: ''}, tt._tree_id_paths)
3090
        self.assertEqual(set(['new-1']), tt._removed_id)
3091
        self.assertEqual(set(['new-2']), tt._removed_contents)
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3092
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3093
    def missing_records(self):
3094
        attribs = self.default_attribs()
3095
        attribs['_id_number'] = 2
3096
        attribs['_non_present_ids'] = {
3097
            'boo': 'new-1',}
3098
        return self.make_records(attribs, [])
3099
3100
    def test_serialize_missing(self):
3101
        tt = self.get_preview()
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3102
        boo_trans_id = tt.trans_id_file_id('boo')
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3103
        self.assertSerializesTo(self.missing_records(), tt)
3104
3105
    def test_deserialize_missing(self):
3106
        tt = self.get_preview()
3107
        tt.deserialize(iter(self.missing_records()))
3108
        self.assertEqual({'boo': 'new-1'}, tt._non_present_ids)
3109
3110
    def make_modification_preview(self):
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3111
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
3112
        LINES_TWO = 'z\nbb\nx\ndd\n'
3113
        tree = self.make_branch_and_tree('tree')
3114
        self.build_tree_contents([('tree/file', LINES_ONE)])
3115
        tree.add('file', 'file-id')
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3116
        return self.get_preview(tree), LINES_TWO
3117
3118
    def modification_records(self):
3119
        attribs = self.default_attribs()
3120
        attribs['_id_number'] = 2
3121
        attribs['_tree_path_ids'] = {
3122
            'file': 'new-1',
3123
            '': 'new-0',}
3124
        attribs['_removed_contents'] = ['new-1']
3125
        contents = [('new-1', 'file',
3126
                     'i 1\nz\n\nc 0 1 1 1\ni 1\nx\n\nc 0 3 3 1\n')]
3127
        return self.make_records(attribs, contents)
3128
3129
    def test_serialize_modification(self):
3130
        tt, LINES = self.make_modification_preview()
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3131
        trans_id = tt.trans_id_file_id('file-id')
3132
        tt.delete_contents(trans_id)
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3133
        tt.create_file(LINES, trans_id)
3134
        self.assertSerializesTo(self.modification_records(), tt)
3135
3136
    def test_deserialize_modification(self):
3137
        tt, LINES = self.make_modification_preview()
3138
        tt.deserialize(iter(self.modification_records()))
3139
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3140
3141
    def make_kind_change_preview(self):
3142
        LINES = 'a\nb\nc\nd\n'
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3143
        tree = self.make_branch_and_tree('tree')
3144
        self.build_tree(['tree/foo/'])
3145
        tree.add('foo', 'foo-id')
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3146
        return self.get_preview(tree), LINES
3147
3148
    def kind_change_records(self):
3149
        attribs = self.default_attribs()
3150
        attribs['_id_number'] = 2
3151
        attribs['_tree_path_ids'] = {
3152
            'foo': 'new-1',
3153
            '': 'new-0',}
3154
        attribs['_removed_contents'] = ['new-1']
3155
        contents = [('new-1', 'file',
3156
                     'i 4\na\nb\nc\nd\n\n')]
3157
        return self.make_records(attribs, contents)
3158
3159
    def test_serialize_kind_change(self):
3160
        tt, LINES = self.make_kind_change_preview()
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3161
        trans_id = tt.trans_id_file_id('foo-id')
3162
        tt.delete_contents(trans_id)
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3163
        tt.create_file(LINES, trans_id)
3164
        self.assertSerializesTo(self.kind_change_records(), tt)
3165
3166
    def test_deserialize_kind_change(self):
3167
        tt, LINES = self.make_kind_change_preview()
3168
        tt.deserialize(iter(self.kind_change_records()))
3169
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3170
3171
    def make_add_contents_preview(self):
3172
        LINES = 'a\nb\nc\nd\n'
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3173
        tree = self.make_branch_and_tree('tree')
3174
        self.build_tree(['tree/foo'])
3175
        tree.add('foo')
3176
        os.unlink('tree/foo')
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3177
        return self.get_preview(tree), LINES
3178
3179
    def add_contents_records(self):
3180
        attribs = self.default_attribs()
3181
        attribs['_id_number'] = 2
3182
        attribs['_tree_path_ids'] = {
3183
            'foo': 'new-1',
3184
            '': 'new-0',}
3185
        contents = [('new-1', 'file',
3186
                     'i 4\na\nb\nc\nd\n\n')]
3187
        return self.make_records(attribs, contents)
3188
3189
    def test_serialize_add_contents(self):
3190
        tt, LINES = self.make_add_contents_preview()
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3191
        trans_id = tt.trans_id_tree_path('foo')
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3192
        tt.create_file(LINES, trans_id)
3193
        self.assertSerializesTo(self.add_contents_records(), tt)
3194
3195
    def test_deserialize_add_contents(self):
3196
        tt, LINES = self.make_add_contents_preview()
3197
        tt.deserialize(iter(self.add_contents_records()))
3198
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3199
3200
    def test_get_parents_lines(self):
3201
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
3202
        LINES_TWO = 'z\nbb\nx\ndd\n'
3203
        tree = self.make_branch_and_tree('tree')
3204
        self.build_tree_contents([('tree/file', LINES_ONE)])
3205
        tree.add('file', 'file-id')
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3206
        tt = self.get_preview(tree)
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3207
        trans_id = tt.trans_id_tree_path('file')
3208
        self.assertEqual((['aa\n', 'bb\n', 'cc\n', 'dd\n'],),
3209
            tt._get_parents_lines(trans_id))
3210
3211
    def test_get_parents_texts(self):
3212
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
3213
        LINES_TWO = 'z\nbb\nx\ndd\n'
3214
        tree = self.make_branch_and_tree('tree')
3215
        self.build_tree_contents([('tree/file', LINES_ONE)])
3216
        tree.add('file', 'file-id')
0.13.18 by Aaron Bentley
Finish converting tests to direct serialize/deserialize tests, clean up
3217
        tt = self.get_preview(tree)
0.13.8 by Aaron Bentley
Integrate serialization into TreeTransforms
3218
        trans_id = tt.trans_id_tree_path('file')
3219
        self.assertEqual((LINES_ONE,),
3220
            tt._get_parents_texts(trans_id))