/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: 2018-02-18 21:42:57 UTC
  • mto: This revision was merged to the branch mainline in revision 6859.
  • Revision ID: jelmer@jelmer.uk-20180218214257-jpevutp1wa30tz3v
Update TODO to reference Breezy, not Bazaar.

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('name').read())
 
166
        self.assertEqual(self.wt.path2id('name'), 'my_pretties')
 
167
        self.assertIs(self.wt.is_executable('name'), True)
 
168
        self.assertIs(self.wt.is_executable('name2'), 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('one', filtered=False)
 
253
        fo.close()
 
254
        fo, st2 = self.wt.get_file_with_stat('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('name'), 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('oz/dorothy/toto').read())
 
392
        self.assertIs(self.wt.is_executable('oz/dorothy/toto'), 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.assertFalse(self.wt.is_versioned('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.assertFalse(self.wt.is_versioned('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.assertFalse(self.wt.is_versioned('parent'))
 
954
        self.assertFalse(self.wt.is_versioned('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('set_on_creation'))
 
1070
        self.assertTrue(wt.is_executable('set_after_creation'))
 
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'))
 
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'))
 
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("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("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("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("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', file_id='foo-id')
 
1614
        bar_trans_id = tt.create_path('bar', tt.root)
 
1615
        create_from_tree(tt, bar_trans_id, tree1, 'bar', file_id='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', file_id='foo-id',
 
1629
                         bytes='qux')
 
1630
        tt.apply()
 
1631
        self.assertFileEqual('qux', 'tree2/foo')
 
1632
 
 
1633
    def test_create_from_tree_symlink(self):
 
1634
        self.requireFeature(SymlinkFeature)
 
1635
        tree1 = self.make_branch_and_tree('tree1')
 
1636
        os.symlink('bar', 'tree1/foo')
 
1637
        tree1.add('foo', 'foo-id')
 
1638
        tt = TreeTransform(self.make_branch_and_tree('tree2'))
 
1639
        foo_trans_id = tt.create_path('foo', tt.root)
 
1640
        create_from_tree(tt, foo_trans_id, tree1, 'foo', file_id='foo-id')
 
1641
        tt.apply()
 
1642
        self.assertEqual('bar', os.readlink('tree2/foo'))
 
1643
 
 
1644
 
89
1645
class TransformGroup(object):
90
1646
 
91
1647
    def __init__(self, dirname, root_id):
94
1650
        self.wt = ControlDir.create_standalone_workingtree(dirname)
95
1651
        self.wt.set_root_id(root_id)
96
1652
        self.b = self.wt.branch
97
 
        self.tt = self.wt.transform()
98
 
        self.root = self.tt.trans_id_tree_path('')
 
1653
        self.tt = TreeTransform(self.wt)
 
1654
        self.root = self.tt.trans_id_tree_file_id(self.wt.get_root_id())
99
1655
 
100
1656
 
101
1657
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)
 
1658
    template = '%s TREE\n%s%s\n%s%s MERGE-SOURCE\n'
 
1659
    return template % ('<' * 7, tree, '=' * 7, merge, '>' * 7)
 
1660
 
 
1661
 
 
1662
class TestInventoryAltered(tests.TestCaseWithTransport):
 
1663
 
 
1664
    def test_inventory_altered_unchanged(self):
 
1665
        tree = self.make_branch_and_tree('tree')
 
1666
        self.build_tree(['tree/foo'])
 
1667
        tree.add('foo', 'foo-id')
 
1668
        with TransformPreview(tree) as tt:
 
1669
            self.assertEqual([], tt._inventory_altered())
 
1670
 
 
1671
    def test_inventory_altered_changed_parent_id(self):
 
1672
        tree = self.make_branch_and_tree('tree')
 
1673
        self.build_tree(['tree/foo'])
 
1674
        tree.add('foo', 'foo-id')
 
1675
        with TransformPreview(tree) as tt:
 
1676
            tt.unversion_file(tt.root)
 
1677
            tt.version_file('new-id', tt.root)
 
1678
            foo_trans_id = tt.trans_id_tree_file_id('foo-id')
 
1679
            foo_tuple = ('foo', foo_trans_id)
 
1680
            root_tuple = ('', tt.root)
 
1681
            self.assertEqual([root_tuple, foo_tuple], tt._inventory_altered())
 
1682
 
 
1683
    def test_inventory_altered_noop_changed_parent_id(self):
 
1684
        tree = self.make_branch_and_tree('tree')
 
1685
        self.build_tree(['tree/foo'])
 
1686
        tree.add('foo', 'foo-id')
 
1687
        with TransformPreview(tree) as tt:
 
1688
            tt.unversion_file(tt.root)
 
1689
            tt.version_file(tree.get_root_id(), tt.root)
 
1690
            foo_trans_id = tt.trans_id_tree_file_id('foo-id')
 
1691
            self.assertEqual([], tt._inventory_altered())
104
1692
 
105
1693
 
106
1694
class TestTransformMerge(TestCaseInTempDir):
108
1696
    def test_text_merge(self):
109
1697
        root_id = generate_ids.gen_root_id()
110
1698
        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')
 
1699
        base.tt.new_file('a', base.root, 'a\nb\nc\nd\be\n', 'a')
 
1700
        base.tt.new_file('b', base.root, 'b1', 'b')
 
1701
        base.tt.new_file('c', base.root, 'c', 'c')
 
1702
        base.tt.new_file('d', base.root, 'd', 'd')
 
1703
        base.tt.new_file('e', base.root, 'e', 'e')
 
1704
        base.tt.new_file('f', base.root, 'f', 'f')
 
1705
        base.tt.new_directory('g', base.root, 'g')
 
1706
        base.tt.new_directory('h', base.root, 'h')
119
1707
        base.tt.apply()
120
1708
        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')
 
1709
        other.tt.new_file('a', other.root, 'y\nb\nc\nd\be\n', 'a')
 
1710
        other.tt.new_file('b', other.root, 'b2', 'b')
 
1711
        other.tt.new_file('c', other.root, 'c2', 'c')
 
1712
        other.tt.new_file('d', other.root, 'd', 'd')
 
1713
        other.tt.new_file('e', other.root, 'e2', 'e')
 
1714
        other.tt.new_file('f', other.root, 'f', 'f')
 
1715
        other.tt.new_file('g', other.root, 'g', 'g')
 
1716
        other.tt.new_file('h', other.root, 'h\ni\nj\nk\n', 'h')
 
1717
        other.tt.new_file('i', other.root, 'h\ni\nj\nk\n', 'i')
130
1718
        other.tt.apply()
131
1719
        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')
 
1720
        this.tt.new_file('a', this.root, 'a\nb\nc\nd\bz\n', 'a')
 
1721
        this.tt.new_file('b', this.root, 'b', 'b')
 
1722
        this.tt.new_file('c', this.root, 'c', 'c')
 
1723
        this.tt.new_file('d', this.root, 'd2', 'd')
 
1724
        this.tt.new_file('e', this.root, 'e2', 'e')
 
1725
        this.tt.new_file('f', this.root, 'f', 'f')
 
1726
        this.tt.new_file('g', this.root, 'g', 'g')
 
1727
        this.tt.new_file('h', this.root, '1\n2\n3\n4\n', 'h')
 
1728
        this.tt.new_file('i', this.root, '1\n2\n3\n4\n', 'i')
141
1729
        this.tt.apply()
142
1730
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
143
1731
 
144
1732
        # 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')
 
1733
        self.assertEqual(this.wt.get_file(this.wt.id2path('a')).read(), 'y\nb\nc\nd\bz\n')
147
1734
        # 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'))
 
1735
        self.assertEqual(this.wt.get_file(this.wt.id2path('b')).read(),
 
1736
                         conflict_text('b', 'b2'))
150
1737
        # OTHER wins
151
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'c')).read(), b'c2')
 
1738
        self.assertEqual(this.wt.get_file(this.wt.id2path('c')).read(), 'c2')
152
1739
        # THIS wins
153
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'd')).read(), b'd2')
 
1740
        self.assertEqual(this.wt.get_file(this.wt.id2path('d')).read(), 'd2')
154
1741
        # Ambigious clean merge
155
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'e')).read(), b'e2')
 
1742
        self.assertEqual(this.wt.get_file(this.wt.id2path('e')).read(), 'e2')
156
1743
        # No change
157
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'f')).read(), b'f')
 
1744
        self.assertEqual(this.wt.get_file(this.wt.id2path('f')).read(), 'f')
158
1745
        # Correct correct results when THIS == OTHER
159
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'g')).read(), b'g')
 
1746
        self.assertEqual(this.wt.get_file(this.wt.id2path('g')).read(), 'g')
160
1747
        # 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'))
 
1748
        self.assertEqual(this.wt.get_file(this.wt.id2path('h')).read(),
 
1749
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
163
1750
        self.assertEqual(this.wt.get_file('h.THIS').read(),
164
 
                         b'1\n2\n3\n4\n')
 
1751
                         '1\n2\n3\n4\n')
165
1752
        self.assertEqual(this.wt.get_file('h.OTHER').read(),
166
 
                         b'h\ni\nj\nk\n')
 
1753
                         'h\ni\nj\nk\n')
167
1754
        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'))
 
1755
        self.assertEqual(this.wt.get_file(this.wt.id2path('i')).read(),
 
1756
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
170
1757
        self.assertEqual(this.wt.get_file('i.THIS').read(),
171
 
                         b'1\n2\n3\n4\n')
 
1758
                         '1\n2\n3\n4\n')
172
1759
        self.assertEqual(this.wt.get_file('i.OTHER').read(),
173
 
                         b'h\ni\nj\nk\n')
 
1760
                         'h\ni\nj\nk\n')
174
1761
        self.assertEqual(os.path.exists(this.wt.abspath('i.BASE')), False)
175
1762
        modified = ['a', 'b', 'c', 'h', 'i']
176
1763
        merge_modified = this.wt.merge_modified()
177
1764
        self.assertSubset(merge_modified, modified)
178
1765
        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')
 
1766
        with file(this.wt.id2abspath('a'), 'wb') as f: f.write('booga')
181
1767
        modified.pop(0)
182
1768
        merge_modified = this.wt.merge_modified()
183
1769
        self.assertSubset(merge_modified, modified)
192
1778
        this = TransformGroup("THIS", root_id)
193
1779
        other = TransformGroup("OTHER", root_id)
194
1780
        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')
 
1781
            tg.tt.new_directory('a', tg.root, 'a')
 
1782
            tg.tt.new_symlink('b', tg.root, 'b', 'b')
 
1783
            tg.tt.new_file('c', tg.root, 'c', 'c')
 
1784
            tg.tt.new_symlink('d', tg.root, tg.name, 'd')
199
1785
        targets = ((base, 'base-e', 'base-f', None, None),
200
1786
                   (this, 'other-e', 'this-f', 'other-g', 'this-h'),
201
1787
                   (other, 'other-e', None, 'other-g', 'other-h'))
203
1789
            for link, target in (('e', e_target), ('f', f_target),
204
1790
                                 ('g', g_target), ('h', h_target)):
205
1791
                if target is not None:
206
 
                    tg.tt.new_symlink(link, tg.root, target,
207
 
                                      link.encode('ascii'))
 
1792
                    tg.tt.new_symlink(link, tg.root, target, link)
208
1793
 
209
1794
        for tg in this, base, other:
210
1795
            tg.tt.apply()
213
1798
        self.assertIs(os.path.islink(this.wt.abspath('b')), True)
214
1799
        self.assertIs(os.path.isfile(this.wt.abspath('c')), True)
215
1800
        for suffix in ('THIS', 'BASE', 'OTHER'):
216
 
            self.assertEqual(os.readlink(
217
 
                this.wt.abspath('d.' + suffix)), suffix)
 
1801
            self.assertEqual(os.readlink(this.wt.abspath('d.'+suffix)), suffix)
218
1802
        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')
 
1803
        self.assertEqual(this.wt.id2path('d'), 'd.OTHER')
 
1804
        self.assertEqual(this.wt.id2path('f'), 'f.THIS')
221
1805
        self.assertEqual(os.readlink(this.wt.abspath('e')), 'other-e')
222
1806
        self.assertIs(os.path.lexists(this.wt.abspath('e.THIS')), False)
223
1807
        self.assertIs(os.path.lexists(this.wt.abspath('e.OTHER')), False)
234
1818
        base = TransformGroup("BASE", root_id)
235
1819
        this = TransformGroup("THIS", root_id)
236
1820
        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')
 
1821
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
 
1822
                                   for t in [base, this, other]]
 
1823
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
 
1824
                                   for t in [base, this, other]]
 
1825
        base.tt.new_directory('c', base_a, 'c')
 
1826
        this.tt.new_directory('c1', this_a, 'c')
 
1827
        other.tt.new_directory('c', other_b, 'c')
 
1828
 
 
1829
        base.tt.new_directory('d', base_a, 'd')
 
1830
        this.tt.new_directory('d1', this_b, 'd')
 
