/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/tests/test_transform.py

  • Committer: Jelmer Vernooij
  • Date: 2017-06-08 23:30:31 UTC
  • mto: This revision was merged to the branch mainline in revision 6690.
  • Revision ID: jelmer@jelmer.uk-20170608233031-3qavls2o7a1pqllj
Update imports.

Show diffs side-by-side

added added

removed removed

Lines of Context:
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
 
import codecs
18
17
import errno
19
 
from io import BytesIO, StringIO
20
18
import os
21
19
import sys
22
20
import time
25
23
    bencode,
26
24
    errors,
27
25
    filters,
 
26
    generate_ids,
28
27
    osutils,
29
28
    revision as _mod_revision,
30
29
    rules,
33
32
    transform,
34
33
    urlutils,
35
34
    )
36
 
from ..bzr import (
37
 
    generate_ids,
38
 
    )
39
35
from ..conflicts import (
40
36
    DeletingParent,
41
37
    DuplicateEntry,
51
47
    DuplicateKey,
52
48
    ExistingLimbo,
53
49
    ExistingPendingDeletion,
 
50
    ImmortalLimbo,
54
51
    ImmortalPendingDeletion,
55
52
    LockError,
 
53
    MalformedTransform,
 
54
    ReusingTransform,
56
55
)
57
56
from ..osutils import (
58
57
    file_kind,
60
59
)
61
60
from ..merge import Merge3Merger, Merger
62
61
from ..mutabletree import MutableTree
 
62
from ..sixish import (
 
63
    BytesIO,
 
64
    )
63
65
from . import (
64
66
    features,
65
67
    TestCaseInTempDir,
78
80
    resolve_conflicts,
79
81
    resolve_checkout,
80
82
    ROOT_PARENT,
81
 
    ImmortalLimbo,
82
 
    MalformedTransform,
83
 
    NoFinalPath,
84
 
    ReusingTransform,
85
 
    TransformRenameFailed,
 
83
    TransformPreview,
 
84
    TreeTransform,
86
85
)
87
86
 
88
87
 
 
88
class TestTreeTransform(tests.TestCaseWithTransport):
 
89
 
 
90
    def setUp(self):
 
91
        super(TestTreeTransform, self).setUp()
 
92
        self.wt = self.make_branch_and_tree('.', format='development-subtree')
 
93
        os.chdir('..')
 
94
 
 
95
    def get_transform(self):
 
96
        transform = TreeTransform(self.wt)
 
97
        self.addCleanup(transform.finalize)
 
98
        return transform, transform.root
 
99
 
 
100
    def get_transform_for_sha1_test(self):
 
101
        trans, root = self.get_transform()
 
102
        self.wt.lock_tree_write()
 
103
        self.addCleanup(self.wt.unlock)
 
104
        contents = ['just some content\n']
 
105
        sha1 = osutils.sha_strings(contents)
 
106
        # Roll back the clock
 
107
        trans._creation_mtime = time.time() - 20.0
 
108
        return trans, root, contents, sha1
 
109
 
 
110
    def test_existing_limbo(self):
 
111
        transform, root = self.get_transform()
 
112
        limbo_name = transform._limbodir
 
113
        deletion_path = transform._deletiondir
 
114
        os.mkdir(pathjoin(limbo_name, 'hehe'))
 
115
        self.assertRaises(ImmortalLimbo, transform.apply)
 
116
        self.assertRaises(LockError, self.wt.unlock)
 
117
        self.assertRaises(ExistingLimbo, self.get_transform)
 
118
        self.assertRaises(LockError, self.wt.unlock)
 
119
        os.rmdir(pathjoin(limbo_name, 'hehe'))
 
120
        os.rmdir(limbo_name)
 
121
        os.rmdir(deletion_path)
 
122
        transform, root = self.get_transform()
 
123
        transform.apply()
 
124
 
 
125
    def test_existing_pending_deletion(self):
 
126
        transform, root = self.get_transform()
 
127
        deletion_path = self._limbodir = urlutils.local_path_from_url(
 
128
            transform._tree._transport.abspath('pending-deletion'))
 
129
        os.mkdir(pathjoin(deletion_path, 'blocking-directory'))
 
130
        self.assertRaises(ImmortalPendingDeletion, transform.apply)
 
131
        self.assertRaises(LockError, self.wt.unlock)
 
132
        self.assertRaises(ExistingPendingDeletion, self.get_transform)
 
133
 
 
134
    def test_build(self):
 
135
        transform, root = self.get_transform()
 
136
        self.wt.lock_tree_write()
 
137
        self.addCleanup(self.wt.unlock)
 
138
        self.assertIs(transform.get_tree_parent(root), ROOT_PARENT)
 
139
        imaginary_id = transform.trans_id_tree_path('imaginary')
 
140
        imaginary_id2 = transform.trans_id_tree_path('imaginary/')
 
141
        self.assertEqual(imaginary_id, imaginary_id2)
 
142
        self.assertEqual(root, transform.get_tree_parent(imaginary_id))
 
143
        self.assertEqual('directory', transform.final_kind(root))
 
144
        self.assertEqual(self.wt.get_root_id(), transform.final_file_id(root))
 
145
        trans_id = transform.create_path('name', root)
 
146
        self.assertIs(transform.final_file_id(trans_id), None)
 
147
        self.assertIs(None, transform.final_kind(trans_id))
 
148
        transform.create_file('contents', trans_id)
 
149
        transform.set_executability(True, trans_id)
 
150
        transform.version_file('my_pretties', trans_id)
 
151
        self.assertRaises(DuplicateKey, transform.version_file,
 
152
                          'my_pretties', trans_id)
 
153
        self.assertEqual(transform.final_file_id(trans_id), 'my_pretties')
 
154
        self.assertEqual(transform.final_parent(trans_id), root)
 
155
        self.assertIs(transform.final_parent(root), ROOT_PARENT)
 
156
        self.assertIs(transform.get_tree_parent(root), ROOT_PARENT)
 
157
        oz_id = transform.create_path('oz', root)
 
158
        transform.create_directory(oz_id)
 
159
        transform.version_file('ozzie', oz_id)
 
160
        trans_id2 = transform.create_path('name2', root)
 
161
        transform.create_file('contents', trans_id2)
 
162
        transform.set_executability(False, trans_id2)
 
163
        transform.version_file('my_pretties2', trans_id2)
 
164
        modified_paths = transform.apply().modified_paths
 
165
        self.assertEqual('contents', self.wt.get_file_byname('name').read())
 
166
        self.assertEqual(self.wt.path2id('name'), 'my_pretties')
 
167
        self.assertIs(self.wt.is_executable('my_pretties'), True)
 
168
        self.assertIs(self.wt.is_executable('my_pretties2'), False)
 
169
        self.assertEqual('directory', file_kind(self.wt.abspath('oz')))
 
170
        self.assertEqual(len(modified_paths), 3)
 
171
        tree_mod_paths = [self.wt.id2abspath(f) for f in
 
172
                          ('ozzie', 'my_pretties', 'my_pretties2')]
 
173
        self.assertSubset(tree_mod_paths, modified_paths)
 
174
        # is it safe to finalize repeatedly?
 
175
        transform.finalize()
 
176
        transform.finalize()
 
177
 
 
178
    def test_apply_informs_tree_of_observed_sha1(self):
 
179
        trans, root, contents, sha1 = self.get_transform_for_sha1_test()
 
180
        trans_id = trans.new_file('file1', root, contents, file_id='file1-id',
 
181
                                  sha1=sha1)
 
182
        calls = []
 
183
        orig = self.wt._observed_sha1
 
184
        def _observed_sha1(*args):
 
185
            calls.append(args)
 
186
            orig(*args)
 
187
        self.wt._observed_sha1 = _observed_sha1
 
188
        trans.apply()
 
189
        self.assertEqual([(None, 'file1', trans._observed_sha1s[trans_id])],
 
190
                         calls)
 
191
 
 
192
    def test_create_file_caches_sha1(self):
 
193
        trans, root, contents, sha1 = self.get_transform_for_sha1_test()
 
194
        trans_id = trans.create_path('file1', root)
 
195
        trans.create_file(contents, trans_id, sha1=sha1)
 
196
        st_val = osutils.lstat(trans._limbo_name(trans_id))
 
197
        o_sha1, o_st_val = trans._observed_sha1s[trans_id]
 
198
        self.assertEqual(o_sha1, sha1)
 
199
        self.assertEqualStat(o_st_val, st_val)
 
200
 
 
201
    def test__apply_insertions_updates_sha1(self):
 
202
        trans, root, contents, sha1 = self.get_transform_for_sha1_test()
 
203
        trans_id = trans.create_path('file1', root)
 
204
        trans.create_file(contents, trans_id, sha1=sha1)
 
205
        st_val = osutils.lstat(trans._limbo_name(trans_id))
 
206
        o_sha1, o_st_val = trans._observed_sha1s[trans_id]
 
207
        self.assertEqual(o_sha1, sha1)
 
208
        self.assertEqualStat(o_st_val, st_val)
 
209
        creation_mtime = trans._creation_mtime + 10.0
 
210
        # We fake a time difference from when the file was created until now it
 
211
        # is being renamed by using os.utime. Note that the change we actually
 
212
        # want to see is the real ctime change from 'os.rename()', but as long
 
213
        # as we observe a new stat value, we should be fine.
 
214
        os.utime(trans._limbo_name(trans_id), (creation_mtime, creation_mtime))
 
215
        trans.apply()
 
216
        new_st_val = osutils.lstat(self.wt.abspath('file1'))
 
217
        o_sha1, o_st_val = trans._observed_sha1s[trans_id]
 
218
        self.assertEqual(o_sha1, sha1)
 
219
        self.assertEqualStat(o_st_val, new_st_val)
 
220
        self.assertNotEqual(st_val.st_mtime, new_st_val.st_mtime)
 
221
 
 
222
    def test_new_file_caches_sha1(self):
 
223
        trans, root, contents, sha1 = self.get_transform_for_sha1_test()
 
224
        trans_id = trans.new_file('file1', root, contents, file_id='file1-id',
 
225
                                  sha1=sha1)
 
226
        st_val = osutils.lstat(trans._limbo_name(trans_id))
 
227
        o_sha1, o_st_val = trans._observed_sha1s[trans_id]
 
228
        self.assertEqual(o_sha1, sha1)
 
229
        self.assertEqualStat(o_st_val, st_val)
 
230
 
 
231
    def test_cancel_creation_removes_observed_sha1(self):
 
232
        trans, root, contents, sha1 = self.get_transform_for_sha1_test()
 
233
        trans_id = trans.new_file('file1', root, contents, file_id='file1-id',
 
234
                                  sha1=sha1)
 
235
        self.assertTrue(trans_id in trans._observed_sha1s)
 
236
        trans.cancel_creation(trans_id)
 
237
        self.assertFalse(trans_id in trans._observed_sha1s)
 
238
 
 
239
    def test_create_files_same_timestamp(self):
 
240
        transform, root = self.get_transform()
 
241
        self.wt.lock_tree_write()
 
242
        self.addCleanup(self.wt.unlock)
 
243
        # Roll back the clock, so that we know everything is being set to the
 
244
        # exact time
 
245
        transform._creation_mtime = creation_mtime = time.time() - 20.0
 
246
        transform.create_file('content-one',
 
247
                              transform.create_path('one', root))
 
248
        time.sleep(1) # *ugly*
 
249
        transform.create_file('content-two',
 
250
                              transform.create_path('two', root))
 
251
        transform.apply()
 
252
        fo, st1 = self.wt.get_file_with_stat(None, path='one', filtered=False)
 
253
        fo.close()
 
254
        fo, st2 = self.wt.get_file_with_stat(None, path='two', filtered=False)
 
255
        fo.close()
 
256
        # We only guarantee 2s resolution
 
257
        self.assertTrue(abs(creation_mtime - st1.st_mtime) < 2.0,
 
258
            "%s != %s within 2 seconds" % (creation_mtime, st1.st_mtime))
 
259
        # But if we have more than that, all files should get the same result
 
260
        self.assertEqual(st1.st_mtime, st2.st_mtime)
 
261
 
 
262
    def test_change_root_id(self):
 
263
        transform, root = self.get_transform()
 
264
        self.assertNotEqual('new-root-id', self.wt.get_root_id())
 
265
        transform.new_directory('', ROOT_PARENT, 'new-root-id')
 
266
        transform.delete_contents(root)
 
267
        transform.unversion_file(root)
 
268
        transform.fixup_new_roots()
 
269
        transform.apply()
 
270
        self.assertEqual('new-root-id', self.wt.get_root_id())
 
271
 
 
272
    def test_change_root_id_add_files(self):
 
273
        transform, root = self.get_transform()
 
274
        self.assertNotEqual('new-root-id', self.wt.get_root_id())
 
275
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'new-root-id')
 
276
        transform.new_file('file', new_trans_id, ['new-contents\n'],
 
277
                           'new-file-id')
 
278
        transform.delete_contents(root)
 
279
        transform.unversion_file(root)
 
280
        transform.fixup_new_roots()
 
281
        transform.apply()
 
282
        self.assertEqual('new-root-id', self.wt.get_root_id())
 
283
        self.assertEqual('new-file-id', self.wt.path2id('file'))
 
284
        self.assertFileEqual('new-contents\n', self.wt.abspath('file'))
 
285
 
 
286
    def test_add_two_roots(self):
 
287
        transform, root = self.get_transform()
 
288
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'new-root-id')
 
289
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'alt-root-id')
 
290
        self.assertRaises(ValueError, transform.fixup_new_roots)
 
291
 
 
292
    def test_retain_existing_root(self):
 
293
        tt, root = self.get_transform()
 
294
        with tt:
 
295
            tt.new_directory('', ROOT_PARENT, 'new-root-id')
 
296
            tt.fixup_new_roots()
 
297
            self.assertNotEqual('new-root-id', tt.final_file_id(tt.root))
 
298
 
 
299
    def test_retain_existing_root_added_file(self):
 
300
        tt, root = self.get_transform()
 
301
        new_trans_id = tt.new_directory('', ROOT_PARENT, 'new-root-id')
 
302
        child = tt.new_directory('child', new_trans_id, 'child-id')
 
303
        tt.fixup_new_roots()
 
304
        self.assertEqual(tt.root, tt.final_parent(child))
 
305
 
 
306
    def test_add_unversioned_root(self):
 
307
        transform, root = self.get_transform()
 
308
        new_trans_id = transform.new_directory('', ROOT_PARENT, None)
 
309
        transform.delete_contents(transform.root)
 
310
        transform.fixup_new_roots()
 
311
        self.assertNotIn(transform.root, transform._new_id)
 
312
 
 
313
    def test_remove_root_fixup(self):
 
314
        transform, root = self.get_transform()
 
315
        old_root_id = self.wt.get_root_id()
 
316
        self.assertNotEqual('new-root-id', old_root_id)
 
317
        transform.delete_contents(root)
 
318
        transform.unversion_file(root)
 
319
        transform.fixup_new_roots()
 
320
        transform.apply()
 
321
        self.assertEqual(old_root_id, self.wt.get_root_id())
 
322
 
 
323
        transform, root = self.get_transform()
 
324
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'new-root-id')
 
325
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'alt-root-id')
 
326
        self.assertRaises(ValueError, transform.fixup_new_roots)
 
327
 
 
328
    def test_fixup_new_roots_permits_empty_tree(self):
 
329
        transform, root = self.get_transform()
 
330
        transform.delete_contents(root)
 
331
        transform.unversion_file(root)
 
332
        transform.fixup_new_roots()
 
333
        self.assertIs(None, transform.final_kind(root))
 
334
        self.assertIs(None, transform.final_file_id(root))
 
335
 
 
336
    def test_apply_retains_root_directory(self):
 
337
        # Do not attempt to delete the physical root directory, because that
 
338
        # is impossible.
 
339
        transform, root = self.get_transform()
 
340
        with transform:
 
341
            transform.delete_contents(root)
 
342
            e = self.assertRaises(AssertionError, self.assertRaises,
 
343
                                  errors.TransformRenameFailed,
 
344
                                  transform.apply)
 
345
        self.assertContainsRe('TransformRenameFailed not raised', str(e))
 
346
 
 
347
    def test_apply_retains_file_id(self):
 
348
        transform, root = self.get_transform()
 
349
        old_root_id = transform.tree_file_id(root)
 
350
        transform.unversion_file(root)
 
351
        transform.apply()
 
352
        self.assertEqual(old_root_id, self.wt.get_root_id())
 
353
 
 
354
    def test_hardlink(self):
 
355
        self.requireFeature(HardlinkFeature)
 
356
        transform, root = self.get_transform()
 
357
        transform.new_file('file1', root, 'contents')
 
358
        transform.apply()
 
359
        target = self.make_branch_and_tree('target')
 
360
        target_transform = TreeTransform(target)
 
361
        trans_id = target_transform.create_path('file1', target_transform.root)
 
362
        target_transform.create_hardlink(self.wt.abspath('file1'), trans_id)
 
363
        target_transform.apply()
 
364
        self.assertPathExists('target/file1')
 
365
        source_stat = os.stat(self.wt.abspath('file1'))
 
366
        target_stat = os.stat('target/file1')
 
367
        self.assertEqual(source_stat, target_stat)
 
368
 
 
369
    def test_convenience(self):
 
370
        transform, root = self.get_transform()
 
371
        self.wt.lock_tree_write()
 
372
        self.addCleanup(self.wt.unlock)
 
373
        trans_id = transform.new_file('name', root, 'contents',
 
374
                                      'my_pretties', True)
 
375
        oz = transform.new_directory('oz', root, 'oz-id')
 
376
        dorothy = transform.new_directory('dorothy', oz, 'dorothy-id')
 
377
        toto = transform.new_file('toto', dorothy, 'toto-contents',
 
378
                                  'toto-id', False)
 
379
 
 
380
        self.assertEqual(len(transform.find_conflicts()), 0)
 
381
        transform.apply()
 
382
        self.assertRaises(ReusingTransform, transform.find_conflicts)
 
383
        self.assertEqual('contents', file(self.wt.abspath('name')).read())
 
384
        self.assertEqual(self.wt.path2id('name'), 'my_pretties')
 
385
        self.assertIs(self.wt.is_executable('my_pretties'), True)
 
386
        self.assertEqual(self.wt.path2id('oz'), 'oz-id')
 
387
        self.assertEqual(self.wt.path2id('oz/dorothy'), 'dorothy-id')
 
388
        self.assertEqual(self.wt.path2id('oz/dorothy/toto'), 'toto-id')
 
389
 
 
390
        self.assertEqual('toto-contents',
 
391
                         self.wt.get_file_byname('oz/dorothy/toto').read())
 
392
        self.assertIs(self.wt.is_executable('toto-id'), False)
 
393
 
 
394
    def test_tree_reference(self):
 
395
        transform, root = self.get_transform()
 
396
        tree = transform._tree
 
397
        trans_id = transform.new_directory('reference', root, 'subtree-id')
 
398
        transform.set_tree_reference('subtree-revision', trans_id)
 
399
        transform.apply()
 
400
        tree.lock_read()
 
401
        self.addCleanup(tree.unlock)
 
402
        self.assertEqual('subtree-revision',
 
403
                         tree.root_inventory['subtree-id'].reference_revision)
 
404
 
 
405
    def test_conflicts(self):
 
406
        transform, root = self.get_transform()
 
407
        trans_id = transform.new_file('name', root, 'contents',
 
408
                                      'my_pretties')
 
409
        self.assertEqual(len(transform.find_conflicts()), 0)
 
410
        trans_id2 = transform.new_file('name', root, 'Crontents', 'toto')
 
411
        self.assertEqual(transform.find_conflicts(),
 
412
                         [('duplicate', trans_id, trans_id2, 'name')])
 
413
        self.assertRaises(MalformedTransform, transform.apply)
 
414
        transform.adjust_path('name', trans_id, trans_id2)
 
415
        self.assertEqual(transform.find_conflicts(),
 
416
                         [('non-directory parent', trans_id)])
 
417
        tinman_id = transform.trans_id_tree_path('tinman')
 
418
        transform.adjust_path('name', tinman_id, trans_id2)
 
419
        self.assertEqual(transform.find_conflicts(),
 
420
                         [('unversioned parent', tinman_id),
 
421
                          ('missing parent', tinman_id)])
 
422
        lion_id = transform.create_path('lion', root)
 
423
        self.assertEqual(transform.find_conflicts(),
 
424
                         [('unversioned parent', tinman_id),
 
425
                          ('missing parent', tinman_id)])
 
426
        transform.adjust_path('name', lion_id, trans_id2)
 
427
        self.assertEqual(transform.find_conflicts(),
 
428
                         [('unversioned parent', lion_id),
 
429
                          ('missing parent', lion_id)])
 
430
        transform.version_file("Courage", lion_id)
 
431
        self.assertEqual(transform.find_conflicts(),
 
432
                         [('missing parent', lion_id),
 
433
                          ('versioning no contents', lion_id)])
 
434
        transform.adjust_path('name2', root, trans_id2)
 
435
        self.assertEqual(transform.find_conflicts(),
 
436
                         [('versioning no contents', lion_id)])
 
437
        transform.create_file('Contents, okay?', lion_id)
 
438
        transform.adjust_path('name2', trans_id2, trans_id2)
 
439
        self.assertEqual(transform.find_conflicts(),
 
440
                         [('parent loop', trans_id2),
 
441
                          ('non-directory parent', trans_id2)])
 
442
        transform.adjust_path('name2', root, trans_id2)
 
443
        oz_id = transform.new_directory('oz', root)
 
444
        transform.set_executability(True, oz_id)
 
445
        self.assertEqual(transform.find_conflicts(),
 
446
                         [('unversioned executability', oz_id)])
 
447
        transform.version_file('oz-id', oz_id)
 
448
        self.assertEqual(transform.find_conflicts(),
 
449
                         [('non-file executability', oz_id)])
 
450
        transform.set_executability(None, oz_id)
 
451
        tip_id = transform.new_file('tip', oz_id, 'ozma', 'tip-id')
 
452
        transform.apply()
 
453
        self.assertEqual(self.wt.path2id('name'), 'my_pretties')
 
454
        self.assertEqual('contents', file(self.wt.abspath('name')).read())
 
455
        transform2, root = self.get_transform()
 
456
        oz_id = transform2.trans_id_tree_file_id('oz-id')
 
457
        newtip = transform2.new_file('tip', oz_id, 'other', 'tip-id')
 
458
        result = transform2.find_conflicts()
 
459
        fp = FinalPaths(transform2)
 
460
        self.assertTrue('oz/tip' in transform2._tree_path_ids)
 
461
        self.assertEqual(fp.get_path(newtip), pathjoin('oz', 'tip'))
 
462
        self.assertEqual(len(result), 2)
 
463
        self.assertEqual((result[0][0], result[0][1]),
 
464
                         ('duplicate', newtip))
 
465
        self.assertEqual((result[1][0], result[1][2]),
 
466
                         ('duplicate id', newtip))
 
467
        transform2.finalize()
 
468
        transform3 = TreeTransform(self.wt)
 
469
        self.addCleanup(transform3.finalize)
 
470
        oz_id = transform3.trans_id_tree_file_id('oz-id')
 
471
        transform3.delete_contents(oz_id)
 
472
        self.assertEqual(transform3.find_conflicts(),
 
473
                         [('missing parent', oz_id)])
 
474
        root_id = transform3.root
 
475
        tip_id = transform3.trans_id_tree_file_id('tip-id')
 
476
        transform3.adjust_path('tip', root_id, tip_id)
 
477
        transform3.apply()
 
478
 
 
479
    def test_conflict_on_case_insensitive(self):
 
480
        tree = self.make_branch_and_tree('tree')
 
481
        # Don't try this at home, kids!
 
482
        # Force the tree to report that it is case sensitive, for conflict
 
483
        # resolution tests
 
484
        tree.case_sensitive = True
 
485
        transform = TreeTransform(tree)
 
486
        self.addCleanup(transform.finalize)
 
487
        transform.new_file('file', transform.root, 'content')
 
488
        transform.new_file('FiLe', transform.root, 'content')
 
489
        result = transform.find_conflicts()
 
490
        self.assertEqual([], result)
 
491
        transform.finalize()
 
492
        # Force the tree to report that it is case insensitive, for conflict
 
493
        # generation tests
 
494
        tree.case_sensitive = False
 
495
        transform = TreeTransform(tree)
 
496
        self.addCleanup(transform.finalize)
 
497
        transform.new_file('file', transform.root, 'content')
 
498
        transform.new_file('FiLe', transform.root, 'content')
 
499
        result = transform.find_conflicts()
 
500
        self.assertEqual([('duplicate', 'new-1', 'new-2', 'file')], result)
 
501
 
 
502
    def test_conflict_on_case_insensitive_existing(self):
 
503
        tree = self.make_branch_and_tree('tree')
 
504
        self.build_tree(['tree/FiLe'])
 
505
        # Don't try this at home, kids!
 
506
        # Force the tree to report that it is case sensitive, for conflict
 
507
        # resolution tests
 
508
        tree.case_sensitive = True
 
509
        transform = TreeTransform(tree)
 
510
        self.addCleanup(transform.finalize)
 
511
        transform.new_file('file', transform.root, 'content')
 
