/brz/remove-bazaar

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

« back to all changes in this revision

Viewing changes to breezy/tests/test_transform.py

  • Committer: Jelmer Vernooij
  • Date: 2017-07-23 22:06:41 UTC
  • mfrom: (6738 trunk)
  • mto: This revision was merged to the branch mainline in revision 6739.
  • Revision ID: jelmer@jelmer.uk-20170723220641-69eczax9bmv8d6kk
Merge trunk, address review comments.

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