1831
        other.tt.new_directory('d', other_a, 'd')
 
1832
 
 
1833
        base.tt.new_directory('e', base_a, 'e')
 
1834
        this.tt.new_directory('e', this_a, 'e')
 
1835
        other.tt.new_directory('e1', other_b, 'e')
 
1836
 
 
1837
        base.tt.new_directory('f', base_a, 'f')
 
1838
        this.tt.new_directory('f1', this_b, 'f')
 
1839
        other.tt.new_directory('f1', other_b, 'f')
256
1840
 
257
1841
        for tg in [this, base, other]:
258
1842
            tg.tt.apply()
259
1843
        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'))
 
1844
        self.assertEqual(this.wt.id2path('c'), pathjoin('b/c1'))
 
1845
        self.assertEqual(this.wt.id2path('d'), pathjoin('b/d1'))
 
1846
        self.assertEqual(this.wt.id2path('e'), pathjoin('b/e1'))
 
1847
        self.assertEqual(this.wt.id2path('f'), pathjoin('b/f1'))
264
1848
 
265
1849
    def test_filename_merge_conflicts(self):
266
1850
        root_id = generate_ids.gen_root_id()
267
1851
        base = TransformGroup("BASE", root_id)
268
1852
        this = TransformGroup("THIS", root_id)
269
1853
        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')
 
1854
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
 
1855
                                   for t in [base, this, other]]
 
1856
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
 
1857
                                   for t in [base, this, other]]
 
1858
 
 
1859
        base.tt.new_file('g', base_a, 'g', 'g')
 
1860
        other.tt.new_file('g1', other_b, 'g1', 'g')
 
1861
 
 
1862
        base.tt.new_file('h', base_a, 'h', 'h')
 
1863
        this.tt.new_file('h1', this_b, 'h1', 'h')
 
1864
 
 
1865
        base.tt.new_file('i', base.root, 'i', 'i')
 
1866
        other.tt.new_directory('i1', this_b, 'i')
283
1867
 
284
1868
        for tg in [this, base, other]:
285
1869
            tg.tt.apply()
286
1870
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
287
1871
 
288
 
        self.assertEqual(this.wt.id2path(b'g'), pathjoin('b/g1.OTHER'))
 
1872
        self.assertEqual(this.wt.id2path('g'), pathjoin('b/g1.OTHER'))
289
1873
        self.assertIs(os.path.lexists(this.wt.abspath('b/g1.BASE')), True)
290
1874
        self.assertIs(os.path.lexists(this.wt.abspath('b/g1.THIS')), False)
291
 
        self.assertEqual(this.wt.id2path(b'h'), pathjoin('b/h1.THIS'))
 
1875
        self.assertEqual(this.wt.id2path('h'), pathjoin('b/h1.THIS'))
292
1876
        self.assertIs(os.path.lexists(this.wt.abspath('b/h1.BASE')), True)
293
1877
        self.assertIs(os.path.lexists(this.wt.abspath('b/h1.OTHER')), False)
294
 
        self.assertEqual(this.wt.id2path(b'i'), pathjoin('b/i1.OTHER'))
 
1878
        self.assertEqual(this.wt.id2path('i'), pathjoin('b/i1.OTHER'))
295
1879
 
296
1880
 
297
1881
class TestBuildTree(tests.TestCaseWithTransport):
301
1885
        os.mkdir('a')
302
1886
        a = ControlDir.create_standalone_workingtree('a')
303
1887
        os.mkdir('a/foo')
304
 
        with open('a/foo/bar', 'wb') as f:
305
 
            f.write(b'contents')
 
1888
        with file('a/foo/bar', 'wb') as f: f.write('contents')
306
1889
        os.symlink('a/foo/bar', 'a/foo/baz')
307
1890
        a.add(['foo', 'foo/bar', 'foo/baz'])
308
1891
        a.commit('initial commit')
312
1895
        self.addCleanup(basis.unlock)
313
1896
        build_tree(basis, b)
314
1897
        self.assertIs(os.path.isdir('b/foo'), True)
315
 
        with open('b/foo/bar', 'rb') as f:
316
 
            self.assertEqual(f.read(), b"contents")
 
1898
        self.assertEqual(file('b/foo/bar', 'rb').read(), "contents")
317
1899
        self.assertEqual(os.readlink('b/foo/baz'), 'a/foo/bar')
318
1900
 
319
1901
    def test_build_with_references(self):
320
1902
        tree = self.make_branch_and_tree('source',
321
 
                                         format='development-subtree')
 
1903
            format='development-subtree')
322
1904
        subtree = self.make_branch_and_tree('source/subtree',
323
 
                                            format='development-subtree')
 
1905
            format='development-subtree')
324
1906
        tree.add_reference(subtree)
325
1907
        tree.commit('a revision')
326
1908
        tree.branch.create_checkout('target')
332
1914
        source = self.make_branch_and_tree('source')
333
1915
        target = self.make_branch_and_tree('target')
334
1916
        self.build_tree(['source/file', 'target/file'])
335
 
        source.add('file', b'new-file')
 
1917
        source.add('file', 'new-file')
336
1918
        source.commit('added file')
337
1919
        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())
 
1920
        self.assertEqual([DuplicateEntry('Moved existing file to',
 
1921
                          'file.moved', 'file', None, 'new-file')],
 
1922
                         target.conflicts())
342
1923
        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())
 
1924
        target_file = file('target2/file', 'wb')
 
1925
        try:
 
1926
            source_file = file('source/file', 'rb')
 
1927
            try:
 
1928
                target_file.write(source_file.read())
 
1929
            finally:
 
1930
                source_file.close()
 
1931
        finally:
 
1932
            target_file.close()
346
1933
        build_tree(source.basis_tree(), target2)
347
1934
        self.assertEqual([], target2.conflicts())
348
1935
 
351
1938
        self.requireFeature(SymlinkFeature)
352
1939
        source = self.make_branch_and_tree('source')
353
1940
        os.symlink('foo', 'source/symlink')
354
 
        source.add('symlink', b'new-symlink')
 
1941
        source.add('symlink', 'new-symlink')
355
1942
        source.commit('added file')
356
1943
        target = self.make_branch_and_tree('target')
357
1944
        os.symlink('bar', 'target/symlink')
358
1945
        build_tree(source.basis_tree(), target)
359
 
        self.assertEqual(
360
 
            [DuplicateEntry('Moved existing file to', 'symlink.moved',
361
 
                            'symlink', None, 'new-symlink')],
 
1946
        self.assertEqual([DuplicateEntry('Moved existing file to',
 
1947
            'symlink.moved', 'symlink', None, 'new-symlink')],
362
1948
            target.conflicts())
363
1949
        target = self.make_branch_and_tree('target2')
364
1950
        os.symlink('foo', 'target2/symlink')
370
1956
        source = self.make_branch_and_tree('source')
371
1957
        target = self.make_branch_and_tree('target')
372
1958
        self.build_tree(['source/dir1/', 'source/dir1/file', 'target/dir1/'])
373
 
        source.add(['dir1', 'dir1/file'], [b'new-dir1', b'new-file'])
 
1959
        source.add(['dir1', 'dir1/file'], ['new-dir1', 'new-file'])
374
1960
        source.commit('added file')
375
1961
        build_tree(source.basis_tree(), target)
376
1962
        self.assertEqual([], target.conflicts())
392
1978
        self.assertPathDoesNotExist('target3/dir1/file')
393
1979
        self.assertPathExists('target3/dir1/file2')
394
1980
        self.assertPathExists('target3/dir1.diverted/file')
395
 
        self.assertEqual(
396
 
            [DuplicateEntry('Diverted to', 'dir1.diverted',
397
 
                            'dir1', 'new-dir1', None)],
 
1981
        self.assertEqual([DuplicateEntry('Diverted to',
 
1982
            'dir1.diverted', 'dir1', 'new-dir1', None)],
398
1983
            target.conflicts())
399
1984
 
400
1985
        target = self.make_branch_and_tree('target4')
404
1989
        self.assertPathExists('target4/dir1/file')
405
1990
        self.assertEqual('directory', file_kind('target4/dir1/file'))
406
1991
        self.assertPathExists('target4/dir1/file.diverted')
407
 
        self.assertEqual(
408
 
            [DuplicateEntry('Diverted to', 'dir1/file.diverted',
409
 
                            'dir1/file', 'new-file', None)],
 
1992
        self.assertEqual([DuplicateEntry('Diverted to',
 
1993
            'dir1/file.diverted', 'dir1/file', 'new-file', None)],
410
1994
            target.conflicts())
411
1995
 
412
1996
    def test_mixed_conflict_handling(self):
414
1998
        source = self.make_branch_and_tree('source')
415
1999
        target = self.make_branch_and_tree('target')
416
2000
        self.build_tree(['source/name', 'target/name/'])
417
 
        source.add('name', b'new-name')
 
2001
        source.add('name', 'new-name')
418
2002
        source.commit('added file')
419
2003
        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())
 
2004
        self.assertEqual([DuplicateEntry('Moved existing file to',
 
2005
            'name.moved', 'name', None, 'new-name')], target.conflicts())
424
2006
 
425
2007
    def test_raises_in_populated(self):
426
2008
        source = self.make_branch_and_tree('source')
431
2013
        self.build_tree(['target/name'])
432
2014
        target.add('name')
433
2015
        self.assertRaises(errors.WorkingTreeAlreadyPopulated,
434
 
                          build_tree, source.basis_tree(), target)
 
2016
            build_tree, source.basis_tree(), target)
435
2017
 
436
2018
    def test_build_tree_rename_count(self):
437
2019
        source = self.make_branch_and_tree('source')
453
2035
    def create_ab_tree(self):
454
2036
        """Create a committed test tree with two files"""
455
2037
        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'])
 
2038
        self.build_tree_contents([('source/file1', 'A')])
 
2039
        self.build_tree_contents([('source/file2', 'B')])
 
2040
        source.add(['file1', 'file2'], ['file1-id', 'file2-id'])
459
2041
        source.commit('commit files')
460
2042
        source.lock_write()
461
2043
        self.addCleanup(source.unlock)
463
2045
 
464
2046
    def test_build_tree_accelerator_tree(self):
465
2047
        source = self.create_ab_tree()
466
 
        self.build_tree_contents([('source/file2', b'C')])
 
2048
        self.build_tree_contents([('source/file2', 'C')])
467
2049
        calls = []
468
2050
        real_source_get_file = source.get_file
469
 
 
470
 
        def get_file(path):
471
 
            calls.append(path)
472
 
            return real_source_get_file(path)
 
2051
        def get_file(path, file_id=None):
 
2052
            calls.append(file_id)
 
2053
            return real_source_get_file(path, file_id)
473
2054
        source.get_file = get_file
474
2055
        target = self.make_branch_and_tree('target')
475
2056
        revision_tree = source.basis_tree()
476
2057
        revision_tree.lock_read()
477
2058
        self.addCleanup(revision_tree.unlock)
478
2059
        build_tree(revision_tree, target, source)
479
 
        self.assertEqual(['file1'], calls)
 
2060
        self.assertEqual(['file1-id'], calls)
480
2061
        target.lock_read()
481
2062
        self.addCleanup(target.unlock)
482
2063
        self.assertEqual([], list(target.iter_changes(revision_tree)))
483
2064
 
484
2065
    def test_build_tree_accelerator_tree_observes_sha1(self):
485
2066
        source = self.create_ab_tree()
486
 
        sha1 = osutils.sha_string(b'A')
 
2067
        sha1 = osutils.sha_string('A')
487
2068
        target = self.make_branch_and_tree('target')
488
2069
        target.lock_write()
489
2070
        self.addCleanup(target.unlock)
490
2071
        state = target.current_dirstate()
491
2072
        state._cutoff_time = time.time() + 60
492
2073
        build_tree(source.basis_tree(), target, source)
493
 
        entry = state._get_entry(0, path_utf8=b'file1')
 
2074
        entry = state._get_entry(0, path_utf8='file1')
494
2075
        self.assertEqual(sha1, entry[1][0][1])
495
2076
 
496
2077
    def test_build_tree_accelerator_tree_missing_file(self):
509
2090
    def test_build_tree_accelerator_wrong_kind(self):
510
2091
        self.requireFeature(SymlinkFeature)
511
2092
        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'])
 
2093
        self.build_tree_contents([('source/file1', '')])
 
2094
        self.build_tree_contents([('source/file2', '')])
 
2095
        source.add(['file1', 'file2'], ['file1-id', 'file2-id'])
515
2096
        source.commit('commit files')
516
2097
        os.unlink('source/file2')
517
 
        self.build_tree_contents([('source/file2/', b'C')])
 
2098
        self.build_tree_contents([('source/file2/', 'C')])
518
2099
        os.unlink('source/file1')
519
2100
        os.symlink('file2', 'source/file1')
520
2101
        calls = []
521
2102
        real_source_get_file = source.get_file
522
 
 
523
 
        def get_file(path):
524
 
            calls.append(path)
525
 
            return real_source_get_file(path)
 
2103
        def get_file(path, file_id=None):
 
2104
            calls.append(file_id)
 
2105
            return real_source_get_file(path, file_id)
526
2106
        source.get_file = get_file
527
2107
        target = self.make_branch_and_tree('target')
528
2108
        revision_tree = source.basis_tree()
561
2141
 
562
2142
    def test_build_tree_accelerator_tree_moved(self):
563
2143
        source = self.make_branch_and_tree('source')
564
 
        self.build_tree_contents([('source/file1', b'A')])
565
 
        source.add(['file1'], [b'file1-id'])
 
2144
        self.build_tree_contents([('source/file1', 'A')])
 
2145
        source.add(['file1'], ['file1-id'])
566
2146
        source.commit('commit files')
567
2147
        source.rename_one('file1', 'file2')
568
2148
        source.lock_read()
579
2159
    def test_build_tree_hardlinks_preserve_execute(self):
580
2160
        self.requireFeature(HardlinkFeature)
581
2161
        source = self.create_ab_tree()
582
 
        tt = source.transform()
583
 
        trans_id = tt.trans_id_tree_path('file1')
 
2162
        tt = TreeTransform(source)
 
2163
        trans_id = tt.trans_id_tree_file_id('file1-id')
584
2164
        tt.set_executability(True, trans_id)
585
2165
        tt.apply()
586
2166
        self.assertTrue(source.is_executable('file1'))
600
2180
        # below, but that looks a bit... hard to read even if it's exactly
601
2181
        # the same thing.
602
2182
        original_registry = filters._reset_registry()
603
 
 
604
2183
        def restore_registry():
605
2184
            filters._reset_registry(original_registry)
606
2185
        self.addCleanup(restore_registry)
607
 
 
608
2186
        def rot13(chunks, context=None):
609
 
            return [
610
 
                codecs.encode(chunk.decode('ascii'), 'rot13').encode('ascii')
611
 
                for chunk in chunks]
 
2187
            return [''.join(chunks).encode('rot13')]
612
2188
        rot13filter = filters.ContentFilter(rot13, rot13)
613
2189
        filters.filter_stacks_registry.register(
614
2190
            'rot13', {'yes': [rot13filter]}.get)
615
2191
        os.mkdir(self.test_home_dir + '/.bazaar')
616
2192
        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
 
 
 
2193
        f = open(rules_filename, 'wb')
 
2194
        f.write('[name %s]\nrot13=yes\n' % (pattern,))
 
2195
        f.close()
620
2196
        def uninstall_rules():
621
2197
            os.remove(rules_filename)
622
2198
            rules.reset_rules()
629
2205
        if it can).