512
        result = transform.find_conflicts()
 
513
        self.assertEqual([], result)
 
514
        transform.finalize()
 
515
        # Force the tree to report that it is case insensitive, for conflict
 
516
        # generation tests
 
517
        tree.case_sensitive = False
 
518
        transform = TreeTransform(tree)
 
519
        self.addCleanup(transform.finalize)
 
520
        transform.new_file('file', transform.root, 'content')
 
521
        result = transform.find_conflicts()
 
522
        self.assertEqual([('duplicate', 'new-1', 'new-2', 'file')], result)
 
523
 
 
524
    def test_resolve_case_insensitive_conflict(self):
 
525
        tree = self.make_branch_and_tree('tree')
 
526
        # Don't try this at home, kids!
 
527
        # Force the tree to report that it is case insensitive, for conflict
 
528
        # resolution tests
 
529
        tree.case_sensitive = False
 
530
        transform = TreeTransform(tree)
 
531
        self.addCleanup(transform.finalize)
 
532
        transform.new_file('file', transform.root, 'content')
 
533
        transform.new_file('FiLe', transform.root, 'content')
 
534
        resolve_conflicts(transform)
 
535
        transform.apply()
 
536
        self.assertPathExists('tree/file')
 
537
        self.assertPathExists('tree/FiLe.moved')
 
538
 
 
539
    def test_resolve_checkout_case_conflict(self):
 
540
        tree = self.make_branch_and_tree('tree')
 
541
        # Don't try this at home, kids!
 
542
        # Force the tree to report that it is case insensitive, for conflict
 
543
        # resolution tests
 
544
        tree.case_sensitive = False
 
545
        transform = TreeTransform(tree)
 
546
        self.addCleanup(transform.finalize)
 
547
        transform.new_file('file', transform.root, 'content')
 
548
        transform.new_file('FiLe', transform.root, 'content')
 
549
        resolve_conflicts(transform,
 
550
                          pass_func=lambda t, c: resolve_checkout(t, c, []))
 
551
        transform.apply()
 
552
        self.assertPathExists('tree/file')
 
553
        self.assertPathExists('tree/FiLe.moved')
 
554
 
 
555
    def test_apply_case_conflict(self):
 
556
        """Ensure that a transform with case conflicts can always be applied"""
 
557
        tree = self.make_branch_and_tree('tree')
 
558
        transform = TreeTransform(tree)
 
559
        self.addCleanup(transform.finalize)
 
560
        transform.new_file('file', transform.root, 'content')
 
561
        transform.new_file('FiLe', transform.root, 'content')
 
562
        dir = transform.new_directory('dir', transform.root)
 
563
        transform.new_file('dirfile', dir, 'content')
 
564
        transform.new_file('dirFiLe', dir, 'content')
 
565
        resolve_conflicts(transform)
 
566
        transform.apply()
 
567
        self.assertPathExists('tree/file')
 
568
        if not os.path.exists('tree/FiLe.moved'):
 
569
            self.assertPathExists('tree/FiLe')
 
570
        self.assertPathExists('tree/dir/dirfile')
 
571
        if not os.path.exists('tree/dir/dirFiLe.moved'):
 
572
            self.assertPathExists('tree/dir/dirFiLe')
 
573
 
 
574
    def test_case_insensitive_limbo(self):
 
575
        tree = self.make_branch_and_tree('tree')
 
576
        # Don't try this at home, kids!
 
577
        # Force the tree to report that it is case insensitive
 
578
        tree.case_sensitive = False
 
579
        transform = TreeTransform(tree)
 
580
        self.addCleanup(transform.finalize)
 
581
        dir = transform.new_directory('dir', transform.root)
 
582
        first = transform.new_file('file', dir, 'content')
 
583
        second = transform.new_file('FiLe', dir, 'content')
 
584
        self.assertContainsRe(transform._limbo_name(first), 'new-1/file')
 
585
        self.assertNotContainsRe(transform._limbo_name(second), 'new-1/FiLe')
 
586
 
 
587
    def test_adjust_path_updates_child_limbo_names(self):
 
588
        tree = self.make_branch_and_tree('tree')
 
589
        transform = TreeTransform(tree)
 
590
        self.addCleanup(transform.finalize)
 
591
        foo_id = transform.new_directory('foo', transform.root)
 
592
        bar_id = transform.new_directory('bar', foo_id)
 
593
        baz_id = transform.new_directory('baz', bar_id)
 
594
        qux_id = transform.new_directory('qux', baz_id)
 
595
        transform.adjust_path('quxx', foo_id, bar_id)
 
596
        self.assertStartsWith(transform._limbo_name(qux_id),
 
597
                              transform._limbo_name(bar_id))
 
598
 
 
599
    def test_add_del(self):
 
600
        start, root = self.get_transform()
 
601
        start.new_directory('a', root, 'a')
 
602
        start.apply()
 
603
        transform, root = self.get_transform()
 
604
        transform.delete_versioned(transform.trans_id_tree_file_id('a'))
 
605
        transform.new_directory('a', root, 'a')
 
606
        transform.apply()
 
607
 
 
608
    def test_unversioning(self):
 
609
        create_tree, root = self.get_transform()
 
610
        parent_id = create_tree.new_directory('parent', root, 'parent-id')
 
611
        create_tree.new_file('child', parent_id, 'child', 'child-id')
 
612
        create_tree.apply()
 
613
        unversion = TreeTransform(self.wt)
 
614
        self.addCleanup(unversion.finalize)
 
615
        parent = unversion.trans_id_tree_path('parent')
 
616
        unversion.unversion_file(parent)
 
617
        self.assertEqual(unversion.find_conflicts(),
 
618
                         [('unversioned parent', parent_id)])
 
619
        file_id = unversion.trans_id_tree_file_id('child-id')
 
620
        unversion.unversion_file(file_id)
 
621
        unversion.apply()
 
622
 
 
623
    def test_name_invariants(self):
 
624
        create_tree, root = self.get_transform()
 
625
        # prepare tree
 
626
        root = create_tree.root
 
627
        create_tree.new_file('name1', root, 'hello1', 'name1')
 
628
        create_tree.new_file('name2', root, 'hello2', 'name2')
 
629
        ddir = create_tree.new_directory('dying_directory', root, 'ddir')
 
630
        create_tree.new_file('dying_file', ddir, 'goodbye1', 'dfile')
 
631
        create_tree.new_file('moving_file', ddir, 'later1', 'mfile')
 
632
        create_tree.new_file('moving_file2', root, 'later2', 'mfile2')
 
633
        create_tree.apply()
 
634
 
 
635
        mangle_tree,root = self.get_transform()
 
636
        root = mangle_tree.root
 
637
        #swap names
 
638
        name1 = mangle_tree.trans_id_tree_file_id('name1')
 
639
        name2 = mangle_tree.trans_id_tree_file_id('name2')
 
640
        mangle_tree.adjust_path('name2', root, name1)
 
641
        mangle_tree.adjust_path('name1', root, name2)
 
642
 
 
643
        #tests for deleting parent directories
 
644
        ddir = mangle_tree.trans_id_tree_file_id('ddir')
 
645
        mangle_tree.delete_contents(ddir)
 
646
        dfile = mangle_tree.trans_id_tree_file_id('dfile')
 
647
        mangle_tree.delete_versioned(dfile)
 
648
        mangle_tree.unversion_file(dfile)
 
649
        mfile = mangle_tree.trans_id_tree_file_id('mfile')
 
650
        mangle_tree.adjust_path('mfile', root, mfile)
 
651
 
 
652
        #tests for adding parent directories
 
653
        newdir = mangle_tree.new_directory('new_directory', root, 'newdir')
 
654
        mfile2 = mangle_tree.trans_id_tree_file_id('mfile2')
 
655
        mangle_tree.adjust_path('mfile2', newdir, mfile2)
 
656
        mangle_tree.new_file('newfile', newdir, 'hello3', 'dfile')
 
657
        self.assertEqual(mangle_tree.final_file_id(mfile2), 'mfile2')
 
658
        self.assertEqual(mangle_tree.final_parent(mfile2), newdir)
 
659
        self.assertEqual(mangle_tree.final_file_id(mfile2), 'mfile2')
 
660
        mangle_tree.apply()
 
661
        self.assertEqual(file(self.wt.abspath('name1')).read(), 'hello2')
 
662
        self.assertEqual(file(self.wt.abspath('name2')).read(), 'hello1')
 
663
        mfile2_path = self.wt.abspath(pathjoin('new_directory','mfile2'))
 
664
        self.assertEqual(mangle_tree.final_parent(mfile2), newdir)
 
665
        self.assertEqual(file(mfile2_path).read(), 'later2')
 
666
        self.assertEqual(self.wt.id2path('mfile2'), 'new_directory/mfile2')
 
667
        self.assertEqual(self.wt.path2id('new_directory/mfile2'), 'mfile2')
 
668
        newfile_path = self.wt.abspath(pathjoin('new_directory','newfile'))
 
669
        self.assertEqual(file(newfile_path).read(), 'hello3')
 
670
        self.assertEqual(self.wt.path2id('dying_directory'), 'ddir')
 
671
        self.assertIs(self.wt.path2id('dying_directory/dying_file'), None)
 
672
        mfile2_path = self.wt.abspath(pathjoin('new_directory','mfile2'))
 
673
 
 
674
    def test_both_rename(self):
 
675
        create_tree,root = self.get_transform()
 
676
        newdir = create_tree.new_directory('selftest', root, 'selftest-id')
 
677
        create_tree.new_file('blackbox.py', newdir, 'hello1', 'blackbox-id')
 
678
        create_tree.apply()
 
679
        mangle_tree,root = self.get_transform()
 
680
        selftest = mangle_tree.trans_id_tree_file_id('selftest-id')
 
681
        blackbox = mangle_tree.trans_id_tree_file_id('blackbox-id')
 
682
        mangle_tree.adjust_path('test', root, selftest)
 
683
        mangle_tree.adjust_path('test_too_much', root, selftest)
 
684
        mangle_tree.set_executability(True, blackbox)
 
685
        mangle_tree.apply()
 
686
 
 
687
    def test_both_rename2(self):
 
688
        create_tree,root = self.get_transform()
 
689
        breezy = create_tree.new_directory('breezy', root, 'breezy-id')
 
690
        tests = create_tree.new_directory('tests', breezy, 'tests-id')
 
691
        blackbox = create_tree.new_directory('blackbox', tests, 'blackbox-id')
 
692
        create_tree.new_file('test_too_much.py', blackbox, 'hello1',
 
693
                             'test_too_much-id')
 
694
        create_tree.apply()
 
695
        mangle_tree,root = self.get_transform()
 
696
        breezy = mangle_tree.trans_id_tree_file_id('breezy-id')
 
697
        tests = mangle_tree.trans_id_tree_file_id('tests-id')
 
698
        test_too_much = mangle_tree.trans_id_tree_file_id('test_too_much-id')
 
699
        mangle_tree.adjust_path('selftest', breezy, tests)
 
700
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much)
 
701
        mangle_tree.set_executability(True, test_too_much)
 
702
        mangle_tree.apply()
 
703
 
 
704
    def test_both_rename3(self):
 
705
        create_tree,root = self.get_transform()
 
706
        tests = create_tree.new_directory('tests', root, 'tests-id')
 
707
        create_tree.new_file('test_too_much.py', tests, 'hello1',
 
708
                             'test_too_much-id')
 
709
        create_tree.apply()
 
710
        mangle_tree,root = self.get_transform()
 
711
        tests = mangle_tree.trans_id_tree_file_id('tests-id')
 
712
        test_too_much = mangle_tree.trans_id_tree_file_id('test_too_much-id')
 
713
        mangle_tree.adjust_path('selftest', root, tests)
 
714
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much)
 
715
        mangle_tree.set_executability(True, test_too_much)
 
716
        mangle_tree.apply()
 
717
 
 
718
    def test_move_dangling_ie(self):
 
719
        create_tree, root = self.get_transform()
 
720
        # prepare tree
 
721
        root = create_tree.root
 
722
        create_tree.new_file('name1', root, 'hello1', 'name1')
 
723
        create_tree.apply()
 
724
        delete_contents, root = self.get_transform()
 
725
        file = delete_contents.trans_id_tree_file_id('name1')
 
726
        delete_contents.delete_contents(file)
 
727
        delete_contents.apply()
 
728
        move_id, root = self.get_transform()
 
729
        name1 = move_id.trans_id_tree_file_id('name1')
 
730
        newdir = move_id.new_directory('dir', root, 'newdir')
 
731
        move_id.adjust_path('name2', newdir, name1)
 
732
        move_id.apply()
 
733
 
 
734
    def test_replace_dangling_ie(self):
 
735
        create_tree, root = self.get_transform()
 
736
        # prepare tree
 
737
        root = create_tree.root
 
738
        create_tree.new_file('name1', root, 'hello1', 'name1')
 
739
        create_tree.apply()
 
740
        delete_contents = TreeTransform(self.wt)
 
741
        self.addCleanup(delete_contents.finalize)
 
742
        file = delete_contents.trans_id_tree_file_id('name1')
 
743
        delete_contents.delete_contents(file)
 
744
        delete_contents.apply()
 
745
        delete_contents.finalize()
 
746
        replace = TreeTransform(self.wt)
 
747
        self.addCleanup(replace.finalize)
 
748
        name2 = replace.new_file('name2', root, 'hello2', 'name1')
 
749
        conflicts = replace.find_conflicts()
 
750
        name1 = replace.trans_id_tree_file_id('name1')
 
751
        self.assertEqual(conflicts, [('duplicate id', name1, name2)])
 
752
        resolve_conflicts(replace)
 
753
        replace.apply()
 
754
 
 
755
    def _test_symlinks(self, link_name1,link_target1,
 
756
                       link_name2, link_target2):
 
757
 
 
758
        def ozpath(p): return 'oz/' + p
 
759
 
 
760
        self.requireFeature(SymlinkFeature)
 
761
        transform, root = self.get_transform()
 
762
        oz_id = transform.new_directory('oz', root, 'oz-id')
 
763
        wizard = transform.new_symlink(link_name1, oz_id, link_target1,
 
764
                                       'wizard-id')
 
765
        wiz_id = transform.create_path(link_name2, oz_id)
 
766
        transform.create_symlink(link_target2, wiz_id)
 
767
        transform.version_file('wiz-id2', wiz_id)
 
768
        transform.set_executability(True, wiz_id)
 
769
        self.assertEqual(transform.find_conflicts(),
 
770
                         [('non-file executability', wiz_id)])
 
771
        transform.set_executability(None, wiz_id)
 
772
        transform.apply()
 
773
        self.assertEqual(self.wt.path2id(ozpath(link_name1)), 'wizard-id')
 
774
        self.assertEqual('symlink',
 
775
                         file_kind(self.wt.abspath(ozpath(link_name1))))
 
776
        self.assertEqual(link_target2,
 
777
                         osutils.readlink(self.wt.abspath(ozpath(link_name2))))
 
778
        self.assertEqual(link_target1,
 
779
                         osutils.readlink(self.wt.abspath(ozpath(link_name1))))
 
780
 
 
781
    def test_symlinks(self):
 
782
        self._test_symlinks('wizard', 'wizard-target',
 
783
                            'wizard2', 'behind_curtain')
 
784
 
 
785
    def test_symlinks_unicode(self):
 
786
        self.requireFeature(features.UnicodeFilenameFeature)
 
787
        self._test_symlinks(u'\N{Euro Sign}wizard',
 
788
                            u'wizard-targ\N{Euro Sign}t',
 
789
                            u'\N{Euro Sign}wizard2',
 
790
                            u'b\N{Euro Sign}hind_curtain')
 
791
 
 
792
    def test_unable_create_symlink(self):
 
793
        def tt_helper():
 
794
            wt = self.make_branch_and_tree('.')
 
795
            tt = TreeTransform(wt)  # TreeTransform obtains write lock
 
796
            try:
 
797
                tt.new_symlink('foo', tt.root, 'bar')
 
798
                tt.apply()
 
799
            finally:
 
800
                wt.unlock()
 
801
        os_symlink = getattr(os, 'symlink', None)
 
802
        os.symlink = None
 
803
        try:
 
804
            err = self.assertRaises(errors.UnableCreateSymlink, tt_helper)
 
805
            self.assertEqual(
 
806
                "Unable to create symlink 'foo' on this platform",
 
807
                str(err))
 
808
        finally:
 
809
            if os_symlink:
 
810
                os.symlink = os_symlink
 
811
 
 
812
    def get_conflicted(self):
 
813
        create,root = self.get_transform()
 
814
        create.new_file('dorothy', root, 'dorothy', 'dorothy-id')
 
815
        oz = create.new_directory('oz', root, 'oz-id')
 
816
        create.new_directory('emeraldcity', oz, 'emerald-id')
 
817
        create.apply()
 
818
        conflicts,root = self.get_transform()
 
819
        # set up duplicate entry, duplicate id
 
820
        new_dorothy = conflicts.new_file('dorothy', root, 'dorothy',
 
821
                                         'dorothy-id')
 
822
        old_dorothy = conflicts.trans_id_tree_file_id('dorothy-id')
 
823
        oz = conflicts.trans_id_tree_file_id('oz-id')
 
824
        # set up DeletedParent parent conflict
 
825
        conflicts.delete_versioned(oz)
 
826
        emerald = conflicts.trans_id_tree_file_id('emerald-id')
 
827
        # set up MissingParent conflict
 
828
        munchkincity = conflicts.trans_id_file_id('munchkincity-id')
 
829
        conflicts.adjust_path('munchkincity', root, munchkincity)
 
830
        conflicts.new_directory('auntem', munchkincity, 'auntem-id')
 
831
        # set up parent loop
 
832
        conflicts.adjust_path('emeraldcity', emerald, emerald)
 
833
        return conflicts, emerald, oz, old_dorothy, new_dorothy
 
834
 
 
835
    def test_conflict_resolution(self):
 
836
        conflicts, emerald, oz, old_dorothy, new_dorothy =\
 
837
            self.get_conflicted()
 
838
        resolve_conflicts(conflicts)
 
839
        self.assertEqual(conflicts.final_name(old_dorothy), 'dorothy.moved')
 
840
        self.assertIs(conflicts.final_file_id(old_dorothy), None)
 
841
        self.assertEqual(conflicts.final_name(new_dorothy), 'dorothy')
 
842
        self.assertEqual(conflicts.final_file_id(new_dorothy), 'dorothy-id')
 
843
        self.assertEqual(conflicts.final_parent(emerald), oz)
 
844
        conflicts.apply()
 
845
 
 
846
    def test_cook_conflicts(self):
 
847
        tt, emerald, oz, old_dorothy, new_dorothy = self.get_conflicted()
 
848
        raw_conflicts = resolve_conflicts(tt)
 
849
        cooked_conflicts = cook_conflicts(raw_conflicts, tt)
 
850
        duplicate = DuplicateEntry('Moved existing file to', 'dorothy.moved',
 
851
                                   'dorothy', None, 'dorothy-id')
 
852
        self.assertEqual(cooked_conflicts[0], duplicate)
 
853
        duplicate_id = DuplicateID('Unversioned existing file',
 
854
                                   'dorothy.moved', 'dorothy', None,
 
855
                                   'dorothy-id')
 
856
        self.assertEqual(cooked_conflicts[1], duplicate_id)
 
857
        missing_parent = MissingParent('Created directory', 'munchkincity',
 
858
                                       'munchkincity-id')
 
859
        deleted_parent = DeletingParent('Not deleting', 'oz', 'oz-id')
 
860
        self.assertEqual(cooked_conflicts[2], missing_parent)
 
861
        unversioned_parent = UnversionedParent('Versioned directory',
 
862
                                               'munchkincity',
 
863
                                               'munchkincity-id')
 
864
        unversioned_parent2 = UnversionedParent('Versioned directory', 'oz',
 
865
                                               'oz-id')
 
866
        self.assertEqual(cooked_conflicts[3], unversioned_parent)
 
867
        parent_loop = ParentLoop('Cancelled move', 'oz/emeraldcity',
 
868
                                 'oz/emeraldcity', 'emerald-id', 'emerald-id')
 
869
        self.assertEqual(cooked_conflicts[4], deleted_parent)
 
870
        self.assertEqual(cooked_conflicts[5], unversioned_parent2)
 
871
        self.assertEqual(cooked_conflicts[6], parent_loop)
 
872
        self.assertEqual(len(cooked_conflicts), 7)
 
873
        tt.finalize()
 
874
 
 
875
    def test_string_conflicts(self):
 
876
        tt, emerald, oz, old_dorothy, new_dorothy = self.get_conflicted()
 
877
        raw_conflicts = resolve_conflicts(tt)
 
878
        cooked_conflicts = cook_conflicts(raw_conflicts, tt)
 
879
        tt.finalize()
 
880
        conflicts_s = [unicode(c) for c in cooked_conflicts]
 
881
        self.assertEqual(len(cooked_conflicts), len(conflicts_s))
 
882
        self.assertEqual(conflicts_s[0], 'Conflict adding file dorothy.  '
 
883
                                         'Moved existing file to '
 
884
                                         'dorothy.moved.')
 
885
        self.assertEqual(conflicts_s[1], 'Conflict adding id to dorothy.  '
 
886
                                         'Unversioned existing file '
 
887
                                         'dorothy.moved.')
 
888
        self.assertEqual(conflicts_s[2], 'Conflict adding files to'
 
889
                                         ' munchkincity.  Created directory.')
 
890
        self.assertEqual(conflicts_s[3], 'Conflict because munchkincity is not'
 
891
                                         ' versioned, but has versioned'
 
892
                                         ' children.  Versioned directory.')
 
893
        self.assertEqualDiff(conflicts_s[4], "Conflict: can't delete oz because it"
 
894
                                         " is not empty.  Not deleting.")
 
895
        self.assertEqual(conflicts_s[5], 'Conflict because oz is not'
 
896
                                         ' versioned, but has versioned'
 
897
                                         ' children.  Versioned directory.')
 
898
        self.assertEqual(conflicts_s[6], 'Conflict moving oz/emeraldcity into'
 
899
                                         ' oz/emeraldcity. Cancelled move.')
 
900
 
 
901
    def prepare_wrong_parent_kind(self):
 
902
        tt, root = self.get_transform()
 
903
        tt.new_file('parent', root, 'contents', 'parent-id')
 
904
        tt.apply()
 
905
        tt, root = self.get_transform()
 
906
        parent_id = tt.trans_id_file_id('parent-id')
 
907
        tt.new_file('child,', parent_id, 'contents2', 'file-id')
 
908
        return tt
 
909
 
 
910
    def test_find_conflicts_wrong_parent_kind(self):
 
911
        tt = self.prepare_wrong_parent_kind()
 
912
        tt.find_conflicts()
 
913
 
 
914
    def test_resolve_conflicts_wrong_existing_parent_kind(self):
 
915
        tt = self.prepare_wrong_parent_kind()
 
916
        raw_conflicts = resolve_conflicts(tt)
 
917
        self.assertEqual({('non-directory parent', 'Created directory',
 
918
                         'new-3')}, raw_conflicts)
 
919
        cooked_conflicts = cook_conflicts(raw_conflicts, tt)
 
920
        self.assertEqual([NonDirectoryParent('Created directory', 'parent.new',
 
921
        'parent-id')], cooked_conflicts)
 
922
        tt.apply()
 
923
        self.assertEqual(None, self.wt.path2id('parent'))
 
924
        self.assertEqual('parent-id', self.wt.path2id('parent.new'))
 
925
 
 
926
    def test_resolve_conflicts_wrong_new_parent_kind(self):
 
927
        tt, root = self.get_transform()
 
928
        parent_id = tt.new_directory('parent', root, 'parent-id')
 
929
        tt.new_file('child,', parent_id, 'contents2', 'file-id')
 
930
        tt.apply()
 
931
        tt, root = self.get_transform()
 
932
        parent_id = tt.trans_id_file_id('parent-id')
 
933
        tt.delete_contents(parent_id)
 
934
        tt.create_file('contents', parent_id)
 
935
        raw_conflicts = resolve_conflicts(tt)
 
936
        self.assertEqual({('non-directory parent', 'Created directory',
 
937
                         'new-3')}, raw_conflicts)
 
938
        tt.apply()
 
939
        self.assertEqual(None, self.wt.path2id('parent'))
 
940
        self.assertEqual('parent-id', self.wt.path2id('parent.new'))
 
941
 
 
942
    def test_resolve_conflicts_wrong_parent_kind_unversioned(self):
 
943
        tt, root = self.get_transform()
 
944
        parent_id = tt.new_directory('parent', root)
 
945
        tt.new_file('child,', parent_id, 'contents2')
 
946
        tt.apply()
 
947
        tt, root = self.get_transform()
 
948
        parent_id = tt.trans_id_tree_path('parent')
 
949
        tt.delete_contents(parent_id)
 
950
        tt.create_file('contents', parent_id)
 
951
        resolve_conflicts(tt)
 
952
        tt.apply()
 
953
        self.assertIs(None, self.wt.path2id('parent'))
 
954
        self.assertIs(None, self.wt.path2id('parent.new'))
 
955
 
 
956
    def test_resolve_conflicts_missing_parent(self):
 
