/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 bzrlib/tests/test_transform.py

  • Committer: Robert Collins
  • Date: 2010-05-11 08:44:59 UTC
  • mfrom: (5221 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100511084459-pb0uinna9zs3wu59
Merge trunk - resolve conflicts.

Show diffs side-by-side

added added

removed removed

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