630
2206
        """
631
2207
        self.requireFeature(HardlinkFeature)
632
 
        self.install_rot13_content_filter(b'file1')
 
2208
        self.install_rot13_content_filter('file1')
633
2209
        source = self.create_ab_tree()
634
2210
        target = self.make_branch_and_tree('target')
635
2211
        revision_tree = source.basis_tree()
648
2224
 
649
2225
    def test_case_insensitive_build_tree_inventory(self):
650
2226
        if (features.CaseInsensitiveFilesystemFeature.available()
651
 
                or features.CaseInsCasePresFilenameFeature.available()):
 
2227
            or features.CaseInsCasePresFilenameFeature.available()):
652
2228
            raise tests.UnavailableFeature('Fully case sensitive filesystem')
653
2229
        source = self.make_branch_and_tree('source')
654
2230
        self.build_tree(['source/file', 'source/FILE'])
655
 
        source.add(['file', 'FILE'], [b'lower-id', b'upper-id'])
 
2231
        source.add(['file', 'FILE'], ['lower-id', 'upper-id'])
656
2232
        source.commit('added files')
657
2233
        # Don't try this at home, kids!
658
2234
        # Force the tree to report that it is case insensitive
659
2235
        target = self.make_branch_and_tree('target')
660
2236
        target.case_sensitive = False
661
2237
        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'))
 
2238
        self.assertEqual('file.moved', target.id2path('lower-id'))
 
2239
        self.assertEqual('FILE', target.id2path('upper-id'))
664
2240
 
665
2241
    def test_build_tree_observes_sha(self):
666
2242
        source = self.make_branch_and_tree('source')
667
2243
        self.build_tree(['source/file1', 'source/dir/', 'source/dir/file2'])
668
2244
        source.add(['file1', 'dir', 'dir/file2'],
669
 
                   [b'file1-id', b'dir-id', b'file2-id'])
 
2245
                   ['file1-id', 'dir-id', 'file2-id'])
670
2246
        source.commit('new files')
671
2247
        target = self.make_branch_and_tree('target')
672
2248
        target.lock_write()
681
2257
        # entry[1] is the state information, entry[1][0] is the state of the
682
2258
        # working tree, entry[1][0][1] is the sha value for the current working
683
2259
        # tree
684
 
        entry1 = state._get_entry(0, path_utf8=b'file1')
 
2260
        entry1 = state._get_entry(0, path_utf8='file1')
685
2261
        self.assertEqual(entry1_sha, entry1[1][0][1])
686
2262
        # The 'size' field must also be set.
687
2263
        self.assertEqual(25, entry1[1][0][2])
688
2264
        entry1_state = entry1[1][0]
689
 
        entry2 = state._get_entry(0, path_utf8=b'dir/file2')
 
2265
        entry2 = state._get_entry(0, path_utf8='dir/file2')
690
2266
        self.assertEqual(entry2_sha, entry2[1][0][1])
691
2267
        self.assertEqual(29, entry2[1][0][2])
692
2268
        entry2_state = entry2[1][0]
693
2269
        # Now, make sure that we don't have to re-read the content. The
694
2270
        # 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'))
 
2271
        self.assertEqual(entry1_sha, target.get_file_sha1('file1', 'file1-id'))
 
2272
        self.assertEqual(entry2_sha,
 
2273
                         target.get_file_sha1('dir/file2', 'file2-id'))
697
2274
        self.assertEqual(entry1_state, entry1[1][0])
698
2275
        self.assertEqual(entry2_state, entry2[1][0])
699
2276
 
709
2286
 
710
2287
    def get_branch_and_transform(self):
711
2288
        branch = self.get_branch()
712
 
        tt = branch.basis_tree().preview_transform()
 
2289
        tt = TransformPreview(branch.basis_tree())
713
2290
        self.addCleanup(tt.finalize)
714
2291
        return branch, tt
715
2292
 
717
2294
        branch = self.get_branch()
718
2295
        basis = branch.repository.revision_tree(
719
2296
            _mod_revision.NULL_REVISION)
720
 
        tt = basis.preview_transform()
 
2297
        tt = TransformPreview(basis)
721
2298
        self.addCleanup(tt.finalize)
722
2299
        e = self.assertRaises(ValueError, tt.commit, branch, '')
723
2300
        self.assertEqual('TreeTransform not based on branch basis: null:',
732
2309
 
733
2310
    def test_merge_parents(self):
734
2311
        branch, tt = self.get_branch_and_transform()
735
 
        tt.commit(branch, 'my message', [b'rev1b', b'rev1c'])
736
 
        self.assertEqual([b'rev1b', b'rev1c'],
 
2312
        rev = tt.commit(branch, 'my message', ['rev1b', 'rev1c'])
 
2313
        self.assertEqual(['rev1b', 'rev1c'],
737
2314
                         branch.basis_tree().get_parent_ids()[1:])
738
2315
 
739
2316
    def test_first_commit(self):
740
2317
        branch = self.make_branch('branch')
741
2318
        branch.lock_write()
742
2319
        self.addCleanup(branch.unlock)
743
 
        tt = branch.basis_tree().preview_transform()
 
2320
        tt = TransformPreview(branch.basis_tree())
744
2321
        self.addCleanup(tt.finalize)
745
 
        tt.new_directory('', ROOT_PARENT, b'TREE_ROOT')
746
 
        tt.commit(branch, 'my message')
 
2322
        tt.new_directory('', ROOT_PARENT, 'TREE_ROOT')
 
2323
        rev = tt.commit(branch, 'my message')
747
2324
        self.assertEqual([], branch.basis_tree().get_parent_ids())
748
2325
        self.assertNotEqual(_mod_revision.NULL_REVISION,
749
2326
                            branch.last_revision())
752
2329
        branch = self.make_branch('branch')
753
2330
        branch.lock_write()
754
2331
        self.addCleanup(branch.unlock)
755
 
        tt = branch.basis_tree().preview_transform()
 
2332
        tt = TransformPreview(branch.basis_tree())
756
2333
        self.addCleanup(tt.finalize)
757
2334
        e = self.assertRaises(ValueError, tt.commit, branch,
758
 
                              'my message', [b'rev1b-id'])
 
2335
                          'my message', ['rev1b-id'])
759
2336
        self.assertEqual('Cannot supply merge parents for first commit.',
760
2337
                         str(e))
761
2338
        self.assertEqual(_mod_revision.NULL_REVISION, branch.last_revision())
762
2339
 
763
2340
    def test_add_files(self):
764
2341
        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')
 
2342
        tt.new_file('file', tt.root, 'contents', 'file-id')
 
2343
        trans_id = tt.new_directory('dir', tt.root, 'dir-id')
767
2344
        if SymlinkFeature.available():
768
 
            tt.new_symlink('symlink', trans_id, 'target', b'symlink-id')
769
 
        tt.commit(branch, 'message')
 
2345
            tt.new_symlink('symlink', trans_id, 'target', 'symlink-id')
 
2346
        rev = tt.commit(branch, 'message')
770
2347
        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'))
 
2348
        self.assertEqual('file', tree.id2path('file-id'))
 
2349
        self.assertEqual('contents', tree.get_file_text('file', 'file-id'))
 
2350
        self.assertEqual('dir', tree.id2path('dir-id'))
774
2351
        if SymlinkFeature.available():
775
 
            self.assertEqual('dir/symlink', tree.id2path(b'symlink-id'))
 
2352
            self.assertEqual('dir/symlink', tree.id2path('symlink-id'))
776
2353
            self.assertEqual('target', tree.get_symlink_target('dir/symlink'))
777
2354
 
778
2355
    def test_add_unversioned(self):
779
2356
        branch, tt = self.get_branch_and_transform()
780
 
        tt.new_file('file', tt.root, [b'contents'])
 
2357
        tt.new_file('file', tt.root, 'contents')
781
2358
        self.assertRaises(errors.StrictCommitFailed, tt.commit, branch,
782
2359
                          'message', strict=True)
783
2360
 
784
2361
    def test_modify_strict(self):
785
2362
        branch, tt = self.get_branch_and_transform()
786
 
        tt.new_file('file', tt.root, [b'contents'], b'file-id')
 
2363
        tt.new_file('file', tt.root, 'contents', 'file-id')
787
2364
        tt.commit(branch, 'message', strict=True)
788
 
        tt = branch.basis_tree().preview_transform()
 
2365
        tt = TransformPreview(branch.basis_tree())
789
2366
        self.addCleanup(tt.finalize)
790
 
        trans_id = tt.trans_id_file_id(b'file-id')
 
2367
        trans_id = tt.trans_id_file_id('file-id')
791
2368
        tt.delete_contents(trans_id)
792
 
        tt.create_file([b'contents'], trans_id)
 
2369
        tt.create_file('contents', trans_id)
793
2370
        tt.commit(branch, 'message', strict=True)
794
2371
 
795
2372
    def test_commit_malformed(self):
798
2375
        In this case, we are adding a file without adding its parent.
799
2376
        """
800
2377
        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,
 
2378
        parent_id = tt.trans_id_file_id('parent-id')
 
2379
        tt.new_file('file', parent_id, 'contents', 'file-id')
 
2380
        self.assertRaises(errors.MalformedTransform, tt.commit, branch,
804
2381
                          'message')
805
2382
 
806
2383
    def test_commit_rich_revision_data(self):
807
2384
        branch, tt = self.get_branch_and_transform()
808
2385
        rev_id = tt.commit(branch, 'message', timestamp=1, timezone=43201,
809
2386
                           committer='me <me@example.com>',
810
 
                           revprops={u'foo': 'bar'}, revision_id=b'revid-1',
 
2387
                           revprops={'foo': 'bar'}, revision_id='revid-1',
811
2388
                           authors=['Author1 <author1@example.com>',
812
 
                                    'Author2 <author2@example.com>',
813
 
                                    ])
814
 
        self.assertEqual(b'revid-1', rev_id)
 
2389
                              'Author2 <author2@example.com>',
 
2390
                               ])
 
2391
        self.assertEqual('revid-1', rev_id)
815
2392
        revision = branch.repository.get_revision(rev_id)
816
2393
        self.assertEqual(1, revision.timestamp)
817
2394
        self.assertEqual(43201, revision.timezone)
880
2457
        mover.rename('c/e', 'c/d')
881
2458
        try:
882
2459
            mover.rename('a', 'c')
883
 
        except errors.FileExists:
 
2460
        except errors.FileExists as e:
884
2461
            mover.rollback()
885
2462
        self.assertPathExists('a')