957
        wt = self.make_branch_and_tree('.')
 
958
        tt = TreeTransform(wt)
 
959
        self.addCleanup(tt.finalize)
 
960
        parent = tt.trans_id_file_id('parent-id')
 
961
        tt.new_file('file', parent, 'Contents')
 
962
        raw_conflicts = resolve_conflicts(tt)
 
963
        # Since the directory doesn't exist it's seen as 'missing'.  So
 
964
        # 'resolve_conflicts' create a conflict asking for it to be created.
 
965
        self.assertLength(1, raw_conflicts)
 
966
        self.assertEqual(('missing parent', 'Created directory', 'new-1'),
 
967
                         raw_conflicts.pop())
 
968
        # apply fail since the missing directory doesn't exist
 
969
        self.assertRaises(errors.NoFinalPath, tt.apply)
 
970
 
 
971
    def test_moving_versioned_directories(self):
 
972
        create, root = self.get_transform()
 
973
        kansas = create.new_directory('kansas', root, 'kansas-id')
 
974
        create.new_directory('house', kansas, 'house-id')
 
975
        create.new_directory('oz', root, 'oz-id')
 
976
        create.apply()
 
977
        cyclone, root = self.get_transform()
 
978
        oz = cyclone.trans_id_tree_file_id('oz-id')
 
979
        house = cyclone.trans_id_tree_file_id('house-id')
 
980
        cyclone.adjust_path('house', oz, house)
 
981
        cyclone.apply()
 
982
 
 
983
    def test_moving_root(self):
 
984
        create, root = self.get_transform()
 
985
        fun = create.new_directory('fun', root, 'fun-id')
 
986
        create.new_directory('sun', root, 'sun-id')
 
987
        create.new_directory('moon', root, 'moon')
 
988
        create.apply()
 
989
        transform, root = self.get_transform()
 
990
        transform.adjust_root_path('oldroot', fun)
 
991
        new_root = transform.trans_id_tree_path('')
 
992
        transform.version_file('new-root', new_root)
 
993
        transform.apply()
 
994
 
 
995
    def test_renames(self):
 
996
        create, root = self.get_transform()
 
997
        old = create.new_directory('old-parent', root, 'old-id')
 
998
        intermediate = create.new_directory('intermediate', old, 'im-id')
 
999
        myfile = create.new_file('myfile', intermediate, 'myfile-text',
 
1000
                                 'myfile-id')
 
1001
        create.apply()
 
1002
        rename, root = self.get_transform()
 
1003
        old = rename.trans_id_file_id('old-id')
 
1004
        rename.adjust_path('new', root, old)
 
1005
        myfile = rename.trans_id_file_id('myfile-id')
 
1006
        rename.set_executability(True, myfile)
 
1007
        rename.apply()
 
1008
 
 
1009
    def test_rename_fails(self):
 
1010
        self.requireFeature(features.not_running_as_root)
 
1011
        # see https://bugs.launchpad.net/bzr/+bug/491763
 
1012
        create, root_id = self.get_transform()
 
1013
        first_dir = create.new_directory('first-dir', root_id, 'first-id')
 
1014
        myfile = create.new_file('myfile', root_id, 'myfile-text',
 
1015
                                 'myfile-id')
 
1016
        create.apply()
 
1017
        if os.name == "posix" and sys.platform != "cygwin":
 
1018
            # posix filesystems fail on renaming if the readonly bit is set
 
1019
            osutils.make_readonly(self.wt.abspath('first-dir'))
 
1020
        elif os.name == "nt":
 
1021
            # windows filesystems fail on renaming open files
 
1022
            self.addCleanup(file(self.wt.abspath('myfile')).close)
 
1023
        else:
 
1024
            self.skipTest("Can't force a permissions error on rename")
 
1025
        # now transform to rename
 
1026
        rename_transform, root_id = self.get_transform()
 
1027
        file_trans_id = rename_transform.trans_id_file_id('myfile-id')
 
1028
        dir_id = rename_transform.trans_id_file_id('first-id')
 
1029
        rename_transform.adjust_path('newname', dir_id, file_trans_id)
 
1030
        e = self.assertRaises(errors.TransformRenameFailed,
 
1031
            rename_transform.apply)
 
1032
        # On nix looks like: 
 
1033
        # "Failed to rename .../work/.bzr/checkout/limbo/new-1
 
1034
        # to .../first-dir/newname: [Errno 13] Permission denied"
 
1035
        # On windows looks like:
 
1036
        # "Failed to rename .../work/myfile to 
 
1037
        # .../work/.bzr/checkout/limbo/new-1: [Errno 13] Permission denied"
 
1038
        # This test isn't concerned with exactly what the error looks like,
 
1039
        # and the strerror will vary across OS and locales, but the assert
 
1040
        # that the exeception attributes are what we expect
 
1041
        self.assertEqual(e.errno, errno.EACCES)
 
1042
        if os.name == "posix":
 
1043
            self.assertEndsWith(e.to_path, "/first-dir/newname")
 
1044
        else:
 
1045
            self.assertEqual(os.path.basename(e.from_path), "myfile")
 
1046
 
 
1047
    def test_set_executability_order(self):
 
1048
        """Ensure that executability behaves the same, no matter what order.
 
1049
 
 
1050
        - create file and set executability simultaneously
 
1051
        - create file and set executability afterward
 
1052
        - unsetting the executability of a file whose executability has not been
 
1053
        declared should throw an exception (this may happen when a
 
1054
        merge attempts to create a file with a duplicate ID)
 
1055
        """
 
1056
        transform, root = self.get_transform()
 
1057
        wt = transform._tree
 
1058
        wt.lock_read()
 
1059
        self.addCleanup(wt.unlock)
 
1060
        transform.new_file('set_on_creation', root, 'Set on creation', 'soc',
 
1061
                           True)
 
1062
        sac = transform.new_file('set_after_creation', root,
 
1063
                                 'Set after creation', 'sac')
 
1064
        transform.set_executability(True, sac)
 
1065
        uws = transform.new_file('unset_without_set', root, 'Unset badly',
 
1066
                                 'uws')
 
1067
        self.assertRaises(KeyError, transform.set_executability, None, uws)
 
1068
        transform.apply()
 
1069
        self.assertTrue(wt.is_executable('soc'))
 
1070
        self.assertTrue(wt.is_executable('sac'))
 
1071
 
 
1072
    def test_preserve_mode(self):
 
1073
        """File mode is preserved when replacing content"""
 
1074
        if sys.platform == 'win32':
 
1075
            raise TestSkipped('chmod has no effect on win32')
 
1076
        transform, root = self.get_transform()
 
1077
        transform.new_file('file1', root, 'contents', 'file1-id', True)
 
1078
        transform.apply()
 
1079
        self.wt.lock_write()
 
1080
        self.addCleanup(self.wt.unlock)
 
1081
        self.assertTrue(self.wt.is_executable('file1-id'))
 
1082
        transform, root = self.get_transform()
 
1083
        file1_id = transform.trans_id_tree_file_id('file1-id')
 
1084
        transform.delete_contents(file1_id)
 
1085
        transform.create_file('contents2', file1_id)
 
1086
        transform.apply()
 
1087
        self.assertTrue(self.wt.is_executable('file1-id'))
 
1088
 
 
1089
    def test__set_mode_stats_correctly(self):
 
1090
        """_set_mode stats to determine file mode."""
 
1091
        if sys.platform == 'win32':
 
1092
            raise TestSkipped('chmod has no effect on win32')
 
1093
 
 
1094
        stat_paths = []
 
1095
        real_stat = os.stat
 
1096
        def instrumented_stat(path):
 
1097
            stat_paths.append(path)
 
1098
            return real_stat(path)
 
1099
 
 
1100
        transform, root = self.get_transform()
 
1101
 
 
1102
        bar1_id = transform.new_file('bar', root, 'bar contents 1\n',
 
1103
                                     file_id='bar-id-1', executable=False)
 
1104
        transform.apply()
 
1105
 
 
1106
        transform, root = self.get_transform()
 
1107
        bar1_id = transform.trans_id_tree_path('bar')
 
1108
        bar2_id = transform.trans_id_tree_path('bar2')
 
1109
        try:
 
1110
            os.stat = instrumented_stat
 
1111
            transform.create_file('bar2 contents\n', bar2_id, mode_id=bar1_id)
 
1112
        finally:
 
1113
            os.stat = real_stat
 
1114
            transform.finalize()
 
1115
 
 
1116
        bar1_abspath = self.wt.abspath('bar')
 
1117
        self.assertEqual([bar1_abspath], stat_paths)
 
1118
 
 
1119
    def test_iter_changes(self):
 
1120
        self.wt.set_root_id('eert_toor')
 
1121
        transform, root = self.get_transform()
 
1122
        transform.new_file('old', root, 'blah', 'id-1', True)
 
1123
        transform.apply()
 
1124
        transform, root = self.get_transform()
 
1125
        try:
 
1126
            self.assertEqual([], list(transform.iter_changes()))
 
1127
            old = transform.trans_id_tree_file_id('id-1')
 
1128
            transform.unversion_file(old)
 
1129
            self.assertEqual([('id-1', ('old', None), False, (True, False),
 
1130
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
1131
                (True, True))], list(transform.iter_changes()))
 
1132
            transform.new_directory('new', root, 'id-1')
 
1133
            self.assertEqual([('id-1', ('old', 'new'), True, (True, True),
 
1134
                ('eert_toor', 'eert_toor'), ('old', 'new'),
 
1135
                ('file', 'directory'),
 
1136
                (True, False))], list(transform.iter_changes()))
 
1137
        finally:
 
1138
            transform.finalize()
 
1139
 
 
1140
    def test_iter_changes_new(self):
 
1141
        self.wt.set_root_id('eert_toor')
 
1142
        transform, root = self.get_transform()
 
1143
        transform.new_file('old', root, 'blah')
 
1144
        transform.apply()
 
1145
        transform, root = self.get_transform()
 
1146
        try:
 
1147
            old = transform.trans_id_tree_path('old')
 
1148
            transform.version_file('id-1', old)
 
1149
            self.assertEqual([('id-1', (None, 'old'), False, (False, True),
 
1150
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
1151
                (False, False))], list(transform.iter_changes()))
 
1152
        finally:
 
1153
            transform.finalize()
 
1154
 
 
1155
    def test_iter_changes_modifications(self):
 
1156
        self.wt.set_root_id('eert_toor')
 
1157
        transform, root = self.get_transform()
 
1158
        transform.new_file('old', root, 'blah', 'id-1')
 
1159
        transform.new_file('new', root, 'blah')
 
1160
        transform.new_directory('subdir', root, 'subdir-id')
 
1161
        transform.apply()
 
1162
        transform, root = self.get_transform()
 
1163
        try:
 
1164
            old = transform.trans_id_tree_path('old')
 
1165
            subdir = transform.trans_id_tree_file_id('subdir-id')
 
1166
            new = transform.trans_id_tree_path('new')
 
1167
            self.assertEqual([], list(transform.iter_changes()))
 
1168
 
 
1169
            #content deletion
 
1170
            transform.delete_contents(old)
 
1171
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
 
1172
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', None),
 
1173
                (False, False))], list(transform.iter_changes()))
 
1174
 
 
1175
            #content change
 
1176
            transform.create_file('blah', old)
 
1177
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
 
1178
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
1179
                (False, False))], list(transform.iter_changes()))
 
1180
            transform.cancel_deletion(old)
 
1181
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
 
1182
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
1183
                (False, False))], list(transform.iter_changes()))
 
1184
            transform.cancel_creation(old)
 
1185
 
 
1186
            # move file_id to a different file
 
1187
            self.assertEqual([], list(transform.iter_changes()))
 
1188
            transform.unversion_file(old)
 
1189
            transform.version_file('id-1', new)
 
1190
            transform.adjust_path('old', root, new)
 
1191
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
 
1192
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
1193
                (False, False))], list(transform.iter_changes()))
 
1194
            transform.cancel_versioning(new)
 
1195
            transform._removed_id = set()
 
1196
 
 
1197
            #execute bit
 
1198
            self.assertEqual([], list(transform.iter_changes()))
 
1199
            transform.set_executability(True, old)
 
1200
            self.assertEqual([('id-1', ('old', 'old'), False, (True, True),
 
1201
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
1202
                (False, True))], list(transform.iter_changes()))
 
1203
            transform.set_executability(None, old)
 
1204
 
 
1205
            # filename
 
1206
            self.assertEqual([], list(transform.iter_changes()))
 
1207
            transform.adjust_path('new', root, old)
 
1208
            transform._new_parent = {}
 
1209
            self.assertEqual([('id-1', ('old', 'new'), False, (True, True),
 
1210
                ('eert_toor', 'eert_toor'), ('old', 'new'), ('file', 'file'),
 
1211
                (False, False))], list(transform.iter_changes()))
 
1212
            transform._new_name = {}
 
1213
 
 
1214
            # parent directory
 
1215
            self.assertEqual([], list(transform.iter_changes()))
 
1216
            transform.adjust_path('new', subdir, old)
 
1217
            transform._new_name = {}
 
1218
            self.assertEqual([('id-1', ('old', 'subdir/old'), False,
 
1219
                (True, True), ('eert_toor', 'subdir-id'), ('old', 'old'),
 
1220
                ('file', 'file'), (False, False))],
 
1221
                list(transform.iter_changes()))
 
1222
            transform._new_path = {}
 
1223
 
 
1224
        finally:
 
1225
            transform.finalize()
 
1226
 
 
1227
    def test_iter_changes_modified_bleed(self):
 
1228
        self.wt.set_root_id('eert_toor')
 
1229
        """Modified flag should not bleed from one change to another"""
 
1230
        # unfortunately, we have no guarantee that file1 (which is modified)
 
1231
        # will be applied before file2.  And if it's applied after file2, it
 
1232
        # obviously can't bleed into file2's change output.  But for now, it
 
1233
        # works.
 
1234
        transform, root = self.get_transform()
 
1235
        transform.new_file('file1', root, 'blah', 'id-1')
 
1236
        transform.new_file('file2', root, 'blah', 'id-2')
 
1237
        transform.apply()
 
1238
        transform, root = self.get_transform()
 
1239
        try:
 
1240
            transform.delete_contents(transform.trans_id_file_id('id-1'))
 
1241
            transform.set_executability(True,
 
1242
            transform.trans_id_file_id('id-2'))
 
1243
            self.assertEqual([('id-1', (u'file1', u'file1'), True, (True, True),
 
1244
                ('eert_toor', 'eert_toor'), ('file1', u'file1'),
 
1245
                ('file', None), (False, False)),
 
1246
                ('id-2', (u'file2', u'file2'), False, (True, True),
 
1247
                ('eert_toor', 'eert_toor'), ('file2', u'file2'),
 
1248
                ('file', 'file'), (False, True))],
 
1249
                list(transform.iter_changes()))
 
1250
        finally:
 
1251
            transform.finalize()
 
1252
 
 
1253
    def test_iter_changes_move_missing(self):
 
1254
        """Test moving ids with no files around"""
 
1255
        self.wt.set_root_id('toor_eert')
 
1256
        # Need two steps because versioning a non-existant file is a conflict.
 
1257
        transform, root = self.get_transform()
 
1258
        transform.new_directory('floater', root, 'floater-id')
 
1259
        transform.apply()
 
1260
        transform, root = self.get_transform()
 
1261
        transform.delete_contents(transform.trans_id_tree_path('floater'))
 
1262
        transform.apply()
 
1263
        transform, root = self.get_transform()
 
1264
        floater = transform.trans_id_tree_path('floater')
 
1265
        try:
 
1266
            transform.adjust_path('flitter', root, floater)
 
1267
            self.assertEqual([('floater-id', ('floater', 'flitter'), False,
 
1268
            (True, True), ('toor_eert', 'toor_eert'), ('floater', 'flitter'),
 
1269
            (None, None), (False, False))], list(transform.iter_changes()))
 
1270
        finally:
 
1271
            transform.finalize()
 
1272
 
 
1273
    def test_iter_changes_pointless(self):
 
1274
        """Ensure that no-ops are not treated as modifications"""
 
1275
        self.wt.set_root_id('eert_toor')
 
1276
        transform, root = self.get_transform()
 
1277
        transform.new_file('old', root, 'blah', 'id-1')
 
1278
        transform.new_directory('subdir', root, 'subdir-id')
 
1279
        transform.apply()
 
1280
        transform, root = self.get_transform()
 
1281
        try:
 
1282
            old = transform.trans_id_tree_path('old')
 
1283
            subdir = transform.trans_id_tree_file_id('subdir-id')
 
1284
            self.assertEqual([], list(transform.iter_changes()))
 
1285
            transform.delete_contents(subdir)
 
1286
            transform.create_directory(subdir)
 
1287
            transform.set_executability(False, old)
 
1288
            transform.unversion_file(old)
 
1289
            transform.version_file('id-1', old)
 
1290
            transform.adjust_path('old', root, old)
 
1291
            self.assertEqual([], list(transform.iter_changes()))
 
1292
        finally:
 
1293
            transform.finalize()
 
1294
 
 
1295
    def test_rename_count(self):
 
1296
        transform, root = self.get_transform()
 
1297
        transform.new_file('name1', root, 'contents')
 
1298
        self.assertEqual(transform.rename_count, 0)
 
1299
        transform.apply()
 
1300
        self.assertEqual(transform.rename_count, 1)
 
1301
        transform2, root = self.get_transform()
 
1302
        transform2.adjust_path('name2', root,
 
1303
                               transform2.trans_id_tree_path('name1'))
 
1304
        self.assertEqual(transform2.rename_count, 0)
 
1305
        transform2.apply()
 
1306
        self.assertEqual(transform2.rename_count, 2)
 
1307
 
 
1308
    def test_change_parent(self):
 
1309
        """Ensure that after we change a parent, the results are still right.
 
1310
 
 
1311
        Renames and parent changes on pending transforms can happen as part
 
1312
        of conflict resolution, and are explicitly permitted by the
 
1313
        TreeTransform API.
 
1314
 
 
1315
        This test ensures they work correctly with the rename-avoidance
 
1316
        optimization.
 
1317
        """
 
1318
        transform, root = self.get_transform()
 
1319
        parent1 = transform.new_directory('parent1', root)
 
1320
        child1 = transform.new_file('child1', parent1, 'contents')
 
1321
        parent2 = transform.new_directory('parent2', root)
 
1322
        transform.adjust_path('child1', parent2, child1)
 
1323
        transform.apply()
 
1324
        self.assertPathDoesNotExist(self.wt.abspath('parent1/child1'))
 
1325
        self.assertPathExists(self.wt.abspath('parent2/child1'))
 
1326
        # rename limbo/new-1 => parent1, rename limbo/new-3 => parent2
 
1327
        # no rename for child1 (counting only renames during apply)
 
1328
        self.assertEqual(2, transform.rename_count)
 
1329
 
 
1330
    def test_cancel_parent(self):
 
1331
        """Cancelling a parent doesn't cause deletion of a non-empty directory
 
1332
 
 
1333
        This is like the test_change_parent, except that we cancel the parent
 
1334
        before adjusting the path.  The transform must detect that the
 
1335
        directory is non-empty, and move children to safe locations.
 
1336
        """
 
1337
        transform, root = self.get_transform()
 
1338
        parent1 = transform.new_directory('parent1', root)
 
1339
        child1 = transform.new_file('child1', parent1, 'contents')
 
1340
        child2 = transform.new_file('child2', parent1, 'contents')
 
1341
        try:
 
1342
            transform.cancel_creation(parent1)
 
1343
        except OSError:
 
1344
            self.fail('Failed to move child1 before deleting parent1')
 
1345
        transform.cancel_creation(child2)
 
1346
        transform.create_directory(parent1)
 
1347
        try:
 
1348
            transform.cancel_creation(parent1)
 
1349
        # If the transform incorrectly believes that child2 is still in
 
1350
        # parent1's limbo directory, it will try to rename it and fail
 
1351
        # because was already moved by the first cancel_creation.
 
1352
        except OSError:
 
1353
            self.fail('Transform still thinks child2 is a child of parent1')
 
1354
        parent2 = transform.new_directory('parent2', root)
 
1355
        transform.adjust_path('child1', parent2, child1)
 
1356
        transform.apply()
 
1357
        self.assertPathDoesNotExist(self.wt.abspath('parent1'))
 
1358
        self.assertPathExists(self.wt.abspath('parent2/child1'))
 
1359
        # rename limbo/new-3 => parent2, rename limbo/new-2 => child1
 
1360
        self.assertEqual(2, transform.rename_count)
 
1361
 
 
1362
    def test_adjust_and_cancel(self):
 
1363
        """Make sure adjust_path keeps track of limbo children properly"""
 
1364
        transform, root = self.get_transform()
 
1365
        parent1 = transform.new_directory('parent1', root)
 
1366
        child1 = transform.new_file('child1', parent1, 'contents')
 
1367
        parent2 = transform.new_directory('parent2', root)
 
1368
        transform.adjust_path('child1', parent2, child1)
 
1369
        transform.cancel_creation(child1)
 
1370
        try:
 
1371
            transform.cancel_creation(parent1)
 
1372
        # if the transform thinks child1 is still in parent1's limbo
 
1373
        # directory, it will attempt to move it and fail.
 
1374
        except OSError:
 
1375
            self.fail('Transform still thinks child1 is a child of parent1')
 
1376
        transform.finalize()
 
1377
 
 
1378
    def test_noname_contents(self):
 
1379
        """TreeTransform should permit deferring naming files."""
 
1380
        transform, root = self.get_transform()
 
1381
        parent = transform.trans_id_file_id('parent-id')
 
1382
        try:
 
1383
            transform.create_directory(parent)
 
1384
        except KeyError:
 
1385
            self.fail("Can't handle contents with no name")
 
1386
        transform.finalize()
 
1387
 
 
1388
    def test_noname_contents_nested(self):
 
1389
        """TreeTransform should permit deferring naming files."""
 
1390
        transform, root = self.get_transform()
 
1391
        parent = transform.trans_id_file_id('parent-id')
 
1392
        try:
 
1393
            transform.create_directory(parent)
 
1394
        except KeyError:
 
1395
            self.fail("Can't handle contents with no name")
 
1396
        child = transform.new_directory('child', parent)
 
1397
        transform.adjust_path('parent', root, parent)
 
1398
        transform.apply()
 
1399
        self.assertPathExists(self.wt.abspath('parent/child'))
 
1400
        self.assertEqual(1, transform.rename_count)
 
1401
 
 
1402
    def test_reuse_name(self):
 
1403
        """Avoid reusing the same limbo name for different files"""
 
1404
        transform, root = self.get_transform()
 
1405
        parent = transform.new_directory('parent', root)
 
1406
        child1 = transform.new_directory('child', parent)
 
1407
        try:
 
1408
            child2 = transform.new_directory('child', parent)
 
1409
        except OSError:
 
1410
            self.fail('Tranform tried to use the same limbo name twice')
 
1411
        transform.adjust_path('child2', parent, child2)
 
1412
        transform.apply()
 
1413
        # limbo/new-1 => parent, limbo/new-3 => parent/child2
 
1414
        # child2 is put into top-level limbo because child1 has already
 
1415
        # claimed the direct limbo path when child2 is created.  There is no
 
1416
        # advantage in renaming files once they're in top-level limbo, except
 
1417
        # as part of apply.
 
1418
        self.assertEqual(2, transform.rename_count)
 
1419
 
 
1420
    def test_reuse_when_first_moved(self):
 
1421
        """Don't avoid direct paths when it is safe to use them"""
 
1422
        transform, root = self.get_transform()
 
1423
        parent = transform.new_directory('parent', root)
 
1424
        child1 = transform.new_directory('child', parent)
 
1425
        transform.adjust_path('child1', parent, child1)
 
1426
        child2 = transform.new_directory('child', parent)
 
1427
        transform.apply()
 
1428
        # limbo/new-1 => parent
 
1429
        self.assertEqual(1, transform.rename_count)
 
1430
 
 
1431
    def test_reuse_after_cancel(self):
 
1432
        """Don't avoid direct paths when it is safe to use them"""
 
1433
        transform, root = self.get_transform()
 
1434
        parent2 = transform.new_directory('parent2', root)
 
1435
        child1 = transform.new_directory('child1', parent2)
 
1436
        transform.cancel_creation(parent2)
 
1437
        transform.create_directory(parent2)
 
1438
        child2 = transform.new_directory('child1', parent2)
 
1439
        transform.adjust_path('child2', parent2, child1)
 
1440
        transform.apply()
 
1441
        # limbo/new-1 => parent2, limbo/new-2 => parent2/child1
 
1442
        self.assertEqual(2, transform.rename_count)
 
1443
 
 
1444
    def test_finalize_order(self):
 
1445
        """Finalize must be done in child-to-parent order"""
 
1446
        transform, root = self.get_transform()
 
1447
        parent = transform.new_directory('parent', root)
 
1448
        child = transform.new_directory('child', parent)
 
1449
        try:
 
1450
            transform.finalize()
 
1451
        except OSError:
 
1452
            self.fail('Tried to remove parent before child1')
 
1453
 
 
1454
    def test_cancel_with_cancelled_child_should_succeed(self):
 
1455
        transform, root = self.get_transform()
 