886
2463
        self.assertPathExists('c/d')
901
2478
 
902
2479
        def rename(self, source, target):
903
2480
            if (self.bad_source is not None and
904
 
                    source.endswith(self.bad_source)):
 
2481
                source.endswith(self.bad_source)):
905
2482
                raise Bogus
906
2483
            elif (self.bad_target is not None and
907
 
                  target.endswith(self.bad_target)):
 
2484
                target.endswith(self.bad_target)):
908
2485
                raise Bogus
909
2486
            else:
910
2487
                _FileMover.rename(self, source, target)
912
2489
    def test_rollback_rename(self):
913
2490
        tree = self.make_branch_and_tree('.')
914
2491
        self.build_tree(['a/', 'a/b'])
915
 
        tt = tree.transform()
 
2492
        tt = TreeTransform(tree)
916
2493
        self.addCleanup(tt.finalize)
917
2494
        a_id = tt.trans_id_tree_path('a')
918
2495
        tt.adjust_path('c', tt.root, a_id)
928
2505
    def test_rollback_rename_into_place(self):
929
2506
        tree = self.make_branch_and_tree('.')
930
2507
        self.build_tree(['a/', 'a/b'])
931
 
        tt = tree.transform()
 
2508
        tt = TreeTransform(tree)
932
2509
        self.addCleanup(tt.finalize)
933
2510
        a_id = tt.trans_id_tree_path('a')
934
2511
        tt.adjust_path('c', tt.root, a_id)
944
2521
    def test_rollback_deletion(self):
945
2522
        tree = self.make_branch_and_tree('.')
946
2523
        self.build_tree(['a/', 'a/b'])
947
 
        tt = tree.transform()
 
2524
        tt = TreeTransform(tree)
948
2525
        self.addCleanup(tt.finalize)
949
2526
        a_id = tt.trans_id_tree_path('a')
950
2527
        tt.delete_contents(a_id)
965
2542
        new_globals = dict(func.__globals__)
966
2543
        new_globals.update(globals)
967
2544
        new_func = types.FunctionType(func.__code__, new_globals,
968
 
                                      func.__name__, func.__defaults__)
 
2545
            func.__name__, func.__defaults__)
969
2546
        setattr(instance, method_name,
970
 
                types.MethodType(new_func, instance))
 
2547
            types.MethodType(new_func, instance, instance.__class__))
971
2548
        self.addCleanup(delattr, instance, method_name)
972
2549
 
973
2550
    @staticmethod
984
2561
    def create_transform_and_root_trans_id(self):
985
2562
        """Setup a transform creating a file in limbo"""
986
2563
        tree = self.make_branch_and_tree('.')
987
 
        tt = tree.transform()
 
2564
        tt = TreeTransform(tree)
988
2565
        return tt, tt.create_path("a", tt.root)
989
2566
 
990
2567
    def create_transform_and_subdir_trans_id(self):
991
2568
        """Setup a transform creating a directory containing a file in limbo"""
992
2569
        tree = self.make_branch_and_tree('.')
993
 
        tt = tree.transform()
 
2570
        tt = TreeTransform(tree)
994
2571
        d_trans_id = tt.create_path("d", tt.root)
995
2572
        tt.create_directory(d_trans_id)
996
2573
        f_trans_id = tt.create_path("a", d_trans_id)
999
2576
 
1000
2577
    def test_root_create_file_open_raises_before_creation(self):
1001
2578
        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)
 
2579
        self._override_globals_in_method(tt, "create_file",
 
2580
            {"open": self._fake_open_raises_before})
 
2581
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
1006
2582
        path = tt._limbo_name(trans_id)
1007
2583
        self.assertPathDoesNotExist(path)
1008
2584
        tt.finalize()
1010
2586
 
1011
2587
    def test_root_create_file_open_raises_after_creation(self):
1012
2588
        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)
 
2589
        self._override_globals_in_method(tt, "create_file",
 
2590
            {"open": self._fake_open_raises_after})
 
2591
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
1017
2592
        path = tt._limbo_name(trans_id)
1018
2593
        self.assertPathExists(path)
1019
2594
        tt.finalize()
1022
2597
 
1023
2598
    def test_subdir_create_file_open_raises_before_creation(self):
1024
2599
        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)
 
2600
        self._override_globals_in_method(tt, "create_file",
 
2601
            {"open": self._fake_open_raises_before})
 
2602
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
1029
2603
        path = tt._limbo_name(trans_id)
1030
2604
        self.assertPathDoesNotExist(path)
1031
2605
        tt.finalize()
1033
2607
 
1034
2608
    def test_subdir_create_file_open_raises_after_creation(self):
1035
2609
        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)
 
2610
        self._override_globals_in_method(tt, "create_file",
 
2611
            {"open": self._fake_open_raises_after})
 
2612
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
1040
2613
        path = tt._limbo_name(trans_id)
1041
2614
        self.assertPathExists(path)
1042
2615
        tt.finalize()
1046
2619
    def test_rename_in_limbo_rename_raises_after_rename(self):
1047
2620
        tt, trans_id = self.create_transform_and_root_trans_id()
1048
2621
        parent1 = tt.new_directory('parent1', tt.root)
1049
 
        child1 = tt.new_file('child1', parent1, [b'contents'])
 
2622
        child1 = tt.new_file('child1', parent1, 'contents')
1050
2623
        parent2 = tt.new_directory('parent2', tt.root)
1051
2624
 
1052
2625
        class FakeOSModule(object):
1054
2627
                os.rename(old, new)
1055
2628
                raise RuntimeError
1056
2629
        self._override_globals_in_method(tt, "_rename_in_limbo",
1057
 
                                         {"os": FakeOSModule()})
 
2630
            {"os": FakeOSModule()})
1058
2631
        self.assertRaises(
1059
2632
            RuntimeError, tt.adjust_path, "child1", parent2, child1)
1060
2633
        path = osutils.pathjoin(tt._limbo_name(parent2), "child1")
1066
2639
    def test_rename_in_limbo_rename_raises_before_rename(self):
1067
2640
        tt, trans_id = self.create_transform_and_root_trans_id()
1068
2641
        parent1 = tt.new_directory('parent1', tt.root)
1069
 
        child1 = tt.new_file('child1', parent1, [b'contents'])
 
2642
        child1 = tt.new_file('child1', parent1, 'contents')
1070
2643
        parent2 = tt.new_directory('parent2', tt.root)
1071
2644
 
1072
2645
        class FakeOSModule(object):
1073
2646
            def rename(self, old, new):
1074
2647
                raise RuntimeError
1075
2648
        self._override_globals_in_method(tt, "_rename_in_limbo",
1076
 
                                         {"os": FakeOSModule()})
 
2649
            {"os": FakeOSModule()})
1077
2650
        self.assertRaises(
1078
2651
            RuntimeError, tt.adjust_path, "child1", parent2, child1)
1079
2652
        path = osutils.pathjoin(tt._limbo_name(parent1), "child1")
1087
2660
 
1088
2661
    def make_tt_with_versioned_dir(self):
1089
2662
        wt = self.make_branch_and_tree('.')
1090
 
        self.build_tree(['dir/', ])
1091
 
        wt.add(['dir'], [b'dir-id'])
 
2663
        self.build_tree(['dir/',])
 
2664
        wt.add(['dir'], ['dir-id'])
1092
2665
        wt.commit('Create dir')
1093
 
        tt = wt.transform()
 
2666
        tt = TreeTransform(wt)
1094
2667
        self.addCleanup(tt.finalize)
1095
2668
        return wt, tt
1096
2669
 
1097
2670
    def test_resolve_create_parent_for_versioned_file(self):
1098
2671
        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')
 
2672
        dir_tid = tt.trans_id_tree_file_id('dir-id')
 
2673
        file_tid = tt.new_file('file', dir_tid, 'Contents', file_id='file-id')
1101
2674
        tt.delete_contents(dir_tid)
1102
2675
        tt.unversion_file(dir_tid)
1103
2676
        conflicts = resolve_conflicts(tt)
1107
2680
 
1108
2681
    def test_non_versioned_file_create_conflict(self):
1109
2682
        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'])
 
2683
        dir_tid = tt.trans_id_tree_file_id('dir-id')
 
2684
        tt.new_file('file', dir_tid, 'Contents')
1112
2685
        tt.delete_contents(dir_tid)
1113
2686
        tt.unversion_file(dir_tid)
1114
2687
        conflicts = resolve_conflicts(tt)
1118
2691
                         conflicts.pop())
1119
2692
 
1120
2693
 
 
2694
A_ENTRY = ('a-id', ('a', 'a'), True, (True, True),
 
2695
                  ('TREE_ROOT', 'TREE_ROOT'), ('a', 'a'), ('file', 'file'),
 
2696
                  (False, False))
 
2697
ROOT_ENTRY = ('TREE_ROOT', ('', ''), False, (True, True), (None, None),
 
2698
              ('', ''), ('directory', 'directory'), (False, False))
 
2699
 
 
2700
 
 
2701
class TestTransformPreview(tests.TestCaseWithTransport):
 
2702
 
 
2703
    def create_tree(self):
 
2704
        tree = self.make_branch_and_tree('.')
 
2705
        self.build_tree_contents([('a', 'content 1')])
 
2706
        tree.set_root_id('TREE_ROOT')
 
2707
        tree.add('a', 'a-id')
 
2708
        tree.commit('rev1', rev_id='rev1')
 
2709
        return tree.branch.repository.revision_tree('rev1')
 
2710
 
 
2711
    def get_empty_preview(self):
 
2712
        repository = self.make_repository('repo')
 
2713
        tree = repository.revision_tree(_mod_revision.NULL_REVISION)
 
2714
        preview = TransformPreview(tree)
 
2715
        self.addCleanup(preview.finalize)
 
2716
        return preview
 
2717
 
 
2718
    def test_transform_preview(self):
 
2719
        revision_tree = self.create_tree()
 
2720
        preview = TransformPreview(revision_tree)
 
2721
        self.addCleanup(preview.finalize)
 
2722
 
 
2723
    def test_transform_preview_tree(self):
 
2724
        revision_tree = self.create_tree()
 
2725
        preview = TransformPreview(revision_tree)
 
2726
        self.addCleanup(preview.finalize)
 
2727
        preview.get_preview_tree()
 
2728
 
 
2729
    def test_transform_new_file(self):
 
2730
        revision_tree = self.create_tree()
 
2731
        preview = TransformPreview(revision_tree)
 
2732
        self.addCleanup(preview.finalize)
 
2733
        preview.new_file('file2', preview.root, 'content B\n', 'file2-id')
 
2734
        preview_tree = preview.get_preview_tree()
 
2735
        self.assertEqual(preview_tree.kind('file2'), 'file')
 
2736
        self.assertEqual(
 
2737
            preview_tree.get_file('file2', 'file2-id').read(), 'content B\n')
 
2738
 
 
2739
    def test_diff_preview_tree(self):
 
2740
        revision_tree = self.create_tree()
 
2741
        preview = TransformPreview(revision_tree)
 
2742
        self.addCleanup(preview.finalize)
 
2743
        preview.new_file('file2', preview.root, 'content B\n', 'file2-id')
 
2744
        preview_tree = preview.get_preview_tree()
 
2745
        out = BytesIO()
 
2746
        show_diff_trees(revision_tree, preview_tree, out)
 
2747
        lines = out.getvalue().splitlines()
 
2748
        self.assertEqual(lines[0], "=== added file 'file2'")
 
2749
        # 3 lines of diff administrivia
 
2750
        self.assertEqual(lines[4], "+content B")
 
2751
 
 
2752
    def test_transform_conflicts(self):
 
2753
        revision_tree = self.create_tree()
 
2754
        preview = TransformPreview(revision_tree)
 
2755
        self.addCleanup(preview.finalize)
 
2756
        preview.new_file('a', preview.root, 'content 2')
 
2757
        resolve_conflicts(preview)
 
2758
        trans_id = preview.trans_id_file_id('a-id')
 
2759
        self.assertEqual('a.moved', preview.final_name(trans_id))
 
2760
 
 
2761
    def get_tree_and_preview_tree(self):
 
2762
        revision_tree = self.create_tree()
 
2763
        preview = TransformPreview(revision_tree)
 
2764
        self.addCleanup(preview.finalize)
 
2765
        a_trans_id = preview.trans_id_file_id('a-id')
 
2766
        preview.delete_contents(a_trans_id)
 
2767
        preview.create_file('b content', a_trans_id)
 
2768
        preview_tree = preview.get_preview_tree()
 
2769
        return revision_tree, preview_tree
 
2770
 
 
2771
    def test_iter_changes(self):
 
2772
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2773
        root = revision_tree.get_root_id()
 
2774
        self.assertEqual([('a-id', ('a', 'a'), True, (True, True),
 
2775
                          (root, root), ('a', 'a'), ('file', 'file'),
 
2776
                          (False, False))],
 
2777
                          list(preview_tree.iter_changes(revision_tree)))
 
2778
 
 
2779
    def test_include_unchanged_succeeds(self):
 
2780
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2781
        changes = preview_tree.iter_changes(revision_tree,
 
2782
                                            include_unchanged=True)
 
2783
        root = revision_tree.get_root_id()
 
2784
 
 
2785
        self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
 
2786
 
 
2787
    def test_specific_files(self):
 
2788
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2789
        changes = preview_tree.iter_changes(revision_tree,
 
2790
                                            specific_files=[''])
 
2791
        self.assertEqual([A_ENTRY], list(changes))
 
2792
 
 
2793
    def test_want_unversioned(self):
 
2794
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2795
        changes = preview_tree.iter_changes(revision_tree,
 
2796
                                            want_unversioned=True)
 
2797
        self.assertEqual([A_ENTRY], list(changes))
 
2798
 
 
2799
    def test_ignore_extra_trees_no_specific_files(self):
 
2800
        # extra_trees is harmless without specific_files, so we'll silently
 
2801
        # accept it, even though we won't use it.
 
2802
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2803
        preview_tree.iter_changes(revision_tree, extra_trees=[preview_tree])
 
2804
 
 
2805
    def test_ignore_require_versioned_no_specific_files(self):
 
2806
        # require_versioned is meaningless without specific_files.
 
2807
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2808
        preview_tree.iter_changes(revision_tree, require_versioned=False)
 
2809
 
 
2810
    def test_ignore_pb(self):
 
2811
        # pb could be supported, but TT.iter_changes doesn't support it.
 
2812
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2813
        preview_tree.iter_changes(revision_tree)
 
2814
 
 
2815
    def test_kind(self):
 
2816
        revision_tree = self.create_tree()
 
2817
        preview = TransformPreview(revision_tree)
 
2818
        self.addCleanup(preview.finalize)
 
2819
        preview.new_file('file', preview.root, 'contents', 'file-id')
 
2820
        preview.new_directory('directory', preview.root, 'dir-id')
 
2821
        preview_tree = preview.get_preview_tree()
 
2822
        self.assertEqual('file', preview_tree.kind('file'))
 
2823
        self.assertEqual('directory', preview_tree.kind('directory'))
 
2824
 
 
2825
    def test_get_file_mtime(self):
 
2826
        preview = self.get_empty_preview()
 
2827
        file_trans_id = preview.new_file('file', preview.root, 'contents',
 
2828
                                         'file-id')
 
2829
        limbo_path = preview._limbo_name(file_trans_id)
 
2830
        preview_tree = preview.get_preview_tree()
 
2831
        self.assertEqual(os.stat(limbo_path).st_mtime,
 
2832
                         preview_tree.get_file_mtime('file', 'file-id'))
 
2833
 
 
2834
    def test_get_file_mtime_renamed(self):
 
2835
        work_tree = self.make_branch_and_tree('tree')
 
2836
        self.build_tree(['tree/file'])
 
2837
        work_tree.add('file', 'file-id')
 
2838
        preview = TransformPreview(work_tree)
 
2839
        self.addCleanup(preview.finalize)
 
2840
        file_trans_id = preview.trans_id_tree_file_id('file-id')
 
2841
        preview.adjust_path('renamed', preview.root, file_trans_id)
 
2842
        preview_tree = preview.get_preview_tree()
 
2843
        preview_mtime = preview_tree.get_file_mtime('renamed', 'file-id')
 
2844
        work_mtime = work_tree.get_file_mtime('file', 'file-id')
 
2845
 
 
2846
    def test_get_file_size(self):
 
2847
        work_tree = self.make_branch_and_tree('tree')
 
2848
        self.build_tree_contents([('tree/old', 'old')])
 
2849
        work_tree.add('old', 'old-id')
 
2850
        preview = TransformPreview(work_tree)
 
2851
        self.addCleanup(preview.finalize)
 
2852
        new_id = preview.new_file('name', preview.root, 'contents', 'new-id',
 
2853
                                  'executable')
 
2854
        tree = preview.get_preview_tree()
 
2855
        self.assertEqual(len('old'), tree.get_file_size('old'))
 
2856
        self.assertEqual(len('contents'), tree.get_file_size('name'))
 
2857
 
 
2858
    def test_get_file(self):
 
2859
        preview = self.get_empty_preview()
 
2860
        preview.new_file('file', preview.root, 'contents', 'file-id')
 
2861
        preview_tree = preview.get_preview_tree()
 
2862
        tree_file = preview_tree.get_file('file')
 
2863
        try:
 
2864
            self.assertEqual('contents', tree_file.read())
 
2865
        finally:
 
2866
            tree_file.close()
 
2867
 
 
2868
    def test_get_symlink_target(self):
 
2869
        self.requireFeature(SymlinkFeature)
 
2870
        preview = self.get_empty_preview()
 
2871
        preview.new_symlink('symlink', preview.root, 'target', 'symlink-id')
 
2872
        preview_tree = preview.get_preview_tree()
 
2873
        self.assertEqual('target',
 
2874
                         preview_tree.get_symlink_target('symlink'))
 
2875
 
 
2876
    def test_all_file_ids(self):
 
2877
        tree = self.make_branch_and_tree('tree')
 
2878
        self.build_tree(['tree/a', 'tree/b', 'tree/c'])
 
2879
        tree.add(['a', 'b', 'c'], ['a-id', 'b-id', 'c-id'])
 
2880
        preview = TransformPreview(tree)
 
2881
        self.addCleanup(preview.finalize)
 
2882
        preview.unversion_file(preview.trans_id_file_id('b-id'))
 
2883
        c_trans_id = preview.trans_id_file_id('c-id')
 
2884
        preview.unversion_file(c_trans_id)
 
2885
        preview.version_file('c-id', c_trans_id)
 
2886
        preview_tree = preview.get_preview_tree()
 
2887
        self.assertEqual({'a-id', 'c-id', tree.get_root_id()},
 
2888
                         preview_tree.all_file_ids())
 
2889
 
 
2890
    def test_path2id_deleted_unchanged(self):
 
2891
        tree = self.make_branch_and_tree('tree')
 
2892
        self.build_tree(['tree/unchanged', 'tree/deleted'])
 
2893
        tree.add(['unchanged', 'deleted'], ['unchanged-id', 'deleted-id'])
 
2894
        preview = TransformPreview(tree)
 
2895
        self.addCleanup(preview.finalize)
 
2896
        preview.unversion_file(preview.trans_id_file_id('deleted-id'))
 
2897
        preview_tree = preview.get_preview_tree()
 
2898
        self.assertEqual('unchanged-id', preview_tree.path2id('unchanged'))
 
2899
        self.assertFalse(preview_tree.is_versioned('deleted'))
 
2900
 
 
2901
    def test_path2id_created(self):
 
2902
        tree = self.make_branch_and_tree('tree')
 
2903
        self.build_tree(['tree/unchanged'])
 
2904
        tree.add(['unchanged'], ['unchanged-id'])
 
2905
        preview = TransformPreview(tree)
 
2906
        self.addCleanup(preview.finalize)
 
2907
        preview.new_file('new', preview.trans_id_file_id('unchanged-id'),
 
2908
            'contents', 'new-id')
 
2909
        preview_tree = preview.get_preview_tree()
 
2910
        self.assertEqual('new-id', preview_tree.path2id('unchanged/new'))
 
2911
 
 
2912
    def test_path2id_moved(self):
 
2913
        tree = self.make_branch_and_tree('tree')
 
2914
        self.build_tree(['tree/old_parent/', 'tree/old_parent/child'])
 
2915
        tree.add(['old_parent', 'old_parent/child'],
 
2916
                 ['old_parent-id', 'child-id'])
 
2917
        preview = TransformPreview(tree)
 
2918
        self.addCleanup(preview.finalize)
 
2919
        new_parent = preview.new_directory('new_parent', preview.root,
 
2920
                                           'new_parent-id')
 
2921
        preview.adjust_path('child', new_parent,
 
2922
                            preview.trans_id_file_id('child-id'))
 
2923
        preview_tree = preview.get_preview_tree()
 
2924
        self.assertFalse(preview_tree.is_versioned('old_parent/child'))
 
2925
        self.assertEqual('child-id', preview_tree.path2id('new_parent/child'))
 
2926
 
 
2927
    def test_path2id_renamed_parent(self):
 
2928
        tree = self.make_branch_and_tree('tree')
 
2929
        self.build_tree(['tree/old_name/', 'tree/old_name/child'])
 
2930
        tree.add(['old_name', 'old_name/child'],
 
2931
                 ['parent-id', 'child-id'])
 
2932
        preview = TransformPreview(tree)
 
2933
        self.addCleanup(preview.finalize)
 
2934
        preview.adjust_path('new_name', preview.root,
 
2935
                            preview.trans_id_file_id('parent-id'))
 
2936
        preview_tree = preview.get_preview_tree()
 
2937
        self.assertFalse(preview_tree.is_versioned('old_name/child'))
 
2938
        self.assertEqual('child-id', preview_tree.path2id('new_name/child'))
 
2939
 
 
2940
    def assertMatchingIterEntries(self, tt, specific_file_ids=None):
 
2941
        preview_tree = tt.get_preview_tree()
 
2942
        preview_result = list(preview_tree.iter_entries_by_dir(
 
2943
                              specific_file_ids))
 
2944
        tree = tt._tree
 
2945
        tt.apply()
 
2946
        actual_result = list(tree.iter_entries_by_dir(specific_file_ids))
 
2947
        self.assertEqual(actual_result, preview_result)
 
2948
 
 
2949
    def test_iter_entries_by_dir_new(self):
 
2950
        tree = self.make_branch_and_tree('tree')
 
2951
        tt = TreeTransform(tree)
 
2952
        tt.new_file('new', tt.root, 'contents', 'new-id')
 
2953
        self.assertMatchingIterEntries(tt)
 
2954
 
 
2955
    def test_iter_entries_by_dir_deleted(self):
 
2956
        tree = self.make_branch_and_tree('tree')
 
2957
        self.build_tree(['tree/deleted'])
 
2958
        tree.add('deleted', 'deleted-id')
 
2959
        tt = TreeTransform(tree)
 
2960
        tt.delete_contents(tt.trans_id_file_id('deleted-id'))
 
2961
        self.assertMatchingIterEntries(tt)
 
2962
 
 
2963
    def test_iter_entries_by_dir_unversioned(self):
 
2964
        tree = self.make_branch_and_tree('tree')
 
2965
        self.build_tree(['tree/removed'])
 
2966
        tree.add('removed', 'removed-id')
 
2967
        tt = TreeTransform(tree)
 
2968
        tt.unversion_file(tt.trans_id_file_id('removed-id'))
 
2969
        self.assertMatchingIterEntries(tt)
 
2970
 
 
2971
    def test_iter_entries_by_dir_moved(self):
 
2972
        tree = self.make_branch_and_tree('tree')
 
2973
        self.build_tree(['tree/moved', 'tree/new_parent/'])
 
2974
        tree.add(['moved', 'new_parent'], ['moved-id', 'new_parent-id'])
 
2975
        tt = TreeTransform(tree)
 
2976
        tt.adjust_path('moved', tt.trans_id_file_id('new_parent-id'),
 
2977
                       tt.trans_id_file_id('moved-id'))
 
2978
        self.assertMatchingIterEntries(tt)
 
2979
 
 
2980
    def test_iter_entries_by_dir_specific_file_ids(self):
 
2981
        tree = self.make_branch_and_tree('tree')
 
2982
        tree.set_root_id('tree-root-id')
 
2983
        self.build_tree(['tree/parent/', 'tree/parent/child'])
 
2984
        tree.add(['parent', 'parent/child'], ['parent-id', 'child-id'])
 
2985
        tt = TreeTransform(tree)
 
2986
        self.assertMatchingIterEntries(tt, ['tree-root-id', 'child-id'])
 
2987
 
 
2988
    def test_symlink_content_summary(self):
 
2989
        self.requireFeature(SymlinkFeature)
 
2990
        preview = self.get_empty_preview()
 
2991
        preview.new_symlink('path', preview.root, 'target', 'path-id')
 
2992
        summary = preview.get_preview_tree().path_content_summary('path')
 
2993
        self.assertEqual(('symlink', None, None, 'target'), summary)
 
2994
 
 
2995
    def test_missing_content_summary(self):
 
2996
        preview = self.get_empty_preview()
 
2997
        summary = preview.get_preview_tree().path_content_summary('path')
 
2998
        self.assertEqual(('missing', None, None, None), summary)
 
2999
 
 
3000
    def test_deleted_content_summary(self):
 
3001
        tree = self.make_branch_and_tree('tree')
 
3002
        self.build_tree(['tree/path/'])
 
3003
        tree.add('path')
 
3004
        preview = TransformPreview(tree)
 
3005
        self.addCleanup(preview.finalize)
 
3006
        preview.delete_contents(preview.trans_id_tree_path('path'))
 
3007
        summary = preview.get_preview_tree().path_content_summary('path')
 
3008
        self.assertEqual(('missing', None, None, None), summary)
 