1456
        parent = transform.new_directory('parent', root)
 
1457
        child = transform.new_directory('child', parent)
 
1458
        transform.cancel_creation(child)
 
1459
        transform.cancel_creation(parent)
 
1460
        transform.finalize()
 
1461
 
 
1462
    def test_rollback_on_directory_clash(self):
 
1463
        def tt_helper():
 
1464
            wt = self.make_branch_and_tree('.')
 
1465
            tt = TreeTransform(wt)  # TreeTransform obtains write lock
 
1466
            try:
 
1467
                foo = tt.new_directory('foo', tt.root)
 
1468
                tt.new_file('bar', foo, 'foobar')
 
1469
                baz = tt.new_directory('baz', tt.root)
 
1470
                tt.new_file('qux', baz, 'quux')
 
1471
                # Ask for a rename 'foo' -> 'baz'
 
1472
                tt.adjust_path('baz', tt.root, foo)
 
1473
                # Lie to tt that we've already resolved all conflicts.
 
1474
                tt.apply(no_conflicts=True)
 
1475
            except:
 
1476
                wt.unlock()
 
1477
                raise
 
1478
        # The rename will fail because the target directory is not empty (but
 
1479
        # raises FileExists anyway).
 
1480
        err = self.assertRaises(errors.FileExists, tt_helper)
 
1481
        self.assertEndsWith(err.path, "/baz")
 
1482
 
 
1483
    def test_two_directories_clash(self):
 
1484
        def tt_helper():
 
1485
            wt = self.make_branch_and_tree('.')
 
1486
            tt = TreeTransform(wt)  # TreeTransform obtains write lock
 
1487
            try:
 
1488
                foo_1 = tt.new_directory('foo', tt.root)
 
1489
                tt.new_directory('bar', foo_1)
 
1490
                # Adding the same directory with a different content
 
1491
                foo_2 = tt.new_directory('foo', tt.root)
 
1492
                tt.new_directory('baz', foo_2)
 
1493
                # Lie to tt that we've already resolved all conflicts.
 
1494
                tt.apply(no_conflicts=True)
 
1495
            except:
 
1496
                wt.unlock()
 
1497
                raise
 
1498
        err = self.assertRaises(errors.FileExists, tt_helper)
 
1499
        self.assertEndsWith(err.path, "/foo")
 
1500
 
 
1501
    def test_two_directories_clash_finalize(self):
 
1502
        def tt_helper():
 
1503
            wt = self.make_branch_and_tree('.')
 
1504
            tt = TreeTransform(wt)  # TreeTransform obtains write lock
 
1505
            try:
 
1506
                foo_1 = tt.new_directory('foo', tt.root)
 
1507
                tt.new_directory('bar', foo_1)
 
1508
                # Adding the same directory with a different content
 
1509
                foo_2 = tt.new_directory('foo', tt.root)
 
1510
                tt.new_directory('baz', foo_2)
 
1511
                # Lie to tt that we've already resolved all conflicts.
 
1512
                tt.apply(no_conflicts=True)
 
1513
            except:
 
1514
                tt.finalize()
 
1515
                raise
 
1516
        err = self.assertRaises(errors.FileExists, tt_helper)
 
1517
        self.assertEndsWith(err.path, "/foo")
 
1518
 
 
1519
    def test_file_to_directory(self):
 
1520
        wt = self.make_branch_and_tree('.')
 
1521
        self.build_tree(['foo'])
 
1522
        wt.add(['foo'])
 
1523
        wt.commit("one")
 
1524
        tt = TreeTransform(wt)
 
1525
        self.addCleanup(tt.finalize)
 
1526
        foo_trans_id = tt.trans_id_tree_path("foo")
 
1527
        tt.delete_contents(foo_trans_id)
 
1528
        tt.create_directory(foo_trans_id)
 
1529
        bar_trans_id = tt.trans_id_tree_path("foo/bar")
 
1530
        tt.create_file(["aa\n"], bar_trans_id)
 
1531
        tt.version_file("bar-1", bar_trans_id)
 
1532
        tt.apply()
 
1533
        self.assertPathExists("foo/bar")
 
1534
        wt.lock_read()
 
1535
        try:
 
1536
            self.assertEqual(wt.kind(wt.path2id("foo")), "directory")
 
1537
        finally:
 
1538
            wt.unlock()
 
1539
        wt.commit("two")
 
1540
        changes = wt.changes_from(wt.basis_tree())
 
1541
        self.assertFalse(changes.has_changed(), changes)
 
1542
 
 
1543
    def test_file_to_symlink(self):
 
1544
        self.requireFeature(SymlinkFeature)
 
1545
        wt = self.make_branch_and_tree('.')
 
1546
        self.build_tree(['foo'])
 
1547
        wt.add(['foo'])
 
1548
        wt.commit("one")
 
1549
        tt = TreeTransform(wt)
 
1550
        self.addCleanup(tt.finalize)
 
1551
        foo_trans_id = tt.trans_id_tree_path("foo")
 
1552
        tt.delete_contents(foo_trans_id)
 
1553
        tt.create_symlink("bar", foo_trans_id)
 
1554
        tt.apply()
 
1555
        self.assertPathExists("foo")
 
1556
        wt.lock_read()
 
1557
        self.addCleanup(wt.unlock)
 
1558
        self.assertEqual(wt.kind(wt.path2id("foo")), "symlink")
 
1559
 
 
1560
    def test_dir_to_file(self):
 
1561
        wt = self.make_branch_and_tree('.')
 
1562
        self.build_tree(['foo/', 'foo/bar'])
 
1563
        wt.add(['foo', 'foo/bar'])
 
1564
        wt.commit("one")
 
1565
        tt = TreeTransform(wt)
 
1566
        self.addCleanup(tt.finalize)
 
1567
        foo_trans_id = tt.trans_id_tree_path("foo")
 
1568
        bar_trans_id = tt.trans_id_tree_path("foo/bar")
 
1569
        tt.delete_contents(foo_trans_id)
 
1570
        tt.delete_versioned(bar_trans_id)
 
1571
        tt.create_file(["aa\n"], foo_trans_id)
 
1572
        tt.apply()
 
1573
        self.assertPathExists("foo")
 
1574
        wt.lock_read()
 
1575
        self.addCleanup(wt.unlock)
 
1576
        self.assertEqual(wt.kind(wt.path2id("foo")), "file")
 
1577
 
 
1578
    def test_dir_to_hardlink(self):
 
1579
        self.requireFeature(HardlinkFeature)
 
1580
        wt = self.make_branch_and_tree('.')
 
1581
        self.build_tree(['foo/', 'foo/bar'])
 
1582
        wt.add(['foo', 'foo/bar'])
 
1583
        wt.commit("one")
 
1584
        tt = TreeTransform(wt)
 
1585
        self.addCleanup(tt.finalize)
 
1586
        foo_trans_id = tt.trans_id_tree_path("foo")
 
1587
        bar_trans_id = tt.trans_id_tree_path("foo/bar")
 
1588
        tt.delete_contents(foo_trans_id)
 
1589
        tt.delete_versioned(bar_trans_id)
 
1590
        self.build_tree(['baz'])
 
1591
        tt.create_hardlink("baz", foo_trans_id)
 
1592
        tt.apply()
 
1593
        self.assertPathExists("foo")
 
1594
        self.assertPathExists("baz")
 
1595
        wt.lock_read()
 
1596
        self.addCleanup(wt.unlock)
 
1597
        self.assertEqual(wt.kind(wt.path2id("foo")), "file")
 
1598
 
 
1599
    def test_no_final_path(self):
 
1600
        transform, root = self.get_transform()
 
1601
        trans_id = transform.trans_id_file_id('foo')
 
1602
        transform.create_file('bar', trans_id)
 
1603
        transform.cancel_creation(trans_id)
 
1604
        transform.apply()
 
1605
 
 
1606
    def test_create_from_tree(self):
 
1607
        tree1 = self.make_branch_and_tree('tree1')
 
1608
        self.build_tree_contents([('tree1/foo/',), ('tree1/bar', 'baz')])
 
1609
        tree1.add(['foo', 'bar'], ['foo-id', 'bar-id'])
 
1610
        tree2 = self.make_branch_and_tree('tree2')
 
1611
        tt = TreeTransform(tree2)
 
1612
        foo_trans_id = tt.create_path('foo', tt.root)
 
1613
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id')
 
1614
        bar_trans_id = tt.create_path('bar', tt.root)
 
1615
        create_from_tree(tt, bar_trans_id, tree1, 'bar-id')
 
1616
        tt.apply()
 
1617
        self.assertEqual('directory', osutils.file_kind('tree2/foo'))
 
1618
        self.assertFileEqual('baz', 'tree2/bar')
 
1619
 
 
1620
    def test_create_from_tree_bytes(self):
 
1621
        """Provided lines are used instead of tree content."""
 
1622
        tree1 = self.make_branch_and_tree('tree1')
 
1623
        self.build_tree_contents([('tree1/foo', 'bar'),])
 
1624
        tree1.add('foo', 'foo-id')
 
1625
        tree2 = self.make_branch_and_tree('tree2')
 
1626
        tt = TreeTransform(tree2)
 
1627
        foo_trans_id = tt.create_path('foo', tt.root)
 
1628
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id', bytes='qux')
 
1629
        tt.apply()
 
1630
        self.assertFileEqual('qux', 'tree2/foo')
 
1631
 
 
1632
    def test_create_from_tree_symlink(self):
 
1633
        self.requireFeature(SymlinkFeature)
 
1634
        tree1 = self.make_branch_and_tree('tree1')
 
1635
        os.symlink('bar', 'tree1/foo')
 
1636
        tree1.add('foo', 'foo-id')
 
1637
        tt = TreeTransform(self.make_branch_and_tree('tree2'))
 
1638
        foo_trans_id = tt.create_path('foo', tt.root)
 
1639
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id')
 
1640
        tt.apply()
 
1641
        self.assertEqual('bar', os.readlink('tree2/foo'))
 
1642
 
 
1643
 
89
1644
class TransformGroup(object):
90
1645
 
91
1646
    def __init__(self, dirname, root_id):
94
1649
        self.wt = ControlDir.create_standalone_workingtree(dirname)
95
1650
        self.wt.set_root_id(root_id)
96
1651
        self.b = self.wt.branch
97
 
        self.tt = self.wt.transform()
98
 
        self.root = self.tt.trans_id_tree_path('')
 
1652
        self.tt = TreeTransform(self.wt)
 
1653
        self.root = self.tt.trans_id_tree_file_id(self.wt.get_root_id())
99
1654
 
100
1655
 
101
1656
def conflict_text(tree, merge):
102
 
    template = b'%s TREE\n%s%s\n%s%s MERGE-SOURCE\n'
103
 
    return template % (b'<' * 7, tree, b'=' * 7, merge, b'>' * 7)
 
1657
    template = '%s TREE\n%s%s\n%s%s MERGE-SOURCE\n'
 
1658
    return template % ('<' * 7, tree, '=' * 7, merge, '>' * 7)
 
1659
 
 
1660
 
 
1661
class TestInventoryAltered(tests.TestCaseWithTransport):
 
1662
 
 
1663
    def test_inventory_altered_unchanged(self):
 
1664
        tree = self.make_branch_and_tree('tree')
 
1665
        self.build_tree(['tree/foo'])
 
1666
        tree.add('foo', 'foo-id')
 
1667
        with TransformPreview(tree) as tt:
 
1668
            self.assertEqual([], tt._inventory_altered())
 
1669
 
 
1670
    def test_inventory_altered_changed_parent_id(self):
 
1671
        tree = self.make_branch_and_tree('tree')
 
1672
        self.build_tree(['tree/foo'])
 
1673
        tree.add('foo', 'foo-id')
 
1674
        with TransformPreview(tree) as tt:
 
1675
            tt.unversion_file(tt.root)
 
1676
            tt.version_file('new-id', tt.root)
 
1677
            foo_trans_id = tt.trans_id_tree_file_id('foo-id')
 
1678
            foo_tuple = ('foo', foo_trans_id)
 
1679
            root_tuple = ('', tt.root)
 
1680
            self.assertEqual([root_tuple, foo_tuple], tt._inventory_altered())
 
1681
 
 
1682
    def test_inventory_altered_noop_changed_parent_id(self):
 
1683
        tree = self.make_branch_and_tree('tree')
 
1684
        self.build_tree(['tree/foo'])
 
1685
        tree.add('foo', 'foo-id')
 
1686
        with TransformPreview(tree) as tt:
 
1687
            tt.unversion_file(tt.root)
 
1688
            tt.version_file(tree.get_root_id(), tt.root)
 
1689
            foo_trans_id = tt.trans_id_tree_file_id('foo-id')
 
1690
            self.assertEqual([], tt._inventory_altered())
104
1691
 
105
1692
 
106
1693
class TestTransformMerge(TestCaseInTempDir):
108
1695
    def test_text_merge(self):
109
1696
        root_id = generate_ids.gen_root_id()
110
1697
        base = TransformGroup("base", root_id)
111
 
        base.tt.new_file('a', base.root, [b'a\nb\nc\nd\be\n'], b'a')
112
 
        base.tt.new_file('b', base.root, [b'b1'], b'b')
113
 
        base.tt.new_file('c', base.root, [b'c'], b'c')
114
 
        base.tt.new_file('d', base.root, [b'd'], b'd')
115
 
        base.tt.new_file('e', base.root, [b'e'], b'e')
116
 
        base.tt.new_file('f', base.root, [b'f'], b'f')
117
 
        base.tt.new_directory('g', base.root, b'g')
118
 
        base.tt.new_directory('h', base.root, b'h')
 
1698
        base.tt.new_file('a', base.root, 'a\nb\nc\nd\be\n', 'a')
 
1699
        base.tt.new_file('b', base.root, 'b1', 'b')
 
1700
        base.tt.new_file('c', base.root, 'c', 'c')
 
1701
        base.tt.new_file('d', base.root, 'd', 'd')
 
1702
        base.tt.new_file('e', base.root, 'e', 'e')
 
1703
        base.tt.new_file('f', base.root, 'f', 'f')
 
1704
        base.tt.new_directory('g', base.root, 'g')
 
1705
        base.tt.new_directory('h', base.root, 'h')
119
1706
        base.tt.apply()
120
1707
        other = TransformGroup("other", root_id)
121
 
        other.tt.new_file('a', other.root, [b'y\nb\nc\nd\be\n'], b'a')
122
 
        other.tt.new_file('b', other.root, [b'b2'], b'b')
123
 
        other.tt.new_file('c', other.root, [b'c2'], b'c')
124
 
        other.tt.new_file('d', other.root, [b'd'], b'd')
125
 
        other.tt.new_file('e', other.root, [b'e2'], b'e')
126
 
        other.tt.new_file('f', other.root, [b'f'], b'f')
127
 
        other.tt.new_file('g', other.root, [b'g'], b'g')
128
 
        other.tt.new_file('h', other.root, [b'h\ni\nj\nk\n'], b'h')
129
 
        other.tt.new_file('i', other.root, [b'h\ni\nj\nk\n'], b'i')
 
1708
        other.tt.new_file('a', other.root, 'y\nb\nc\nd\be\n', 'a')
 
1709
        other.tt.new_file('b', other.root, 'b2', 'b')
 
1710
        other.tt.new_file('c', other.root, 'c2', 'c')
 
1711
        other.tt.new_file('d', other.root, 'd', 'd')
 
1712
        other.tt.new_file('e', other.root, 'e2', 'e')
 
1713
        other.tt.new_file('f', other.root, 'f', 'f')
 
1714
        other.tt.new_file('g', other.root, 'g', 'g')
 
1715
        other.tt.new_file('h', other.root, 'h\ni\nj\nk\n', 'h')
 
1716
        other.tt.new_file('i', other.root, 'h\ni\nj\nk\n', 'i')
130
1717
        other.tt.apply()
131
1718
        this = TransformGroup("this", root_id)
132
 
        this.tt.new_file('a', this.root, [b'a\nb\nc\nd\bz\n'], b'a')
133
 
        this.tt.new_file('b', this.root, [b'b'], b'b')
134
 
        this.tt.new_file('c', this.root, [b'c'], b'c')
135
 
        this.tt.new_file('d', this.root, [b'd2'], b'd')
136
 
        this.tt.new_file('e', this.root, [b'e2'], b'e')
137
 
        this.tt.new_file('f', this.root, [b'f'], b'f')
138
 
        this.tt.new_file('g', this.root, [b'g'], b'g')
139
 
        this.tt.new_file('h', this.root, [b'1\n2\n3\n4\n'], b'h')
140
 
        this.tt.new_file('i', this.root, [b'1\n2\n3\n4\n'], b'i')
 
1719
        this.tt.new_file('a', this.root, 'a\nb\nc\nd\bz\n', 'a')
 
1720
        this.tt.new_file('b', this.root, 'b', 'b')
 
1721
        this.tt.new_file('c', this.root, 'c', 'c')
 
1722
        this.tt.new_file('d', this.root, 'd2', 'd')
 
1723
        this.tt.new_file('e', this.root, 'e2', 'e')
 
1724
        this.tt.new_file('f', this.root, 'f', 'f')
 
1725
        this.tt.new_file('g', this.root, 'g', 'g')
 
1726
        this.tt.new_file('h', this.root, '1\n2\n3\n4\n', 'h')
 
1727
        this.tt.new_file('i', this.root, '1\n2\n3\n4\n', 'i')
141
1728
        this.tt.apply()
142
1729
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
143
1730
 
144
1731
        # textual merge
145
 
        with this.wt.get_file(this.wt.id2path(b'a')) as f:
146
 
            self.assertEqual(f.read(), b'y\nb\nc\nd\bz\n')
 
1732
        self.assertEqual(this.wt.get_file('a').read(), 'y\nb\nc\nd\bz\n')
147
1733
        # three-way text conflict
148
 
        with this.wt.get_file(this.wt.id2path(b'b')) as f:
149
 
            self.assertEqual(f.read(), conflict_text(b'b', b'b2'))
 
1734
        self.assertEqual(this.wt.get_file('b').read(),
 
1735
                         conflict_text('b', 'b2'))
150
1736
        # OTHER wins
151
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'c')).read(), b'c2')
 
1737
        self.assertEqual(this.wt.get_file('c').read(), 'c2')
152
1738
        # THIS wins
153
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'd')).read(), b'd2')
 
1739
        self.assertEqual(this.wt.get_file('d').read(), 'd2')
154
1740
        # Ambigious clean merge
155
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'e')).read(), b'e2')
 
1741
        self.assertEqual(this.wt.get_file('e').read(), 'e2')
156
1742
        # No change
157
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'f')).read(), b'f')
 
1743
        self.assertEqual(this.wt.get_file('f').read(), 'f')
158
1744
        # Correct correct results when THIS == OTHER
159
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'g')).read(), b'g')
 
1745
        self.assertEqual(this.wt.get_file('g').read(), 'g')
160
1746
        # Text conflict when THIS & OTHER are text and BASE is dir
161
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'h')).read(),
162
 
                         conflict_text(b'1\n2\n3\n4\n', b'h\ni\nj\nk\n'))
163
 
        self.assertEqual(this.wt.get_file('h.THIS').read(),
164
 
                         b'1\n2\n3\n4\n')
165
 
        self.assertEqual(this.wt.get_file('h.OTHER').read(),
166
 
                         b'h\ni\nj\nk\n')
 
1747
        self.assertEqual(this.wt.get_file('h').read(),
 
1748
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
 
1749
        self.assertEqual(this.wt.get_file_byname('h.THIS').read(),
 
1750
                         '1\n2\n3\n4\n')
 
1751
        self.assertEqual(this.wt.get_file_byname('h.OTHER').read(),
 
1752
                         'h\ni\nj\nk\n')
167
1753
        self.assertEqual(file_kind(this.wt.abspath('h.BASE')), 'directory')
168
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'i')).read(),
169
 
                         conflict_text(b'1\n2\n3\n4\n', b'h\ni\nj\nk\n'))
170
 
        self.assertEqual(this.wt.get_file('i.THIS').read(),
171
 
                         b'1\n2\n3\n4\n')
172
 
        self.assertEqual(this.wt.get_file('i.OTHER').read(),
173
 
                         b'h\ni\nj\nk\n')
 
1754
        self.assertEqual(this.wt.get_file('i').read(),
 
1755
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
 
1756
        self.assertEqual(this.wt.get_file_byname('i.THIS').read(),
 
1757
                         '1\n2\n3\n4\n')
 
1758
        self.assertEqual(this.wt.get_file_byname('i.OTHER').read(),
 
1759
                         'h\ni\nj\nk\n')
174
1760
        self.assertEqual(os.path.exists(this.wt.abspath('i.BASE')), False)
175
1761
        modified = ['a', 'b', 'c', 'h', 'i']
176
1762
        merge_modified = this.wt.merge_modified()
177
1763
        self.assertSubset(merge_modified, modified)
178
1764
        self.assertEqual(len(merge_modified), len(modified))
179
 
        with open(this.wt.abspath(this.wt.id2path(b'a')), 'wb') as f:
180
 
            f.write(b'booga')
 
1765
        with file(this.wt.id2abspath('a'), 'wb') as f: f.write('booga')
181
1766
        modified.pop(0)
182
1767
        merge_modified = this.wt.merge_modified()
183
1768
        self.assertSubset(merge_modified, modified)
192
1777
        this = TransformGroup("THIS", root_id)
193
1778
        other = TransformGroup("OTHER", root_id)
194
1779
        for tg in this, base, other:
195
 
            tg.tt.new_directory('a', tg.root, b'a')
196
 
            tg.tt.new_symlink('b', tg.root, 'b', b'b')
197
 
            tg.tt.new_file('c', tg.root, [b'c'], b'c')
198
 
            tg.tt.new_symlink('d', tg.root, tg.name, b'd')
 
1780
            tg.tt.new_directory('a', tg.root, 'a')
 
1781
            tg.tt.new_symlink('b', tg.root, 'b', 'b')
 
1782
            tg.tt.new_file('c', tg.root, 'c', 'c')
 
1783
            tg.tt.new_symlink('d', tg.root, tg.name, 'd')
199
1784
        targets = ((base, 'base-e', 'base-f', None, None),
200
1785
                   (this, 'other-e', 'this-f', 'other-g', 'this-h'),
201
1786
                   (other, 'other-e', None, 'other-g', 'other-h'))
203
1788
            for link, target in (('e', e_target), ('f', f_target),
204
1789
                                 ('g', g_target), ('h', h_target)):
205
1790
                if target is not None:
206
 
                    tg.tt.new_symlink(link, tg.root, target,
207
 
                                      link.encode('ascii'))
 
1791
                    tg.tt.new_symlink(link, tg.root, target, link)
208
1792
 
209
1793
        for tg in this, base, other:
210
1794
            tg.tt.apply()
213
1797
        self.assertIs(os.path.islink(this.wt.abspath('b')), True)
214
1798
        self.assertIs(os.path.isfile(this.wt.abspath('c')), True)
215
1799
        for suffix in ('THIS', 'BASE', 'OTHER'):
216
 
            self.assertEqual(os.readlink(
217
 
                this.wt.abspath('d.' + suffix)), suffix)
 
1800
            self.assertEqual(os.readlink(this.wt.abspath('d.'+suffix)), suffix)
218
1801
        self.assertIs(os.path.lexists(this.wt.abspath('d')), False)
219
 
        self.assertEqual(this.wt.id2path(b'd'), 'd.OTHER')
220
 
        self.assertEqual(this.wt.id2path(b'f'), 'f.THIS')
 
1802
        self.assertEqual(this.wt.id2path('d'), 'd.OTHER')
 
1803
        self.assertEqual(this.wt.id2path('f'), 'f.THIS')
221
1804
        self.assertEqual(os.readlink(this.wt.abspath('e')), 'other-e')
222
1805
        self.assertIs(os.path.lexists(this.wt.abspath('e.THIS')), False)
223
1806
        self.assertIs(os.path.lexists(this.wt.abspath('e.OTHER')), False)
234
1817
        base = TransformGroup("BASE", root_id)
235
1818
        this = TransformGroup("THIS", root_id)
236
1819
        other = TransformGroup("OTHER", root_id)
237
 
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, b'a')
238
 
                                   for t in [base, this, other]]
239
 
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, b'b')
240
 
                                   for t in [base, this, other]]
241
 
        base.tt.new_directory('c', base_a, b'c')
242
 
        this.tt.new_directory('c1', this_a, b'c')
243
 
        other.tt.new_directory('c', other_b, b'c')
244
 
 
245
 
        base.tt.new_directory('d', base_a, b'd')
246
 
        this.tt.new_directory('d1', this_b, b'd')
247
 
        other.tt.new_directory('d', other_a, b'd')
248
 
 
249
 
        base.tt.new_directory('e', base_a, b'e')
250
 
        this.tt.new_directory('e', this_a, b'e')
251
 
        other.tt.new_directory('e1', other_b, b'e')
252
 
 
253
 
        base.tt.new_directory('f', base_a, b'f')
254
 
        this.tt.new_directory('f1', this_b, b'f')
255
 
        other.tt.new_directory('f1', other_b, b'f')
 
1820
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
 
1821
                                   for t in [base, this, other]]
 
1822
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
 
1823
                                   for t in [base, this, other]]
 
1824
        base.tt.new_directory('c', base_a, 'c')
 
1825
        this.tt.new_directory('c1', this_a, 'c')
 
1826
        other.tt.new_directory('c', other_b, 'c')
 
1827
 
 
1828
        base.tt.new_directory('d', base_a, 'd')
 
1829
        this.tt.new_directory('d1', this_b, 'd')
 
1830
        other.tt.new_directory('d', other_a, 'd')
 
1831
 
 
1832
        base.tt.new_directory('e', base_a, 'e')
 
1833
        this.tt.new_directory('e', this_a, 'e')
 
1834
        other.tt.new_directory('e1', other_b, 'e')
 
1835
 
 
1836
        base.tt.new_directory('f', base_a, 'f')
 
1837
        this.tt.new_directory('f1', this_b, 'f')
 
1838
        other.tt.new_directory('f1', other_b, 'f')
256
1839
 
257
1840
        for tg in [this, base, other]:
258
1841
            tg.tt.apply()
259
1842
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
260
 
        self.assertEqual(this.wt.id2path(b'c'), pathjoin('b/c1'))
261
 
        self.assertEqual(this.wt.id2path(b'd'), pathjoin('b/d1'))
262
 
        self.assertEqual(this.wt.id2path(b'e'), pathjoin('b/e1'))
263
 
        self.assertEqual(this.wt.id2path(b'f'), pathjoin('b/f1'))
 
1843
        self.assertEqual(this.wt.id2path('c'), pathjoin('b/c1'))
 
1844
        self.assertEqual(this.wt.id2path('d'), pathjoin('b/d1'))
 
1845
        self.assertEqual(this.wt.id2path('e'), pathjoin('b/e1'))
 
1846
        self.assertEqual(this.wt.id2path('f'), pathjoin('b/f1'))
264
1847
 
265
1848
    def test_filename_merge_conflicts(self):
266
1849
        root_id = generate_ids.gen_root_id()
267
1850
        base = TransformGroup("BASE", root_id)
268
1851
        this = TransformGroup("THIS", root_id)
269
1852
        other = TransformGroup("OTHER", root_id)
270
 
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, b'a')
271
 
                                   for t in [base, this, other]]
272
 
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, b'b')
273
 
                                   for t in [base, this, other]]
274
 
 
275
 
        base.tt.new_file('g', base_a, [b'g'], b'g')
276
 
        other.tt.new_file('g1', other_b, [b'g1'], b'g')
277
 
 
278
 
        base.tt.new_file('h', base_a, [b'h'], b'h')
279
 
        this.tt.new_file('h1', this_b, [b'h1'], b'h')
280
 
 
281
 
        base.tt.new_file('i', base.root, [b'i'], b'i')
282
 
        other.tt.new_directory('i1', this_b, b'i')
 
1853
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
 
1854
                                   for t in [base, this, other]]
 
1855
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
 
1856
                                   for t in [base, this, other]]
 
1857
 
 
1858
        base.tt.new_file('g', base_a, 'g', 'g')
 
1859
        other.tt.new_file('g1', other_b, 'g1', 'g')
 
1860
 
 
1861
        base.tt.new_file('h', base_a, 'h', 'h')
 
1862
        this.tt.new_file('h1', this_b, 'h1', 'h')
 
1863
 
 
1864
        base.tt.new_file('i', base.root, 'i', 'i')
 
1865
        other.tt.new_directory('i1', this_b, 'i')
283
1866
 
284
1867
        for tg in [this, base, other]:
285
1868
            tg.tt.apply()
286
1869
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
287
1870
 
288
 
        self.assertEqual(this.wt.id2path(b'g'), pathjoin('b/g1.OTHER'))
 
1871
        self.assertEqual(this.wt.id2path('g'), pathjoin('b/g1.OTHER'))
289
1872
        self.assertIs(os.path.lexists(this.wt.abspath('b/g1.BASE')), True)
290
1873
        self.assertIs(os.path.lexists(this.wt.abspath('b/g1.THIS')), False)
291
 
        self.assertEqual(this.wt.id2path(b'h'), pathjoin('b/h1.THIS'))
 
1874
        self.assertEqual(this.wt.id2path('h'), pathjoin('b/h1.THIS'))
292
1875
        self.assertIs(os.path.lexists(this.wt.abspath('b/h1.BASE')), True)
293
1876
        self.assertIs(os.path.lexists(this.wt.abspath('b/h1.OTHER')), False)
294
 
        self.assertEqual(this.wt.id2path(b'i'), pathjoin('b/i1.OTHER'))
 
1877
        self.assertEqual(this.wt.id2path('i'), pathjoin('b/i1.OTHER'))
295
1878
 
296
1879
 
297
1880
class TestBuildTree(tests.TestCaseWithTransport):
301
1884
        os.mkdir('a')
302
1885
        a = ControlDir.create_standalone_workingtree('a')
303
1886
        os.mkdir('a/foo')
304
 
        with open('a/foo/bar', 'wb') as f:
305
 
            f.write(b'contents')
 
1887
        with file('a/foo/bar', 'wb') as f: f.write('contents')
306
1888
        os.symlink('a/foo/bar', 'a/foo/baz')
307
1889
        a.add(['foo', 'foo/bar', 'foo/baz'])
308
1890
        a.commit('initial commit')
312
1894
        self.addCleanup(basis.unlock)
313
1895
        build_tree(basis, b)
314
1896
        self.assertIs(os.path.isdir('b/foo'), True)
315
 
        with open('b/foo/bar', 'rb') as f:
316
 
            self.assertEqual(f.read(), b"contents")
 
1897
        self.assertEqual(file('b/foo/bar', 'rb').read(), "contents")
317
1898
        self.assertEqual(os.readlink('b/foo/baz'), 'a/foo/bar')
318
1899
 
319
1900
    def test_build_with_references(self):
320
1901
        tree = self.make_branch_and_tree('source',
321
 
                                         format='development-subtree')
 
1902
            format='development-subtree')
322
1903
        subtree = self.make_branch_and_tree('source/subtree',
323
 
                                            format='development-subtree')
 
1904
            format='development-subtree')
324
1905
        tree.add_reference(subtree)
325
1906
        tree.commit('a revision')
326
1907
        tree.branch.create_checkout('target')
332
1913
        source = self.make_branch_and_tree('source')
333
1914
        target = self.make_branch_and_tree('target')
334
1915
        self.build_tree(['source/file', 'target/file'])
335
 
        source.add('file', b'new-file')
 
1916
        source.add('file', 'new-file')
336
1917
        source.commit('added file')
337
1918
        build_tree(source.basis_tree(), target)
338
 
        self.assertEqual(
339
 
            [DuplicateEntry('Moved existing file to', 'file.moved',
340
 
                            'file', None, 'new-file')],
341
 
            target.conflicts())
 
1919
        self.assertEqual([DuplicateEntry('Moved existing file to',
 
1920
                          'file.moved', 'file', None, 'new-file')],
 
1921
                         target.conflicts())
342
1922
        target2 = self.make_branch_and_tree('target2')
343
 
        with open('target2/file', 'wb') as target_file, \
344
 
                open('source/file', 'rb') as source_file:
345
 
            target_file.write(source_file.read())
 
1923
        target_file = file('target2/file', 'wb')
 
1924
        try:
 
1925
            source_file = file('source/file', 'rb')
 
1926
            try:
 
1927
                target_file.write(source_file.read())
 
1928
            finally:
 
1929
                source_file.close()
 
1930
        finally:
 
1931
            target_file.close()
346
1932
        build_tree(source.basis_tree(), target2)
347
1933
        self.assertEqual([], target2.conflicts())
348
1934
 
351
1937
        self.requireFeature(SymlinkFeature)
352
1938
        source = self.make_branch_and_tree('source')
353
1939
        os.symlink('foo', 'source/symlink')
354
 
        source.add('symlink', b'new-symlink')
 
1940
        source.add('symlink', 'new-symlink')
355
1941
        source.commit('added file')
356
1942
        target = self.make_branch_and_tree('target')
357
1943
        os.symlink('bar', 'target/symlink')
358
1944
        build_tree(source.basis_tree(), target)
359
 
        self.assertEqual(
360
 
            [DuplicateEntry('Moved existing file to', 'symlink.moved',
361
 
                            'symlink', None, 'new-symlink')],
 
1945
        self.assertEqual([DuplicateEntry('Moved existing file to',
 
1946
            'symlink.moved', 'symlink', None, 'new-symlink')],
362
1947
            target.conflicts())
363
1948
        target = self.make_branch_and_tree('target2')
364
1949
        os.symlink('foo', 'target2/symlink')
370
1955
        source = self.make_branch_and_tree('source')
371
1956
        target = self.make_branch_and_tree('target')
372
1957
        self.build_tree(['source/dir1/', 'source/dir1/file', 'target/dir1/'])
373
 
        source.add(['dir1', 'dir1/file'], [b'new-dir1', b'new-file'])
 
1958
        source.add(['dir1', 'dir1/file'], ['new-dir1', 'new-file'])
374
1959
        source.commit('added file')
375
1960
        build_tree(source.basis_tree(), target)
376
1961
        self.assertEqual([], target.conflicts())
392
1977
        self.assertPathDoesNotExist('target3/dir1/file')
393
1978
        self.assertPathExists('target3/dir1/file2')
394
1979
        self.assertPathExists('target3/dir1.diverted/file')
395
 
        self.assertEqual(
396
 
            [DuplicateEntry('Diverted to', 'dir1.diverted',
397
 
                            'dir1', 'new-dir1', None)],
 
1980
        self.assertEqual([DuplicateEntry('Diverted to',
 
1981
            'dir1.diverted', 'dir1', 'new-dir1', None)],
398
1982
            target.conflicts())
399
1983
 
400
1984
        target = self.make_branch_and_tree('target4')
404
1988
        self.assertPathExists('target4/dir1/file')
405
1989
        self.assertEqual('directory', file_kind('target4/dir1/file'))
406
1990
        self.assertPathExists('target4/dir1/file.diverted')
407
 
        self.assertEqual(
408
 
            [DuplicateEntry('Diverted to', 'dir1/file.diverted',
409
 
                            'dir1/file', 'new-file', None)],
 
1991
        self.assertEqual([DuplicateEntry('Diverted to',
 
1992
            'dir1/file.diverted', 'dir1/file', 'new-file', None)],
410
1993
            target.conflicts())
411
1994
 
412
1995
    def test_mixed_conflict_handling(self):
414
1997
        source = self.make_branch_and_tree('source')
415
1998
        target = self.make_branch_and_tree('target')
416
1999
        self.build_tree(['source/name', 'target/name/'])
417
 
        source.add('name', b'new-name')
 
2000
        source.add('name', 'new-name')
418
2001
        source.commit('added file')
419
2002
        build_tree(source.basis_tree(), target)
420
 
        self.assertEqual(
421
 
            [DuplicateEntry('Moved existing file to',
422
 
                            'name.moved', 'name', None, 'new-name')],
423
 
            target.conflicts())
 
2003
        self.assertEqual([DuplicateEntry('Moved existing file to',
 
2004
            'name.moved', 'name', None, 'new-name')], target.conflicts())
424
2005
 
425
2006
    def test_raises_in_populated(self):
426
2007
        source = self.make_branch_and_tree('source')
431
2012
        self.build_tree(['target/name'])
432
2013
        target.add('name')
433
2014
        self.assertRaises(errors.WorkingTreeAlreadyPopulated,
434
 
                          build_tree, source.basis_tree(), target)
 
2015
            build_tree, source.basis_tree(), target)
435
2016
 
436
2017
    def test_build_tree_rename_count(self):
437
2018
        source = self.make_branch_and_tree('source')
453
2034
    def create_ab_tree(self):
454
2035
        """Create a committed test tree with two files"""
455
2036
        source = self.make_branch_and_tree('source')
456
 
        self.build_tree_contents([('source/file1', b'A')])
457
 
        self.build_tree_contents([('source/file2', b'B')])
458
 
        source.add(['file1', 'file2'], [b'file1-id', b'file2-id'])
 
2037
        self.build_tree_contents([('source/file1', 'A')])
 
2038
        self.build_tree_contents([('source/file2', 'B')])
 
2039
        source.add(['file1', 'file2'], ['file1-id', 'file2-id'])
459
2040
        source.commit('commit files')
460
2041
        source.lock_write()
461
2042
        self.addCleanup(source.unlock)
463
2044
 
464
2045
    def test_build_tree_accelerator_tree(self):
465
2046
        source = self.create_ab_tree()
466
 
        self.build_tree_contents([('source/file2', b'C')])
 
2047
        self.build_tree_contents([('source/file2', 'C')])
467
2048
        calls = []
468
2049
        real_source_get_file = source.get_file
469
 
 
470
 
        def get_file(path):
471
 
            calls.append(path)
472
 
            return real_source_get_file(path)
 
2050
        def get_file(file_id, path=None):
 
2051
            calls.append(file_id)
 
2052
            return real_source_get_file(file_id, path)
473
2053
        source.get_file = get_file
474
2054
        target = self.make_branch_and_tree('target')
475
2055
        revision_tree = source.basis_tree()
476
2056
        revision_tree.lock_read()
477
2057
        self.addCleanup(revision_tree.unlock)
478
2058
        build_tree(revision_tree, target, source)
479
 
        self.assertEqual(['file1'], calls)
 
2059
        self.assertEqual(['file1-id'], calls)
480
2060
        target.lock_read()
481
2061
        self.addCleanup(target.unlock)
482
2062
        self.assertEqual([], list(target.iter_changes(revision_tree)))
483
2063
 
484
2064
    def test_build_tree_accelerator_tree_observes_sha1(self):
485
2065
        source = self.create_ab_tree()
486
 
        sha1 = osutils.sha_string(b'A')
 
2066
        sha1 = osutils.sha_string('A')
487
2067
        target = self.make_branch_and_tree('target')
488
2068
        target.lock_write()
489
2069
        self.addCleanup(target.unlock)
490
2070
        state = target.current_dirstate()
491
2071
        state._cutoff_time = time.time() + 60
492
2072
        build_tree(source.basis_tree(), target, source)
493
 
        entry = state._get_entry(0, path_utf8=b'file1')
 
2073
        entry = state._get_entry(0, path_utf8='file1')
494
2074
        self.assertEqual(sha1, entry[1][0][1])
495
2075
 
496
2076
    def test_build_tree_accelerator_tree_missing_file(self):
509
2089
    def test_build_tree_accelerator_wrong_kind(self):
510
2090
        self.requireFeature(SymlinkFeature)
511
2091
        source = self.make_branch_and_tree('source')
512
 
        self.build_tree_contents([('source/file1', b'')])
513
 
        self.build_tree_contents([('source/file2', b'')])
514
 
        source.add(['file1', 'file2'], [b'file1-id', b'file2-id'])
 
2092
        self.build_tree_contents([('source/file1', '')])
 
2093
        self.build_tree_contents([('source/file2', '')])
 
2094
        source.add(['file1', 'file2'], ['file1-id', 'file2-id'])
515
2095
        source.commit('commit files')
516
2096
        os.unlink('source/file2')
517
 
        self.build_tree_contents([('source/file2/', b'C')])
 
2097
        self.build_tree_contents([('source/file2/', 'C')])
518
2098
        os.unlink('source/file1')
519
2099
        os.symlink('file2', 'source/file1')
520
2100
        calls = []
521
2101
        real_source_get_file = source.get_file
522
 
 
523
 
        def get_file(path):
524
 
            calls.append(path)
525
 
            return real_source_get_file(path)
 
2102
        def get_file(file_id, path=None):
 
2103
            calls.append(file_id)
 
2104
            return real_source_get_file(file_id, path)
526
2105
        source.get_file = get_file
527
2106
        target = self.make_branch_and_tree('target')
528
2107
        revision_tree = source.basis_tree()
561
2140
 
562
2141
    def test_build_tree_accelerator_tree_moved(self):
563
2142
        source = self.make_branch_and_tree('source')
564
 
        self.build_tree_contents([('source/file1', b'A')])
565
 
        source.add(['file1'], [b'file1-id'])
 
2143
        self.build_tree_contents([('source/file1', 'A')])
 
2144
        source.add(['file1'], ['file1-id'])
566
2145
        source.commit('commit files')
567
2146
        source.rename_one('file1', 'file2')
568
2147
        source.lock_read()
579
2158
    def test_build_tree_hardlinks_preserve_execute(self):
580
2159
        self.requireFeature(HardlinkFeature)
581
2160
        source = self.create_ab_tree()
582
 
        tt = source.transform()
583
 
        trans_id = tt.trans_id_tree_path('file1')
 
2161
        tt = TreeTransform(source)
 
2162
        trans_id = tt.trans_id_tree_file_id('file1-id')
584
2163
        tt.set_executability(True, trans_id)
585
2164
        tt.apply()
586
 
        self.assertTrue(source.is_executable('file1'))
 
2165
        self.assertTrue(source.is_executable('file1-id'))
587
2166
        target = self.make_branch_and_tree('target')
588
2167
        revision_tree = source.basis_tree()
589
2168
        revision_tree.lock_read()
592
2171
        target.lock_read()
593
2172
        self.addCleanup(target.unlock)
594
2173
        self.assertEqual([], list(target.iter_changes(revision_tree)))
595
 
        self.assertTrue(source.is_executable('file1'))
 
2174
        self.assertTrue(source.is_executable('file1-id'))
596
2175
 
597
2176
    def install_rot13_content_filter(self, pattern):
598
2177
        # We could use
600
2179
        # below, but that looks a bit... hard to read even if it's exactly
601
2180
        # the same thing.
602
2181
        original_registry = filters._reset_registry()
603
 
 
604
2182
        def restore_registry():
605
2183
            filters._reset_registry(original_registry)
606
2184
        self.addCleanup(restore_registry)
607
 
 
608
2185
        def rot13(chunks, context=None):
609
 
            return [
610
 
                codecs.encode(chunk.decode('ascii'), 'rot13').encode('ascii')
611
 
                for chunk in chunks]
 
2186
            return [''.join(chunks).encode('rot13')]
612
2187
        rot13filter = filters.ContentFilter(rot13, rot13)
613
2188
        filters.filter_stacks_registry.register(
614
2189
            'rot13', {'yes': [rot13filter]}.get)
615
2190
        os.mkdir(self.test_home_dir + '/.bazaar')
616
2191
        rules_filename = self.test_home_dir + '/.bazaar/rules'
617
 
        with open(rules_filename, 'wb') as f:
618
 
            f.write(b'[name %s]\nrot13=yes\n' % (pattern,))
619
 
 
 
2192
        f = open(rules_filename, 'wb')
 
2193
        f.write('[name %s]\nrot13=yes\n' % (pattern,))
 
2194
        f.close()
620
2195
        def uninstall_rules():
621
2196
            os.remove(rules_filename)
622
2197
            rules.reset_rules()
629
2204
        if it can).
630
2205
        """
631
2206
        self.requireFeature(HardlinkFeature)
632
 
        self.install_rot13_content_filter(b'file1')
 
2207
        self.install_rot13_content_filter('file1')
633
2208
        source = self.create_ab_tree()
634
2209
        target = self.make_branch_and_tree('target')
635
2210
        revision_tree = source.basis_tree()
648
2223
 
649
2224
    def test_case_insensitive_build_tree_inventory(self):
650
2225
        if (features.CaseInsensitiveFilesystemFeature.available()
651
 
                or features.CaseInsCasePresFilenameFeature.available()):
 
2226
            or features.CaseInsCasePresFilenameFeature.available()):
652
2227
            raise tests.UnavailableFeature('Fully case sensitive filesystem')
653
2228
        source = self.make_branch_and_tree('source')
654
2229
        self.build_tree(['source/file', 'source/FILE'])
655
 
        source.add(['file', 'FILE'], [b'lower-id', b'upper-id'])
 
2230
        source.add(['file', 'FILE'], ['lower-id', 'upper-id'])
656
2231
        source.commit('added files')
657
2232
        # Don't try this at home, kids!
658
2233
        # Force the tree to report that it is case insensitive
659
2234
        target = self.make_branch_and_tree('target')
660
2235
        target.case_sensitive = False
661
2236
        build_tree(source.basis_tree(), target, source, delta_from_tree=True)
662
 
        self.assertEqual('file.moved', target.id2path(b'lower-id'))
663
 
        self.assertEqual('FILE', target.id2path(b'upper-id'))
 
2237
        self.assertEqual('file.moved', target.id2path('lower-id'))
 
2238
        self.assertEqual('FILE', target.id2path('upper-id'))
664
2239
 
665
2240
    def test_build_tree_observes_sha(self):
666
2241
        source = self.make_branch_and_tree('source')
667
2242
        self.build_tree(['source/file1', 'source/dir/', 'source/dir/file2'])
668
2243
        source.add(['file1', 'dir', 'dir/file2'],
669
 
                   [b'file1-id', b'dir-id', b'file2-id'])
 
2244
                   ['file1-id', 'dir-id', 'file2-id'])
670
2245
        source.commit('new files')
671
2246
        target = self.make_branch_and_tree('target')
672
2247
        target.lock_write()
681
2256
        # entry[1] is the state information, entry[1][0] is the state of the
682
2257
        # working tree, entry[1][0][1] is the sha value for the current working
683
2258
        # tree
684
 
        entry1 = state._get_entry(0, path_utf8=b'file1')
 
2259
        entry1 = state._get_entry(0, path_utf8='file1')
685
2260
        self.assertEqual(entry1_sha, entry1[1][0][1])
686
2261
        # The 'size' field must also be set.
687
2262
        self.assertEqual(25, entry1[1][0][2])
688
2263
        entry1_state = entry1[1][0]
689
 
        entry2 = state._get_entry(0, path_utf8=b'dir/file2')
 
2264
        entry2 = state._get_entry(0, path_utf8='dir/file2')
690
2265
        self.assertEqual(entry2_sha, entry2[1][0][1])
691
2266
        self.assertEqual(29, entry2[1][0][2])
692
2267
        entry2_state = entry2[1][0]
693
2268
        # Now, make sure that we don't have to re-read the content. The
694
2269
        # packed_stat should match exactly.
695
 
        self.assertEqual(entry1_sha, target.get_file_sha1('file1'))
696
 
        self.assertEqual(entry2_sha, target.get_file_sha1('dir/file2'))
 
2270
        self.assertEqual(entry1_sha, target.get_file_sha1('file1-id', 'file1'))
 
2271
        self.assertEqual(entry2_sha,
 
2272
                         target.get_file_sha1('file2-id', 'dir/file2'))
697
2273
        self.assertEqual(entry1_state, entry1[1][0])
698
2274
        self.assertEqual(entry2_state, entry2[1][0])
699
2275
 
709
2285
 
710
2286
    def get_branch_and_transform(self):
711
2287
        branch = self.get_branch()
712
 
        tt = branch.basis_tree().preview_transform()
 
2288
        tt = TransformPreview(branch.basis_tree())
713
2289
        self.addCleanup(tt.finalize)
714
2290
        return branch, tt
715
2291
 
717
2293
        branch = self.get_branch()
718
2294
        basis = branch.repository.revision_tree(
719
2295
            _mod_revision.NULL_REVISION)
720
 
        tt = basis.preview_transform()
 
2296
        tt = TransformPreview(basis)
721
2297
        self.addCleanup(tt.finalize)
722
2298
        e = self.assertRaises(ValueError, tt.commit, branch, '')
723
2299
        self.assertEqual('TreeTransform not based on branch basis: null:',
732
2308
 
733
2309
    def test_merge_parents(self):
734
2310
        branch, tt = self.get_branch_and_transform()
735
 
        tt.commit(branch, 'my message', [b'rev1b', b'rev1c'])
736
 
        self.assertEqual([b'rev1b', b'rev1c'],
 
2311
        rev = tt.commit(branch, 'my message', ['rev1b', 'rev1c'])
 
2312
        self.assertEqual(['rev1b', 'rev1c'],
737
2313
                         branch.basis_tree().get_parent_ids()[1:])
738
2314
 
739
2315
    def test_first_commit(self):
740
2316
        branch = self.make_branch('branch')
741
2317
        branch.lock_write()
742
2318
        self.addCleanup(branch.unlock)
743
 
        tt = branch.basis_tree().preview_transform()
 
2319
        tt = TransformPreview(branch.basis_tree())
744
2320
        self.addCleanup(tt.finalize)
745
 
        tt.new_directory('', ROOT_PARENT, b'TREE_ROOT')
746
 
        tt.commit(branch, 'my message')
 
2321
        tt.new_directory('', ROOT_PARENT, 'TREE_ROOT')
 
2322
        rev = tt.commit(branch, 'my message')
747
2323
        self.assertEqual([], branch.basis_tree().get_parent_ids())
748
2324
        self.assertNotEqual(_mod_revision.NULL_REVISION,
749
2325
                            branch.last_revision())
752
2328
        branch = self.make_branch('branch')
753
2329
        branch.lock_write()
754
2330
        self.addCleanup(branch.unlock)
755
 
        tt = branch.basis_tree().preview_transform()
 
2331
        tt = TransformPreview(branch.basis_tree())
756
2332
        self.addCleanup(tt.finalize)
757
2333
        e = self.assertRaises(ValueError, tt.commit, branch,
758
 
                              'my message', [b'rev1b-id'])
 
2334
                          'my message', ['rev1b-id'])
759
2335
        self.assertEqual('Cannot supply merge parents for first commit.',
760
2336
                         str(e))
761
2337
        self.assertEqual(_mod_revision.NULL_REVISION, branch.last_revision())
762
2338
 
763
2339
    def test_add_files(self):
764
2340
        branch, tt = self.get_branch_and_transform()
765
 
        tt.new_file('file', tt.root, [b'contents'], b'file-id')
766
 
        trans_id = tt.new_directory('dir', tt.root, b'dir-id')
 
2341
        tt.new_file('file', tt.root, 'contents', 'file-id')
 
2342
        trans_id = tt.new_directory('dir', tt.root, 'dir-id')
767
2343
        if SymlinkFeature.available():
768
 
            tt.new_symlink('symlink', trans_id, 'target', b'symlink-id')
769
 
        tt.commit(branch, 'message')
 
2344
            tt.new_symlink('symlink', trans_id, 'target', 'symlink-id')
 
2345
        rev = tt.commit(branch, 'message')
770
2346
        tree = branch.basis_tree()
771
 
        self.assertEqual('file', tree.id2path(b'file-id'))
772
 
        self.assertEqual(b'contents', tree.get_file_text('file'))
773
 
        self.assertEqual('dir', tree.id2path(b'dir-id'))
 
2347
        self.assertEqual('file', tree.id2path('file-id'))
 
2348
        self.assertEqual('contents', tree.get_file_text('file-id'))
 
2349
        self.assertEqual('dir', tree.id2path('dir-id'))
774
2350
        if SymlinkFeature.available():
775
 
            self.assertEqual('dir/symlink', tree.id2path(b'symlink-id'))
776
 
            self.assertEqual('target', tree.get_symlink_target('dir/symlink'))
 
2351
            self.assertEqual('dir/symlink', tree.id2path('symlink-id'))
 
2352
            self.assertEqual('target', tree.get_symlink_target('symlink-id'))
777
2353
 
778
2354
    def test_add_unversioned(self):
779
2355
        branch, tt = self.get_branch_and_transform()
780
 
        tt.new_file('file', tt.root, [b'contents'])
 
2356
        tt.new_file('file', tt.root, 'contents')
781
2357
        self.assertRaises(errors.StrictCommitFailed, tt.commit, branch,
782
2358
                          'message', strict=True)
783
2359
 
784
2360
    def test_modify_strict(self):
785
2361
        branch, tt = self.get_branch_and_transform()
786
 
        tt.new_file('file', tt.root, [b'contents'], b'file-id')
 
2362
        tt.new_file('file', tt.root, 'contents', 'file-id')
787
2363
        tt.commit(branch, 'message', strict=True)
788
 
        tt = branch.basis_tree().preview_transform()
 
2364
        tt = TransformPreview(branch.basis_tree())
789
2365
        self.addCleanup(tt.finalize)
790
 
        trans_id = tt.trans_id_file_id(b'file-id')
 
2366
        trans_id = tt.trans_id_file_id('file-id')
791
2367
        tt.delete_contents(trans_id)
792
 
        tt.create_file([b'contents'], trans_id)
 
2368
        tt.create_file('contents', trans_id)
793
2369
        tt.commit(branch, 'message', strict=True)
794
2370
 
795
2371
    def test_commit_malformed(self):
798
2374
        In this case, we are adding a file without adding its parent.
799
2375
        """
800
2376
        branch, tt = self.get_branch_and_transform()
801
 
        parent_id = tt.trans_id_file_id(b'parent-id')
802
 
        tt.new_file('file', parent_id, [b'contents'], b'file-id')
803
 
        self.assertRaises(MalformedTransform, tt.commit, branch,
 
2377
        parent_id = tt.trans_id_file_id('parent-id')
 
2378
        tt.new_file('file', parent_id, 'contents', 'file-id')
 
2379
        self.assertRaises(errors.MalformedTransform, tt.commit, branch,
804
2380
                          'message')
805
2381
 
806
2382
    def test_commit_rich_revision_data(self):
807
2383
        branch, tt = self.get_branch_and_transform()
808
2384
        rev_id = tt.commit(branch, 'message', timestamp=1, timezone=43201,
809
2385
                           committer='me <me@example.com>',
810
 
                           revprops={u'foo': 'bar'}, revision_id=b'revid-1',
 
2386
                           revprops={'foo': 'bar'}, revision_id='revid-1',
811
2387
                           authors=['Author1 <author1@example.com>',
812
 
                                    'Author2 <author2@example.com>',
813
 
                                    ])
814
 
        self.assertEqual(b'revid-1', rev_id)
 
2388
                              'Author2 <author2@example.com>',
 
2389
                               ])
 