3009
 
 
3010
    def test_file_content_summary_executable(self):
 
3011
        preview = self.get_empty_preview()
 
3012
        path_id = preview.new_file('path', preview.root, 'contents', 'path-id')
 
3013
        preview.set_executability(True, path_id)
 
3014
        summary = preview.get_preview_tree().path_content_summary('path')
 
3015
        self.assertEqual(4, len(summary))
 
3016
        self.assertEqual('file', summary[0])
 
3017
        # size must be known
 
3018
        self.assertEqual(len('contents'), summary[1])
 
3019
        # executable
 
3020
        self.assertEqual(True, summary[2])
 
3021
        # will not have hash (not cheap to determine)
 
3022
        self.assertIs(None, summary[3])
 
3023
 
 
3024
    def test_change_executability(self):
 
3025
        tree = self.make_branch_and_tree('tree')
 
3026
        self.build_tree(['tree/path'])
 
3027
        tree.add('path')
 
3028
        preview = TransformPreview(tree)
 
3029
        self.addCleanup(preview.finalize)
 
3030
        path_id = preview.trans_id_tree_path('path')
 
3031
        preview.set_executability(True, path_id)
 
3032
        summary = preview.get_preview_tree().path_content_summary('path')
 
3033
        self.assertEqual(True, summary[2])
 
3034
 
 
3035
    def test_file_content_summary_non_exec(self):
 
3036
        preview = self.get_empty_preview()
 
3037
        preview.new_file('path', preview.root, 'contents', 'path-id')
 
3038
        summary = preview.get_preview_tree().path_content_summary('path')
 
3039
        self.assertEqual(4, len(summary))
 
3040
        self.assertEqual('file', summary[0])
 
3041
        # size must be known
 
3042
        self.assertEqual(len('contents'), summary[1])
 
3043
        # not executable
 
3044
        self.assertEqual(False, summary[2])
 
3045
        # will not have hash (not cheap to determine)
 
3046
        self.assertIs(None, summary[3])
 
3047
 
 
3048
    def test_dir_content_summary(self):
 
3049
        preview = self.get_empty_preview()
 
3050
        preview.new_directory('path', preview.root, 'path-id')
 
3051
        summary = preview.get_preview_tree().path_content_summary('path')
 
3052
        self.assertEqual(('directory', None, None, None), summary)
 
3053
 
 
3054
    def test_tree_content_summary(self):
 
3055
        preview = self.get_empty_preview()
 
3056
        path = preview.new_directory('path', preview.root, 'path-id')
 
3057
        preview.set_tree_reference('rev-1', path)
 
3058
        summary = preview.get_preview_tree().path_content_summary('path')
 
3059
        self.assertEqual(4, len(summary))
 
3060
        self.assertEqual('tree-reference', summary[0])
 
3061
 
 
3062
    def test_annotate(self):
 
3063
        tree = self.make_branch_and_tree('tree')
 
3064
        self.build_tree_contents([('tree/file', 'a\n')])
 
3065
        tree.add('file', 'file-id')
 
3066
        tree.commit('a', rev_id='one')
 
3067
        self.build_tree_contents([('tree/file', 'a\nb\n')])
 
3068
        preview = TransformPreview(tree)
 
3069
        self.addCleanup(preview.finalize)
 
3070
        file_trans_id = preview.trans_id_file_id('file-id')
 
3071
        preview.delete_contents(file_trans_id)
 
3072
        preview.create_file('a\nb\nc\n', file_trans_id)
 
3073
        preview_tree = preview.get_preview_tree()
 
3074
        expected = [
 
3075
            ('one', 'a\n'),
 
3076
            ('me:', 'b\n'),
 
3077
            ('me:', 'c\n'),
 
3078
        ]
 
3079
        annotation = preview_tree.annotate_iter('file', default_revision='me:')
 
3080
        self.assertEqual(expected, annotation)
 
3081
 
 
3082
    def test_annotate_missing(self):
 
3083
        preview = self.get_empty_preview()
 
3084
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
 
3085
        preview_tree = preview.get_preview_tree()
 
3086
        expected = [
 
3087
            ('me:', 'a\n'),
 
3088
            ('me:', 'b\n'),
 
3089
            ('me:', 'c\n'),
 
3090
         ]
 
3091
        annotation = preview_tree.annotate_iter('file', default_revision='me:')
 
3092
        self.assertEqual(expected, annotation)
 
3093
 
 
3094
    def test_annotate_rename(self):
 
3095
        tree = self.make_branch_and_tree('tree')
 
3096
        self.build_tree_contents([('tree/file', 'a\n')])
 
3097
        tree.add('file', 'file-id')
 
3098
        tree.commit('a', rev_id='one')
 
3099
        preview = TransformPreview(tree)
 
3100
        self.addCleanup(preview.finalize)
 
3101
        file_trans_id = preview.trans_id_file_id('file-id')
 
3102
        preview.adjust_path('newname', preview.root, file_trans_id)
 
3103
        preview_tree = preview.get_preview_tree()
 
3104
        expected = [
 
3105
            ('one', 'a\n'),
 
3106
        ]
 
3107
        annotation = preview_tree.annotate_iter('file', default_revision='me:')
 
3108
        self.assertEqual(expected, annotation)
 
3109
 
 
3110
    def test_annotate_deleted(self):
 
3111
        tree = self.make_branch_and_tree('tree')
 
3112
        self.build_tree_contents([('tree/file', 'a\n')])
 
3113
        tree.add('file', 'file-id')
 
3114
        tree.commit('a', rev_id='one')
 
3115
        self.build_tree_contents([('tree/file', 'a\nb\n')])
 
3116
        preview = TransformPreview(tree)
 
3117
        self.addCleanup(preview.finalize)
 
3118
        file_trans_id = preview.trans_id_file_id('file-id')
 
3119
        preview.delete_contents(file_trans_id)
 
3120
        preview_tree = preview.get_preview_tree()
 
3121
        annotation = preview_tree.annotate_iter('file', default_revision='me:')
 
3122
        self.assertIs(None, annotation)
 
3123
 
 
3124
    def test_stored_kind(self):
 
3125
        preview = self.get_empty_preview()
 
3126
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
 
3127
        preview_tree = preview.get_preview_tree()
 
3128
        self.assertEqual('file', preview_tree.stored_kind('file'))
 
3129
 
 
3130
    def test_is_executable(self):
 
3131
        preview = self.get_empty_preview()
 
3132
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
 
3133
        preview.set_executability(True, preview.trans_id_file_id('file-id'))
 
3134
        preview_tree = preview.get_preview_tree()
 
3135
        self.assertEqual(True, preview_tree.is_executable('file'))
 
3136
 
 
3137
    def test_get_set_parent_ids(self):
 
3138
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
3139
        self.assertEqual([], preview_tree.get_parent_ids())
 
3140
        preview_tree.set_parent_ids(['rev-1'])
 
3141
        self.assertEqual(['rev-1'], preview_tree.get_parent_ids())
 
3142
 
 
3143
    def test_plan_file_merge(self):
 
3144
        work_a = self.make_branch_and_tree('wta')
 
3145
        self.build_tree_contents([('wta/file', 'a\nb\nc\nd\n')])
 
3146
        work_a.add('file', 'file-id')
 
3147
        base_id = work_a.commit('base version')
 
3148
        tree_b = work_a.controldir.sprout('wtb').open_workingtree()
 
3149
        preview = TransformPreview(work_a)
 
3150
        self.addCleanup(preview.finalize)
 
3151
        trans_id = preview.trans_id_file_id('file-id')
 
3152
        preview.delete_contents(trans_id)
 
3153
        preview.create_file('b\nc\nd\ne\n', trans_id)
 
3154
        self.build_tree_contents([('wtb/file', 'a\nc\nd\nf\n')])
 
3155
        tree_a = preview.get_preview_tree()
 
3156
        tree_a.set_parent_ids([base_id])
 
3157
        self.assertEqual([
 
3158
            ('killed-a', 'a\n'),
 
3159
            ('killed-b', 'b\n'),
 
3160
            ('unchanged', 'c\n'),
 
3161
            ('unchanged', 'd\n'),
 
3162
            ('new-a', 'e\n'),
 
3163
            ('new-b', 'f\n'),
 
3164
        ], list(tree_a.plan_file_merge('file-id', tree_b)))
 
3165
 
 
3166
    def test_plan_file_merge_revision_tree(self):
 
3167
        work_a = self.make_branch_and_tree('wta')
 
3168
        self.build_tree_contents([('wta/file', 'a\nb\nc\nd\n')])
 
3169
        work_a.add('file', 'file-id')
 
3170
        base_id = work_a.commit('base version')
 
3171
        tree_b = work_a.controldir.sprout('wtb').open_workingtree()
 
3172
        preview = TransformPreview(work_a.basis_tree())
 
3173
        self.addCleanup(preview.finalize)
 
3174
        trans_id = preview.trans_id_file_id('file-id')
 
3175
        preview.delete_contents(trans_id)
 
3176
        preview.create_file('b\nc\nd\ne\n', trans_id)
 
3177
        self.build_tree_contents([('wtb/file', 'a\nc\nd\nf\n')])
 
3178
        tree_a = preview.get_preview_tree()
 
3179
        tree_a.set_parent_ids([base_id])
 
3180
        self.assertEqual([
 
3181
            ('killed-a', 'a\n'),
 
3182
            ('killed-b', 'b\n'),
 
3183
            ('unchanged', 'c\n'),
 
3184
            ('unchanged', 'd\n'),
 
3185
            ('new-a', 'e\n'),
 
3186
            ('new-b', 'f\n'),
 
3187
        ], list(tree_a.plan_file_merge('file-id', tree_b)))
 
3188
 
 
3189
    def test_walkdirs(self):
 
3190
        preview = self.get_empty_preview()
 
3191
        root = preview.new_directory('', ROOT_PARENT, 'tree-root')
 
3192
        # FIXME: new_directory should mark root.
 
3193
        preview.fixup_new_roots()
 
3194
        preview_tree = preview.get_preview_tree()
 
3195
        file_trans_id = preview.new_file('a', preview.root, 'contents',
 
3196
                                         'a-id')
 
3197
        expected = [(('', 'tree-root'),
 
3198
                    [('a', 'a', 'file', None, 'a-id', 'file')])]
 
3199
        self.assertEqual(expected, list(preview_tree.walkdirs()))
 
3200
 
 
3201
    def test_extras(self):
 
3202
        work_tree = self.make_branch_and_tree('tree')
 
3203
        self.build_tree(['tree/removed-file', 'tree/existing-file',
 
3204
                         'tree/not-removed-file'])
 
3205
        work_tree.add(['removed-file', 'not-removed-file'])
 
3206
        preview = TransformPreview(work_tree)
 
3207
        self.addCleanup(preview.finalize)
 
3208
        preview.new_file('new-file', preview.root, 'contents')
 
3209
        preview.new_file('new-versioned-file', preview.root, 'contents',
 
3210
                         'new-versioned-id')
 
3211
        tree = preview.get_preview_tree()
 
3212
        preview.unversion_file(preview.trans_id_tree_path('removed-file'))
 
3213
        self.assertEqual({'new-file', 'removed-file', 'existing-file'},
 
3214
                         set(tree.extras()))
 
3215
 
 
3216
    def test_merge_into_preview(self):
 
3217
        work_tree = self.make_branch_and_tree('tree')
 
3218
        self.build_tree_contents([('tree/file', 'b\n')])
 
3219
        work_tree.add('file', 'file-id')
 
3220
        work_tree.commit('first commit')
 
3221
        child_tree = work_tree.controldir.sprout('child').open_workingtree()
 
3222
        self.build_tree_contents([('child/file', 'b\nc\n')])
 
3223
        child_tree.commit('child commit')
 
3224
        child_tree.lock_write()
 
3225
        self.addCleanup(child_tree.unlock)
 
3226
        work_tree.lock_write()
 
3227
        self.addCleanup(work_tree.unlock)
 
3228
        preview = TransformPreview(work_tree)
 
3229
        self.addCleanup(preview.finalize)
 
3230
        file_trans_id = preview.trans_id_file_id('file-id')
 
3231
        preview.delete_contents(file_trans_id)
 
3232
        preview.create_file('a\nb\n', file_trans_id)
 
3233
        preview_tree = preview.get_preview_tree()
 
3234
        merger = Merger.from_revision_ids(preview_tree,
 
3235
                                          child_tree.branch.last_revision(),
 
3236
                                          other_branch=child_tree.branch,
 
3237
                                          tree_branch=work_tree.branch)
 
3238
        merger.merge_type = Merge3Merger
 
3239
        tt = merger.make_merger().make_preview_transform()
 
3240
        self.addCleanup(tt.finalize)
 
3241
        final_tree = tt.get_preview_tree()
 
3242
        self.assertEqual(
 
3243
                'a\nb\nc\n',
 
3244
                final_tree.get_file_text(final_tree.id2path('file-id')))
 
3245
 
 
3246
    def test_merge_preview_into_workingtree(self):
 