2390
        self.assertEqual('revid-1', rev_id)
815
2391
        revision = branch.repository.get_revision(rev_id)
816
2392
        self.assertEqual(1, revision.timestamp)
817
2393
        self.assertEqual(43201, revision.timezone)
880
2456
        mover.rename('c/e', 'c/d')
881
2457
        try:
882
2458
            mover.rename('a', 'c')
883
 
        except errors.FileExists:
 
2459
        except errors.FileExists as e:
884
2460
            mover.rollback()
885
2461
        self.assertPathExists('a')
886
2462
        self.assertPathExists('c/d')
901
2477
 
902
2478
        def rename(self, source, target):
903
2479
            if (self.bad_source is not None and
904
 
                    source.endswith(self.bad_source)):
 
2480
                source.endswith(self.bad_source)):
905
2481
                raise Bogus
906
2482
            elif (self.bad_target is not None and
907
 
                  target.endswith(self.bad_target)):
 
2483
                target.endswith(self.bad_target)):
908
2484
                raise Bogus
909
2485
            else:
910
2486
                _FileMover.rename(self, source, target)
912
2488
    def test_rollback_rename(self):
913
2489
        tree = self.make_branch_and_tree('.')
914
2490
        self.build_tree(['a/', 'a/b'])
915
 
        tt = tree.transform()
 
2491
        tt = TreeTransform(tree)
916
2492
        self.addCleanup(tt.finalize)
917
2493
        a_id = tt.trans_id_tree_path('a')
918
2494
        tt.adjust_path('c', tt.root, a_id)
928
2504
    def test_rollback_rename_into_place(self):
929
2505
        tree = self.make_branch_and_tree('.')
930
2506
        self.build_tree(['a/', 'a/b'])
931
 
        tt = tree.transform()
 
2507
        tt = TreeTransform(tree)
932
2508
        self.addCleanup(tt.finalize)
933
2509
        a_id = tt.trans_id_tree_path('a')
934
2510
        tt.adjust_path('c', tt.root, a_id)
944
2520
    def test_rollback_deletion(self):
945
2521
        tree = self.make_branch_and_tree('.')
946
2522
        self.build_tree(['a/', 'a/b'])
947
 
        tt = tree.transform()
 
2523
        tt = TreeTransform(tree)
948
2524
        self.addCleanup(tt.finalize)
949
2525
        a_id = tt.trans_id_tree_path('a')
950
2526
        tt.delete_contents(a_id)
965
2541
        new_globals = dict(func.__globals__)
966
2542
        new_globals.update(globals)
967
2543
        new_func = types.FunctionType(func.__code__, new_globals,
968
 
                                      func.__name__, func.__defaults__)
 
2544
            func.__name__, func.__defaults__)
969
2545
        setattr(instance, method_name,
970
 
                types.MethodType(new_func, instance))
 
2546
            types.MethodType(new_func, instance, instance.__class__))
971
2547
        self.addCleanup(delattr, instance, method_name)
972
2548
 
973
2549
    @staticmethod
984
2560
    def create_transform_and_root_trans_id(self):
985
2561
        """Setup a transform creating a file in limbo"""
986
2562
        tree = self.make_branch_and_tree('.')
987
 
        tt = tree.transform()
 
2563
        tt = TreeTransform(tree)
988
2564
        return tt, tt.create_path("a", tt.root)
989
2565
 
990
2566
    def create_transform_and_subdir_trans_id(self):
991
2567
        """Setup a transform creating a directory containing a file in limbo"""
992
2568
        tree = self.make_branch_and_tree('.')
993
 
        tt = tree.transform()
 
2569
        tt = TreeTransform(tree)
994
2570
        d_trans_id = tt.create_path("d", tt.root)
995
2571
        tt.create_directory(d_trans_id)
996
2572
        f_trans_id = tt.create_path("a", d_trans_id)
999
2575
 
1000
2576
    def test_root_create_file_open_raises_before_creation(self):
1001
2577
        tt, trans_id = self.create_transform_and_root_trans_id()
1002
 
        self._override_globals_in_method(
1003
 
            tt, "create_file", {"open": self._fake_open_raises_before})
1004
 
        self.assertRaises(RuntimeError, tt.create_file,
1005
 
                          [b"contents"], trans_id)
 
2578
        self._override_globals_in_method(tt, "create_file",
 
2579
            {"open": self._fake_open_raises_before})
 
2580
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
1006
2581
        path = tt._limbo_name(trans_id)
1007
2582
        self.assertPathDoesNotExist(path)
1008
2583
        tt.finalize()
1010
2585
 
1011
2586
    def test_root_create_file_open_raises_after_creation(self):
1012
2587
        tt, trans_id = self.create_transform_and_root_trans_id()
1013
 
        self._override_globals_in_method(
1014
 
            tt, "create_file", {"open": self._fake_open_raises_after})
1015
 
        self.assertRaises(RuntimeError, tt.create_file,
1016
 
                          [b"contents"], trans_id)
 
2588
        self._override_globals_in_method(tt, "create_file",
 
2589
            {"open": self._fake_open_raises_after})
 
2590
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
1017
2591
        path = tt._limbo_name(trans_id)
1018
2592
        self.assertPathExists(path)
1019
2593
        tt.finalize()
1022
2596
 
1023
2597
    def test_subdir_create_file_open_raises_before_creation(self):
1024
2598
        tt, trans_id = self.create_transform_and_subdir_trans_id()
1025
 
        self._override_globals_in_method(
1026
 
            tt, "create_file", {"open": self._fake_open_raises_before})
1027
 
        self.assertRaises(RuntimeError, tt.create_file,
1028
 
                          [b"contents"], trans_id)
 
2599
        self._override_globals_in_method(tt, "create_file",
 
2600
            {"open": self._fake_open_raises_before})
 
2601
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
1029
2602
        path = tt._limbo_name(trans_id)
1030
2603
        self.assertPathDoesNotExist(path)
1031
2604
        tt.finalize()
1033
2606
 
1034
2607
    def test_subdir_create_file_open_raises_after_creation(self):
1035
2608
        tt, trans_id = self.create_transform_and_subdir_trans_id()
1036
 
        self._override_globals_in_method(
1037
 
            tt, "create_file", {"open": self._fake_open_raises_after})
1038
 
        self.assertRaises(RuntimeError, tt.create_file,
1039
 
                          [b"contents"], trans_id)
 
2609
        self._override_globals_in_method(tt, "create_file",
 
2610
            {"open": self._fake_open_raises_after})
 
2611
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
1040
2612
        path = tt._limbo_name(trans_id)
1041
2613
        self.assertPathExists(path)
1042
2614
        tt.finalize()
1046
2618
    def test_rename_in_limbo_rename_raises_after_rename(self):
1047
2619
        tt, trans_id = self.create_transform_and_root_trans_id()
1048
2620
        parent1 = tt.new_directory('parent1', tt.root)
1049
 
        child1 = tt.new_file('child1', parent1, [b'contents'])
 
2621
        child1 = tt.new_file('child1', parent1, 'contents')
1050
2622
        parent2 = tt.new_directory('parent2', tt.root)
1051
2623
 
1052
2624
        class FakeOSModule(object):
1054
2626
                os.rename(old, new)
1055
2627
                raise RuntimeError
1056
2628
        self._override_globals_in_method(tt, "_rename_in_limbo",
1057
 
                                         {"os": FakeOSModule()})
 
2629
            {"os": FakeOSModule()})
1058
2630
        self.assertRaises(
1059
2631
            RuntimeError, tt.adjust_path, "child1", parent2, child1)
1060
2632
        path = osutils.pathjoin(tt._limbo_name(parent2), "child1")
1066
2638
    def test_rename_in_limbo_rename_raises_before_rename(self):
1067
2639
        tt, trans_id = self.create_transform_and_root_trans_id()
1068
2640
        parent1 = tt.new_directory('parent1', tt.root)
1069
 
        child1 = tt.new_file('child1', parent1, [b'contents'])
 
2641
        child1 = tt.new_file('child1', parent1, 'contents')
1070
2642
        parent2 = tt.new_directory('parent2', tt.root)
1071
2643
 
1072
2644
        class FakeOSModule(object):
1073
2645
            def rename(self, old, new):
1074
2646
                raise RuntimeError
1075
2647
        self._override_globals_in_method(tt, "_rename_in_limbo",
1076
 
                                         {"os": FakeOSModule()})
 
2648
            {"os": FakeOSModule()})
1077
2649
        self.assertRaises(
1078
2650
            RuntimeError, tt.adjust_path, "child1", parent2, child1)
1079
2651
        path = osutils.pathjoin(tt._limbo_name(parent1), "child1")
1087
2659
 
1088
2660
    def make_tt_with_versioned_dir(self):
1089
2661
        wt = self.make_branch_and_tree('.')
1090
 
        self.build_tree(['dir/', ])
1091
 
        wt.add(['dir'], [b'dir-id'])
 
2662
        self.build_tree(['dir/',])
 
2663
        wt.add(['dir'], ['dir-id'])
1092
2664
        wt.commit('Create dir')
1093
 
        tt = wt.transform()
 
2665
        tt = TreeTransform(wt)
1094
2666
        self.addCleanup(tt.finalize)
1095
2667
        return wt, tt
1096
2668
 
1097
2669
    def test_resolve_create_parent_for_versioned_file(self):
1098
2670
        wt, tt = self.make_tt_with_versioned_dir()
1099
 
        dir_tid = tt.trans_id_tree_path('dir')
1100
 
        tt.new_file('file', dir_tid, [b'Contents'], file_id=b'file-id')
 
2671
        dir_tid = tt.trans_id_tree_file_id('dir-id')
 
2672
        file_tid = tt.new_file('file', dir_tid, 'Contents', file_id='file-id')
1101
2673
        tt.delete_contents(dir_tid)
1102
2674
        tt.unversion_file(dir_tid)
1103
2675
        conflicts = resolve_conflicts(tt)
1107
2679
 
1108
2680
    def test_non_versioned_file_create_conflict(self):
1109
2681
        wt, tt = self.make_tt_with_versioned_dir()
1110
 
        dir_tid = tt.trans_id_tree_path('dir')
1111
 
        tt.new_file('file', dir_tid, [b'Contents'])
 
2682
        dir_tid = tt.trans_id_tree_file_id('dir-id')
 
2683
        tt.new_file('file', dir_tid, 'Contents')
1112
2684
        tt.delete_contents(dir_tid)
1113
2685
        tt.unversion_file(dir_tid)
1114
2686
        conflicts = resolve_conflicts(tt)
1118
2690
                         conflicts.pop())
1119
2691
 
1120
2692
 
 
2693
A_ENTRY = ('a-id', ('a', 'a'), True, (True, True),
 
2694
                  ('TREE_ROOT', 'TREE_ROOT'), ('a', 'a'), ('file', 'file'),
 
2695
                  (False, False))
 
2696
ROOT_ENTRY = ('TREE_ROOT', ('', ''), False, (True, True), (None, None),
 
2697
              ('', ''), ('directory', 'directory'), (False, False))
 
2698
 
 
2699
 
 
2700
class TestTransformPreview(tests.TestCaseWithTransport):
 
2701
 
 
2702
    def create_tree(self):
 
2703
        tree = self.make_branch_and_tree('.')
 
2704
        self.build_tree_contents([('a', 'content 1')])
 
2705
        tree.set_root_id('TREE_ROOT')
 
2706
        tree.add('a', 'a-id')
 
2707
        tree.commit('rev1', rev_id='rev1')
 
2708
        return tree.branch.repository.revision_tree('rev1')
 
2709
 
 
2710
    def get_empty_preview(self):
 
2711
        repository = self.make_repository('repo')
 
2712
        tree = repository.revision_tree(_mod_revision.NULL_REVISION)
 
2713
        preview = TransformPreview(tree)
 
2714
        self.addCleanup(preview.finalize)
 
2715
        return preview
 
2716
 
 
2717
    def test_transform_preview(self):
 
2718
        revision_tree = self.create_tree()
 
2719
        preview = TransformPreview(revision_tree)
 
2720
        self.addCleanup(preview.finalize)
 
2721
 
 
2722
    def test_transform_preview_tree(self):
 
2723
        revision_tree = self.create_tree()
 
2724
        preview = TransformPreview(revision_tree)
 
2725
        self.addCleanup(preview.finalize)
 
2726
        preview.get_preview_tree()
 
2727
 
 
2728
    def test_transform_new_file(self):
 
2729
        revision_tree = self.create_tree()
 
2730
        preview = TransformPreview(revision_tree)
 
2731
        self.addCleanup(preview.finalize)
 
2732
        preview.new_file('file2', preview.root, 'content B\n', 'file2-id')
 
2733
        preview_tree = preview.get_preview_tree()
 
2734
        self.assertEqual(preview_tree.kind('file2-id'), 'file')
 
2735
        self.assertEqual(
 
2736
            preview_tree.get_file('file2-id').read(), 'content B\n')
 
2737
 
 
2738
    def test_diff_preview_tree(self):
 
2739
        revision_tree = self.create_tree()
 
2740
        preview = TransformPreview(revision_tree)
 
2741
        self.addCleanup(preview.finalize)
 
2742
        preview.new_file('file2', preview.root, 'content B\n', 'file2-id')
 
2743
        preview_tree = preview.get_preview_tree()
 
2744
        out = BytesIO()
 
2745
        show_diff_trees(revision_tree, preview_tree, out)
 
2746
        lines = out.getvalue().splitlines()
 
2747
        self.assertEqual(lines[0], "=== added file 'file2'")
 
2748
        # 3 lines of diff administrivia
 
2749
        self.assertEqual(lines[4], "+content B")
 
2750
 
 
2751
    def test_transform_conflicts(self):
 
2752
        revision_tree = self.create_tree()
 
2753
        preview = TransformPreview(revision_tree)
 
2754
        self.addCleanup(preview.finalize)
 
2755
        preview.new_file('a', preview.root, 'content 2')
 
2756
        resolve_conflicts(preview)
 
2757
        trans_id = preview.trans_id_file_id('a-id')
 
2758
        self.assertEqual('a.moved', preview.final_name(trans_id))
 
2759
 
 
2760
    def get_tree_and_preview_tree(self):
 
2761
        revision_tree = self.create_tree()
 
2762
        preview = TransformPreview(revision_tree)
 
2763
        self.addCleanup(preview.finalize)
 
2764
        a_trans_id = preview.trans_id_file_id('a-id')
 
2765
        preview.delete_contents(a_trans_id)
 
2766
        preview.create_file('b content', a_trans_id)
 
2767
        preview_tree = preview.get_preview_tree()
 
2768
        return revision_tree, preview_tree
 
2769
 
 
2770
    def test_iter_changes(self):
 
2771
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2772
        root = revision_tree.get_root_id()
 
2773
        self.assertEqual([('a-id', ('a', 'a'), True, (True, True),
 
2774
                          (root, root), ('a', 'a'), ('file', 'file'),
 
2775
                          (False, False))],
 
2776
                          list(preview_tree.iter_changes(revision_tree)))
 
2777
 
 
2778
    def test_include_unchanged_succeeds(self):
 
2779
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2780
        changes = preview_tree.iter_changes(revision_tree,
 
2781
                                            include_unchanged=True)
 
2782
        root = revision_tree.get_root_id()
 
2783
 
 
2784
        self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
 
2785
 
 
2786
    def test_specific_files(self):
 
2787
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2788
        changes = preview_tree.iter_changes(revision_tree,
 
2789
                                            specific_files=[''])
 
2790
        self.assertEqual([A_ENTRY], list(changes))
 
2791
 
 
2792
    def test_want_unversioned(self):
 
2793
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2794
        changes = preview_tree.iter_changes(revision_tree,
 
2795
                                            want_unversioned=True)
 
2796
        self.assertEqual([A_ENTRY], list(changes))
 
2797
 
 
2798
    def test_ignore_extra_trees_no_specific_files(self):
 
2799
        # extra_trees is harmless without specific_files, so we'll silently
 
2800
        # accept it, even though we won't use it.
 
2801
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2802
        preview_tree.iter_changes(revision_tree, extra_trees=[preview_tree])
 
2803
 
 
2804
    def test_ignore_require_versioned_no_specific_files(self):
 
2805
        # require_versioned is meaningless without specific_files.
 
2806
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2807
        preview_tree.iter_changes(revision_tree, require_versioned=False)
 
2808
 
 
2809
    def test_ignore_pb(self):
 
2810
        # pb could be supported, but TT.iter_changes doesn't support it.
 
2811
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2812
        preview_tree.iter_changes(revision_tree)
 
2813
 
 
2814
    def test_kind(self):
 
2815
        revision_tree = self.create_tree()
 
2816
        preview = TransformPreview(revision_tree)
 
2817
        self.addCleanup(preview.finalize)
 
2818
        preview.new_file('file', preview.root, 'contents', 'file-id')
 
2819
        preview.new_directory('directory', preview.root, 'dir-id')
 
2820
        preview_tree = preview.get_preview_tree()
 
2821
        self.assertEqual('file', preview_tree.kind('file-id'))
 
2822
        self.assertEqual('directory', preview_tree.kind('dir-id'))
 
2823
 
 
2824
    def test_get_file_mtime(self):
 
2825
        preview = self.get_empty_preview()
 
2826
        file_trans_id = preview.new_file('file', preview.root, 'contents',
 
2827
                                         'file-id')
 
2828
        limbo_path = preview._limbo_name(file_trans_id)
 
2829
        preview_tree = preview.get_preview_tree()
 
2830
        self.assertEqual(os.stat(limbo_path).st_mtime,
 
2831
                         preview_tree.get_file_mtime('file-id'))
 
2832
 
 
2833
    def test_get_file_mtime_renamed(self):
 
2834
        work_tree = self.make_branch_and_tree('tree')
 
2835
        self.build_tree(['tree/file'])
 
2836
        work_tree.add('file', 'file-id')
 
2837
        preview = TransformPreview(work_tree)
 
2838
        self.addCleanup(preview.finalize)
 
2839
        file_trans_id = preview.trans_id_tree_file_id('file-id')
 
2840
        preview.adjust_path('renamed', preview.root, file_trans_id)
 
2841
        preview_tree = preview.get_preview_tree()
 
2842
        preview_mtime = preview_tree.get_file_mtime('file-id', 'renamed')
 
2843
        work_mtime = work_tree.get_file_mtime('file-id', 'file')
 
2844
 
 
2845
    def test_get_file_size(self):
 
2846
        work_tree = self.make_branch_and_tree('tree')
 
2847
        self.build_tree_contents([('tree/old', 'old')])
 
2848
        work_tree.add('old', 'old-id')
 
2849
        preview = TransformPreview(work_tree)
 
2850
        self.addCleanup(preview.finalize)
 
2851
        new_id = preview.new_file('name', preview.root, 'contents', 'new-id',
 
2852
                                  'executable')
 
2853
        tree = preview.get_preview_tree()
 
2854
        self.assertEqual(len('old'), tree.get_file_size('old-id'))
 
2855
        self.assertEqual(len('contents'), tree.get_file_size('new-id'))
 
2856
 
 
2857
    def test_get_file(self):
 
2858
        preview = self.get_empty_preview()
 
2859
        preview.new_file('file', preview.root, 'contents', 'file-id')
 
2860
        preview_tree = preview.get_preview_tree()
 
2861
        tree_file = preview_tree.get_file('file-id')
 
2862
        try:
 
2863
            self.assertEqual('contents', tree_file.read())
 
2864
        finally:
 
2865
            tree_file.close()
 
2866
 
 
2867
    def test_get_symlink_target(self):
 
2868
        self.requireFeature(SymlinkFeature)
 
2869
        preview = self.get_empty_preview()
 
2870
        preview.new_symlink('symlink', preview.root, 'target', 'symlink-id')
 
2871
        preview_tree = preview.get_preview_tree()
 
2872
        self.assertEqual('target',
 
2873
                         preview_tree.get_symlink_target('symlink-id'))
 
2874
 
 
2875
    def test_all_file_ids(self):
 
2876
        tree = self.make_branch_and_tree('tree')
 
2877
        self.build_tree(['tree/a', 'tree/b', 'tree/c'])
 
2878
        tree.add(['a', 'b', 'c'], ['a-id', 'b-id', 'c-id'])
 
2879
        preview = TransformPreview(tree)
 
2880
        self.addCleanup(preview.finalize)
 
2881
        preview.unversion_file(preview.trans_id_file_id('b-id'))
 
2882
        c_trans_id = preview.trans_id_file_id('c-id')
 
2883
        preview.unversion_file(c_trans_id)
 
2884
        preview.version_file('c-id', c_trans_id)
 
2885
        preview_tree = preview.get_preview_tree()
 
2886
        self.assertEqual({'a-id', 'c-id', tree.get_root_id()},
 
2887
                         preview_tree.all_file_ids())
 
2888
 
 
2889
    def test_path2id_deleted_unchanged(self):
 
2890
        tree = self.make_branch_and_tree('tree')
 
2891
        self.build_tree(['tree/unchanged', 'tree/deleted'])
 
2892
        tree.add(['unchanged', 'deleted'], ['unchanged-id', 'deleted-id'])
 
2893
        preview = TransformPreview(tree)
 
2894
        self.addCleanup(preview.finalize)
 
2895
        preview.unversion_file(preview.trans_id_file_id('deleted-id'))
 
2896
        preview_tree = preview.get_preview_tree()
 
2897
        self.assertEqual('unchanged-id', preview_tree.path2id('unchanged'))
 
2898
        self.assertIs(None, preview_tree.path2id('deleted'))
 
2899
 
 
2900
    def test_path2id_created(self):
 
2901
        tree = self.make_branch_and_tree('tree')
 
2902
        self.build_tree(['tree/unchanged'])
 
2903
        tree.add(['unchanged'], ['unchanged-id'])
 
2904
        preview = TransformPreview(tree)
 
2905
        self.addCleanup(preview.finalize)
 
2906
        preview.new_file('new', preview.trans_id_file_id('unchanged-id'),
 
2907
            'contents', 'new-id')
 
2908
        preview_tree = preview.get_preview_tree()
 
2909
        self.assertEqual('new-id', preview_tree.path2id('unchanged/new'))
 
2910
 
 
2911
    def test_path2id_moved(self):
 
2912
        tree = self.make_branch_and_tree('tree')
 
2913
        self.build_tree(['tree/old_parent/', 'tree/old_parent/child'])
 
2914
        tree.add(['old_parent', 'old_parent/child'],
 
2915
                 ['old_parent-id', 'child-id'])
 
2916
        preview = TransformPreview(tree)
 
2917
        self.addCleanup(preview.finalize)
 
2918
        new_parent = preview.new_directory('new_parent', preview.root,
 
2919
                                           'new_parent-id')
 
2920
        preview.adjust_path('child', new_parent,
 
2921
                            preview.trans_id_file_id('child-id'))
 
2922
        preview_tree = preview.get_preview_tree()
 
2923
        self.assertIs(None, preview_tree.path2id('old_parent/child'))
 
2924
        self.assertEqual('child-id', preview_tree.path2id('new_parent/child'))
 
2925
 
 
2926
    def test_path2id_renamed_parent(self):
 
2927
        tree = self.make_branch_and_tree('tree')
 
2928
        self.build_tree(['tree/old_name/', 'tree/old_name/child'])
 
2929
        tree.add(['old_name', 'old_name/child'],
 
2930
                 ['parent-id', 'child-id'])
 
2931
        preview = TransformPreview(tree)
 
2932
        self.addCleanup(preview.finalize)
 
2933
        preview.adjust_path('new_name', preview.root,
 
2934
                            preview.trans_id_file_id('parent-id'))
 
2935
        preview_tree = preview.get_preview_tree()
 
2936
        self.assertIs(None, preview_tree.path2id('old_name/child'))
 
2937
        self.assertEqual('child-id', preview_tree.path2id('new_name/child'))
 
2938
 
 
2939
    def assertMatchingIterEntries(self, tt, specific_file_ids=None):
 
2940
        preview_tree = tt.get_preview_tree()
 
2941
        preview_result = list(preview_tree.iter_entries_by_dir(
 
2942
                              specific_file_ids))
 
2943
        tree = tt._tree
 
2944
        tt.apply()
 
2945
        actual_result = list(tree.iter_entries_by_dir(specific_file_ids))
 
2946
        self.assertEqual(actual_result, preview_result)
 
2947
 
 
2948
    def test_iter_entries_by_dir_new(self):
 
2949
        tree = self.make_branch_and_tree('tree')
 
2950
        tt = TreeTransform(tree)
 
2951
        tt.new_file('new', tt.root, 'contents', 'new-id')
 
2952
        self.assertMatchingIterEntries(tt)
 
2953
 
 
2954
    def test_iter_entries_by_dir_deleted(self):
 
2955
        tree = self.make_branch_and_tree('tree')
 
2956
        self.build_tree(['tree/deleted'])
 
2957
        tree.add('deleted', 'deleted-id')
 
2958
        tt = TreeTransform(tree)
 
2959
        tt.delete_contents(tt.trans_id_file_id('deleted-id'))
 
2960
        self.assertMatchingIterEntries(tt)
 
2961
 
 
2962
    def test_iter_entries_by_dir_unversioned(self):
 
2963
        tree = self.make_branch_and_tree('tree')
 
2964
        self.build_tree(['tree/removed'])
 
2965
        tree.add('removed', 'removed-id')
 
2966
        tt = TreeTransform(tree)
 
2967
        tt.unversion_file(tt.trans_id_file_id('removed-id'))
 
2968
        self.assertMatchingIterEntries(tt)
 
2969
 
 
2970
    def test_iter_entries_by_dir_moved(self):
 
2971
        tree = self.make_branch_and_tree('tree')
 
2972
        self.build_tree(['tree/moved', 'tree/new_parent/'])
 
2973
        tree.add(['moved', 'new_parent'], ['moved-id', 'new_parent-id'])
 
2974
        tt = TreeTransform(tree)
 
2975
        tt.adjust_path('moved', tt.trans_id_file_id('new_parent-id'),
 
2976
                       tt.trans_id_file_id('moved-id'))
 
2977
        self.assertMatchingIterEntries(tt)
 
2978
 
 
2979
    def test_iter_entries_by_dir_specific_file_ids(self):
 
2980
        tree = self.make_branch_and_tree('tree')
 
2981
        tree.set_root_id('tree-root-id')
 
2982
        self.build_tree(['tree/parent/', 'tree/parent/child'])
 
2983
        tree.add(['parent', 'parent/child'], ['parent-id', 'child-id'])
 
2984
        tt = TreeTransform(tree)
 
2985
        self.assertMatchingIterEntries(tt, ['tree-root-id', 'child-id'])
 
2986
 
 
2987
    def test_symlink_content_summary(self):
 
2988
        self.requireFeature(SymlinkFeature)
 
2989
        preview = self.get_empty_preview()
 
2990
        preview.new_symlink('path', preview.root, 'target', 'path-id')
 
2991
        summary = preview.get_preview_tree().path_content_summary('path')
 
2992
        self.assertEqual(('symlink', None, None, 'target'), summary)
 
2993
 
 
2994
    def test_missing_content_summary(self):
 
2995
        preview = self.get_empty_preview()
 
2996
        summary = preview.get_preview_tree().path_content_summary('path')
 
2997
        self.assertEqual(('missing', None, None, None), summary)
 
2998
 
 
2999
    def test_deleted_content_summary(self):
 
3000
        tree = self.make_branch_and_tree('tree')
 
3001
        self.build_tree(['tree/path/'])
 
3002
        tree.add('path')
 
3003
        preview = TransformPreview(tree)
 
3004
        self.addCleanup(preview.finalize)
 
3005
        preview.delete_contents(preview.trans_id_tree_path('path'))
 
3006
        summary = preview.get_preview_tree().path_content_summary('path')
 
3007
        self.assertEqual(('missing', None, None, None), summary)
 
3008
 
 
3009
    def test_file_content_summary_executable(self):
 
3010
        preview = self.get_empty_preview()
 
3011
        path_id = preview.new_file('path', preview.root, 'contents', 'path-id')
 
3012
        preview.set_executability(True, path_id)
 
3013
        summary = preview.get_preview_tree().path_content_summary('path')
 
3014
        self.assertEqual(4, len(summary))
 
3015
        self.assertEqual('file', summary[0])
 
3016
        # size must be known
 
3017
        self.assertEqual(len('contents'), summary[1])
 
3018
        # executable
 
3019
        self.assertEqual(True, summary[2])
 
3020
        # will not have hash (not cheap to determine)
 
3021
        self.assertIs(None, summary[3])
 
3022
 
 
3023
    def test_change_executability(self):
 
3024
        tree = self.make_branch_and_tree('tree')
 
3025
        self.build_tree(['tree/path'])
 
3026
        tree.add('path')
 
3027
        preview = TransformPreview(tree)
 
3028
        self.addCleanup(preview.finalize)
 
3029
        path_id = preview.trans_id_tree_path('path')
 
3030
        preview.set_executability(True, path_id)
 
3031
        summary = preview.get_preview_tree().path_content_summary('path')
 
3032
        self.assertEqual(True, summary[2])
 
3033
 
 
3034
    def test_file_content_summary_non_exec(self):
 
3035
        preview = self.get_empty_preview()
 
3036
        preview.new_file('path', preview.root, 'contents', 'path-id')
 
3037
        summary = preview.get_preview_tree().path_content_summary('path')
 
3038
        self.assertEqual(4, len(summary))
 
3039
        self.assertEqual('file', summary[0])
 
3040
        # size must be known
 
3041
        self.assertEqual(len('contents'), summary[1])
 
3042
        # not executable
 
3043
        self.assertEqual(False, summary[2])
 
3044
        # will not have hash (not cheap to determine)
 
3045
        self.assertIs(None, summary[3])
 
3046
 
 
3047
    def test_dir_content_summary(self):
 
3048
        preview = self.get_empty_preview()
 
3049
        preview.new_directory('path', preview.root, 'path-id')
 
3050
        summary = preview.get_preview_tree().path_content_summary('path')
 
3051
        self.assertEqual(('directory', None, None, None), summary)
 
3052
 
 
3053
    def test_tree_content_summary(self):
 
3054
        preview = self.get_empty_preview()
 
3055
        path = preview.new_directory('path', preview.root, 'path-id')
 
3056
        preview.set_tree_reference('rev-1', path)
 
3057
        summary = preview.get_preview_tree().path_content_summary('path')
 
3058
        self.assertEqual(4, len(summary))
 
3059
        self.assertEqual('tree-reference', summary[0])
 
3060
 
 
3061
    def test_annotate(self):
 
3062
        tree = self.make_branch_and_tree('tree')
 
3063
        self.build_tree_contents([('tree/file', 'a\n')])
 
3064
        tree.add('file', 'file-id')
 
3065
        tree.commit('a', rev_id='one')
 
3066
        self.build_tree_contents([('tree/file', 'a\nb\n')])
 
3067
        preview = TransformPreview(tree)
 
3068
        self.addCleanup(preview.finalize)
 
3069
        file_trans_id = preview.trans_id_file_id('file-id')
 
3070
        preview.delete_contents(file_trans_id)
 
3071
        preview.create_file('a\nb\nc\n', file_trans_id)
 
3072
        preview_tree = preview.get_preview_tree()
 
3073
        expected = [
 
3074
            ('one', 'a\n'),
 
3075
            ('me:', 'b\n'),
 
3076
            ('me:', 'c\n'),
 
3077
        ]
 
3078
        annotation = preview_tree.annotate_iter('file-id', 'me:')
 
3079
        self.assertEqual(expected, annotation)
 
3080
 
 
3081
    def test_annotate_missing(self):
 
3082
        preview = self.get_empty_preview()
 
3083
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
 
3084
        preview_tree = preview.get_preview_tree()
 
3085
        expected = [
 
3086
            ('me:', 'a\n'),
 
3087
            ('me:', 'b\n'),
 
3088
            ('me:', 'c\n'),
 
3089
         ]
 
3090
        annotation = preview_tree.annotate_iter('file-id', 'me:')
 
3091
        self.assertEqual(expected, annotation)
 
3092
 
 
3093
    def test_annotate_rename(self):
 
3094
        tree = self.make_branch_and_tree('tree')
 
3095
        self.build_tree_contents([('tree/file', 'a\n')])
 
3096
        tree.add('file', 'file-id')
 
3097
        tree.commit('a', rev_id='one')
 
3098
        preview = TransformPreview(tree)
 
3099
        self.addCleanup(preview.finalize)
 
3100
        file_trans_id = preview.trans_id_file_id('file-id')
 
3101
        preview.adjust_path('newname', preview.root, file_trans_id)
 
3102
        preview_tree = preview.get_preview_tree()
 
3103
        expected = [
 
3104
            ('one', 'a\n'),
 
3105
        ]
 
3106
        annotation = preview_tree.annotate_iter('file-id', 'me:')
 
3107
        self.assertEqual(expected, annotation)
 
3108
 
 
3109
    def test_annotate_deleted(self):
 
3110
        tree = self.make_branch_and_tree('tree')
 
3111
        self.build_tree_contents([('tree/file', 'a\n')])
 
3112
        tree.add('file', 'file-id')
 
3113
        tree.commit('a', rev_id='one')
 
3114
        self.build_tree_contents([('tree/file', 'a\nb\n')])
 
3115
        preview = TransformPreview(tree)
 
3116
        self.addCleanup(preview.finalize)
 
3117
        file_trans_id = preview.trans_id_file_id('file-id')
 
3118
        preview.delete_contents(file_trans_id)
 
3119
        preview_tree = preview.get_preview_tree()
 
3120
        annotation = preview_tree.annotate_iter('file-id', 'me:')
 
3121
        self.assertIs(None, annotation)
 
3122
 
 
3123
    def test_stored_kind(self):
 
3124
        preview = self.get_empty_preview()
 
3125
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
 
3126
        preview_tree = preview.get_preview_tree()
 
3127
        self.assertEqual('file', preview_tree.stored_kind('file-id'))
 
3128
 
 
3129
    def test_is_executable(self):
 
3130
        preview = self.get_empty_preview()
 
3131
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
 
3132
        preview.set_executability(True, preview.trans_id_file_id('file-id'))
 
3133
        preview_tree = preview.get_preview_tree()
 
3134
        self.assertEqual(True, preview_tree.is_executable('file-id'))
 
3135
 
 
3136
    def test_get_set_parent_ids(self):
 
3137
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
3138
        self.assertEqual([], preview_tree.get_parent_ids())
 
3139
        preview_tree.set_parent_ids(['rev-1'])
 
3140
        self.assertEqual(['rev-1'], preview_tree.get_parent_ids())
 
3141
 
 
3142
    def test_plan_file_merge(self):
 
3143
        work_a = self.make_branch_and_tree('wta')
 
3144
        self.build_tree_contents([('wta/file', 'a\nb\nc\nd\n')])
 
3145
        work_a.add('file', 'file-id')
 
3146
        base_id = work_a.commit('base version')
 
3147
        tree_b = work_a.bzrdir.sprout('wtb').open_workingtree()
 
3148
        preview = TransformPreview(work_a)
 
3149
        self.addCleanup(preview.finalize)
 
3150
        trans_id = preview.trans_id_file_id('file-id')
 
3151
        preview.delete_contents(trans_id)
 
3152
        preview.create_file('b\nc\nd\ne\n', trans_id)
 
3153
        self.build_tree_contents([('wtb/file', 'a\nc\nd\nf\n')])
 
3154
        tree_a = preview.get_preview_tree()
 
3155
        tree_a.set_parent_ids([base_id])
 
3156
        self.assertEqual([
 
3157
            ('killed-a', 'a\n'),
 
3158
            ('killed-b', 'b\n'),
 
3159
            ('unchanged', 'c\n'),
 
3160
            ('unchanged', 'd\n'),
 
3161
            ('new-a', 'e\n'),
 
3162
            ('new-b', 'f\n'),
 
3163
        ], list(tree_a.plan_file_merge('file-id', tree_b)))
 
3164
 
 
3165
    def test_plan_file_merge_revision_tree(self):
 
3166
        work_a = self.make_branch_and_tree('wta')
 
3167
        self.build_tree_contents([('wta/file', 'a\nb\nc\nd\n')])
 
3168
        work_a.add('file', 'file-id')
 
3169
        base_id = work_a.commit('base version')
 
3170
        tree_b = work_a.bzrdir.sprout('wtb').open_workingtree()
 
3171
        preview = TransformPreview(work_a.basis_tree())
 
3172
        self.addCleanup(preview.finalize)
 
3173
        trans_id = preview.trans_id_file_id('file-id')
 
3174
        preview.delete_contents(trans_id)
 
3175
        preview.create_file('b\nc\nd\ne\n', trans_id)
 
3176
        self.build_tree_contents([('wtb/file', 'a\nc\nd\nf\n')])
 
3177
        tree_a = preview.get_preview_tree()
 
3178
        tree_a.set_parent_ids([base_id])
 
3179
        self.assertEqual([
 
3180
            ('killed-a', 'a\n'),
 
3181
            ('killed-b', 'b\n'),
 
3182
            ('unchanged', 'c\n'),
 
3183
            ('unchanged', 'd\n'),
 
3184
            ('new-a', 'e\n'),
 
3185
            ('new-b', 'f\n'),
 
3186
        ], list(tree_a.plan_file_merge('file-id', tree_b)))
 
3187
 
 
3188
    def test_walkdirs(self):
 
3189
        preview = self.get_empty_preview()
 
3190
        root = preview.new_directory('', ROOT_PARENT, 'tree-root')
 
3191
        # FIXME: new_directory should mark root.
 
3192
        preview.fixup_new_roots()
 
3193
        preview_tree = preview.get_preview_tree()
 
3194
        file_trans_id = preview.new_file('a', preview.root, 'contents',
 
3195
                                         'a-id')
 
3196
        expected = [(('', 'tree-root'),
 
3197
                    [('a', 'a', 'file', None, 'a-id', 'file')])]
 
3198
        self.assertEqual(expected, list(preview_tree.walkdirs()))
 
3199
 
 
3200
    def test_extras(self):
 
3201
        work_tree = self.make_branch_and_tree('tree')
 
3202
        self.build_tree(['tree/removed-file', 'tree/existing-file',
 
3203
                         'tree/not-removed-file'])
 
3204
        work_tree.add(['removed-file', 'not-removed-file'])
 
3205
        preview = TransformPreview(work_tree)
 
3206
        self.addCleanup(preview.finalize)
 
3207
        preview.new_file('new-file', preview.root, 'contents')
 
3208
        preview.new_file('new-versioned-file', preview.root, 'contents',
 
3209
                         'new-versioned-id')
 
3210
        tree = preview.get_preview_tree()
 
3211
        preview.unversion_file(preview.trans_id_tree_path('removed-file'))
 
3212
        self.assertEqual({'new-file', 'removed-file', 'existing-file'},
 
3213
                         set(tree.extras()))
 
3214
 
 
3215
    def test_merge_into_preview(self):
 
3216
        work_tree = self.make_branch_and_tree('tree')
 
3217
        self.build_tree_contents([('tree/file','b\n')])
 
3218
        work_tree.add('file', 'file-id')
 
3219
        work_tree.commit('first commit')
 
3220
        child_tree = work_tree.bzrdir.sprout('child').open_workingtree()
 
3221
        self.build_tree_contents([('child/file','b\nc\n')])
 
3222
        child_tree.commit('child commit')
 
3223
        child_tree.lock_write()
 
3224
        self.addCleanup(child_tree.unlock)
 
3225
        work_tree.lock_write()
 
3226
        self.addCleanup(work_tree.unlock)
 
3227
        preview = TransformPreview(work_tree)
 
3228
        self.addCleanup(preview.finalize)
 
3229
        file_trans_id = preview.trans_id_file_id('file-id')
 
3230
        preview.delete_contents(file_trans_id)
 
3231
        preview.create_file('a\nb\n', file_trans_id)
 
3232
        preview_tree = preview.get_preview_tree()
 
3233
        merger = Merger.from_revision_ids(None, preview_tree,
 
3234
                                          child_tree.branch.last_revision(),
 
3235
                                          other_branch=child_tree.branch,
 
3236
                                          tree_branch=work_tree.branch)
 
3237
        merger.merge_type = Merge3Merger
 
3238
        tt = merger.make_merger().make_preview_transform()
 
3239
        self.addCleanup(tt.finalize)
 
3240
        final_tree = tt.get_preview_tree()
 
3241
        self.assertEqual('a\nb\nc\n', final_tree.get_file_text('file-id'))
 
3242
 
 
3243
    def test_merge_preview_into_workingtree(self):
 
3244
        tree = self.make_branch_and_tree('tree')
 
3245
        tree.set_root_id('TREE_ROOT')
 
3246
        tt = TransformPreview(tree)
 
3247
        self.addCleanup(tt.finalize)
 
3248
        tt.new_file('name', tt.root, 'content', 'file-id')
 
3249
        tree2 = self.make_branch_and_tree('tree2')
 
3250
        tree2.set_root_id('TREE_ROOT')
 
3251
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
 
3252
                                         None, tree.basis_tree())
 
3253
        merger.merge_type = Merge3Merger
 
3254
        merger.do_merge()
 
3255
 
 
3256
    def test_merge_preview_into_workingtree_handles_conflicts(self):
 
3257
        tree = self.make_branch_and_tree('tree')
 
3258
        self.build_tree_contents([('tree/foo', 'bar')])
 
3259
        tree.add('foo', 'foo-id')
 
3260
        tree.commit('foo')
 
3261
        tt = TransformPreview(tree)
 
3262
        self.addCleanup(tt.finalize)
 
3263
        trans_id = tt.trans_id_file_id('foo-id')
 
3264
        tt.delete_contents(trans_id)
 
3265
        tt.create_file('baz', trans_id)
 
3266
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
 
3267
        self.build_tree_contents([('tree2/foo', 'qux')])
 
3268
        pb = None
 
3269
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
 
3270
                                         pb, tree.basis_tree())
 
3271
        merger.merge_type = Merge3Merger
 
3272
        merger.do_merge()
 
3273
 
 
3274
    def test_has_filename(self):
 
3275
        wt = self.make_branch_and_tree('tree')
 
3276
        self.build_tree(['tree/unmodified', 'tree/removed', 'tree/modified'])
 
3277
        tt = TransformPreview(wt)
 
3278
        removed_id = tt.trans_id_tree_path('removed')
 
3279
        tt.delete_contents(removed_id)
 
3280
        tt.new_file('new', tt.root, 'contents')
 
3281
        modified_id = tt.trans_id_tree_path('modified')
 
3282
        tt.delete_contents(modified_id)
 
3283
        tt.create_file('modified-contents', modified_id)
 
3284
        self.addCleanup(tt.finalize)
 
3285
        tree = tt.get_preview_tree()
 
3286
        self.assertTrue(tree.has_filename('unmodified'))
 
3287
        self.assertFalse(tree.has_filename('not-present'))
 
3288
        self.assertFalse(tree.has_filename('removed'))
 
3289
        self.assertTrue(tree.has_filename('new'))
 
3290
        self.assertTrue(tree.has_filename('modified'))
 
3291
 
 
3292
    def test_is_executable(self):
 
3293
        tree = self.make_branch_and_tree('tree')
 
3294
        preview = TransformPreview(tree)
 
3295
        self.addCleanup(preview.finalize)
 
3296
        preview.new_file('foo', preview.root, 'bar', 'baz-id')
 
3297
        preview_tree = preview.get_preview_tree()
 
3298
        self.assertEqual(False, preview_tree.is_executable('baz-id',
 
3299
                                                           'tree/foo'))
 
3300
        self.assertEqual(False, preview_tree.is_executable('baz-id'))
 
3301
 
 
3302
    def test_commit_preview_tree(self):
 
3303
        tree = self.make_branch_and_tree('tree')
 
3304
        rev_id = tree.commit('rev1')
 
3305
        tree.branch.lock_write()
 
3306
        self.addCleanup(tree.branch.unlock)
 
3307
        tt = TransformPreview(tree)
 
3308
        tt.new_file('file', tt.root, 'contents', 'file_id')
 
3309
        self.addCleanup(tt.finalize)
 
3310
        preview = tt.get_preview_tree()
 
3311
        preview.set_parent_ids([rev_id])
 
3312
        builder = tree.branch.get_commit_builder([rev_id])
 
3313
        list(builder.record_iter_changes(preview, rev_id, tt.iter_changes()))
 
3314
        builder.finish_inventory()
 
3315
        rev2_id = builder.commit('rev2')
 
3316
        rev2_tree = tree.branch.repository.revision_tree(rev2_id)
 
3317
        self.assertEqual('contents', rev2_tree.get_file_text('file_id'))
 
3318
 
 
3319
    def test_ascii_limbo_paths(self):
 
3320
        self.requireFeature(features.UnicodeFilenameFeature)
 
3321
        branch = self.make_branch('any')
 
3322
        tree = branch.repository.revision_tree(_mod_revision.NULL_REVISION)
 
3323
        tt = TransformPreview(tree)
 
3324
        self.addCleanup(tt.finalize)
 
3325
        foo_id = tt.new_directory('', ROOT_PARENT)
 
3326
        bar_id = tt.new_file(u'\u1234bar', foo_id, 'contents')
 
3327
        limbo_path = tt._limbo_name(bar_id)
 
3328
        self.assertEqual(limbo_path.encode('ascii', 'replace'), limbo_path)
 
3329
 
 
3330
 