3247
        tree = self.make_branch_and_tree('tree')
 
3248
        tree.set_root_id('TREE_ROOT')
 
3249
        tt = TransformPreview(tree)
 
3250
        self.addCleanup(tt.finalize)
 
3251
        tt.new_file('name', tt.root, 'content', 'file-id')
 
3252
        tree2 = self.make_branch_and_tree('tree2')
 
3253
        tree2.set_root_id('TREE_ROOT')
 
3254
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
 
3255
                                         tree.basis_tree())
 
3256
        merger.merge_type = Merge3Merger
 
3257
        merger.do_merge()
 
3258
 
 
3259
    def test_merge_preview_into_workingtree_handles_conflicts(self):
 
3260
        tree = self.make_branch_and_tree('tree')
 
3261
        self.build_tree_contents([('tree/foo', 'bar')])
 
3262
        tree.add('foo', 'foo-id')
 
3263
        tree.commit('foo')
 
3264
        tt = TransformPreview(tree)
 
3265
        self.addCleanup(tt.finalize)
 
3266
        trans_id = tt.trans_id_file_id('foo-id')
 
3267
        tt.delete_contents(trans_id)
 
3268
        tt.create_file('baz', trans_id)
 
3269
        tree2 = tree.controldir.sprout('tree2').open_workingtree()
 
3270
        self.build_tree_contents([('tree2/foo', 'qux')])
 
3271
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
 
3272
                                         tree.basis_tree())
 
3273
        merger.merge_type = Merge3Merger
 
3274
        merger.do_merge()
 
3275
 
 
3276
    def test_has_filename(self):
 
3277
        wt = self.make_branch_and_tree('tree')
 
3278
        self.build_tree(['tree/unmodified', 'tree/removed', 'tree/modified'])
 
3279
        tt = TransformPreview(wt)
 
3280
        removed_id = tt.trans_id_tree_path('removed')
 
3281
        tt.delete_contents(removed_id)
 
3282
        tt.new_file('new', tt.root, 'contents')
 
3283
        modified_id = tt.trans_id_tree_path('modified')
 
3284
        tt.delete_contents(modified_id)
 
3285
        tt.create_file('modified-contents', modified_id)
 
3286
        self.addCleanup(tt.finalize)
 
3287
        tree = tt.get_preview_tree()
 
3288
        self.assertTrue(tree.has_filename('unmodified'))
 
3289
        self.assertFalse(tree.has_filename('not-present'))
 
3290
        self.assertFalse(tree.has_filename('removed'))
 
3291
        self.assertTrue(tree.has_filename('new'))
 
3292
        self.assertTrue(tree.has_filename('modified'))
 
3293
 
 
3294
    def test_is_executable(self):
 
3295
        tree = self.make_branch_and_tree('tree')
 
3296
        preview = TransformPreview(tree)
 
3297
        self.addCleanup(preview.finalize)
 
3298
        preview.new_file('foo', preview.root, 'bar', 'baz-id')
 
3299
        preview_tree = preview.get_preview_tree()
 
3300
        self.assertEqual(False, preview_tree.is_executable('tree/foo', 'baz-id'))
 
3301
        self.assertEqual(False, preview_tree.is_executable('tree/foo'))
 
3302
 
 
3303
    def test_commit_preview_tree(self):
 
3304
        tree = self.make_branch_and_tree('tree')
 
3305
        rev_id = tree.commit('rev1')
 
3306
        tree.branch.lock_write()
 
3307
        self.addCleanup(tree.branch.unlock)
 
3308
        tt = TransformPreview(tree)
 
3309
        tt.new_file('file', tt.root, 'contents', 'file_id')
 
3310
        self.addCleanup(tt.finalize)
 
3311
        preview = tt.get_preview_tree()
 
3312
        preview.set_parent_ids([rev_id])
 
3313
        builder = tree.branch.get_commit_builder([rev_id])
 
3314
        list(builder.record_iter_changes(preview, rev_id, tt.iter_changes()))
 
3315
        builder.finish_inventory()
 
3316
        rev2_id = builder.commit('rev2')
 
3317
        rev2_tree = tree.branch.repository.revision_tree(rev2_id)
 
3318
        self.assertEqual('contents', rev2_tree.get_file_text('file'))
 
3319
 
 
3320
    def test_ascii_limbo_paths(self):
 
3321
        self.requireFeature(features.UnicodeFilenameFeature)
 
3322
        branch = self.make_branch('any')
 
3323
        tree = branch.repository.revision_tree(_mod_revision.NULL_REVISION)
 
3324
        tt = TransformPreview(tree)
 
3325
        self.addCleanup(tt.finalize)
 
3326
        foo_id = tt.new_directory('', ROOT_PARENT)
 
3327
        bar_id = tt.new_file(u'\u1234bar', foo_id, 'contents')
 
3328
        limbo_path = tt._limbo_name(bar_id)
 
3329
        self.assertEqual(limbo_path.encode('ascii', 'replace'), limbo_path)
 
3330
 
 
3331
 
1121
3332
class FakeSerializer(object):
1122
3333
    """Serializer implementation that simply returns the input.
1123
3334
 
1135
3346
    def get_preview(self, tree=None):
1136
3347
        if tree is None:
1137
3348
            tree = self.make_branch_and_tree('tree')
1138
 
        tt = tree.preview_transform()
 
3349
        tt = TransformPreview(tree)
1139
3350
        self.addCleanup(tt.finalize)
1140
3351
        return tt
1141
3352
 
1146
3357
    @staticmethod
1147
3358
    def default_attribs():
1148
3359
        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': {},
 
3360
            '_id_number': 1,
 
3361
            '_new_name': {},
 
3362
            '_new_parent': {},
 
3363
            '_new_executability': {},
 
3364
            '_new_id': {},
 
3365
            '_tree_path_ids': {'': 'new-0'},
 
3366
            '_removed_id': [],
 
3367
            '_removed_contents': [],
 
3368
            '_non_present_ids': {},
1158
3369
            }
1159
3370
 
1160
3371
    def make_records(self, attribs, contents):
1161
3372
        records = [
1162
 
            ((((b'attribs'),),), bencode.bencode(attribs))]
 
3373
            (((('attribs'),),), bencode.bencode(attribs))]
1163
3374
        records.extend([(((n, k),), c) for n, k, c in contents])
1164
3375
        return records
1165
3376
 
1166
3377
    def creation_records(self):
1167
3378
        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}
 
3379
        attribs['_id_number'] = 3
 
3380
        attribs['_new_name'] = {
 
3381
            'new-1': u'foo\u1234'.encode('utf-8'), 'new-2': 'qux'}
 
3382
        attribs['_new_id'] = {'new-1': 'baz', 'new-2': 'quxx'}
 
3383
        attribs['_new_parent'] = {'new-1': 'new-0', 'new-2': 'new-0'}
 
3384
        attribs['_new_executability'] = {'new-1': 1}
1174
3385
        contents = [
1175
 
            (b'new-1', b'file', b'i 1\nbar\n'),
1176
 
            (b'new-2', b'directory', b''),
 
3386
            ('new-1', 'file', 'i 1\nbar\n'),
 
3387
            ('new-2', 'directory', ''),
1177
3388
            ]
1178
3389
        return self.make_records(attribs, contents)
1179
3390
 
1180
3391
    def test_serialize_creation(self):
1181
3392
        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')
 
3393
        tt.new_file(u'foo\u1234', tt.root, 'bar', 'baz', True)
 
3394
        tt.new_directory('qux', tt.root, 'quxx')
1184
3395
        self.assertSerializesTo(self.creation_records(), tt)
1185
3396
 
1186
3397
    def test_deserialize_creation(self):
1189
3400
        self.assertEqual(3, tt._id_number)
1190
3401
        self.assertEqual({'new-1': u'foo\u1234',
1191
3402
                          'new-2': 'qux'}, tt._new_name)
1192
 
        self.assertEqual({'new-1': b'baz', 'new-2': b'quxx'}, tt._new_id)
 
3403
        self.assertEqual({'new-1': 'baz', 'new-2': 'quxx'}, tt._new_id)
1193
3404
        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)
 
3405
        self.assertEqual({'baz': 'new-1', 'quxx': 'new-2'}, tt._r_new_id)
1195
3406
        self.assertEqual({'new-1': True}, tt._new_executability)
1196
3407
        self.assertEqual({'new-1': 'file',
1197
3408
                          'new-2': 'directory'}, tt._new_contents)
1198
 
        with open(tt._limbo_name('new-1'), 'rb') as foo_limbo:
 
3409
        foo_limbo = open(tt._limbo_name('new-1'), 'rb')
 
3410
        try:
1199
3411
            foo_content = foo_limbo.read()
1200
 
        self.assertEqual(b'bar', foo_content)
 
3412
        finally:
 
3413
            foo_limbo.close()
 
3414
        self.assertEqual('bar', foo_content)
1201
3415
 
1202
3416
    def symlink_creation_records(self):
1203
3417
        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'))]
 
3418
        attribs['_id_number'] = 2
 
3419
        attribs['_new_name'] = {'new-1': u'foo\u1234'.encode('utf-8')}
 
3420
        attribs['_new_parent'] = {'new-1': 'new-0'}
 
3421
        contents = [('new-1', 'symlink', u'bar\u1234'.encode('utf-8'))]
1208
3422
        return self.make_records(attribs, contents)
1209
3423
 
1210
3424
    def test_serialize_symlink_creation(self):
1224
3438
    def make_destruction_preview(self):
1225
3439
        tree = self.make_branch_and_tree('.')
1226
3440
        self.build_tree([u'foo\u1234', 'bar'])
1227
 
        tree.add([u'foo\u1234', 'bar'], [b'foo-id', b'bar-id'])
 
3441
        tree.add([u'foo\u1234', 'bar'], ['foo-id', 'bar-id'])
1228
3442
        return self.get_preview(tree)
1229
3443
 
1230
3444
    def destruction_records(self):
1231
3445
        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',
 
3446
        attribs['_id_number'] = 3
 
3447
        attribs['_removed_id'] = ['new-1']
 
3448
        attribs['_removed_contents'] = ['new-2']
 
3449
        attribs['_tree_path_ids'] = {
 
3450
            '': 'new-0',
 
3451
            u'foo\u1234'.encode('utf-8'): 'new-1',
 
3452
            'bar': 'new-2',
1239
3453
            }
1240
3454
        return self.make_records(attribs, [])
1241
3455
 
1242
3456
    def test_serialize_destruction(self):
1243
3457
        tt = self.make_destruction_preview()
1244
 
        foo_trans_id = tt.trans_id_tree_path(u'foo\u1234')
 
3458
        foo_trans_id = tt.trans_id_tree_file_id('foo-id')
1245
3459
        tt.unversion_file(foo_trans_id)
1246
 
        bar_trans_id = tt.trans_id_tree_path('bar')
 
3460
        bar_trans_id = tt.trans_id_tree_file_id('bar-id')
1247
3461
        tt.delete_contents(bar_trans_id)
1248
3462
        self.assertSerializesTo(self.destruction_records(), tt)
1249
3463
 
1261
3475
 
1262
3476
    def missing_records(self):
1263
3477
        attribs = self.default_attribs()
1264
 
        attribs[b'_id_number'] = 2
1265
 
        attribs[b'_non_present_ids'] = {
1266
 
            b'boo': b'new-1', }
 
3478
        attribs['_id_number'] = 2
 
3479
        attribs['_non_present_ids'] = {
 
3480
            'boo': 'new-1',}
1267
3481
        return self.make_records(attribs, [])
1268
3482
 
1269
3483
    def test_serialize_missing(self):
1270
3484
        tt = self.get_preview()
1271
 
        tt.trans_id_file_id(b'boo')
 
3485
        boo_trans_id = tt.trans_id_file_id('boo')
1272
3486
        self.assertSerializesTo(self.missing_records(), tt)
1273
3487
 
1274
3488
    def test_deserialize_missing(self):
1275
3489
        tt = self.get_preview()
1276
3490
        tt.deserialize(iter(self.missing_records()))
1277
 
        self.assertEqual({b'boo': 'new-1'}, tt._non_present_ids)
 
3491
        self.assertEqual({'boo': 'new-1'}, tt._non_present_ids)
1278
3492
 
1279
3493
    def make_modification_preview(self):
1280
 
        LINES_ONE = b'aa\nbb\ncc\ndd\n'
1281
 
        LINES_TWO = b'z\nbb\nx\ndd\n'
 
3494
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
 
3495
        LINES_TWO = 'z\nbb\nx\ndd\n'
1282
3496
        tree = self.make_branch_and_tree('tree')
1283
3497
        self.build_tree_contents([('tree/file', LINES_ONE)])
1284
 
        tree.add('file', b'file-id')
1285
 
        return self.get_preview(tree), [LINES_TWO]
 
3498
        tree.add('file', 'file-id')
 
3499
        return self.get_preview(tree), LINES_TWO
1286
3500
 
1287
3501
    def modification_records(self):
1288
3502
        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')]
 
3503
        attribs['_id_number'] = 2
 
3504
        attribs['_tree_path_ids'] = {
 
3505
            'file': 'new-1',
 
3506
            '': 'new-0',}
 
3507
        attribs['_removed_contents'] = ['new-1']
 
3508
        contents = [('new-1', 'file',
 
3509
                     'i 1\nz\n\nc 0 1 1 1\ni 1\nx\n\nc 0 3 3 1\n')]
1296
3510
        return self.make_records(attribs, contents)
1297
3511
 
1298
3512
    def test_serialize_modification(self):
1299
3513
        tt, LINES = self.make_modification_preview()
1300
 
        trans_id = tt.trans_id_file_id(b'file-id')
 
3514
        trans_id = tt.trans_id_file_id('file-id')
1301
3515
        tt.delete_contents(trans_id)
1302
3516
        tt.create_file(LINES, trans_id)
1303
3517
        self.assertSerializesTo(self.modification_records(), tt)
1305
3519
    def test_deserialize_modification(self):
1306
3520
        tt, LINES = self.make_modification_preview()
1307
3521
        tt.deserialize(iter(self.modification_records()))
1308
 
        self.assertFileEqual(b''.join(LINES), tt._limbo_name('new-1'))
 
3522
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
1309
3523
 
1310
3524
    def make_kind_change_preview(self):
1311
 
        LINES = b'a\nb\nc\nd\n'
 
3525
        LINES = 'a\nb\nc\nd\n'
1312
3526
        tree = self.make_branch_and_tree('tree')
1313
3527
        self.build_tree(['tree/foo/'])
1314
 
        tree.add('foo', b'foo-id')
1315
 
        return self.get_preview(tree), [LINES]
 
3528
        tree.add('foo', 'foo-id')
 
3529
        return self.get_preview(tree), LINES
1316
3530
 
1317
3531
    def kind_change_records(self):
1318
3532
        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')]
 
3533
        attribs['_id_number'] = 2
 
3534
        attribs['_tree_path_ids'] = {
 
3535
            'foo': 'new-1',
 
3536
            '': 'new-0',}
 
3537
        attribs['_removed_contents'] = ['new-1']
 
3538
        contents = [('new-1', 'file',
 
3539
                     'i 4\na\nb\nc\nd\n\n')]
1326
3540
        return self.make_records(attribs, contents)
1327
3541
 
1328
3542
    def test_serialize_kind_change(self):
1329
3543
        tt, LINES = self.make_kind_change_preview()
1330
 
        trans_id = tt.trans_id_file_id(b'foo-id')
 
3544
        trans_id = tt.trans_id_file_id('foo-id')
1331
3545
        tt.delete_contents(trans_id)
1332
3546
        tt.create_file(LINES, trans_id)
1333
3547
        self.assertSerializesTo(self.kind_change_records(), tt)
1335
3549
    def test_deserialize_kind_change(self):
1336
3550
        tt, LINES = self.make_kind_change_preview()
1337
3551
        tt.deserialize(iter(self.kind_change_records()))
1338
 
        self.assertFileEqual(b''.join(LINES), tt._limbo_name('new-1'))
 
3552
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
1339
3553
 
1340
3554
    def make_add_contents_preview(self):
1341
 
        LINES = b'a\nb\nc\nd\n'
 
3555
        LINES = 'a\nb\nc\nd\n'
1342
3556
        tree = self.make_branch_and_tree('tree')
1343
3557
        self.build_tree(['tree/foo'])
1344
3558
        tree.add('foo')
1347
3561
 
1348
3562
    def add_contents_records(self):
1349
3563
        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')]
 
3564
        attribs['_id_number'] = 2
 
3565
        attribs['_tree_path_ids'] = {
 
3566
            'foo': 'new-1',
 
3567
            '': 'new-0',}
 
3568
        contents = [('new-1', 'file',
 
3569
                     'i 4\na\nb\nc\nd\n\n')]
1356
3570
        return self.make_records(attribs, contents)
1357
3571
 
1358
3572
    def test_serialize_add_contents(self):
1359
3573
        tt, LINES = self.make_add_contents_preview()
1360
3574
        trans_id = tt.trans_id_tree_path('foo')
1361
 
        tt.create_file([LINES], trans_id)
 
3575
        tt.create_file(LINES, trans_id)
1362
3576
        self.assertSerializesTo(self.add_contents_records(), tt)
1363
3577
 
1364
3578
    def test_deserialize_add_contents(self):
1367
3581
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
1368
3582
 
1369
3583
    def test_get_parents_lines(self):
1370
 
        LINES_ONE = b'aa\nbb\ncc\ndd\n'
 
3584
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
 
3585
        LINES_TWO = 'z\nbb\nx\ndd\n'
1371
3586
        tree = self.make_branch_and_tree('tree')
1372
3587
        self.build_tree_contents([('tree/file', LINES_ONE)])
1373
 
        tree.add('file', b'file-id')
 
3588
        tree.add('file', 'file-id')
1374
3589
        tt = self.get_preview(tree)
1375
3590
        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))
 
3591
        self.assertEqual((['aa\n', 'bb\n', 'cc\n', 'dd\n'],),
 
3592
            tt._get_parents_lines(trans_id))
1378
3593
 
1379
3594
    def test_get_parents_texts(self):
1380
 
        LINES_ONE = b'aa\nbb\ncc\ndd\n'
 
3595
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
 
3596
        LINES_TWO = 'z\nbb\nx\ndd\n'
1381
3597
        tree = self.make_branch_and_tree('tree')
1382
3598
        self.build_tree_contents([('tree/file', LINES_ONE)])
1383
 
        tree.add('file', b'file-id')
 
3599
        tree.add('file', 'file-id')
1384
3600
        tt = self.get_preview(tree)
1385
3601
        trans_id = tt.trans_id_tree_path('file')
1386
3602
        self.assertEqual((LINES_ONE,),
1387
 
                         tt._get_parents_texts(trans_id))
 
3603
            tt._get_parents_texts(trans_id))
1388
3604
 
1389
3605
 
1390
3606
class TestOrphan(tests.TestCaseWithTransport):
1391
3607
 
1392
3608
    def test_no_orphan_for_transform_preview(self):
1393
3609
        tree = self.make_branch_and_tree('tree')
1394
 
        tt = tree.preview_transform()
 
3610
        tt = transform.TransformPreview(tree)
1395
3611
        self.addCleanup(tt.finalize)
1396
3612
        self.assertRaises(NotImplementedError, tt.new_orphan, 'foo', 'bar')
1397
3613
 
1398
3614
    def _set_orphan_policy(self, wt, policy):
1399
 
        wt.branch.get_config_stack().set('transform.orphan_policy',
1400
 
                                         policy)
 
3615
        wt.branch.get_config_stack().set('bzr.transform.orphan_policy',
 
3616
                                               policy)
1401
3617
 
1402
3618
    def _prepare_orphan(self, wt):
1403
3619
        self.build_tree(['dir/', 'dir/file', 'dir/foo'])
1404
 
        wt.add(['dir', 'dir/file'], [b'dir-id', b'file-id'])
 
3620
        wt.add(['dir', 'dir/file'], ['dir-id', 'file-id'])
1405
3621
        wt.commit('add dir and file ignoring foo')
1406
 
        tt = wt.transform()
 
3622
        tt = transform.TreeTransform(wt)
1407
3623
        self.addCleanup(tt.finalize)
1408
3624
        # dir and bar are deleted
1409
3625
        dir_tid = tt.trans_id_tree_path('dir')
1424
3640
        self._set_orphan_policy(wt, 'move')
1425
3641
        tt, orphan_tid = self._prepare_orphan(wt)
1426
3642
        warnings = []
1427
 
 
1428
3643
        def warning(*args):
1429
3644
            warnings.append(args[0] % args[1:])
1430
3645
        self.overrideAttr(trace, 'warning', warning)
1431
3646
        remaining_conflicts = resolve_conflicts(tt)
1432
3647
        self.assertEqual(['dir/foo has been orphaned in brz-orphans'],
1433
 
                         warnings)
 
3648
                          warnings)
1434
3649
        # Yeah for resolved conflicts !
1435
3650
        self.assertLength(0, remaining_conflicts)
1436
3651
        # We have a new orphan
1437
3652
        self.assertEqual('foo.~1~', tt.final_name(orphan_tid))
1438
3653
        self.assertEqual('brz-orphans',
1439
 
                         tt.final_name(tt.final_parent(orphan_tid)))
 
3654
                          tt.final_name(tt.final_parent(orphan_tid)))
1440
3655
 
1441
3656
    def test_never_orphan(self):
1442
3657
        wt = self.make_branch_and_tree('.')
1467
3682
        self._set_orphan_policy(wt, 'donttouchmypreciouuus')
1468
3683
        tt, orphan_tid = self._prepare_orphan(wt)
1469
3684
        warnings = []
1470
 
 
1471
3685
        def warning(*args):
1472
3686
            warnings.append(args[0] % args[1:])
1473
3687
        self.overrideAttr(trace, 'warning', warning)
1487
3701
        self.wt = self.make_branch_and_tree('.')
1488
3702
        os.chdir('..')
1489
3703
 
1490
 
    def transform(self):
1491
 
        transform = self.wt.transform()
 
3704
    def get_transform(self):
 
3705
        transform = TreeTransform(self.wt)
1492
3706
        self.addCleanup(transform.finalize)
1493
3707
        return transform, transform.root
1494
3708
 
1495
3709
    def test_pre_commit_hooks(self):
1496
3710
        calls = []
1497
 
 
1498
3711
        def record_pre_transform(tree, tt):
1499
3712
            calls.append((tree, tt))
1500
 
        MutableTree.hooks.install_named_hook(
1501
 
            'pre_transform', record_pre_transform, "Pre transform")
1502
 
        transform, root = self.transform()
 
3713
        MutableTree.hooks.install_named_hook('pre_transform',
 
3714
            record_pre_transform, "Pre transform")
 
3715
        transform, root = self.get_transform()
1503
3716
        old_root_id = transform.tree_file_id(root)
1504
3717
        transform.apply()
1505
 
        self.assertEqual(old_root_id, self.wt.path2id(''))
 
3718
        self.assertEqual(old_root_id, self.wt.get_root_id())
1506
3719
        self.assertEqual([(self.wt, transform)], calls)
1507
3720
 
1508
3721
    def test_post_commit_hooks(self):
1509
3722
        calls = []
1510
 
 
1511
3723
        def record_post_transform(tree, tt):
1512
3724
            calls.append((tree, tt))
1513
 
        MutableTree.hooks.install_named_hook(
1514
 
            'post_transform', record_post_transform, "Post transform")
1515
 
        transform, root = self.transform()
 
3725
        MutableTree.hooks.install_named_hook('post_transform',
 
3726
            record_post_transform, "Post transform")
 
3727
        transform, root = self.get_transform()
1516
3728
        old_root_id = transform.tree_file_id(root)
1517
3729
        transform.apply()
1518
 
        self.assertEqual(old_root_id, self.wt.path2id(''))
 
3730
        self.assertEqual(old_root_id, self.wt.get_root_id())
1519
3731
        self.assertEqual([(self.wt, transform)], calls)
1520
3732
 
1521
3733
 
1528
3740
        self.parent_tree = self.make_branch_and_tree('parent')
1529
3741
        self.parent_tree.lock_write()
1530
3742
        self.addCleanup(self.parent_tree.unlock)
1531
 
        self.build_tree_contents([('parent/foo', b'bar')])
1532
 
        self.parent_tree.add('foo')
 
3743
        self.build_tree_contents([('parent/foo', 'bar')])
 
3744
        self.parent_tree.add('foo', 'foo-id')
1533
3745
        self.parent_tree.commit('added foo')
1534
3746
        child_controldir = self.parent_tree.controldir.sprout('child')
1535
3747
        self.child_tree = child_controldir.open_workingtree()
1541
3753
 
1542
3754
    def test_link_fails_if_modified(self):
1543
3755
        """If the file to be linked has modified text, don't link."""
1544
 
        self.build_tree_contents([('child/foo', b'baz')])
 
3756
        self.build_tree_contents([('child/foo', 'baz')])
1545
3757
        transform.link_tree(self.child_tree, self.parent_tree)
1546
3758
        self.assertFalse(self.hardlinked())
1547
3759
 
1548
3760
    def test_link_fails_if_execute_bit_changed(self):
1549
3761
        """If the file to be linked has modified execute bit, don't link."""
1550
 
        tt = self.child_tree.transform()
 
3762
        tt = TreeTransform(self.child_tree)
1551
3763
        try:
1552
 
            trans_id = tt.trans_id_tree_path('foo')
 
3764
            trans_id = tt.trans_id_tree_file_id('foo-id')
1553
3765
            tt.set_executability(True, trans_id)
1554
3766
            tt.apply()
1555
3767
        finally:
1561
3773
        """If the file to be linked is unmodified, link"""
1562
3774
        transform.link_tree(self.child_tree, self.parent_tree)
1563
3775
        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))