1121
3331
class FakeSerializer(object):
1122
3332
    """Serializer implementation that simply returns the input.
1123
3333
 
1135
3345
    def get_preview(self, tree=None):
1136
3346
        if tree is None:
1137
3347
            tree = self.make_branch_and_tree('tree')
1138
 
        tt = tree.preview_transform()
 
3348
        tt = TransformPreview(tree)
1139
3349
        self.addCleanup(tt.finalize)
1140
3350
        return tt
1141
3351
 
1146
3356
    @staticmethod
1147
3357
    def default_attribs():
1148
3358
        return {
1149
 
            b'_id_number': 1,
1150
 
            b'_new_name': {},
1151
 
            b'_new_parent': {},
1152
 
            b'_new_executability': {},
1153
 
            b'_new_id': {},
1154
 
            b'_tree_path_ids': {b'': b'new-0'},
1155
 
            b'_removed_id': [],
1156
 
            b'_removed_contents': [],
1157
 
            b'_non_present_ids': {},
 
3359
            '_id_number': 1,
 
3360
            '_new_name': {},
 
3361
            '_new_parent': {},
 
3362
            '_new_executability': {},
 
3363
            '_new_id': {},
 
3364
            '_tree_path_ids': {'': 'new-0'},
 
3365
            '_removed_id': [],
 
3366
            '_removed_contents': [],
 
3367
            '_non_present_ids': {},
1158
3368
            }
1159
3369
 
1160
3370
    def make_records(self, attribs, contents):
1161
3371
        records = [
1162
 
            ((((b'attribs'),),), bencode.bencode(attribs))]
 
3372
            (((('attribs'),),), bencode.bencode(attribs))]
1163
3373
        records.extend([(((n, k),), c) for n, k, c in contents])
1164
3374
        return records
1165
3375
 
1166
3376
    def creation_records(self):
1167
3377
        attribs = self.default_attribs()
1168
 
        attribs[b'_id_number'] = 3
1169
 
        attribs[b'_new_name'] = {
1170
 
            b'new-1': u'foo\u1234'.encode('utf-8'), b'new-2': b'qux'}
1171
 
        attribs[b'_new_id'] = {b'new-1': b'baz', b'new-2': b'quxx'}
1172
 
        attribs[b'_new_parent'] = {b'new-1': b'new-0', b'new-2': b'new-0'}
1173
 
        attribs[b'_new_executability'] = {b'new-1': 1}
 
3378
        attribs['_id_number'] = 3
 
3379
        attribs['_new_name'] = {
 
3380
            'new-1': u'foo\u1234'.encode('utf-8'), 'new-2': 'qux'}
 
3381
        attribs['_new_id'] = {'new-1': 'baz', 'new-2': 'quxx'}
 
3382
        attribs['_new_parent'] = {'new-1': 'new-0', 'new-2': 'new-0'}
 
3383
        attribs['_new_executability'] = {'new-1': 1}
1174
3384
        contents = [
1175
 
            (b'new-1', b'file', b'i 1\nbar\n'),
1176
 
            (b'new-2', b'directory', b''),
 
3385
            ('new-1', 'file', 'i 1\nbar\n'),
 
3386
            ('new-2', 'directory', ''),
1177
3387
            ]
1178
3388
        return self.make_records(attribs, contents)
1179
3389
 
1180
3390
    def test_serialize_creation(self):
1181
3391
        tt = self.get_preview()
1182
 
        tt.new_file(u'foo\u1234', tt.root, [b'bar'], b'baz', True)
1183
 
        tt.new_directory('qux', tt.root, b'quxx')
 
3392
        tt.new_file(u'foo\u1234', tt.root, 'bar', 'baz', True)
 
3393
        tt.new_directory('qux', tt.root, 'quxx')
1184
3394
        self.assertSerializesTo(self.creation_records(), tt)
1185
3395
 
1186
3396
    def test_deserialize_creation(self):
1189
3399
        self.assertEqual(3, tt._id_number)
1190
3400
        self.assertEqual({'new-1': u'foo\u1234',
1191
3401
                          'new-2': 'qux'}, tt._new_name)
1192
 
        self.assertEqual({'new-1': b'baz', 'new-2': b'quxx'}, tt._new_id)
 
3402
        self.assertEqual({'new-1': 'baz', 'new-2': 'quxx'}, tt._new_id)
1193
3403
        self.assertEqual({'new-1': tt.root, 'new-2': tt.root}, tt._new_parent)
1194
 
        self.assertEqual({b'baz': 'new-1', b'quxx': 'new-2'}, tt._r_new_id)
 
3404
        self.assertEqual({'baz': 'new-1', 'quxx': 'new-2'}, tt._r_new_id)
1195
3405
        self.assertEqual({'new-1': True}, tt._new_executability)
1196
3406
        self.assertEqual({'new-1': 'file',
1197
3407
                          'new-2': 'directory'}, tt._new_contents)
1198
 
        with open(tt._limbo_name('new-1'), 'rb') as foo_limbo:
 
3408
        foo_limbo = open(tt._limbo_name('new-1'), 'rb')
 
3409
        try:
1199
3410
            foo_content = foo_limbo.read()
1200
 
        self.assertEqual(b'bar', foo_content)
 
3411
        finally:
 
3412
            foo_limbo.close()
 
3413
        self.assertEqual('bar', foo_content)
1201
3414
 
1202
3415
    def symlink_creation_records(self):
1203
3416
        attribs = self.default_attribs()
1204
 
        attribs[b'_id_number'] = 2
1205
 
        attribs[b'_new_name'] = {b'new-1': u'foo\u1234'.encode('utf-8')}
1206
 
        attribs[b'_new_parent'] = {b'new-1': b'new-0'}
1207
 
        contents = [(b'new-1', b'symlink', u'bar\u1234'.encode('utf-8'))]
 
3417
        attribs['_id_number'] = 2
 
3418
        attribs['_new_name'] = {'new-1': u'foo\u1234'.encode('utf-8')}
 
3419
        attribs['_new_parent'] = {'new-1': 'new-0'}
 
3420
        contents = [('new-1', 'symlink', u'bar\u1234'.encode('utf-8'))]
1208
3421
        return self.make_records(attribs, contents)
1209
3422
 
1210
3423
    def test_serialize_symlink_creation(self):
1224
3437
    def make_destruction_preview(self):
1225
3438
        tree = self.make_branch_and_tree('.')
1226
3439
        self.build_tree([u'foo\u1234', 'bar'])
1227
 
        tree.add([u'foo\u1234', 'bar'], [b'foo-id', b'bar-id'])
 
3440
        tree.add([u'foo\u1234', 'bar'], ['foo-id', 'bar-id'])
1228
3441
        return self.get_preview(tree)
1229
3442
 
1230
3443
    def destruction_records(self):
1231
3444
        attribs = self.default_attribs()
1232
 
        attribs[b'_id_number'] = 3
1233
 
        attribs[b'_removed_id'] = [b'new-1']
1234
 
        attribs[b'_removed_contents'] = [b'new-2']
1235
 
        attribs[b'_tree_path_ids'] = {
1236
 
            b'': b'new-0',
1237
 
            u'foo\u1234'.encode('utf-8'): b'new-1',
1238
 
            b'bar': b'new-2',
 
3445
        attribs['_id_number'] = 3
 
3446
        attribs['_removed_id'] = ['new-1']
 
3447
        attribs['_removed_contents'] = ['new-2']
 
3448
        attribs['_tree_path_ids'] = {
 
3449
            '': 'new-0',
 
3450
            u'foo\u1234'.encode('utf-8'): 'new-1',
 
3451
            'bar': 'new-2',
1239
3452
            }
1240
3453
        return self.make_records(attribs, [])
1241
3454
 
1242
3455
    def test_serialize_destruction(self):
1243
3456
        tt = self.make_destruction_preview()
1244
 
        foo_trans_id = tt.trans_id_tree_path(u'foo\u1234')
 
3457
        foo_trans_id = tt.trans_id_tree_file_id('foo-id')
1245
3458
        tt.unversion_file(foo_trans_id)
1246
 
        bar_trans_id = tt.trans_id_tree_path('bar')
 
3459
        bar_trans_id = tt.trans_id_tree_file_id('bar-id')
1247
3460
        tt.delete_contents(bar_trans_id)
1248
3461
        self.assertSerializesTo(self.destruction_records(), tt)
1249
3462
 
1261
3474
 
1262
3475
    def missing_records(self):
1263
3476
        attribs = self.default_attribs()
1264
 
        attribs[b'_id_number'] = 2
1265
 
        attribs[b'_non_present_ids'] = {
1266
 
            b'boo': b'new-1', }
 
3477
        attribs['_id_number'] = 2
 
3478
        attribs['_non_present_ids'] = {
 
3479
            'boo': 'new-1',}
1267
3480
        return self.make_records(attribs, [])
1268
3481
 
1269
3482
    def test_serialize_missing(self):
1270
3483
        tt = self.get_preview()
1271
 
        tt.trans_id_file_id(b'boo')
 
3484
        boo_trans_id = tt.trans_id_file_id('boo')
1272
3485
        self.assertSerializesTo(self.missing_records(), tt)
1273
3486
 
1274
3487
    def test_deserialize_missing(self):
1275
3488
        tt = self.get_preview()
1276
3489
        tt.deserialize(iter(self.missing_records()))
1277
 
        self.assertEqual({b'boo': 'new-1'}, tt._non_present_ids)
 
3490
        self.assertEqual({'boo': 'new-1'}, tt._non_present_ids)
1278
3491
 
1279
3492
    def make_modification_preview(self):
1280
 
        LINES_ONE = b'aa\nbb\ncc\ndd\n'
1281
 
        LINES_TWO = b'z\nbb\nx\ndd\n'
 
3493
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
 
3494
        LINES_TWO = 'z\nbb\nx\ndd\n'
1282
3495
        tree = self.make_branch_and_tree('tree')
1283
3496
        self.build_tree_contents([('tree/file', LINES_ONE)])
1284
 
        tree.add('file', b'file-id')
1285
 
        return self.get_preview(tree), [LINES_TWO]
 
3497
        tree.add('file', 'file-id')
 
3498
        return self.get_preview(tree), LINES_TWO
1286
3499
 
1287
3500
    def modification_records(self):
1288
3501
        attribs = self.default_attribs()
1289
 
        attribs[b'_id_number'] = 2
1290
 
        attribs[b'_tree_path_ids'] = {
1291
 
            b'file': b'new-1',
1292
 
            b'': b'new-0', }
1293
 
        attribs[b'_removed_contents'] = [b'new-1']
1294
 
        contents = [(b'new-1', b'file',
1295
 
                     b'i 1\nz\n\nc 0 1 1 1\ni 1\nx\n\nc 0 3 3 1\n')]
 
3502
        attribs['_id_number'] = 2
 
3503
        attribs['_tree_path_ids'] = {
 
3504
            'file': 'new-1',
 
3505
            '': 'new-0',}
 
3506
        attribs['_removed_contents'] = ['new-1']
 
3507
        contents = [('new-1', 'file',
 
3508
                     'i 1\nz\n\nc 0 1 1 1\ni 1\nx\n\nc 0 3 3 1\n')]
1296
3509
        return self.make_records(attribs, contents)
1297
3510
 
1298
3511
    def test_serialize_modification(self):
1299
3512
        tt, LINES = self.make_modification_preview()
1300
 
        trans_id = tt.trans_id_file_id(b'file-id')
 
3513
        trans_id = tt.trans_id_file_id('file-id')
1301
3514
        tt.delete_contents(trans_id)
1302
3515
        tt.create_file(LINES, trans_id)
1303
3516
        self.assertSerializesTo(self.modification_records(), tt)
1305
3518
    def test_deserialize_modification(self):
1306
3519
        tt, LINES = self.make_modification_preview()
1307
3520
        tt.deserialize(iter(self.modification_records()))
1308
 
        self.assertFileEqual(b''.join(LINES), tt._limbo_name('new-1'))
 
3521
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
1309
3522
 
1310
3523
    def make_kind_change_preview(self):
1311
 
        LINES = b'a\nb\nc\nd\n'
 
3524
        LINES = 'a\nb\nc\nd\n'
1312
3525
        tree = self.make_branch_and_tree('tree')
1313
3526
        self.build_tree(['tree/foo/'])
1314
 
        tree.add('foo', b'foo-id')
1315
 
        return self.get_preview(tree), [LINES]
 
3527
        tree.add('foo', 'foo-id')
 
3528
        return self.get_preview(tree), LINES
1316
3529
 
1317
3530
    def kind_change_records(self):
1318
3531
        attribs = self.default_attribs()
1319
 
        attribs[b'_id_number'] = 2
1320
 
        attribs[b'_tree_path_ids'] = {
1321
 
            b'foo': b'new-1',
1322
 
            b'': b'new-0', }
1323
 
        attribs[b'_removed_contents'] = [b'new-1']
1324
 
        contents = [(b'new-1', b'file',
1325
 
                     b'i 4\na\nb\nc\nd\n\n')]
 
3532
        attribs['_id_number'] = 2
 
3533
        attribs['_tree_path_ids'] = {
 
3534
            'foo': 'new-1',
 
3535
            '': 'new-0',}
 
3536
        attribs['_removed_contents'] = ['new-1']
 
3537
        contents = [('new-1', 'file',
 
3538
                     'i 4\na\nb\nc\nd\n\n')]
1326
3539
        return self.make_records(attribs, contents)
1327
3540
 
1328
3541
    def test_serialize_kind_change(self):
1329
3542
        tt, LINES = self.make_kind_change_preview()
1330
 
        trans_id = tt.trans_id_file_id(b'foo-id')
 
3543
        trans_id = tt.trans_id_file_id('foo-id')
1331
3544
        tt.delete_contents(trans_id)
1332
3545
        tt.create_file(LINES, trans_id)
1333
3546
        self.assertSerializesTo(self.kind_change_records(), tt)
1335
3548
    def test_deserialize_kind_change(self):
1336
3549
        tt, LINES = self.make_kind_change_preview()
1337
3550
        tt.deserialize(iter(self.kind_change_records()))
1338
 
        self.assertFileEqual(b''.join(LINES), tt._limbo_name('new-1'))
 
3551
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
1339
3552
 
1340
3553
    def make_add_contents_preview(self):
1341
 
        LINES = b'a\nb\nc\nd\n'
 
3554
        LINES = 'a\nb\nc\nd\n'
1342
3555
        tree = self.make_branch_and_tree('tree')
1343
3556
        self.build_tree(['tree/foo'])
1344
3557
        tree.add('foo')
1347
3560
 
1348
3561
    def add_contents_records(self):
1349
3562
        attribs = self.default_attribs()
1350
 
        attribs[b'_id_number'] = 2
1351
 
        attribs[b'_tree_path_ids'] = {
1352
 
            b'foo': b'new-1',
1353
 
            b'': b'new-0', }
1354
 
        contents = [(b'new-1', b'file',
1355
 
                     b'i 4\na\nb\nc\nd\n\n')]
 
3563
        attribs['_id_number'] = 2
 
3564
        attribs['_tree_path_ids'] = {
 
3565
            'foo': 'new-1',
 
3566
            '': 'new-0',}
 
3567
        contents = [('new-1', 'file',
 
3568
                     'i 4\na\nb\nc\nd\n\n')]
1356
3569
        return self.make_records(attribs, contents)
1357
3570
 
1358
3571
    def test_serialize_add_contents(self):
1359
3572
        tt, LINES = self.make_add_contents_preview()
1360
3573
        trans_id = tt.trans_id_tree_path('foo')
1361
 
        tt.create_file([LINES], trans_id)
 
3574
        tt.create_file(LINES, trans_id)
1362
3575
        self.assertSerializesTo(self.add_contents_records(), tt)
1363
3576
 
1364
3577
    def test_deserialize_add_contents(self):
1367
3580
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
1368
3581
 
1369
3582
    def test_get_parents_lines(self):
1370
 
        LINES_ONE = b'aa\nbb\ncc\ndd\n'
 
3583
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
 
3584
        LINES_TWO = 'z\nbb\nx\ndd\n'
1371
3585
        tree = self.make_branch_and_tree('tree')
1372
3586
        self.build_tree_contents([('tree/file', LINES_ONE)])
1373
 
        tree.add('file', b'file-id')
 
3587
        tree.add('file', 'file-id')
1374
3588
        tt = self.get_preview(tree)
1375
3589
        trans_id = tt.trans_id_tree_path('file')
1376
 
        self.assertEqual(([b'aa\n', b'bb\n', b'cc\n', b'dd\n'],),
1377
 
                         tt._get_parents_lines(trans_id))
 
3590
        self.assertEqual((['aa\n', 'bb\n', 'cc\n', 'dd\n'],),
 
3591
            tt._get_parents_lines(trans_id))
1378
3592
 
1379
3593
    def test_get_parents_texts(self):
1380
 
        LINES_ONE = b'aa\nbb\ncc\ndd\n'
 
3594
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
 
3595
        LINES_TWO = 'z\nbb\nx\ndd\n'
1381
3596
        tree = self.make_branch_and_tree('tree')
1382
3597
        self.build_tree_contents([('tree/file', LINES_ONE)])
1383
 
        tree.add('file', b'file-id')
 
3598
        tree.add('file', 'file-id')
1384
3599
        tt = self.get_preview(tree)
1385
3600
        trans_id = tt.trans_id_tree_path('file')
1386
3601
        self.assertEqual((LINES_ONE,),
1387
 
                         tt._get_parents_texts(trans_id))
 
3602
            tt._get_parents_texts(trans_id))
1388
3603
 
1389
3604
 
1390
3605
class TestOrphan(tests.TestCaseWithTransport):
1391
3606
 
1392
3607
    def test_no_orphan_for_transform_preview(self):
1393
3608
        tree = self.make_branch_and_tree('tree')
1394
 
        tt = tree.preview_transform()
 
3609
        tt = transform.TransformPreview(tree)
1395
3610
        self.addCleanup(tt.finalize)
1396
3611
        self.assertRaises(NotImplementedError, tt.new_orphan, 'foo', 'bar')
1397
3612
 
1398
3613
    def _set_orphan_policy(self, wt, policy):
1399
 
        wt.branch.get_config_stack().set('transform.orphan_policy',
1400
 
                                         policy)
 
3614
        wt.branch.get_config_stack().set('bzr.transform.orphan_policy',
 
3615
                                               policy)
1401
3616
 
1402
3617
    def _prepare_orphan(self, wt):
1403
3618
        self.build_tree(['dir/', 'dir/file', 'dir/foo'])
1404
 
        wt.add(['dir', 'dir/file'], [b'dir-id', b'file-id'])
 
3619
        wt.add(['dir', 'dir/file'], ['dir-id', 'file-id'])
1405
3620
        wt.commit('add dir and file ignoring foo')
1406
 
        tt = wt.transform()
 
3621
        tt = transform.TreeTransform(wt)
1407
3622
        self.addCleanup(tt.finalize)
1408
3623
        # dir and bar are deleted
1409
3624
        dir_tid = tt.trans_id_tree_path('dir')
1424
3639
        self._set_orphan_policy(wt, 'move')
1425
3640
        tt, orphan_tid = self._prepare_orphan(wt)
1426
3641
        warnings = []
1427
 
 
1428
3642
        def warning(*args):
1429
3643
            warnings.append(args[0] % args[1:])
1430
3644
        self.overrideAttr(trace, 'warning', warning)
1431
3645
        remaining_conflicts = resolve_conflicts(tt)
1432
 
        self.assertEqual(['dir/foo has been orphaned in brz-orphans'],
1433
 
                         warnings)
 
3646
        self.assertEqual(['dir/foo has been orphaned in bzr-orphans'],
 
3647
                          warnings)
1434
3648
        # Yeah for resolved conflicts !
1435
3649
        self.assertLength(0, remaining_conflicts)
1436
3650
        # We have a new orphan
1437
3651
        self.assertEqual('foo.~1~', tt.final_name(orphan_tid))
1438
 
        self.assertEqual('brz-orphans',
1439
 
                         tt.final_name(tt.final_parent(orphan_tid)))
 
3652
        self.assertEqual('bzr-orphans',
 
3653
                          tt.final_name(tt.final_parent(orphan_tid)))
1440
3654
 
1441
3655
    def test_never_orphan(self):
1442
3656
        wt = self.make_branch_and_tree('.')
1467
3681
        self._set_orphan_policy(wt, 'donttouchmypreciouuus')
1468
3682
        tt, orphan_tid = self._prepare_orphan(wt)
1469
3683
        warnings = []
1470
 
 
1471
3684
        def warning(*args):
1472
3685
            warnings.append(args[0] % args[1:])
1473
3686
        self.overrideAttr(trace, 'warning', warning)
1487
3700
        self.wt = self.make_branch_and_tree('.')
1488
3701
        os.chdir('..')
1489
3702
 
1490
 
    def transform(self):
1491
 
        transform = self.wt.transform()
 
3703
    def get_transform(self):
 
3704
        transform = TreeTransform(self.wt)
1492
3705
        self.addCleanup(transform.finalize)
1493
3706
        return transform, transform.root
1494
3707
 
1495
3708
    def test_pre_commit_hooks(self):
1496
3709
        calls = []
1497
 
 
1498
3710
        def record_pre_transform(tree, tt):
1499
3711
            calls.append((tree, tt))
1500
 
        MutableTree.hooks.install_named_hook(
1501
 
            'pre_transform', record_pre_transform, "Pre transform")
1502
 
        transform, root = self.transform()
 
3712
        MutableTree.hooks.install_named_hook('pre_transform',
 
3713
            record_pre_transform, "Pre transform")
 
3714
        transform, root = self.get_transform()
1503
3715
        old_root_id = transform.tree_file_id(root)
1504
3716
        transform.apply()
1505
 
        self.assertEqual(old_root_id, self.wt.path2id(''))
 
3717
        self.assertEqual(old_root_id, self.wt.get_root_id())
1506
3718
        self.assertEqual([(self.wt, transform)], calls)
1507
3719
 
1508
3720
    def test_post_commit_hooks(self):
1509
3721
        calls = []
1510
 
 
1511
3722
        def record_post_transform(tree, tt):
1512
3723
            calls.append((tree, tt))
1513
 
        MutableTree.hooks.install_named_hook(
1514
 
            'post_transform', record_post_transform, "Post transform")
1515
 
        transform, root = self.transform()
 
3724
        MutableTree.hooks.install_named_hook('post_transform',
 
3725
            record_post_transform, "Post transform")
 
3726
        transform, root = self.get_transform()
1516
3727
        old_root_id = transform.tree_file_id(root)
1517
3728
        transform.apply()
1518
 
        self.assertEqual(old_root_id, self.wt.path2id(''))
 
3729
        self.assertEqual(old_root_id, self.wt.get_root_id())
1519
3730
        self.assertEqual([(self.wt, transform)], calls)
1520
 
 
1521
 
 
1522
 
class TestLinkTree(tests.TestCaseWithTransport):
1523
 
 
1524
 
    _test_needs_features = [HardlinkFeature]
1525
 
 
1526
 
    def setUp(self):
1527
 
        tests.TestCaseWithTransport.setUp(self)
1528
 
        self.parent_tree = self.make_branch_and_tree('parent')
1529
 
        self.parent_tree.lock_write()
1530
 
        self.addCleanup(self.parent_tree.unlock)
1531
 
        self.build_tree_contents([('parent/foo', b'bar')])
1532
 
        self.parent_tree.add('foo')
1533
 
        self.parent_tree.commit('added foo')
1534
 
        child_controldir = self.parent_tree.controldir.sprout('child')
1535
 
        self.child_tree = child_controldir.open_workingtree()
1536
 
 
1537
 
    def hardlinked(self):
1538
 
        parent_stat = os.lstat(self.parent_tree.abspath('foo'))
1539
 
        child_stat = os.lstat(self.child_tree.abspath('foo'))
1540
 
        return parent_stat.st_ino == child_stat.st_ino
1541
 
 
1542
 
    def test_link_fails_if_modified(self):
1543
 
        """If the file to be linked has modified text, don't link."""
1544
 
        self.build_tree_contents([('child/foo', b'baz')])
1545
 
        transform.link_tree(self.child_tree, self.parent_tree)
1546
 
        self.assertFalse(self.hardlinked())
1547
 
 
1548
 
    def test_link_fails_if_execute_bit_changed(self):
1549
 
        """If the file to be linked has modified execute bit, don't link."""
1550
 
        tt = self.child_tree.transform()
1551
 
        try:
1552
 
            trans_id = tt.trans_id_tree_path('foo')
1553
 
            tt.set_executability(True, trans_id)
1554
 
            tt.apply()
1555
 
        finally:
1556
 
            tt.finalize()
1557
 
        transform.link_tree(self.child_tree, self.parent_tree)
1558
 
        self.assertFalse(self.hardlinked())
1559
 
 
1560
 
    def test_link_succeeds_if_unmodified(self):
1561
 
        """If the file to be linked is unmodified, link"""
1562
 
        transform.link_tree(self.child_tree, self.parent_tree)
1563
 
        self.assertTrue(self.hardlinked())
1564
 
 
1565
 
 
1566
 
class ErrorTests(tests.TestCase):
1567
 
 
1568
 
    def test_transform_rename_failed(self):
1569
 
        e = TransformRenameFailed(u"from", u"to", "readonly file", 2)
1570
 
        self.assertEqual(
1571
 
            u"Failed to rename from to to: readonly file",
1572
 
            str(e))