/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: Marius Kruger
  • Date: 2010-07-10 21:28:56 UTC
  • mto: (5384.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 5385.
  • Revision ID: marius.kruger@enerweb.co.za-20100710212856-uq4ji3go0u5se7hx
* Update documentation
* add NEWS

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
855
    def test_rename_fails(self):
1026
 
        self.requireFeature(features.not_running_as_root)
1027
856
        # see https://bugs.launchpad.net/bzr/+bug/491763
1028
857
        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')
 
858
        first_dir = create.new_directory('first-dir', root_id, 'first-id')
 
859
        myfile = create.new_file('myfile', root_id, 'myfile-text',
 
860
                                 'myfile-id')
1031
861
        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")
 
862
        # make the file and directory readonly in the hope this will prevent
 
863
        # renames
 
864
        osutils.make_readonly(self.wt.abspath('first-dir'))
 
865
        osutils.make_readonly(self.wt.abspath('myfile'))
1040
866
        # now transform to rename
1041
867
        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')
 
868
        file_trans_id = rename_transform.trans_id_file_id('myfile-id')
 
869
        dir_id = rename_transform.trans_id_file_id('first-id')
1044
870
        rename_transform.adjust_path('newname', dir_id, file_trans_id)
1045
871
        e = self.assertRaises(errors.TransformRenameFailed,
1046
 
                              rename_transform.apply)
1047
 
        # On nix looks like:
 
872
            rename_transform.apply)
 
873
        # Looks like: 
1048
874
        # "Failed to rename .../work/.bzr/checkout/limbo/new-1
1049
875
        # 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")
 
876
        # so the first filename is not visible in it; we expect a strerror but
 
877
        # it may vary per OS and language so it's not checked here
 
878
        self.assertContainsRe(str(e),
 
879
            "Failed to rename .*first-dir.newname:")
1061
880
 
1062
881
    def test_set_executability_order(self):
1063
882
        """Ensure that executability behaves the same, no matter what order.
1064
883
 
1065
884
        - create file and set executability simultaneously
1066
885
        - create file and set executability afterward
1067
 
        - unsetting the executability of a file whose executability has not
1068
 
          been
 
886
        - unsetting the executability of a file whose executability has not been
1069
887
        declared should throw an exception (this may happen when a
1070
888
        merge attempts to create a file with a duplicate ID)
1071
889
        """
1073
891
        wt = transform._tree
1074
892
        wt.lock_read()
1075
893
        self.addCleanup(wt.unlock)
1076
 
        transform.new_file('set_on_creation', root, [b'Set on creation'],
1077
 
                           b'soc', True)
 
894
        transform.new_file('set_on_creation', root, 'Set on creation', 'soc',
 
895
                           True)
1078
896
        sac = transform.new_file('set_after_creation', root,
1079
 
                                 [b'Set after creation'], b'sac')
 
897
                                 'Set after creation', 'sac')
1080
898
        transform.set_executability(True, sac)
1081
 
        uws = transform.new_file('unset_without_set', root, [b'Unset badly'],
1082
 
                                 b'uws')
 
899
        uws = transform.new_file('unset_without_set', root, 'Unset badly',
 
900
                                 'uws')
1083
901
        self.assertRaises(KeyError, transform.set_executability, None, uws)
1084
902
        transform.apply()
1085
 
        self.assertTrue(wt.is_executable('set_on_creation'))
1086
 
        self.assertTrue(wt.is_executable('set_after_creation'))
 
903
        self.assertTrue(wt.is_executable('soc'))
 
904
        self.assertTrue(wt.is_executable('sac'))
1087
905
 
1088
906
    def test_preserve_mode(self):
1089
907
        """File mode is preserved when replacing content"""
1090
908
        if sys.platform == 'win32':
1091
909
            raise TestSkipped('chmod has no effect on win32')
1092
910
        transform, root = self.get_transform()
1093
 
        transform.new_file('file1', root, [b'contents'], b'file1-id', True)
 
911
        transform.new_file('file1', root, 'contents', 'file1-id', True)
1094
912
        transform.apply()
1095
913
        self.wt.lock_write()
1096
914
        self.addCleanup(self.wt.unlock)
1097
 
        self.assertTrue(self.wt.is_executable('file1'))
 
915
        self.assertTrue(self.wt.is_executable('file1-id'))
1098
916
        transform, root = self.get_transform()
1099
 
        file1_id = transform.trans_id_tree_path('file1')
 
917
        file1_id = transform.trans_id_tree_file_id('file1-id')
1100
918
        transform.delete_contents(file1_id)
1101
 
        transform.create_file([b'contents2'], file1_id)
 
919
        transform.create_file('contents2', file1_id)
1102
920
        transform.apply()
1103
 
        self.assertTrue(self.wt.is_executable('file1'))
 
921
        self.assertTrue(self.wt.is_executable('file1-id'))
1104
922
 
1105
923
    def test__set_mode_stats_correctly(self):
1106
924
        """_set_mode stats to determine file mode."""
1109
927
 
1110
928
        stat_paths = []
1111
929
        real_stat = os.stat
1112
 
 
1113
930
        def instrumented_stat(path):
1114
931
            stat_paths.append(path)
1115
932
            return real_stat(path)
1116
933
 
1117
934
        transform, root = self.get_transform()
1118
935
 
1119
 
        bar1_id = transform.new_file('bar', root, [b'bar contents 1\n'],
1120
 
                                     file_id=b'bar-id-1', executable=False)
 
936
        bar1_id = transform.new_file('bar', root, 'bar contents 1\n',
 
937
                                     file_id='bar-id-1', executable=False)
1121
938
        transform.apply()
1122
939
 
1123
940
        transform, root = self.get_transform()
1125
942
        bar2_id = transform.trans_id_tree_path('bar2')
1126
943
        try:
1127
944
            os.stat = instrumented_stat
1128
 
            transform.create_file([b'bar2 contents\n'],
1129
 
                                  bar2_id, mode_id=bar1_id)
 
945
            transform.create_file('bar2 contents\n', bar2_id, mode_id=bar1_id)
1130
946
        finally:
1131
947
            os.stat = real_stat
1132
948
            transform.finalize()
1135
951
        self.assertEqual([bar1_abspath], stat_paths)
1136
952
 
1137
953
    def test_iter_changes(self):
1138
 
        self.wt.set_root_id(b'eert_toor')
 
954
        self.wt.set_root_id('eert_toor')
1139
955
        transform, root = self.get_transform()
1140
 
        transform.new_file('old', root, [b'blah'], b'id-1', True)
 
956
        transform.new_file('old', root, 'blah', 'id-1', True)
1141
957
        transform.apply()
1142
958
        transform, root = self.get_transform()
1143
959
        try:
1144
960
            self.assertEqual([], list(transform.iter_changes()))
1145
 
            old = transform.trans_id_tree_path('old')
 
961
            old = transform.trans_id_tree_file_id('id-1')
1146
962
            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()))
 
963
            self.assertEqual([('id-1', ('old', None), False, (True, False),
 
964
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
965
                (True, True))], list(transform.iter_changes()))
 
966
            transform.new_directory('new', root, 'id-1')
 
967
            self.assertEqual([('id-1', ('old', 'new'), True, (True, True),
 
968
                ('eert_toor', 'eert_toor'), ('old', 'new'),
 
969
                ('file', 'directory'),
 
970
                (True, False))], list(transform.iter_changes()))
1158
971
        finally:
1159
972
            transform.finalize()
1160
973
 
1161
974
    def test_iter_changes_new(self):
1162
 
        self.wt.set_root_id(b'eert_toor')
 
975
        self.wt.set_root_id('eert_toor')
1163
976
        transform, root = self.get_transform()
1164
 
        transform.new_file('old', root, [b'blah'])
 
977
        transform.new_file('old', root, 'blah')
1165
978
        transform.apply()
1166
979
        transform, root = self.get_transform()
1167
980
        try:
1168
981
            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()))
 
982
            transform.version_file('id-1', old)
 
983
            self.assertEqual([('id-1', (None, 'old'), False, (False, True),
 
984
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
985
                (False, False))], list(transform.iter_changes()))
1175
986
        finally:
1176
987
            transform.finalize()
1177
988
 
1178
989
    def test_iter_changes_modifications(self):
1179
 
        self.wt.set_root_id(b'eert_toor')
 
990
        self.wt.set_root_id('eert_toor')
1180
991
        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')
 
992
        transform.new_file('old', root, 'blah', 'id-1')
 
993
        transform.new_file('new', root, 'blah')
 
994
        transform.new_directory('subdir', root, 'subdir-id')
1184
995
        transform.apply()
1185
996
        transform, root = self.get_transform()
1186
997
        try:
1187
998
            old = transform.trans_id_tree_path('old')
1188
 
            subdir = transform.trans_id_tree_path('subdir')
 
999
            subdir = transform.trans_id_tree_file_id('subdir-id')
1189
1000
            new = transform.trans_id_tree_path('new')
1190
1001
            self.assertEqual([], list(transform.iter_changes()))
1191
1002
 
1192
 
            # content deletion
 
1003
            #content deletion
1193
1004
            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()))
 
1005
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
 
1006
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', None),
 
1007
                (False, False))], list(transform.iter_changes()))
1199
1008
 
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()))
 
1009
            #content change
 
1010
            transform.create_file('blah', old)
 
1011
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
 
1012
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
1013
                (False, False))], list(transform.iter_changes()))
1207
1014
            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()))
 
1015
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
 
1016
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
1017
                (False, False))], list(transform.iter_changes()))
1213
1018
            transform.cancel_creation(old)
1214
1019
 
1215
1020
            # move file_id to a different file
1216
1021
            self.assertEqual([], list(transform.iter_changes()))
1217
1022
            transform.unversion_file(old)
1218
 
            transform.version_file(b'id-1', new)
 
1023
            transform.version_file('id-1', new)
1219
1024
            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()))
 
1025
            self.assertEqual([('id-1', ('old', 'old'), True, (True, True),
 
1026
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
1027
                (False, False))], list(transform.iter_changes()))
1225
1028
            transform.cancel_versioning(new)
1226
1029
            transform._removed_id = set()
1227
1030
 
1228
 
            # execute bit
 
1031
            #execute bit
1229
1032
            self.assertEqual([], list(transform.iter_changes()))
1230
1033
            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()))
 
1034
            self.assertEqual([('id-1', ('old', 'old'), False, (True, True),
 
1035
                ('eert_toor', 'eert_toor'), ('old', 'old'), ('file', 'file'),
 
1036
                (False, True))], list(transform.iter_changes()))
1236
1037
            transform.set_executability(None, old)
1237
1038
 
1238
1039
            # filename
1239
1040
            self.assertEqual([], list(transform.iter_changes()))
1240
1041
            transform.adjust_path('new', root, old)
1241
1042
            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()))
 
1043
            self.assertEqual([('id-1', ('old', 'new'), False, (True, True),
 
1044
                ('eert_toor', 'eert_toor'), ('old', 'new'), ('file', 'file'),
 
1045
                (False, False))], list(transform.iter_changes()))
1247
1046
            transform._new_name = {}
1248
1047
 
1249
1048
            # parent directory
1250
1049
            self.assertEqual([], list(transform.iter_changes()))
1251
1050
            transform.adjust_path('new', subdir, old)
1252
1051
            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()))
 
1052
            self.assertEqual([('id-1', ('old', 'subdir/old'), False,
 
1053
                (True, True), ('eert_toor', 'subdir-id'), ('old', 'old'),
 
1054
                ('file', 'file'), (False, False))],
 
1055
                list(transform.iter_changes()))
1258
1056
            transform._new_path = {}
1259
1057
 
1260
1058
        finally:
1261
1059
            transform.finalize()
1262
1060
 
1263
1061
    def test_iter_changes_modified_bleed(self):
1264
 
        self.wt.set_root_id(b'eert_toor')
 
1062
        self.wt.set_root_id('eert_toor')
1265
1063
        """Modified flag should not bleed from one change to another"""
1266
1064
        # unfortunately, we have no guarantee that file1 (which is modified)
1267
1065
        # will be applied before file2.  And if it's applied after file2, it
1268
1066
        # obviously can't bleed into file2's change output.  But for now, it
1269
1067
        # works.
1270
1068
        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')
 
1069
        transform.new_file('file1', root, 'blah', 'id-1')
 
1070
        transform.new_file('file2', root, 'blah', 'id-2')
1273
1071
        transform.apply()
1274
1072
        transform, root = self.get_transform()
1275
1073
        try:
1276
 
            transform.delete_contents(transform.trans_id_file_id(b'id-1'))
 
1074
            transform.delete_contents(transform.trans_id_file_id('id-1'))
1277
1075
            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)],
 
1076
            transform.trans_id_file_id('id-2'))
 
1077
            self.assertEqual([('id-1', (u'file1', u'file1'), True, (True, True),
 
1078
                ('eert_toor', 'eert_toor'), ('file1', u'file1'),
 
1079
                ('file', None), (False, False)),
 
1080
                ('id-2', (u'file2', u'file2'), False, (True, True),
 
1081
                ('eert_toor', 'eert_toor'), ('file2', u'file2'),
 
1082
                ('file', 'file'), (False, True))],
1286
1083
                list(transform.iter_changes()))
1287
1084
        finally:
1288
1085
            transform.finalize()
1289
1086
 
1290
1087
    def test_iter_changes_move_missing(self):
1291
1088
        """Test moving ids with no files around"""
1292
 
        self.wt.set_root_id(b'toor_eert')
 
1089
        self.wt.set_root_id('toor_eert')
1293
1090
        # Need two steps because versioning a non-existant file is a conflict.
1294
1091
        transform, root = self.get_transform()
1295
 
        transform.new_directory('floater', root, b'floater-id')
 
1092
        transform.new_directory('floater', root, 'floater-id')
1296
1093
        transform.apply()
1297
1094
        transform, root = self.get_transform()
1298
1095
        transform.delete_contents(transform.trans_id_tree_path('floater'))
1301
1098
        floater = transform.trans_id_tree_path('floater')
1302
1099
        try:
1303
1100
            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()))
 
1101
            self.assertEqual([('floater-id', ('floater', 'flitter'), False,
 
1102
            (True, True), ('toor_eert', 'toor_eert'), ('floater', 'flitter'),
 
1103
            (None, None), (False, False))], list(transform.iter_changes()))
1310
1104
        finally:
1311
1105
            transform.finalize()
1312
1106
 
1313
1107
    def test_iter_changes_pointless(self):
1314
1108
        """Ensure that no-ops are not treated as modifications"""
1315
 
        self.wt.set_root_id(b'eert_toor')
 
1109
        self.wt.set_root_id('eert_toor')
1316
1110
        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')
 
1111
        transform.new_file('old', root, 'blah', 'id-1')
 
1112
        transform.new_directory('subdir', root, 'subdir-id')
1319
1113
        transform.apply()
1320
1114
        transform, root = self.get_transform()
1321
1115
        try:
1322
1116
            old = transform.trans_id_tree_path('old')
1323
 
            subdir = transform.trans_id_tree_path('subdir')
 
1117
            subdir = transform.trans_id_tree_file_id('subdir-id')
1324
1118
            self.assertEqual([], list(transform.iter_changes()))
1325
1119
            transform.delete_contents(subdir)
1326
1120
            transform.create_directory(subdir)
1327
1121
            transform.set_executability(False, old)
1328
1122
            transform.unversion_file(old)
1329
 
            transform.version_file(b'id-1', old)
 
1123
            transform.version_file('id-1', old)
1330
1124
            transform.adjust_path('old', root, old)
1331
1125
            self.assertEqual([], list(transform.iter_changes()))
1332
1126
        finally:
1334
1128
 
1335
1129
    def test_rename_count(self):
1336
1130
        transform, root = self.get_transform()
1337
 
        transform.new_file('name1', root, [b'contents'])
 
1131
        transform.new_file('name1', root, 'contents')
1338
1132
        self.assertEqual(transform.rename_count, 0)
1339
1133
        transform.apply()
1340
1134
        self.assertEqual(transform.rename_count, 1)
1357
1151
        """
1358
1152
        transform, root = self.get_transform()
1359
1153
        parent1 = transform.new_directory('parent1', root)
1360
 
        child1 = transform.new_file('child1', parent1, [b'contents'])
 
1154
        child1 = transform.new_file('child1', parent1, 'contents')
1361
1155
        parent2 = transform.new_directory('parent2', root)
1362
1156
        transform.adjust_path('child1', parent2, child1)
1363
1157
        transform.apply()
1364
 
        self.assertPathDoesNotExist(self.wt.abspath('parent1/child1'))
1365
 
        self.assertPathExists(self.wt.abspath('parent2/child1'))
 
1158
        self.failIfExists(self.wt.abspath('parent1/child1'))
 
1159
        self.failUnlessExists(self.wt.abspath('parent2/child1'))
1366
1160
        # rename limbo/new-1 => parent1, rename limbo/new-3 => parent2
1367
1161
        # no rename for child1 (counting only renames during apply)
1368
 
        self.assertEqual(2, transform.rename_count)
 
1162
        self.failUnlessEqual(2, transform.rename_count)
1369
1163
 
1370
1164
    def test_cancel_parent(self):
1371
1165
        """Cancelling a parent doesn't cause deletion of a non-empty directory
1376
1170
        """
1377
1171
        transform, root = self.get_transform()
1378
1172
        parent1 = transform.new_directory('parent1', root)
1379
 
        child1 = transform.new_file('child1', parent1, [b'contents'])
1380
 
        child2 = transform.new_file('child2', parent1, [b'contents'])
 
1173
        child1 = transform.new_file('child1', parent1, 'contents')
 
1174
        child2 = transform.new_file('child2', parent1, 'contents')
1381
1175
        try:
1382
1176
            transform.cancel_creation(parent1)
1383
1177
        except OSError:
1394
1188
        parent2 = transform.new_directory('parent2', root)
1395
1189
        transform.adjust_path('child1', parent2, child1)
1396
1190
        transform.apply()
1397
 
        self.assertPathDoesNotExist(self.wt.abspath('parent1'))
1398
 
        self.assertPathExists(self.wt.abspath('parent2/child1'))
 
1191
        self.failIfExists(self.wt.abspath('parent1'))
 
1192
        self.failUnlessExists(self.wt.abspath('parent2/child1'))
1399
1193
        # rename limbo/new-3 => parent2, rename limbo/new-2 => child1
1400
 
        self.assertEqual(2, transform.rename_count)
 
1194
        self.failUnlessEqual(2, transform.rename_count)
1401
1195
 
1402
1196
    def test_adjust_and_cancel(self):
1403
1197
        """Make sure adjust_path keeps track of limbo children properly"""
1404
1198
        transform, root = self.get_transform()
1405
1199
        parent1 = transform.new_directory('parent1', root)
1406
 
        child1 = transform.new_file('child1', parent1, [b'contents'])
 
1200
        child1 = transform.new_file('child1', parent1, 'contents')
1407
1201
        parent2 = transform.new_directory('parent2', root)
1408
1202
        transform.adjust_path('child1', parent2, child1)
1409
1203
        transform.cancel_creation(child1)
1418
1212
    def test_noname_contents(self):
1419
1213
        """TreeTransform should permit deferring naming files."""
1420
1214
        transform, root = self.get_transform()
1421
 
        parent = transform.trans_id_file_id(b'parent-id')
 
1215
        parent = transform.trans_id_file_id('parent-id')
1422
1216
        try:
1423
1217
            transform.create_directory(parent)
1424
1218
        except KeyError:
1428
1222
    def test_noname_contents_nested(self):
1429
1223
        """TreeTransform should permit deferring naming files."""
1430
1224
        transform, root = self.get_transform()
1431
 
        parent = transform.trans_id_file_id(b'parent-id')
 
1225
        parent = transform.trans_id_file_id('parent-id')
1432
1226
        try:
1433
1227
            transform.create_directory(parent)
1434
1228
        except KeyError:
1435
1229
            self.fail("Can't handle contents with no name")
1436
 
        transform.new_directory('child', parent)
 
1230
        child = transform.new_directory('child', parent)
1437
1231
        transform.adjust_path('parent', root, parent)
1438
1232
        transform.apply()
1439
 
        self.assertPathExists(self.wt.abspath('parent/child'))
 
1233
        self.failUnlessExists(self.wt.abspath('parent/child'))
1440
1234
        self.assertEqual(1, transform.rename_count)
1441
1235
 
1442
1236
    def test_reuse_name(self):
1443
1237
        """Avoid reusing the same limbo name for different files"""
1444
1238
        transform, root = self.get_transform()
1445
1239
        parent = transform.new_directory('parent', root)
1446
 
        transform.new_directory('child', parent)
 
1240
        child1 = transform.new_directory('child', parent)
1447
1241
        try:
1448
1242
            child2 = transform.new_directory('child', parent)
1449
1243
        except OSError:
1463
1257
        parent = transform.new_directory('parent', root)
1464
1258
        child1 = transform.new_directory('child', parent)
1465
1259
        transform.adjust_path('child1', parent, child1)
1466
 
        transform.new_directory('child', parent)
 
1260
        child2 = transform.new_directory('child', parent)
1467
1261
        transform.apply()
1468
1262
        # limbo/new-1 => parent
1469
1263
        self.assertEqual(1, transform.rename_count)
1475
1269
        child1 = transform.new_directory('child1', parent2)
1476
1270
        transform.cancel_creation(parent2)
1477
1271
        transform.create_directory(parent2)
1478
 
        transform.new_directory('child1', parent2)
 
1272
        child2 = transform.new_directory('child1', parent2)
1479
1273
        transform.adjust_path('child2', parent2, child1)
1480
1274
        transform.apply()
1481
1275
        # limbo/new-1 => parent2, limbo/new-2 => parent2/child1
1485
1279
        """Finalize must be done in child-to-parent order"""
1486
1280
        transform, root = self.get_transform()
1487
1281
        parent = transform.new_directory('parent', root)
1488
 
        transform.new_directory('child', parent)
 
1282
        child = transform.new_directory('child', parent)
1489
1283
        try:
1490
1284
            transform.finalize()
1491
1285
        except OSError:
1505
1299
            tt = TreeTransform(wt)  # TreeTransform obtains write lock
1506
1300
            try:
1507
1301
                foo = tt.new_directory('foo', tt.root)
1508
 
                tt.new_file('bar', foo, [b'foobar'])
 
1302
                tt.new_file('bar', foo, 'foobar')
1509
1303
                baz = tt.new_directory('baz', tt.root)
1510
 
                tt.new_file('qux', baz, [b'quux'])
 
1304
                tt.new_file('qux', baz, 'quux')
1511
1305
                # Ask for a rename 'foo' -> 'baz'
1512
1306
                tt.adjust_path('baz', tt.root, foo)
1513
1307
                # Lie to tt that we've already resolved all conflicts.
1514
1308
                tt.apply(no_conflicts=True)
1515
 
            except BaseException:
 
1309
            except:
1516
1310
                wt.unlock()
1517
1311
                raise
1518
1312
        # The rename will fail because the target directory is not empty (but
1519
1313
        # raises FileExists anyway).
1520
1314
        err = self.assertRaises(errors.FileExists, tt_helper)
1521
 
        self.assertEndsWith(err.path, "/baz")
 
1315
        self.assertContainsRe(str(err),
 
1316
            "^File exists: .+/baz")
1522
1317
 
1523
1318
    def test_two_directories_clash(self):
1524
1319
        def tt_helper():
1532
1327
                tt.new_directory('baz', foo_2)
1533
1328
                # Lie to tt that we've already resolved all conflicts.
1534
1329
                tt.apply(no_conflicts=True)
1535
 
            except BaseException:
 
1330
            except:
1536
1331
                wt.unlock()
1537
1332
                raise
1538
1333
        err = self.assertRaises(errors.FileExists, tt_helper)
1539
 
        self.assertEndsWith(err.path, "/foo")
 
1334
        self.assertContainsRe(str(err),
 
1335
            "^File exists: .+/foo")
1540
1336
 
1541
1337
    def test_two_directories_clash_finalize(self):
1542
1338
        def tt_helper():
1550
1346
                tt.new_directory('baz', foo_2)
1551
1347
                # Lie to tt that we've already resolved all conflicts.
1552
1348
                tt.apply(no_conflicts=True)
1553
 
            except BaseException:
 
1349
            except:
1554
1350
                tt.finalize()
1555
1351
                raise
1556
1352
        err = self.assertRaises(errors.FileExists, tt_helper)
1557
 
        self.assertEndsWith(err.path, "/foo")
 
1353
        self.assertContainsRe(str(err),
 
1354
            "^File exists: .+/foo")
1558
1355
 
1559
1356
    def test_file_to_directory(self):
1560
1357
        wt = self.make_branch_and_tree('.')
1567
1364
        tt.delete_contents(foo_trans_id)
1568
1365
        tt.create_directory(foo_trans_id)
1569
1366
        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)
 
1367
        tt.create_file(["aa\n"], bar_trans_id)
 
1368
        tt.version_file("bar-1", bar_trans_id)
1572
1369
        tt.apply()
1573
 
        self.assertPathExists("foo/bar")
 
1370
        self.failUnlessExists("foo/bar")
1574
1371
        wt.lock_read()
1575
1372
        try:
1576
 
            self.assertEqual(wt.kind("foo"), "directory")
 
1373
            self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
 
1374
                    "directory")
1577
1375
        finally:
1578
1376
            wt.unlock()
1579
1377
        wt.commit("two")
1592
1390
        tt.delete_contents(foo_trans_id)
1593
1391
        tt.create_symlink("bar", foo_trans_id)
1594
1392
        tt.apply()
1595
 
        self.assertPathExists("foo")
 
1393
        self.failUnlessExists("foo")
1596
1394
        wt.lock_read()
1597
1395
        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')
 
1396
        self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
 
1397
                "symlink")
1617
1398
 
1618
1399
    def test_dir_to_file(self):
1619
1400
        wt = self.make_branch_and_tree('.')
1626
1407
        bar_trans_id = tt.trans_id_tree_path("foo/bar")
1627
1408
        tt.delete_contents(foo_trans_id)
1628
1409
        tt.delete_versioned(bar_trans_id)
1629
 
        tt.create_file([b"aa\n"], foo_trans_id)
 
1410
        tt.create_file(["aa\n"], foo_trans_id)
1630
1411
        tt.apply()
1631
 
        self.assertPathExists("foo")
 
1412
        self.failUnlessExists("foo")
1632
1413
        wt.lock_read()
1633
1414
        self.addCleanup(wt.unlock)
1634
 
        self.assertEqual(wt.kind("foo"), "file")
 
1415
        self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
 
1416
                "file")
1635
1417
 
1636
1418
    def test_dir_to_hardlink(self):
1637
1419
        self.requireFeature(HardlinkFeature)
1648
1430
        self.build_tree(['baz'])
1649
1431
        tt.create_hardlink("baz", foo_trans_id)
1650
1432
        tt.apply()
1651
 
        self.assertPathExists("foo")
1652
 
        self.assertPathExists("baz")
 
1433
        self.failUnlessExists("foo")
 
1434
        self.failUnlessExists("baz")
1653
1435
        wt.lock_read()
1654
1436
        self.addCleanup(wt.unlock)
1655
 
        self.assertEqual(wt.kind("foo"), "file")
 
1437
        self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
 
1438
                "file")
1656
1439
 
1657
1440
    def test_no_final_path(self):
1658
1441
        transform, root = self.get_transform()
1659
 
        trans_id = transform.trans_id_file_id(b'foo')
1660
 
        transform.create_file([b'bar'], trans_id)
 
1442
        trans_id = transform.trans_id_file_id('foo')
 
1443
        transform.create_file('bar', trans_id)
1661
1444
        transform.cancel_creation(trans_id)
1662
1445
        transform.apply()
1663
1446
 
1664
1447
    def test_create_from_tree(self):
1665
1448
        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'])
 
1449
        self.build_tree_contents([('tree1/foo/',), ('tree1/bar', 'baz')])
 
1450
        tree1.add(['foo', 'bar'], ['foo-id', 'bar-id'])
1668
1451
        tree2 = self.make_branch_and_tree('tree2')
1669
1452
        tt = TreeTransform(tree2)
1670
1453
        foo_trans_id = tt.create_path('foo', tt.root)
1671
 
        create_from_tree(tt, foo_trans_id, tree1, 'foo')
 
1454
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id')
1672
1455
        bar_trans_id = tt.create_path('bar', tt.root)
1673
 
        create_from_tree(tt, bar_trans_id, tree1, 'bar')
 
1456
        create_from_tree(tt, bar_trans_id, tree1, 'bar-id')
1674
1457
        tt.apply()
1675
1458
        self.assertEqual('directory', osutils.file_kind('tree2/foo'))
1676
 
        self.assertFileEqual(b'baz', 'tree2/bar')
 
1459
        self.assertFileEqual('baz', 'tree2/bar')
1677
1460
 
1678
1461
    def test_create_from_tree_bytes(self):
1679
1462
        """Provided lines are used instead of tree content."""
1680
1463
        tree1 = self.make_branch_and_tree('tree1')
1681
 
        self.build_tree_contents([('tree1/foo', b'bar'), ])
1682
 
        tree1.add('foo', b'foo-id')
 
1464
        self.build_tree_contents([('tree1/foo', 'bar'),])
 
1465
        tree1.add('foo', 'foo-id')
1683
1466
        tree2 = self.make_branch_and_tree('tree2')
1684
1467
        tt = TreeTransform(tree2)
1685
1468
        foo_trans_id = tt.create_path('foo', tt.root)
1686
 
        create_from_tree(tt, foo_trans_id, tree1, 'foo', chunks=[b'qux'])
 
1469
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id', bytes='qux')
1687
1470
        tt.apply()
1688
 
        self.assertFileEqual(b'qux', 'tree2/foo')
 
1471
        self.assertFileEqual('qux', 'tree2/foo')
1689
1472
 
1690
1473
    def test_create_from_tree_symlink(self):
1691
1474
        self.requireFeature(SymlinkFeature)
1692
1475
        tree1 = self.make_branch_and_tree('tree1')
1693
1476
        os.symlink('bar', 'tree1/foo')
1694
 
        tree1.add('foo', b'foo-id')
 
1477
        tree1.add('foo', 'foo-id')
1695
1478
        tt = TreeTransform(self.make_branch_and_tree('tree2'))
1696
1479
        foo_trans_id = tt.create_path('foo', tt.root)
1697
 
        create_from_tree(tt, foo_trans_id, tree1, 'foo')
 
1480
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id')
1698
1481
        tt.apply()
1699
1482
        self.assertEqual('bar', os.readlink('tree2/foo'))
1700
1483
 
1704
1487
    def __init__(self, dirname, root_id):
1705
1488
        self.name = dirname
1706
1489
        os.mkdir(dirname)
1707
 
        self.wt = ControlDir.create_standalone_workingtree(dirname)
 
1490
        self.wt = BzrDir.create_standalone_workingtree(dirname)
1708
1491
        self.wt.set_root_id(root_id)
1709
1492
        self.b = self.wt.branch
1710
1493
        self.tt = TreeTransform(self.wt)
1711
 
        self.root = self.tt.trans_id_tree_path('')
 
1494
        self.root = self.tt.trans_id_tree_file_id(self.wt.get_root_id())
1712
1495
 
1713
1496
 
1714
1497
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())
 
1498
    template = '%s TREE\n%s%s\n%s%s MERGE-SOURCE\n'
 
1499
    return template % ('<' * 7, tree, '=' * 7, merge, '>' * 7)
1749
1500
 
1750
1501
 
1751
1502
class TestTransformMerge(TestCaseInTempDir):
1753
1504
    def test_text_merge(self):
1754
1505
        root_id = generate_ids.gen_root_id()
1755
1506
        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')
 
1507
        base.tt.new_file('a', base.root, 'a\nb\nc\nd\be\n', 'a')
 
1508
        base.tt.new_file('b', base.root, 'b1', 'b')
 
1509
        base.tt.new_file('c', base.root, 'c', 'c')
 
1510
        base.tt.new_file('d', base.root, 'd', 'd')
 
1511
        base.tt.new_file('e', base.root, 'e', 'e')
 
1512
        base.tt.new_file('f', base.root, 'f', 'f')
 
1513
        base.tt.new_directory('g', base.root, 'g')
 
1514
        base.tt.new_directory('h', base.root, 'h')
1764
1515
        base.tt.apply()
1765
1516
        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')
 
1517
        other.tt.new_file('a', other.root, 'y\nb\nc\nd\be\n', 'a')
 
1518
        other.tt.new_file('b', other.root, 'b2', 'b')
 
1519
        other.tt.new_file('c', other.root, 'c2', 'c')
 
1520
        other.tt.new_file('d', other.root, 'd', 'd')
 
1521
        other.tt.new_file('e', other.root, 'e2', 'e')
 
1522
        other.tt.new_file('f', other.root, 'f', 'f')
 
1523
        other.tt.new_file('g', other.root, 'g', 'g')
 
1524
        other.tt.new_file('h', other.root, 'h\ni\nj\nk\n', 'h')
 
1525
        other.tt.new_file('i', other.root, 'h\ni\nj\nk\n', 'i')
1775
1526
        other.tt.apply()
1776
1527
        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')
 
1528
        this.tt.new_file('a', this.root, 'a\nb\nc\nd\bz\n', 'a')
 
1529
        this.tt.new_file('b', this.root, 'b', 'b')
 
1530
        this.tt.new_file('c', this.root, 'c', 'c')
 
1531
        this.tt.new_file('d', this.root, 'd2', 'd')
 
1532
        this.tt.new_file('e', this.root, 'e2', 'e')
 
1533
        this.tt.new_file('f', this.root, 'f', 'f')
 
1534
        this.tt.new_file('g', this.root, 'g', 'g')
 
1535
        this.tt.new_file('h', this.root, '1\n2\n3\n4\n', 'h')
 
1536
        this.tt.new_file('i', this.root, '1\n2\n3\n4\n', 'i')
1786
1537
        this.tt.apply()
1787
1538
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
1788
1539
 
1789
1540
        # 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')
 
1541
        self.assertEqual(this.wt.get_file('a').read(), 'y\nb\nc\nd\bz\n')
1792
1542
        # 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'))
 
1543
        self.assertEqual(this.wt.get_file('b').read(),
 
1544
                         conflict_text('b', 'b2'))
1795
1545
        # OTHER wins
1796
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'c')).read(), b'c2')
 
1546
        self.assertEqual(this.wt.get_file('c').read(), 'c2')
1797
1547
        # THIS wins
1798
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'd')).read(), b'd2')
 
1548
        self.assertEqual(this.wt.get_file('d').read(), 'd2')
1799
1549
        # Ambigious clean merge
1800
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'e')).read(), b'e2')
 
1550
        self.assertEqual(this.wt.get_file('e').read(), 'e2')
1801
1551
        # No change
1802
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'f')).read(), b'f')
 
1552
        self.assertEqual(this.wt.get_file('f').read(), 'f')
1803
1553
        # Correct correct results when THIS == OTHER
1804
 
        self.assertEqual(this.wt.get_file(this.wt.id2path(b'g')).read(), b'g')
 
1554
        self.assertEqual(this.wt.get_file('g').read(), 'g')
1805
1555
        # 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')
 
1556
        self.assertEqual(this.wt.get_file('h').read(),
 
1557
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
 
1558
        self.assertEqual(this.wt.get_file_byname('h.THIS').read(),
 
1559
                         '1\n2\n3\n4\n')
 
1560
        self.assertEqual(this.wt.get_file_byname('h.OTHER').read(),
 
1561
                         'h\ni\nj\nk\n')
1812
1562
        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')
 
1563
        self.assertEqual(this.wt.get_file('i').read(),
 
1564
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
 
1565
        self.assertEqual(this.wt.get_file_byname('i.THIS').read(),
 
1566
                         '1\n2\n3\n4\n')
 
1567
        self.assertEqual(this.wt.get_file_byname('i.OTHER').read(),
 
1568
                         'h\ni\nj\nk\n')
1819
1569
        self.assertEqual(os.path.exists(this.wt.abspath('i.BASE')), False)
1820
1570
        modified = ['a', 'b', 'c', 'h', 'i']
1821
1571
        merge_modified = this.wt.merge_modified()
1822
1572
        self.assertSubset(merge_modified, modified)
1823
1573
        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')
 
1574
        file(this.wt.id2abspath('a'), 'wb').write('booga')
1826
1575
        modified.pop(0)
1827
1576
        merge_modified = this.wt.merge_modified()
1828
1577
        self.assertSubset(merge_modified, modified)
1837
1586
        this = TransformGroup("THIS", root_id)
1838
1587
        other = TransformGroup("OTHER", root_id)
1839
1588
        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')
 
1589
            tg.tt.new_directory('a', tg.root, 'a')
 
1590
            tg.tt.new_symlink('b', tg.root, 'b', 'b')
 
1591
            tg.tt.new_file('c', tg.root, 'c', 'c')
 
1592
            tg.tt.new_symlink('d', tg.root, tg.name, 'd')
1844
1593
        targets = ((base, 'base-e', 'base-f', None, None),
1845
1594
                   (this, 'other-e', 'this-f', 'other-g', 'this-h'),
1846
1595
                   (other, 'other-e', None, 'other-g', 'other-h'))
1848
1597
            for link, target in (('e', e_target), ('f', f_target),
1849
1598
                                 ('g', g_target), ('h', h_target)):
1850
1599
                if target is not None:
1851
 
                    tg.tt.new_symlink(link, tg.root, target,
1852
 
                                      link.encode('ascii'))
 
1600
                    tg.tt.new_symlink(link, tg.root, target, link)
1853
1601
 
1854
1602
        for tg in this, base, other:
1855
1603
            tg.tt.apply()
1858
1606
        self.assertIs(os.path.islink(this.wt.abspath('b')), True)
1859
1607
        self.assertIs(os.path.isfile(this.wt.abspath('c')), True)
1860
1608
        for suffix in ('THIS', 'BASE', 'OTHER'):
1861
 
            self.assertEqual(os.readlink(
1862
 
                this.wt.abspath('d.' + suffix)), suffix)
 
1609
            self.assertEqual(os.readlink(this.wt.abspath('d.'+suffix)), suffix)
1863
1610
        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')
 
1611
        self.assertEqual(this.wt.id2path('d'), 'd.OTHER')
 
1612
        self.assertEqual(this.wt.id2path('f'), 'f.THIS')
1866
1613
        self.assertEqual(os.readlink(this.wt.abspath('e')), 'other-e')
1867
1614
        self.assertIs(os.path.lexists(this.wt.abspath('e.THIS')), False)
1868
1615
        self.assertIs(os.path.lexists(this.wt.abspath('e.OTHER')), False)
1879
1626
        base = TransformGroup("BASE", root_id)
1880
1627
        this = TransformGroup("THIS", root_id)
1881
1628
        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')
 
1629
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
 
1630
                                   for t in [base, this, other]]
 
1631
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
 
1632
                                   for t in [base, this, other]]
 
1633
        base.tt.new_directory('c', base_a, 'c')
 
1634
        this.tt.new_directory('c1', this_a, 'c')
 
1635
        other.tt.new_directory('c', other_b, 'c')
 
1636
 
 
1637
        base.tt.new_directory('d', base_a, 'd')
 
1638
        this.tt.new_directory('d1', this_b, 'd')
 
1639
        other.tt.new_directory('d', other_a, 'd')
 
1640
 
 
1641
        base.tt.new_directory('e', base_a, 'e')
 
1642
        this.tt.new_directory('e', this_a, 'e')
 
1643
        other.tt.new_directory('e1', other_b, 'e')
 
1644
 
 
1645
        base.tt.new_directory('f', base_a, 'f')
 
1646
        this.tt.new_directory('f1', this_b, 'f')
 
1647
        other.tt.new_directory('f1', other_b, 'f')
1901
1648
 
1902
1649
        for tg in [this, base, other]:
1903
1650
            tg.tt.apply()
1904
1651
        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'))
 
1652
        self.assertEqual(this.wt.id2path('c'), pathjoin('b/c1'))
 
1653
        self.assertEqual(this.wt.id2path('d'), pathjoin('b/d1'))
 
1654
        self.assertEqual(this.wt.id2path('e'), pathjoin('b/e1'))
 
1655
        self.assertEqual(this.wt.id2path('f'), pathjoin('b/f1'))
1909
1656
 
1910
1657
    def test_filename_merge_conflicts(self):
1911
1658
        root_id = generate_ids.gen_root_id()
1912
1659
        base = TransformGroup("BASE", root_id)
1913
1660
        this = TransformGroup("THIS", root_id)
1914
1661
        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')
 
1662
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
 
1663
                                   for t in [base, this, other]]
 
1664
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
 
1665
                                   for t in [base, this, other]]
 
1666
 
 
1667
        base.tt.new_file('g', base_a, 'g', 'g')
 
1668
        other.tt.new_file('g1', other_b, 'g1', 'g')
 
1669
 
 
1670
        base.tt.new_file('h', base_a, 'h', 'h')
 
1671
        this.tt.new_file('h1', this_b, 'h1', 'h')
 
1672
 
 
1673
        base.tt.new_file('i', base.root, 'i', 'i')
 
1674
        other.tt.new_directory('i1', this_b, 'i')
1928
1675
 
1929
1676
        for tg in [this, base, other]:
1930
1677
            tg.tt.apply()
1931
1678
        Merge3Merger(this.wt, this.wt, base.wt, other.wt)
1932
1679
 
1933
 
        self.assertEqual(this.wt.id2path(b'g'), pathjoin('b/g1.OTHER'))
 
1680
        self.assertEqual(this.wt.id2path('g'), pathjoin('b/g1.OTHER'))
1934
1681
        self.assertIs(os.path.lexists(this.wt.abspath('b/g1.BASE')), True)
1935
1682
        self.assertIs(os.path.lexists(this.wt.abspath('b/g1.THIS')), False)
1936
 
        self.assertEqual(this.wt.id2path(b'h'), pathjoin('b/h1.THIS'))
 
1683
        self.assertEqual(this.wt.id2path('h'), pathjoin('b/h1.THIS'))
1937
1684
        self.assertIs(os.path.lexists(this.wt.abspath('b/h1.BASE')), True)
1938
1685
        self.assertIs(os.path.lexists(this.wt.abspath('b/h1.OTHER')), False)
1939
 
        self.assertEqual(this.wt.id2path(b'i'), pathjoin('b/i1.OTHER'))
 
1686
        self.assertEqual(this.wt.id2path('i'), pathjoin('b/i1.OTHER'))
1940
1687
 
1941
1688
 
1942
1689
class TestBuildTree(tests.TestCaseWithTransport):
1944
1691
    def test_build_tree_with_symlinks(self):
1945
1692
        self.requireFeature(SymlinkFeature)
1946
1693
        os.mkdir('a')
1947
 
        a = ControlDir.create_standalone_workingtree('a')
 
1694
        a = BzrDir.create_standalone_workingtree('a')
1948
1695
        os.mkdir('a/foo')
1949
 
        with open('a/foo/bar', 'wb') as f:
1950
 
            f.write(b'contents')
 
1696
        file('a/foo/bar', 'wb').write('contents')
1951
1697
        os.symlink('a/foo/bar', 'a/foo/baz')
1952
1698
        a.add(['foo', 'foo/bar', 'foo/baz'])
1953
1699
        a.commit('initial commit')
1954
 
        b = ControlDir.create_standalone_workingtree('b')
 
1700
        b = BzrDir.create_standalone_workingtree('b')
1955
1701
        basis = a.basis_tree()
1956
1702
        basis.lock_read()
1957
1703
        self.addCleanup(basis.unlock)
1958
1704
        build_tree(basis, b)
1959
1705
        self.assertIs(os.path.isdir('b/foo'), True)
1960
 
        with open('b/foo/bar', 'rb') as f:
1961
 
            self.assertEqual(f.read(), b"contents")
 
1706
        self.assertEqual(file('b/foo/bar', 'rb').read(), "contents")
1962
1707
        self.assertEqual(os.readlink('b/foo/baz'), 'a/foo/bar')
1963
1708
 
1964
1709
    def test_build_with_references(self):
1965
1710
        tree = self.make_branch_and_tree('source',
1966
 
                                         format='development-subtree')
 
1711
            format='dirstate-with-subtree')
1967
1712
        subtree = self.make_branch_and_tree('source/subtree',
1968
 
                                            format='development-subtree')
 
1713
            format='dirstate-with-subtree')
1969
1714
        tree.add_reference(subtree)
1970
1715
        tree.commit('a revision')
1971
1716
        tree.branch.create_checkout('target')
1972
 
        self.assertPathExists('target')
1973
 
        self.assertPathExists('target/subtree')
 
1717
        self.failUnlessExists('target')
 
1718
        self.failUnlessExists('target/subtree')
1974
1719
 
1975
1720
    def test_file_conflict_handling(self):
1976
1721
        """Ensure that when building trees, conflict handling is done"""
1977
1722
        source = self.make_branch_and_tree('source')
1978
1723
        target = self.make_branch_and_tree('target')
1979
1724
        self.build_tree(['source/file', 'target/file'])
1980
 
        source.add('file', b'new-file')
 
1725
        source.add('file', 'new-file')
1981
1726
        source.commit('added file')
1982
1727
        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())
 
1728
        self.assertEqual([DuplicateEntry('Moved existing file to',
 
1729
                          'file.moved', 'file', None, 'new-file')],
 
1730
                         target.conflicts())
1987
1731
        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())
 
1732
        target_file = file('target2/file', 'wb')
 
1733
        try:
 
1734
            source_file = file('source/file', 'rb')
 
1735
            try:
 
1736
                target_file.write(source_file.read())
 
1737
            finally:
 
1738
                source_file.close()
 
1739
        finally:
 
1740
            target_file.close()
1991
1741
        build_tree(source.basis_tree(), target2)
1992
1742
        self.assertEqual([], target2.conflicts())
1993
1743
 
1996
1746
        self.requireFeature(SymlinkFeature)
1997
1747
        source = self.make_branch_and_tree('source')
1998
1748
        os.symlink('foo', 'source/symlink')
1999
 
        source.add('symlink', b'new-symlink')
 
1749
        source.add('symlink', 'new-symlink')
2000
1750
        source.commit('added file')
2001
1751
        target = self.make_branch_and_tree('target')
2002
1752
        os.symlink('bar', 'target/symlink')
2003
1753
        build_tree(source.basis_tree(), target)
2004
 
        self.assertEqual(
2005
 
            [DuplicateEntry('Moved existing file to', 'symlink.moved',
2006
 
                            'symlink', None, 'new-symlink')],
 
1754
        self.assertEqual([DuplicateEntry('Moved existing file to',
 
1755
            'symlink.moved', 'symlink', None, 'new-symlink')],
2007
1756
            target.conflicts())
2008
1757
        target = self.make_branch_and_tree('target2')
2009
1758
        os.symlink('foo', 'target2/symlink')
2015
1764
        source = self.make_branch_and_tree('source')
2016
1765
        target = self.make_branch_and_tree('target')
2017
1766
        self.build_tree(['source/dir1/', 'source/dir1/file', 'target/dir1/'])
2018
 
        source.add(['dir1', 'dir1/file'], [b'new-dir1', b'new-file'])
 
1767
        source.add(['dir1', 'dir1/file'], ['new-dir1', 'new-file'])
2019
1768
        source.commit('added file')
2020
1769
        build_tree(source.basis_tree(), target)
2021
1770
        self.assertEqual([], target.conflicts())
2022
 
        self.assertPathExists('target/dir1/file')
 
1771
        self.failUnlessExists('target/dir1/file')
2023
1772
 
2024
1773
        # Ensure contents are merged
2025
1774
        target = self.make_branch_and_tree('target2')
2026
1775
        self.build_tree(['target2/dir1/', 'target2/dir1/file2'])
2027
1776
        build_tree(source.basis_tree(), target)
2028
1777
        self.assertEqual([], target.conflicts())
2029
 
        self.assertPathExists('target2/dir1/file2')
2030
 
        self.assertPathExists('target2/dir1/file')
 
1778
        self.failUnlessExists('target2/dir1/file2')
 
1779
        self.failUnlessExists('target2/dir1/file')
2031
1780
 
2032
1781
        # Ensure new contents are suppressed for existing branches
2033
1782
        target = self.make_branch_and_tree('target3')
2034
1783
        self.make_branch('target3/dir1')
2035
1784
        self.build_tree(['target3/dir1/file2'])
2036
1785
        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)],
 
1786
        self.failIfExists('target3/dir1/file')
 
1787
        self.failUnlessExists('target3/dir1/file2')
 
1788
        self.failUnlessExists('target3/dir1.diverted/file')
 
1789
        self.assertEqual([DuplicateEntry('Diverted to',
 
1790
            'dir1.diverted', 'dir1', 'new-dir1', None)],
2043
1791
            target.conflicts())
2044
1792
 
2045
1793
        target = self.make_branch_and_tree('target4')
2046
1794
        self.build_tree(['target4/dir1/'])
2047
1795
        self.make_branch('target4/dir1/file')
2048
1796
        build_tree(source.basis_tree(), target)
2049
 
        self.assertPathExists('target4/dir1/file')
 
1797
        self.failUnlessExists('target4/dir1/file')
2050
1798
        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)],
 
1799
        self.failUnlessExists('target4/dir1/file.diverted')
 
1800
        self.assertEqual([DuplicateEntry('Diverted to',
 
1801
            'dir1/file.diverted', 'dir1/file', 'new-file', None)],
2055
1802
            target.conflicts())
2056
1803
 
2057
1804
    def test_mixed_conflict_handling(self):
2059
1806
        source = self.make_branch_and_tree('source')
2060
1807
        target = self.make_branch_and_tree('target')
2061
1808
        self.build_tree(['source/name', 'target/name/'])
2062
 
        source.add('name', b'new-name')
 
1809
        source.add('name', 'new-name')
2063
1810
        source.commit('added file')
2064
1811
        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())
 
1812
        self.assertEqual([DuplicateEntry('Moved existing file to',
 
1813
            'name.moved', 'name', None, 'new-name')], target.conflicts())
2069
1814
 
2070
1815
    def test_raises_in_populated(self):
2071
1816
        source = self.make_branch_and_tree('source')
2076
1821
        self.build_tree(['target/name'])
2077
1822
        target.add('name')
2078
1823
        self.assertRaises(errors.WorkingTreeAlreadyPopulated,
2079
 
                          build_tree, source.basis_tree(), target)
 
1824
            build_tree, source.basis_tree(), target)
2080
1825
 
2081
1826
    def test_build_tree_rename_count(self):
2082
1827
        source = self.make_branch_and_tree('source')
2098
1843
    def create_ab_tree(self):
2099
1844
        """Create a committed test tree with two files"""
2100
1845
        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'])
 
1846
        self.build_tree_contents([('source/file1', 'A')])
 
1847
        self.build_tree_contents([('source/file2', 'B')])
 
1848
        source.add(['file1', 'file2'], ['file1-id', 'file2-id'])
2104
1849
        source.commit('commit files')
2105
1850
        source.lock_write()
2106
1851
        self.addCleanup(source.unlock)
2108
1853
 
2109
1854
    def test_build_tree_accelerator_tree(self):
2110
1855
        source = self.create_ab_tree()
2111
 
        self.build_tree_contents([('source/file2', b'C')])
 
1856
        self.build_tree_contents([('source/file2', 'C')])
2112
1857
        calls = []
2113
1858
        real_source_get_file = source.get_file
2114
 
 
2115
 
        def get_file(path):
2116
 
            calls.append(path)
2117
 
            return real_source_get_file(path)
 
1859
        def get_file(file_id, path=None):
 
1860
            calls.append(file_id)
 
1861
            return real_source_get_file(file_id, path)
2118
1862
        source.get_file = get_file
2119
1863
        target = self.make_branch_and_tree('target')
2120
1864
        revision_tree = source.basis_tree()
2121
1865
        revision_tree.lock_read()
2122
1866
        self.addCleanup(revision_tree.unlock)
2123
1867
        build_tree(revision_tree, target, source)
2124
 
        self.assertEqual(['file1'], calls)
 
1868
        self.assertEqual(['file1-id'], calls)
2125
1869
        target.lock_read()
2126
1870
        self.addCleanup(target.unlock)
2127
1871
        self.assertEqual([], list(target.iter_changes(revision_tree)))
2128
1872
 
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
1873
    def test_build_tree_accelerator_tree_missing_file(self):
2142
1874
        source = self.create_ab_tree()
2143
1875
        os.unlink('source/file1')
2154
1886
    def test_build_tree_accelerator_wrong_kind(self):
2155
1887
        self.requireFeature(SymlinkFeature)
2156
1888
        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'])
 
1889
        self.build_tree_contents([('source/file1', '')])
 
1890
        self.build_tree_contents([('source/file2', '')])
 
1891
        source.add(['file1', 'file2'], ['file1-id', 'file2-id'])
2160
1892
        source.commit('commit files')
2161
1893
        os.unlink('source/file2')
2162
 
        self.build_tree_contents([('source/file2/', b'C')])
 
1894
        self.build_tree_contents([('source/file2/', 'C')])
2163
1895
        os.unlink('source/file1')
2164
1896
        os.symlink('file2', 'source/file1')
2165
1897
        calls = []
2166
1898
        real_source_get_file = source.get_file
2167
 
 
2168
 
        def get_file(path):
2169
 
            calls.append(path)
2170
 
            return real_source_get_file(path)
 
1899
        def get_file(file_id, path=None):
 
1900
            calls.append(file_id)
 
1901
            return real_source_get_file(file_id, path)
2171
1902
        source.get_file = get_file
2172
1903
        target = self.make_branch_and_tree('target')
2173
1904
        revision_tree = source.basis_tree()
2206
1937
 
2207
1938
    def test_build_tree_accelerator_tree_moved(self):
2208
1939
        source = self.make_branch_and_tree('source')
2209
 
        self.build_tree_contents([('source/file1', b'A')])
2210
 
        source.add(['file1'], [b'file1-id'])
 
1940
        self.build_tree_contents([('source/file1', 'A')])
 
1941
        source.add(['file1'], ['file1-id'])
2211
1942
        source.commit('commit files')
2212
1943
        source.rename_one('file1', 'file2')
2213
1944
        source.lock_read()
2225
1956
        self.requireFeature(HardlinkFeature)
2226
1957
        source = self.create_ab_tree()
2227
1958
        tt = TreeTransform(source)
2228
 
        trans_id = tt.trans_id_tree_path('file1')
 
1959
        trans_id = tt.trans_id_tree_file_id('file1-id')
2229
1960
        tt.set_executability(True, trans_id)
2230
1961
        tt.apply()
2231
 
        self.assertTrue(source.is_executable('file1'))
 
1962
        self.assertTrue(source.is_executable('file1-id'))
2232
1963
        target = self.make_branch_and_tree('target')
2233
1964
        revision_tree = source.basis_tree()
2234
1965
        revision_tree.lock_read()
2237
1968
        target.lock_read()
2238
1969
        self.addCleanup(target.unlock)
2239
1970
        self.assertEqual([], list(target.iter_changes(revision_tree)))
2240
 
        self.assertTrue(source.is_executable('file1'))
 
1971
        self.assertTrue(source.is_executable('file1-id'))
2241
1972
 
2242
1973
    def install_rot13_content_filter(self, pattern):
2243
1974
        # We could use
2245
1976
        # below, but that looks a bit... hard to read even if it's exactly
2246
1977
        # the same thing.
2247
1978
        original_registry = filters._reset_registry()
2248
 
 
2249
1979
        def restore_registry():
2250
1980
            filters._reset_registry(original_registry)
2251
1981
        self.addCleanup(restore_registry)
2252
 
 
2253
1982
        def rot13(chunks, context=None):
2254
 
            return [
2255
 
                codecs.encode(chunk.decode('ascii'), 'rot13').encode('ascii')
2256
 
                for chunk in chunks]
 
1983
            return [''.join(chunks).encode('rot13')]
2257
1984
        rot13filter = filters.ContentFilter(rot13, rot13)
2258
 
        filters.filter_stacks_registry.register(
2259
 
            'rot13', {'yes': [rot13filter]}.get)
 
1985
        filters.register_filter_stack_map('rot13', {'yes': [rot13filter]}.get)
2260
1986
        os.mkdir(self.test_home_dir + '/.bazaar')
2261
1987
        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
 
 
 
1988
        f = open(rules_filename, 'wb')
 
1989
        f.write('[name %s]\nrot13=yes\n' % (pattern,))
 
1990
        f.close()
2265
1991
        def uninstall_rules():
2266
1992
            os.remove(rules_filename)
2267
1993
            rules.reset_rules()
2274
2000
        if it can).
2275
2001
        """
2276
2002
        self.requireFeature(HardlinkFeature)
2277
 
        self.install_rot13_content_filter(b'file1')
 
2003
        self.install_rot13_content_filter('file1')
2278
2004
        source = self.create_ab_tree()
2279
2005
        target = self.make_branch_and_tree('target')
2280
2006
        revision_tree = source.basis_tree()
2292
2018
        self.assertEqualStat(source_stat, target_stat)
2293
2019
 
2294
2020
    def test_case_insensitive_build_tree_inventory(self):
2295
 
        if (features.CaseInsensitiveFilesystemFeature.available()
2296
 
                or features.CaseInsCasePresFilenameFeature.available()):
 
2021
        if (tests.CaseInsensitiveFilesystemFeature.available()
 
2022
            or tests.CaseInsCasePresFilenameFeature.available()):
2297
2023
            raise tests.UnavailableFeature('Fully case sensitive filesystem')
2298
2024
        source = self.make_branch_and_tree('source')
2299
2025
        self.build_tree(['source/file', 'source/FILE'])
2300
 
        source.add(['file', 'FILE'], [b'lower-id', b'upper-id'])
 
2026
        source.add(['file', 'FILE'], ['lower-id', 'upper-id'])
2301
2027
        source.commit('added files')
2302
2028
        # Don't try this at home, kids!
2303
2029
        # Force the tree to report that it is case insensitive
2304
2030
        target = self.make_branch_and_tree('target')
2305
2031
        target.case_sensitive = False
2306
2032
        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])
 
2033
        self.assertEqual('file.moved', target.id2path('lower-id'))
 
2034
        self.assertEqual('FILE', target.id2path('upper-id'))
2344
2035
 
2345
2036
 
2346
2037
class TestCommitTransform(tests.TestCaseWithTransport):
2377
2068
 
2378
2069
    def test_merge_parents(self):
2379
2070
        branch, tt = self.get_branch_and_transform()
2380
 
        tt.commit(branch, 'my message', [b'rev1b', b'rev1c'])
2381
 
        self.assertEqual([b'rev1b', b'rev1c'],
 
2071
        rev = tt.commit(branch, 'my message', ['rev1b', 'rev1c'])
 
2072
        self.assertEqual(['rev1b', 'rev1c'],
2382
2073
                         branch.basis_tree().get_parent_ids()[1:])
2383
2074
 
2384
2075
    def test_first_commit(self):
2387
2078
        self.addCleanup(branch.unlock)
2388
2079
        tt = TransformPreview(branch.basis_tree())
2389
2080
        self.addCleanup(tt.finalize)
2390
 
        tt.new_directory('', ROOT_PARENT, b'TREE_ROOT')
2391
 
        tt.commit(branch, 'my message')
 
2081
        tt.new_directory('', ROOT_PARENT, 'TREE_ROOT')
 
2082
        rev = tt.commit(branch, 'my message')
2392
2083
        self.assertEqual([], branch.basis_tree().get_parent_ids())
2393
2084
        self.assertNotEqual(_mod_revision.NULL_REVISION,
2394
2085
                            branch.last_revision())
2400
2091
        tt = TransformPreview(branch.basis_tree())
2401
2092
        self.addCleanup(tt.finalize)
2402
2093
        e = self.assertRaises(ValueError, tt.commit, branch,
2403
 
                              'my message', [b'rev1b-id'])
 
2094
                          'my message', ['rev1b-id'])
2404
2095
        self.assertEqual('Cannot supply merge parents for first commit.',
2405
2096
                         str(e))
2406
2097
        self.assertEqual(_mod_revision.NULL_REVISION, branch.last_revision())
2407
2098
 
2408
2099
    def test_add_files(self):
2409
2100
        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')
 
2101
        tt.new_file('file', tt.root, 'contents', 'file-id')
 
2102
        trans_id = tt.new_directory('dir', tt.root, 'dir-id')
2412
2103
        if SymlinkFeature.available():
2413
 
            tt.new_symlink('symlink', trans_id, 'target', b'symlink-id')
2414
 
        tt.commit(branch, 'message')
 
2104
            tt.new_symlink('symlink', trans_id, 'target', 'symlink-id')
 
2105
        rev = tt.commit(branch, 'message')
2415
2106
        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'))
 
2107
        self.assertEqual('file', tree.id2path('file-id'))
 
2108
        self.assertEqual('contents', tree.get_file_text('file-id'))
 
2109
        self.assertEqual('dir', tree.id2path('dir-id'))
2419
2110
        if SymlinkFeature.available():
2420
 
            self.assertEqual('dir/symlink', tree.id2path(b'symlink-id'))
2421
 
            self.assertEqual('target', tree.get_symlink_target('dir/symlink'))
 
2111
            self.assertEqual('dir/symlink', tree.id2path('symlink-id'))
 
2112
            self.assertEqual('target', tree.get_symlink_target('symlink-id'))
2422
2113
 
2423
2114
    def test_add_unversioned(self):
2424
2115
        branch, tt = self.get_branch_and_transform()
2425
 
        tt.new_file('file', tt.root, [b'contents'])
 
2116
        tt.new_file('file', tt.root, 'contents')
2426
2117
        self.assertRaises(errors.StrictCommitFailed, tt.commit, branch,
2427
2118
                          'message', strict=True)
2428
2119
 
2429
2120
    def test_modify_strict(self):
2430
2121
        branch, tt = self.get_branch_and_transform()
2431
 
        tt.new_file('file', tt.root, [b'contents'], b'file-id')
 
2122
        tt.new_file('file', tt.root, 'contents', 'file-id')
2432
2123
        tt.commit(branch, 'message', strict=True)
2433
2124
        tt = TransformPreview(branch.basis_tree())
2434
2125
        self.addCleanup(tt.finalize)
2435
 
        trans_id = tt.trans_id_file_id(b'file-id')
 
2126
        trans_id = tt.trans_id_file_id('file-id')
2436
2127
        tt.delete_contents(trans_id)
2437
 
        tt.create_file([b'contents'], trans_id)
 
2128
        tt.create_file('contents', trans_id)
2438
2129
        tt.commit(branch, 'message', strict=True)
2439
2130
 
2440
2131
    def test_commit_malformed(self):
2443
2134
        In this case, we are adding a file without adding its parent.
2444
2135
        """
2445
2136
        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')
 
2137
        parent_id = tt.trans_id_file_id('parent-id')
 
2138
        tt.new_file('file', parent_id, 'contents', 'file-id')
2448
2139
        self.assertRaises(errors.MalformedTransform, tt.commit, branch,
2449
2140
                          'message')
2450
2141
 
2452
2143
        branch, tt = self.get_branch_and_transform()
2453
2144
        rev_id = tt.commit(branch, 'message', timestamp=1, timezone=43201,
2454
2145
                           committer='me <me@example.com>',
2455
 
                           revprops={u'foo': 'bar'}, revision_id=b'revid-1',
 
2146
                           revprops={'foo': 'bar'}, revision_id='revid-1',
2456
2147
                           authors=['Author1 <author1@example.com>',
2457
 
                                    'Author2 <author2@example.com>',
2458
 
                                    ])
2459
 
        self.assertEqual(b'revid-1', rev_id)
 
2148
                              'Author2 <author2@example.com>',
 
2149
                               ])
 
2150
        self.assertEqual('revid-1', rev_id)
2460
2151
        revision = branch.repository.get_revision(rev_id)
2461
2152
        self.assertEqual(1, revision.timestamp)
2462
2153
        self.assertEqual(43201, revision.timezone)
2481
2172
        self.assertEqual('tree', revision.properties['branch-nick'])
2482
2173
 
2483
2174
 
 
2175
class MockTransform(object):
 
2176
 
 
2177
    def has_named_child(self, by_parent, parent_id, name):
 
2178
        for child_id in by_parent[parent_id]:
 
2179
            if child_id == '0':
 
2180
                if name == "name~":
 
2181
                    return True
 
2182
            elif name == "name.~%s~" % child_id:
 
2183
                return True
 
2184
        return False
 
2185
 
 
2186
 
 
2187
class MockEntry(object):
 
2188
    def __init__(self):
 
2189
        object.__init__(self)
 
2190
        self.name = "name"
 
2191
 
 
2192
 
 
2193
class TestGetBackupName(TestCase):
 
2194
    def test_get_backup_name(self):
 
2195
        tt = MockTransform()
 
2196
        name = get_backup_name(MockEntry(), {'a':[]}, 'a', tt)
 
2197
        self.assertEqual(name, 'name.~1~')
 
2198
        name = get_backup_name(MockEntry(), {'a':['1']}, 'a', tt)
 
2199
        self.assertEqual(name, 'name.~2~')
 
2200
        name = get_backup_name(MockEntry(), {'a':['2']}, 'a', tt)
 
2201
        self.assertEqual(name, 'name.~1~')
 
2202
        name = get_backup_name(MockEntry(), {'a':['2'], 'b':[]}, 'b', tt)
 
2203
        self.assertEqual(name, 'name.~1~')
 
2204
        name = get_backup_name(MockEntry(), {'a':['1', '2', '3']}, 'a', tt)
 
2205
        self.assertEqual(name, 'name.~4~')
 
2206
 
 
2207
 
2484
2208
class TestFileMover(tests.TestCaseWithTransport):
2485
2209
 
2486
2210
    def test_file_mover(self):
2487
2211
        self.build_tree(['a/', 'a/b', 'c/', 'c/d'])
2488
2212
        mover = _FileMover()
2489
2213
        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')
 
2214
        self.failUnlessExists('q')
 
2215
        self.failIfExists('a')
 
2216
        self.failUnlessExists('q/b')
 
2217
        self.failUnlessExists('c')
 
2218
        self.failUnlessExists('c/d')
2495
2219
 
2496
2220
    def test_pre_delete_rollback(self):
2497
2221
        self.build_tree(['a/'])
2498
2222
        mover = _FileMover()
2499
2223
        mover.pre_delete('a', 'q')
2500
 
        self.assertPathExists('q')
2501
 
        self.assertPathDoesNotExist('a')
 
2224
        self.failUnlessExists('q')
 
2225
        self.failIfExists('a')
2502
2226
        mover.rollback()
2503
 
        self.assertPathDoesNotExist('q')
2504
 
        self.assertPathExists('a')
 
2227
        self.failIfExists('q')
 
2228
        self.failUnlessExists('a')
2505
2229
 
2506
2230
    def test_apply_deletions(self):
2507
2231
        self.build_tree(['a/', 'b/'])
2508
2232
        mover = _FileMover()
2509
2233
        mover.pre_delete('a', 'q')
2510
2234
        mover.pre_delete('b', 'r')
2511
 
        self.assertPathExists('q')
2512
 
        self.assertPathExists('r')
2513
 
        self.assertPathDoesNotExist('a')
2514
 
        self.assertPathDoesNotExist('b')
 
2235
        self.failUnlessExists('q')
 
2236
        self.failUnlessExists('r')
 
2237
        self.failIfExists('a')
 
2238
        self.failIfExists('b')
2515
2239
        mover.apply_deletions()
2516
 
        self.assertPathDoesNotExist('q')
2517
 
        self.assertPathDoesNotExist('r')
2518
 
        self.assertPathDoesNotExist('a')
2519
 
        self.assertPathDoesNotExist('b')
 
2240
        self.failIfExists('q')
 
2241
        self.failIfExists('r')
 
2242
        self.failIfExists('a')
 
2243
        self.failIfExists('b')
2520
2244
 
2521
2245
    def test_file_mover_rollback(self):
2522
2246
        self.build_tree(['a/', 'a/b', 'c/', 'c/d/', 'c/e/'])
2525
2249
        mover.rename('c/e', 'c/d')
2526
2250
        try:
2527
2251
            mover.rename('a', 'c')
2528
 
        except errors.FileExists:
 
2252
        except errors.FileExists, e:
2529
2253
            mover.rollback()
2530
 
        self.assertPathExists('a')
2531
 
        self.assertPathExists('c/d')
 
2254
        self.failUnlessExists('a')
 
2255
        self.failUnlessExists('c/d')
2532
2256
 
2533
2257
 
2534
2258
class Bogus(Exception):
2546
2270
 
2547
2271
        def rename(self, source, target):
2548
2272
            if (self.bad_source is not None and
2549
 
                    source.endswith(self.bad_source)):
 
2273
                source.endswith(self.bad_source)):
2550
2274
                raise Bogus
2551
2275
            elif (self.bad_target is not None and
2552
 
                  target.endswith(self.bad_target)):
 
2276
                target.endswith(self.bad_target)):
2553
2277
                raise Bogus
2554
2278
            else:
2555
2279
                _FileMover.rename(self, source, target)
2564
2288
        tt.adjust_path('d', a_id, tt.trans_id_tree_path('a/b'))
2565
2289
        self.assertRaises(Bogus, tt.apply,
2566
2290
                          _mover=self.ExceptionFileMover(bad_source='a'))
2567
 
        self.assertPathExists('a')
2568
 
        self.assertPathExists('a/b')
 
2291
        self.failUnlessExists('a')
 
2292
        self.failUnlessExists('a/b')
2569
2293
        tt.apply()
2570
 
        self.assertPathExists('c')
2571
 
        self.assertPathExists('c/d')
 
2294
        self.failUnlessExists('c')
 
2295
        self.failUnlessExists('c/d')
2572
2296
 
2573
2297
    def test_rollback_rename_into_place(self):
2574
2298
        tree = self.make_branch_and_tree('.')
2580
2304
        tt.adjust_path('d', a_id, tt.trans_id_tree_path('a/b'))
2581
2305
        self.assertRaises(Bogus, tt.apply,
2582
2306
                          _mover=self.ExceptionFileMover(bad_target='c/d'))
2583
 
        self.assertPathExists('a')
2584
 
        self.assertPathExists('a/b')
 
2307
        self.failUnlessExists('a')
 
2308
        self.failUnlessExists('a/b')
2585
2309
        tt.apply()
2586
 
        self.assertPathExists('c')
2587
 
        self.assertPathExists('c/d')
 
2310
        self.failUnlessExists('c')
 
2311
        self.failUnlessExists('c/d')
2588
2312
 
2589
2313
    def test_rollback_deletion(self):
2590
2314
        tree = self.make_branch_and_tree('.')
2596
2320
        tt.adjust_path('d', tt.root, tt.trans_id_tree_path('a/b'))
2597
2321
        self.assertRaises(Bogus, tt.apply,
2598
2322
                          _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):
 
2323
        self.failUnlessExists('a')
 
2324
        self.failUnlessExists('a/b')
 
2325
 
 
2326
    def test_resolve_no_parent(self):
2738
2327
        wt = self.make_branch_and_tree('.')
2739
 
        self.build_tree(['dir/', ])
2740
 
        wt.add(['dir'], [b'dir-id'])
2741
 
        wt.commit('Create dir')
2742
2328
        tt = TreeTransform(wt)
2743
2329
        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)
 
2330
        parent = tt.trans_id_file_id('parent-id')
 
2331
        tt.new_file('file', parent, 'Contents')
 
2332
        resolve_conflicts(tt)
 
2333
 
 
2334
 
 
2335
A_ENTRY = ('a-id', ('a', 'a'), True, (True, True),
 
2336
                  ('TREE_ROOT', 'TREE_ROOT'), ('a', 'a'), ('file', 'file'),
 
2337
                  (False, False))
 
2338
ROOT_ENTRY = ('TREE_ROOT', ('', ''), False, (True, True), (None, None),
 
2339
              ('', ''), ('directory', 'directory'), (False, None))
2775
2340
 
2776
2341
 
2777
2342
class TestTransformPreview(tests.TestCaseWithTransport):
2778
2343
 
2779
2344
    def create_tree(self):
2780
2345
        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')
 
2346
        self.build_tree_contents([('a', 'content 1')])
 
2347
        tree.set_root_id('TREE_ROOT')
 
2348
        tree.add('a', 'a-id')
 
2349
        tree.commit('rev1', rev_id='rev1')
 
2350
        return tree.branch.repository.revision_tree('rev1')
2786
2351
 
2787
2352
    def get_empty_preview(self):
2788
2353
        repository = self.make_repository('repo')
2806
2371
        revision_tree = self.create_tree()
2807
2372
        preview = TransformPreview(revision_tree)
2808
2373
        self.addCleanup(preview.finalize)
2809
 
        preview.new_file('file2', preview.root, [b'content B\n'], b'file2-id')
 
2374
        preview.new_file('file2', preview.root, 'content B\n', 'file2-id')
2810
2375
        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')
 
2376
        self.assertEqual(preview_tree.kind('file2-id'), 'file')
 
2377
        self.assertEqual(
 
2378
            preview_tree.get_file('file2-id').read(), 'content B\n')
2814
2379
 
2815
2380
    def test_diff_preview_tree(self):
2816
2381
        revision_tree = self.create_tree()
2817
2382
        preview = TransformPreview(revision_tree)
2818
2383
        self.addCleanup(preview.finalize)
2819
 
        preview.new_file('file2', preview.root, [b'content B\n'], b'file2-id')
 
2384
        preview.new_file('file2', preview.root, 'content B\n', 'file2-id')
2820
2385
        preview_tree = preview.get_preview_tree()
2821
 
        out = BytesIO()
 
2386
        out = StringIO()
2822
2387
        show_diff_trees(revision_tree, preview_tree, out)
2823
2388
        lines = out.getvalue().splitlines()
2824
 
        self.assertEqual(lines[0], b"=== added file 'file2'")
 
2389
        self.assertEqual(lines[0], "=== added file 'file2'")
2825
2390
        # 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')
 
2391
        self.assertEqual(lines[4], "+content B")
2855
2392
 
2856
2393
    def test_transform_conflicts(self):
2857
2394
        revision_tree = self.create_tree()
2858
2395
        preview = TransformPreview(revision_tree)
2859
2396
        self.addCleanup(preview.finalize)
2860
 
        preview.new_file('a', preview.root, [b'content 2'])
 
2397
        preview.new_file('a', preview.root, 'content 2')
2861
2398
        resolve_conflicts(preview)
2862
 
        trans_id = preview.trans_id_file_id(b'a-id')
 
2399
        trans_id = preview.trans_id_file_id('a-id')
2863
2400
        self.assertEqual('a.moved', preview.final_name(trans_id))
2864
2401
 
2865
2402
    def get_tree_and_preview_tree(self):
2866
2403
        revision_tree = self.create_tree()
2867
2404
        preview = TransformPreview(revision_tree)
2868
2405
        self.addCleanup(preview.finalize)
2869
 
        a_trans_id = preview.trans_id_file_id(b'a-id')
 
2406
        a_trans_id = preview.trans_id_file_id('a-id')
2870
2407
        preview.delete_contents(a_trans_id)
2871
 
        preview.create_file([b'b content'], a_trans_id)
 
2408
        preview.create_file('b content', a_trans_id)
2872
2409
        preview_tree = preview.get_preview_tree()
2873
2410
        return revision_tree, preview_tree
2874
2411
 
2875
2412
    def test_iter_changes(self):
2876
2413
        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)))
 
2414
        root = revision_tree.inventory.root.file_id
 
2415
        self.assertEqual([('a-id', ('a', 'a'), True, (True, True),
 
2416
                          (root, root), ('a', 'a'), ('file', 'file'),
 
2417
                          (False, False))],
 
2418
                          list(preview_tree.iter_changes(revision_tree)))
2882
2419
 
2883
2420
    def test_include_unchanged_succeeds(self):
2884
2421
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
2885
2422
        changes = preview_tree.iter_changes(revision_tree,
2886
2423
                                            include_unchanged=True)
 
2424
        root = revision_tree.inventory.root.file_id
 
2425
 
2887
2426
        self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
2888
2427
 
2889
2428
    def test_specific_files(self):
2890
2429
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
2891
2430
        changes = preview_tree.iter_changes(revision_tree,
2892
2431
                                            specific_files=[''])
2893
 
        self.assertEqual([A_ENTRY], list(changes))
 
2432
        self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
2894
2433
 
2895
2434
    def test_want_unversioned(self):
2896
2435
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
2897
2436
        changes = preview_tree.iter_changes(revision_tree,
2898
2437
                                            want_unversioned=True)
2899
 
        self.assertEqual([A_ENTRY], list(changes))
 
2438
        self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
2900
2439
 
2901
2440
    def test_ignore_extra_trees_no_specific_files(self):
2902
2441
        # extra_trees is harmless without specific_files, so we'll silently
2918
2457
        revision_tree = self.create_tree()
2919
2458
        preview = TransformPreview(revision_tree)
2920
2459
        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')
 
2460
        preview.new_file('file', preview.root, 'contents', 'file-id')
 
2461
        preview.new_directory('directory', preview.root, 'dir-id')
2923
2462
        preview_tree = preview.get_preview_tree()
2924
 
        self.assertEqual('file', preview_tree.kind('file'))
2925
 
        self.assertEqual('directory', preview_tree.kind('directory'))
 
2463
        self.assertEqual('file', preview_tree.kind('file-id'))
 
2464
        self.assertEqual('directory', preview_tree.kind('dir-id'))
2926
2465
 
2927
2466
    def test_get_file_mtime(self):
2928
2467
        preview = self.get_empty_preview()
2929
 
        file_trans_id = preview.new_file('file', preview.root, [b'contents'],
2930
 
                                         b'file-id')
 
2468
        file_trans_id = preview.new_file('file', preview.root, 'contents',
 
2469
                                         'file-id')
2931
2470
        limbo_path = preview._limbo_name(file_trans_id)
2932
2471
        preview_tree = preview.get_preview_tree()
2933
2472
        self.assertEqual(os.stat(limbo_path).st_mtime,
2934
 
                         preview_tree.get_file_mtime('file'))
 
2473
                         preview_tree.get_file_mtime('file-id'))
2935
2474
 
2936
2475
    def test_get_file_mtime_renamed(self):
2937
2476
        work_tree = self.make_branch_and_tree('tree')
2938
2477
        self.build_tree(['tree/file'])
2939
 
        work_tree.add('file', b'file-id')
 
2478
        work_tree.add('file', 'file-id')
2940
2479
        preview = TransformPreview(work_tree)
2941
2480
        self.addCleanup(preview.finalize)
2942
 
        file_trans_id = preview.trans_id_tree_path('file')
 
2481
        file_trans_id = preview.trans_id_tree_file_id('file-id')
2943
2482
        preview.adjust_path('renamed', preview.root, file_trans_id)
2944
2483
        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'))
 
2484
        preview_mtime = preview_tree.get_file_mtime('file-id', 'renamed')
 
2485
        work_mtime = work_tree.get_file_mtime('file-id', 'file')
2959
2486
 
2960
2487
    def test_get_file(self):
2961
2488
        preview = self.get_empty_preview()
2962
 
        preview.new_file('file', preview.root, [b'contents'], b'file-id')
 
2489
        preview.new_file('file', preview.root, 'contents', 'file-id')
2963
2490
        preview_tree = preview.get_preview_tree()
2964
 
        with preview_tree.get_file('file') as tree_file:
2965
 
            self.assertEqual(b'contents', tree_file.read())
 
2491
        tree_file = preview_tree.get_file('file-id')
 
2492
        try:
 
2493
            self.assertEqual('contents', tree_file.read())
 
2494
        finally:
 
2495
            tree_file.close()
2966
2496
 
2967
2497
    def test_get_symlink_target(self):
2968
2498
        self.requireFeature(SymlinkFeature)
2969
2499
        preview = self.get_empty_preview()
2970
 
        preview.new_symlink('symlink', preview.root, 'target', b'symlink-id')
 
2500
        preview.new_symlink('symlink', preview.root, 'target', 'symlink-id')
2971
2501
        preview_tree = preview.get_preview_tree()
2972
2502
        self.assertEqual('target',
2973
 
                         preview_tree.get_symlink_target('symlink'))
 
2503
                         preview_tree.get_symlink_target('symlink-id'))
2974
2504
 
2975
2505
    def test_all_file_ids(self):
2976
2506
        tree = self.make_branch_and_tree('tree')
2977
2507
        self.build_tree(['tree/a', 'tree/b', 'tree/c'])
2978
 
        tree.add(['a', 'b', 'c'], [b'a-id', b'b-id', b'c-id'])
 
2508
        tree.add(['a', 'b', 'c'], ['a-id', 'b-id', 'c-id'])
2979
2509
        preview = TransformPreview(tree)
2980
2510
        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')
 
2511
        preview.unversion_file(preview.trans_id_file_id('b-id'))
 
2512
        c_trans_id = preview.trans_id_file_id('c-id')
2983
2513
        preview.unversion_file(c_trans_id)
2984
 
        preview.version_file(b'c-id', c_trans_id)
 
2514
        preview.version_file('c-id', c_trans_id)
2985
2515
        preview_tree = preview.get_preview_tree()
2986
 
        self.assertEqual({b'a-id', b'c-id', tree.path2id('')},
 
2516
        self.assertEqual(set(['a-id', 'c-id', tree.get_root_id()]),
2987
2517
                         preview_tree.all_file_ids())
2988
2518
 
2989
2519
    def test_path2id_deleted_unchanged(self):
2990
2520
        tree = self.make_branch_and_tree('tree')
2991
2521
        self.build_tree(['tree/unchanged', 'tree/deleted'])
2992
 
        tree.add(['unchanged', 'deleted'], [b'unchanged-id', b'deleted-id'])
 
2522
        tree.add(['unchanged', 'deleted'], ['unchanged-id', 'deleted-id'])
2993
2523
        preview = TransformPreview(tree)
2994
2524
        self.addCleanup(preview.finalize)
2995
 
        preview.unversion_file(preview.trans_id_file_id(b'deleted-id'))
 
2525
        preview.unversion_file(preview.trans_id_file_id('deleted-id'))
2996
2526
        preview_tree = preview.get_preview_tree()
2997
 
        self.assertEqual(b'unchanged-id', preview_tree.path2id('unchanged'))
2998
 
        self.assertFalse(preview_tree.is_versioned('deleted'))
 
2527
        self.assertEqual('unchanged-id', preview_tree.path2id('unchanged'))
 
2528
        self.assertIs(None, preview_tree.path2id('deleted'))
2999
2529
 
3000
2530
    def test_path2id_created(self):
3001
2531
        tree = self.make_branch_and_tree('tree')
3002
2532
        self.build_tree(['tree/unchanged'])
3003
 
        tree.add(['unchanged'], [b'unchanged-id'])
 
2533
        tree.add(['unchanged'], ['unchanged-id'])
3004
2534
        preview = TransformPreview(tree)
3005
2535
        self.addCleanup(preview.finalize)
3006
 
        preview.new_file('new', preview.trans_id_file_id(b'unchanged-id'),
3007
 
                         [b'contents'], b'new-id')
 
2536
        preview.new_file('new', preview.trans_id_file_id('unchanged-id'),
 
2537
            'contents', 'new-id')
3008
2538
        preview_tree = preview.get_preview_tree()
3009
 
        self.assertEqual(b'new-id', preview_tree.path2id('unchanged/new'))
 
2539
        self.assertEqual('new-id', preview_tree.path2id('unchanged/new'))
3010
2540
 
3011
2541
    def test_path2id_moved(self):
3012
2542
        tree = self.make_branch_and_tree('tree')
3013
2543
        self.build_tree(['tree/old_parent/', 'tree/old_parent/child'])
3014
2544
        tree.add(['old_parent', 'old_parent/child'],
3015
 
                 [b'old_parent-id', b'child-id'])
 
2545
                 ['old_parent-id', 'child-id'])
3016
2546
        preview = TransformPreview(tree)
3017
2547
        self.addCleanup(preview.finalize)
3018
2548
        new_parent = preview.new_directory('new_parent', preview.root,
3019
 
                                           b'new_parent-id')
 
2549
                                           'new_parent-id')
3020
2550
        preview.adjust_path('child', new_parent,
3021
 
                            preview.trans_id_file_id(b'child-id'))
 
2551
                            preview.trans_id_file_id('child-id'))
3022
2552
        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'))
 
2553
        self.assertIs(None, preview_tree.path2id('old_parent/child'))
 
2554
        self.assertEqual('child-id', preview_tree.path2id('new_parent/child'))
3025
2555
 
3026
2556
    def test_path2id_renamed_parent(self):
3027
2557
        tree = self.make_branch_and_tree('tree')
3028
2558
        self.build_tree(['tree/old_name/', 'tree/old_name/child'])
3029
2559
        tree.add(['old_name', 'old_name/child'],
3030
 
                 [b'parent-id', b'child-id'])
 
2560
                 ['parent-id', 'child-id'])
3031
2561
        preview = TransformPreview(tree)
3032
2562
        self.addCleanup(preview.finalize)
3033
2563
        preview.adjust_path('new_name', preview.root,
3034
 
                            preview.trans_id_file_id(b'parent-id'))
 
2564
                            preview.trans_id_file_id('parent-id'))
3035
2565
        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'))
 
2566
        self.assertIs(None, preview_tree.path2id('old_name/child'))
 
2567
        self.assertEqual('child-id', preview_tree.path2id('new_name/child'))
3038
2568
 
3039
 
    def assertMatchingIterEntries(self, tt, specific_files=None):
 
2569
    def assertMatchingIterEntries(self, tt, specific_file_ids=None):
3040
2570
        preview_tree = tt.get_preview_tree()
3041
2571
        preview_result = list(preview_tree.iter_entries_by_dir(
3042
 
                              specific_files=specific_files))
 
2572
                              specific_file_ids))
3043
2573
        tree = tt._tree
3044
2574
        tt.apply()
3045
 
        actual_result = list(tree.iter_entries_by_dir(
3046
 
            specific_files=specific_files))
 
2575
        actual_result = list(tree.iter_entries_by_dir(specific_file_ids))
3047
2576
        self.assertEqual(actual_result, preview_result)
3048
2577
 
3049
2578
    def test_iter_entries_by_dir_new(self):
3050
2579
        tree = self.make_branch_and_tree('tree')
3051
2580
        tt = TreeTransform(tree)
3052
 
        tt.new_file('new', tt.root, [b'contents'], b'new-id')
 
2581
        tt.new_file('new', tt.root, 'contents', 'new-id')
3053
2582
        self.assertMatchingIterEntries(tt)
3054
2583
 
3055
2584
    def test_iter_entries_by_dir_deleted(self):
3056
2585
        tree = self.make_branch_and_tree('tree')
3057
2586
        self.build_tree(['tree/deleted'])
3058
 
        tree.add('deleted', b'deleted-id')
 
2587
        tree.add('deleted', 'deleted-id')
3059
2588
        tt = TreeTransform(tree)
3060
 
        tt.delete_contents(tt.trans_id_file_id(b'deleted-id'))
 
2589
        tt.delete_contents(tt.trans_id_file_id('deleted-id'))
3061
2590
        self.assertMatchingIterEntries(tt)
3062
2591
 
3063
2592
    def test_iter_entries_by_dir_unversioned(self):
3064
2593
        tree = self.make_branch_and_tree('tree')
3065
2594
        self.build_tree(['tree/removed'])
3066
 
        tree.add('removed', b'removed-id')
 
2595
        tree.add('removed', 'removed-id')
3067
2596
        tt = TreeTransform(tree)
3068
 
        tt.unversion_file(tt.trans_id_file_id(b'removed-id'))
 
2597
        tt.unversion_file(tt.trans_id_file_id('removed-id'))
3069
2598
        self.assertMatchingIterEntries(tt)
3070
2599
 
3071
2600
    def test_iter_entries_by_dir_moved(self):
3072
2601
        tree = self.make_branch_and_tree('tree')
3073
2602
        self.build_tree(['tree/moved', 'tree/new_parent/'])
3074
 
        tree.add(['moved', 'new_parent'], [b'moved-id', b'new_parent-id'])
 
2603
        tree.add(['moved', 'new_parent'], ['moved-id', 'new_parent-id'])
3075
2604
        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'))
 
2605
        tt.adjust_path('moved', tt.trans_id_file_id('new_parent-id'),
 
2606
                       tt.trans_id_file_id('moved-id'))
3078
2607
        self.assertMatchingIterEntries(tt)
3079
2608
 
3080
 
    def test_iter_entries_by_dir_specific_files(self):
 
2609
    def test_iter_entries_by_dir_specific_file_ids(self):
3081
2610
        tree = self.make_branch_and_tree('tree')
3082
 
        tree.set_root_id(b'tree-root-id')
 
2611
        tree.set_root_id('tree-root-id')
3083
2612
        self.build_tree(['tree/parent/', 'tree/parent/child'])
3084
 
        tree.add(['parent', 'parent/child'], [b'parent-id', b'child-id'])
 
2613
        tree.add(['parent', 'parent/child'], ['parent-id', 'child-id'])
3085
2614
        tt = TreeTransform(tree)
3086
 
        self.assertMatchingIterEntries(tt, ['', 'parent/child'])
 
2615
        self.assertMatchingIterEntries(tt, ['tree-root-id', 'child-id'])
3087
2616
 
3088
2617
    def test_symlink_content_summary(self):
3089
2618
        self.requireFeature(SymlinkFeature)
3090
2619
        preview = self.get_empty_preview()
3091
 
        preview.new_symlink('path', preview.root, 'target', b'path-id')
 
2620
        preview.new_symlink('path', preview.root, 'target', 'path-id')
3092
2621
        summary = preview.get_preview_tree().path_content_summary('path')
3093
2622
        self.assertEqual(('symlink', None, None, 'target'), summary)
3094
2623
 
3109
2638
 
3110
2639
    def test_file_content_summary_executable(self):
3111
2640
        preview = self.get_empty_preview()
3112
 
        path_id = preview.new_file('path', preview.root, [
3113
 
                                   b'contents'], b'path-id')
 
2641
        path_id = preview.new_file('path', preview.root, 'contents', 'path-id')
3114
2642
        preview.set_executability(True, path_id)
3115
2643
        summary = preview.get_preview_tree().path_content_summary('path')
3116
2644
        self.assertEqual(4, len(summary))
3135
2663
 
3136
2664
    def test_file_content_summary_non_exec(self):
3137
2665
        preview = self.get_empty_preview()
3138
 
        preview.new_file('path', preview.root, [b'contents'], b'path-id')
 
2666
        preview.new_file('path', preview.root, 'contents', 'path-id')
3139
2667
        summary = preview.get_preview_tree().path_content_summary('path')
3140
2668
        self.assertEqual(4, len(summary))
3141
2669
        self.assertEqual('file', summary[0])
3148
2676
 
3149
2677
    def test_dir_content_summary(self):
3150
2678
        preview = self.get_empty_preview()
3151
 
        preview.new_directory('path', preview.root, b'path-id')
 
2679
        preview.new_directory('path', preview.root, 'path-id')
3152
2680
        summary = preview.get_preview_tree().path_content_summary('path')
3153
2681
        self.assertEqual(('directory', None, None, None), summary)
3154
2682
 
3155
2683
    def test_tree_content_summary(self):
3156
2684
        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)
 
2685
        path = preview.new_directory('path', preview.root, 'path-id')
 
2686
        preview.set_tree_reference('rev-1', path)
3159
2687
        summary = preview.get_preview_tree().path_content_summary('path')
3160
2688
        self.assertEqual(4, len(summary))
3161
2689
        self.assertEqual('tree-reference', summary[0])
3162
2690
 
3163
2691
    def test_annotate(self):
3164
2692
        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')])
 
2693
        self.build_tree_contents([('tree/file', 'a\n')])
 
2694
        tree.add('file', 'file-id')
 
2695
        tree.commit('a', rev_id='one')
 
2696
        self.build_tree_contents([('tree/file', 'a\nb\n')])
3169
2697
        preview = TransformPreview(tree)
3170
2698
        self.addCleanup(preview.finalize)
3171
 
        file_trans_id = preview.trans_id_file_id(b'file-id')
 
2699
        file_trans_id = preview.trans_id_file_id('file-id')
3172
2700
        preview.delete_contents(file_trans_id)
3173
 
        preview.create_file([b'a\nb\nc\n'], file_trans_id)
 
2701
        preview.create_file('a\nb\nc\n', file_trans_id)
3174
2702
        preview_tree = preview.get_preview_tree()
3175
2703
        expected = [
3176
 
            (b'one', b'a\n'),
3177
 
            (b'me:', b'b\n'),
3178
 
            (b'me:', b'c\n'),
 
2704
            ('one', 'a\n'),
 
2705
            ('me:', 'b\n'),
 
2706
            ('me:', 'c\n'),
3179
2707
        ]
3180
 
        annotation = preview_tree.annotate_iter(
3181
 
            'file', default_revision=b'me:')
 
2708
        annotation = preview_tree.annotate_iter('file-id', 'me:')
3182
2709
        self.assertEqual(expected, annotation)
3183
2710
 
3184
2711
    def test_annotate_missing(self):
3185
2712
        preview = self.get_empty_preview()
3186
 
        preview.new_file('file', preview.root, [b'a\nb\nc\n'], b'file-id')
 
2713
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
3187
2714
        preview_tree = preview.get_preview_tree()
3188
2715
        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:')
 
2716
            ('me:', 'a\n'),
 
2717
            ('me:', 'b\n'),
 
2718
            ('me:', 'c\n'),
 
2719
         ]
 
2720
        annotation = preview_tree.annotate_iter('file-id', 'me:')
3195
2721
        self.assertEqual(expected, annotation)
3196
2722
 
3197
2723
    def test_annotate_rename(self):
3198
2724
        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')
 
2725
        self.build_tree_contents([('tree/file', 'a\n')])
 
2726
        tree.add('file', 'file-id')
 
2727
        tree.commit('a', rev_id='one')
3202
2728
        preview = TransformPreview(tree)
3203
2729
        self.addCleanup(preview.finalize)
3204
 
        file_trans_id = preview.trans_id_file_id(b'file-id')
 
2730
        file_trans_id = preview.trans_id_file_id('file-id')
3205
2731
        preview.adjust_path('newname', preview.root, file_trans_id)
3206
2732
        preview_tree = preview.get_preview_tree()
3207
2733
        expected = [
3208
 
            (b'one', b'a\n'),
 
2734
            ('one', 'a\n'),
3209
2735
        ]
3210
 
        annotation = preview_tree.annotate_iter(
3211
 
            'file', default_revision=b'me:')
 
2736
        annotation = preview_tree.annotate_iter('file-id', 'me:')
3212
2737
        self.assertEqual(expected, annotation)
3213
2738
 
3214
2739
    def test_annotate_deleted(self):
3215
2740
        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')])
 
2741
        self.build_tree_contents([('tree/file', 'a\n')])
 
2742
        tree.add('file', 'file-id')
 
2743
        tree.commit('a', rev_id='one')
 
2744
        self.build_tree_contents([('tree/file', 'a\nb\n')])
3220
2745
        preview = TransformPreview(tree)
3221
2746
        self.addCleanup(preview.finalize)
3222
 
        file_trans_id = preview.trans_id_file_id(b'file-id')
 
2747
        file_trans_id = preview.trans_id_file_id('file-id')
3223
2748
        preview.delete_contents(file_trans_id)
3224
2749
        preview_tree = preview.get_preview_tree()
3225
 
        annotation = preview_tree.annotate_iter(
3226
 
            'file', default_revision=b'me:')
 
2750
        annotation = preview_tree.annotate_iter('file-id', 'me:')
3227
2751
        self.assertIs(None, annotation)
3228
2752
 
3229
2753
    def test_stored_kind(self):
3230
2754
        preview = self.get_empty_preview()
3231
 
        preview.new_file('file', preview.root, [b'a\nb\nc\n'], b'file-id')
 
2755
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
3232
2756
        preview_tree = preview.get_preview_tree()
3233
 
        self.assertEqual('file', preview_tree.stored_kind('file'))
 
2757
        self.assertEqual('file', preview_tree.stored_kind('file-id'))
3234
2758
 
3235
2759
    def test_is_executable(self):
3236
2760
        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'))
 
2761
        preview.new_file('file', preview.root, 'a\nb\nc\n', 'file-id')
 
2762
        preview.set_executability(True, preview.trans_id_file_id('file-id'))
3239
2763
        preview_tree = preview.get_preview_tree()
3240
 
        self.assertEqual(True, preview_tree.is_executable('file'))
 
2764
        self.assertEqual(True, preview_tree.is_executable('file-id'))
3241
2765
 
3242
2766
    def test_get_set_parent_ids(self):
3243
2767
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
3244
2768
        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())
 
2769
        preview_tree.set_parent_ids(['rev-1'])
 
2770
        self.assertEqual(['rev-1'], preview_tree.get_parent_ids())
3247
2771
 
3248
2772
    def test_plan_file_merge(self):
3249
2773
        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')
 
2774
        self.build_tree_contents([('wta/file', 'a\nb\nc\nd\n')])
 
2775
        work_a.add('file', 'file-id')
3252
2776
        base_id = work_a.commit('base version')
3253
 
        tree_b = work_a.controldir.sprout('wtb').open_workingtree()
 
2777
        tree_b = work_a.bzrdir.sprout('wtb').open_workingtree()
3254
2778
        preview = TransformPreview(work_a)
3255
2779
        self.addCleanup(preview.finalize)
3256
 
        trans_id = preview.trans_id_file_id(b'file-id')
 
2780
        trans_id = preview.trans_id_file_id('file-id')
3257
2781
        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')])
 
2782
        preview.create_file('b\nc\nd\ne\n', trans_id)
 
2783
        self.build_tree_contents([('wtb/file', 'a\nc\nd\nf\n')])
3260
2784
        tree_a = preview.get_preview_tree()
3261
2785
        tree_a.set_parent_ids([base_id])
3262
2786
        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)))
 
2787
            ('killed-a', 'a\n'),
 
2788
            ('killed-b', 'b\n'),
 
2789
            ('unchanged', 'c\n'),
 
2790
            ('unchanged', 'd\n'),
 
2791
            ('new-a', 'e\n'),
 
2792
            ('new-b', 'f\n'),
 
2793
        ], list(tree_a.plan_file_merge('file-id', tree_b)))
3270
2794
 
3271
2795
    def test_plan_file_merge_revision_tree(self):
3272
2796
        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')
 
2797
        self.build_tree_contents([('wta/file', 'a\nb\nc\nd\n')])
 
2798
        work_a.add('file', 'file-id')
3275
2799
        base_id = work_a.commit('base version')
3276
 
        tree_b = work_a.controldir.sprout('wtb').open_workingtree()
 
2800
        tree_b = work_a.bzrdir.sprout('wtb').open_workingtree()
3277
2801
        preview = TransformPreview(work_a.basis_tree())
3278
2802
        self.addCleanup(preview.finalize)
3279
 
        trans_id = preview.trans_id_file_id(b'file-id')
 
2803
        trans_id = preview.trans_id_file_id('file-id')
3280
2804
        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')])
 
2805
        preview.create_file('b\nc\nd\ne\n', trans_id)
 
2806
        self.build_tree_contents([('wtb/file', 'a\nc\nd\nf\n')])
3283
2807
        tree_a = preview.get_preview_tree()
3284
2808
        tree_a.set_parent_ids([base_id])
3285
2809
        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)))
 
2810
            ('killed-a', 'a\n'),
 
2811
            ('killed-b', 'b\n'),
 
2812
            ('unchanged', 'c\n'),
 
2813
            ('unchanged', 'd\n'),
 
2814
            ('new-a', 'e\n'),
 
2815
            ('new-b', 'f\n'),
 
2816
        ], list(tree_a.plan_file_merge('file-id', tree_b)))
3293
2817
 
3294
2818
    def test_walkdirs(self):
3295
2819
        preview = self.get_empty_preview()
3296
 
        preview.new_directory('', ROOT_PARENT, b'tree-root')
 
2820
        root = preview.new_directory('', ROOT_PARENT, 'tree-root')
3297
2821
        # FIXME: new_directory should mark root.
3298
2822
        preview.fixup_new_roots()
3299
2823
        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')])]
 
2824
        file_trans_id = preview.new_file('a', preview.root, 'contents',
 
2825
                                         'a-id')
 
2826
        expected = [(('', 'tree-root'),
 
2827
                    [('a', 'a', 'file', None, 'a-id', 'file')])]
3303
2828
        self.assertEqual(expected, list(preview_tree.walkdirs()))
3304
2829
 
3305
2830
    def test_extras(self):
3309
2834
        work_tree.add(['removed-file', 'not-removed-file'])
3310
2835
        preview = TransformPreview(work_tree)
3311
2836
        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')
 
2837
        preview.new_file('new-file', preview.root, 'contents')
 
2838
        preview.new_file('new-versioned-file', preview.root, 'contents',
 
2839
                         'new-versioned-id')
3315
2840
        tree = preview.get_preview_tree()
3316
2841
        preview.unversion_file(preview.trans_id_tree_path('removed-file'))
3317
 
        self.assertEqual({'new-file', 'removed-file', 'existing-file'},
 
2842
        self.assertEqual(set(['new-file', 'removed-file', 'existing-file']),
3318
2843
                         set(tree.extras()))
3319
2844
 
3320
2845
    def test_merge_into_preview(self):
3321
2846
        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')
 
2847
        self.build_tree_contents([('tree/file','b\n')])
 
2848
        work_tree.add('file', 'file-id')
3324
2849
        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')])
 
2850
        child_tree = work_tree.bzrdir.sprout('child').open_workingtree()
 
2851
        self.build_tree_contents([('child/file','b\nc\n')])
3327
2852
        child_tree.commit('child commit')
3328
2853
        child_tree.lock_write()
3329
2854
        self.addCleanup(child_tree.unlock)
3331
2856
        self.addCleanup(work_tree.unlock)
3332
2857
        preview = TransformPreview(work_tree)
3333
2858
        self.addCleanup(preview.finalize)
3334
 
        file_trans_id = preview.trans_id_file_id(b'file-id')
 
2859
        file_trans_id = preview.trans_id_file_id('file-id')
3335
2860
        preview.delete_contents(file_trans_id)
3336
 
        preview.create_file([b'a\nb\n'], file_trans_id)
 
2861
        preview.create_file('a\nb\n', file_trans_id)
3337
2862
        preview_tree = preview.get_preview_tree()
3338
 
        merger = Merger.from_revision_ids(preview_tree,
 
2863
        merger = Merger.from_revision_ids(None, preview_tree,
3339
2864
                                          child_tree.branch.last_revision(),
3340
2865
                                          other_branch=child_tree.branch,
3341
2866
                                          tree_branch=work_tree.branch)
3343
2868
        tt = merger.make_merger().make_preview_transform()
3344
2869
        self.addCleanup(tt.finalize)
3345
2870
        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')))
 
2871
        self.assertEqual('a\nb\nc\n', final_tree.get_file_text('file-id'))
3349
2872
 
3350
2873
    def test_merge_preview_into_workingtree(self):
3351
2874
        tree = self.make_branch_and_tree('tree')
3352
 
        tree.set_root_id(b'TREE_ROOT')
 
2875
        tree.set_root_id('TREE_ROOT')
3353
2876
        tt = TransformPreview(tree)
3354
2877
        self.addCleanup(tt.finalize)
3355
 
        tt.new_file('name', tt.root, [b'content'], b'file-id')
 
2878
        tt.new_file('name', tt.root, 'content', 'file-id')
3356
2879
        tree2 = self.make_branch_and_tree('tree2')
3357
 
        tree2.set_root_id(b'TREE_ROOT')
 
2880
        tree2.set_root_id('TREE_ROOT')
3358
2881
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
3359
 
                                         tree.basis_tree())
 
2882
                                         None, tree.basis_tree())
3360
2883
        merger.merge_type = Merge3Merger
3361
2884
        merger.do_merge()
3362
2885
 
3363
2886
    def test_merge_preview_into_workingtree_handles_conflicts(self):
3364
2887
        tree = self.make_branch_and_tree('tree')
3365
 
        self.build_tree_contents([('tree/foo', b'bar')])
3366
 
        tree.add('foo', b'foo-id')
 
2888
        self.build_tree_contents([('tree/foo', 'bar')])
 
2889
        tree.add('foo', 'foo-id')
3367
2890
        tree.commit('foo')
3368
2891
        tt = TransformPreview(tree)
3369
2892
        self.addCleanup(tt.finalize)
3370
 
        trans_id = tt.trans_id_file_id(b'foo-id')
 
2893
        trans_id = tt.trans_id_file_id('foo-id')
3371
2894
        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')])
 
2895
        tt.create_file('baz', trans_id)
 
2896
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
 
2897
        self.build_tree_contents([('tree2/foo', 'qux')])
 
2898
        pb = None
3375
2899
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
3376
 
                                         tree.basis_tree())
 
2900
                                         pb, tree.basis_tree())
3377
2901
        merger.merge_type = Merge3Merger
3378
2902
        merger.do_merge()
3379
2903
 
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
2904
    def test_is_executable(self):
3399
2905
        tree = self.make_branch_and_tree('tree')
3400
2906
        preview = TransformPreview(tree)
3401
2907
        self.addCleanup(preview.finalize)
3402
 
        preview.new_file('foo', preview.root, [b'bar'], b'baz-id')
 
2908
        preview.new_file('foo', preview.root, 'bar', 'baz-id')
3403
2909
        preview_tree = preview.get_preview_tree()
3404
 
        self.assertEqual(False, preview_tree.is_executable('tree/foo'))
 
2910
        self.assertEqual(False, preview_tree.is_executable('baz-id',
 
2911
                                                           'tree/foo'))
 
2912
        self.assertEqual(False, preview_tree.is_executable('baz-id'))
3405
2913
 
3406
2914
    def test_commit_preview_tree(self):
3407
2915
        tree = self.make_branch_and_tree('tree')
3409
2917
        tree.branch.lock_write()
3410
2918
        self.addCleanup(tree.branch.unlock)
3411
2919
        tt = TransformPreview(tree)
3412
 
        tt.new_file('file', tt.root, [b'contents'], b'file_id')
 
2920
        tt.new_file('file', tt.root, 'contents', 'file_id')
3413
2921
        self.addCleanup(tt.finalize)
3414
2922
        preview = tt.get_preview_tree()
3415
2923
        preview.set_parent_ids([rev_id])
3418
2926
        builder.finish_inventory()
3419
2927
        rev2_id = builder.commit('rev2')
3420
2928
        rev2_tree = tree.branch.repository.revision_tree(rev2_id)
3421
 
        self.assertEqual(b'contents', rev2_tree.get_file_text('file'))
 
2929
        self.assertEqual('contents', rev2_tree.get_file_text('file_id'))
3422
2930
 
3423
2931
    def test_ascii_limbo_paths(self):
3424
 
        self.requireFeature(features.UnicodeFilenameFeature)
 
2932
        self.requireFeature(tests.UnicodeFilenameFeature)
3425
2933
        branch = self.make_branch('any')
3426
2934
        tree = branch.repository.revision_tree(_mod_revision.NULL_REVISION)
3427
2935
        tt = TransformPreview(tree)
3428
2936
        self.addCleanup(tt.finalize)
3429
2937
        foo_id = tt.new_directory('', ROOT_PARENT)
3430
 
        bar_id = tt.new_file(u'\u1234bar', foo_id, [b'contents'])
 
2938
        bar_id = tt.new_file(u'\u1234bar', foo_id, 'contents')
3431
2939
        limbo_path = tt._limbo_name(bar_id)
3432
 
        self.assertEqual(limbo_path, limbo_path)
 
2940
        self.assertEqual(limbo_path.encode('ascii', 'replace'), limbo_path)
3433
2941
 
3434
2942
 
3435
2943
class FakeSerializer(object):
3444
2952
 
3445
2953
class TestSerializeTransform(tests.TestCaseWithTransport):
3446
2954
 
3447
 
    _test_needs_features = [features.UnicodeFilenameFeature]
 
2955
    _test_needs_features = [tests.UnicodeFilenameFeature]
3448
2956
 
3449
2957
    def get_preview(self, tree=None):
3450
2958
        if tree is None:
3460
2968
    @staticmethod
3461
2969
    def default_attribs():
3462
2970
        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': {},
 
2971
            '_id_number': 1,
 
2972
            '_new_name': {},
 
2973
            '_new_parent': {},
 
2974
            '_new_executability': {},
 
2975
            '_new_id': {},
 
2976
            '_tree_path_ids': {'': 'new-0'},
 
2977
            '_removed_id': [],
 
2978
            '_removed_contents': [],
 
2979
            '_non_present_ids': {},
3472
2980
            }
3473
2981
 
3474
2982
    def make_records(self, attribs, contents):
3475
2983
        records = [
3476
 
            ((((b'attribs'),),), bencode.bencode(attribs))]
 
2984
            (((('attribs'),),), bencode.bencode(attribs))]
3477
2985
        records.extend([(((n, k),), c) for n, k, c in contents])
3478
2986
        return records
3479
2987
 
3480
2988
    def creation_records(self):
3481
2989
        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}
 
2990
        attribs['_id_number'] = 3
 
2991
        attribs['_new_name'] = {
 
2992
            'new-1': u'foo\u1234'.encode('utf-8'), 'new-2': 'qux'}
 
2993
        attribs['_new_id'] = {'new-1': 'baz', 'new-2': 'quxx'}
 
2994
        attribs['_new_parent'] = {'new-1': 'new-0', 'new-2': 'new-0'}
 
2995
        attribs['_new_executability'] = {'new-1': 1}
3488
2996
        contents = [
3489
 
            (b'new-1', b'file', b'i 1\nbar\n'),
3490
 
            (b'new-2', b'directory', b''),
 
2997
            ('new-1', 'file', 'i 1\nbar\n'),
 
2998
            ('new-2', 'directory', ''),
3491
2999
            ]
3492
3000
        return self.make_records(attribs, contents)
3493
3001
 
3494
3002
    def test_serialize_creation(self):
3495
3003
        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')
 
3004
        tt.new_file(u'foo\u1234', tt.root, 'bar', 'baz', True)
 
3005
        tt.new_directory('qux', tt.root, 'quxx')
3498
3006
        self.assertSerializesTo(self.creation_records(), tt)
3499
3007
 
3500
3008
    def test_deserialize_creation(self):
3503
3011
        self.assertEqual(3, tt._id_number)
3504
3012
        self.assertEqual({'new-1': u'foo\u1234',
3505
3013
                          'new-2': 'qux'}, tt._new_name)
3506
 
        self.assertEqual({'new-1': b'baz', 'new-2': b'quxx'}, tt._new_id)
 
3014
        self.assertEqual({'new-1': 'baz', 'new-2': 'quxx'}, tt._new_id)
3507
3015
        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)
 
3016
        self.assertEqual({'baz': 'new-1', 'quxx': 'new-2'}, tt._r_new_id)
3509
3017
        self.assertEqual({'new-1': True}, tt._new_executability)
3510
3018
        self.assertEqual({'new-1': 'file',
3511
3019
                          'new-2': 'directory'}, tt._new_contents)
3514
3022
            foo_content = foo_limbo.read()
3515
3023
        finally:
3516
3024
            foo_limbo.close()
3517
 
        self.assertEqual(b'bar', foo_content)
 
3025
        self.assertEqual('bar', foo_content)
3518
3026
 
3519
3027
    def symlink_creation_records(self):
3520
3028
        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'))]
 
3029
        attribs['_id_number'] = 2
 
3030
        attribs['_new_name'] = {'new-1': u'foo\u1234'.encode('utf-8')}
 
3031
        attribs['_new_parent'] = {'new-1': 'new-0'}
 
3032
        contents = [('new-1', 'symlink', u'bar\u1234'.encode('utf-8'))]
3525
3033
        return self.make_records(attribs, contents)
3526
3034
 
3527
3035
    def test_serialize_symlink_creation(self):
3528
 
        self.requireFeature(features.SymlinkFeature)
 
3036
        self.requireFeature(tests.SymlinkFeature)
3529
3037
        tt = self.get_preview()
3530
3038
        tt.new_symlink(u'foo\u1234', tt.root, u'bar\u1234')
3531
3039
        self.assertSerializesTo(self.symlink_creation_records(), tt)
3532
3040
 
3533
3041
    def test_deserialize_symlink_creation(self):
3534
 
        self.requireFeature(features.SymlinkFeature)
 
3042
        self.requireFeature(tests.SymlinkFeature)
3535
3043
        tt = self.get_preview()
3536
3044
        tt.deserialize(iter(self.symlink_creation_records()))
3537
3045
        abspath = tt._limbo_name('new-1')
3541
3049
    def make_destruction_preview(self):
3542
3050
        tree = self.make_branch_and_tree('.')
3543
3051
        self.build_tree([u'foo\u1234', 'bar'])
3544
 
        tree.add([u'foo\u1234', 'bar'], [b'foo-id', b'bar-id'])
 
3052
        tree.add([u'foo\u1234', 'bar'], ['foo-id', 'bar-id'])
3545
3053
        return self.get_preview(tree)
3546
3054
 
3547
3055
    def destruction_records(self):
3548
3056
        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',
 
3057
        attribs['_id_number'] = 3
 
3058
        attribs['_removed_id'] = ['new-1']
 
3059
        attribs['_removed_contents'] = ['new-2']
 
3060
        attribs['_tree_path_ids'] = {
 
3061
            '': 'new-0',
 
3062
            u'foo\u1234'.encode('utf-8'): 'new-1',
 
3063
            'bar': 'new-2',
3556
3064
            }
3557
3065
        return self.make_records(attribs, [])
3558
3066
 
3559
3067
    def test_serialize_destruction(self):
3560
3068
        tt = self.make_destruction_preview()
3561
 
        foo_trans_id = tt.trans_id_tree_path(u'foo\u1234')
 
3069
        foo_trans_id = tt.trans_id_tree_file_id('foo-id')
3562
3070
        tt.unversion_file(foo_trans_id)
3563
 
        bar_trans_id = tt.trans_id_tree_path('bar')
 
3071
        bar_trans_id = tt.trans_id_tree_file_id('bar-id')
3564
3072
        tt.delete_contents(bar_trans_id)
3565
3073
        self.assertSerializesTo(self.destruction_records(), tt)
3566
3074
 
3573
3081
        self.assertEqual({'new-1': u'foo\u1234',
3574
3082
                          'new-2': 'bar',
3575
3083
                          tt.root: ''}, tt._tree_id_paths)
3576
 
        self.assertEqual({'new-1'}, tt._removed_id)
3577
 
        self.assertEqual({'new-2'}, tt._removed_contents)
 
3084
        self.assertEqual(set(['new-1']), tt._removed_id)
 
3085
        self.assertEqual(set(['new-2']), tt._removed_contents)
3578
3086
 
3579
3087
    def missing_records(self):
3580
3088
        attribs = self.default_attribs()
3581
 
        attribs[b'_id_number'] = 2
3582
 
        attribs[b'_non_present_ids'] = {
3583
 
            b'boo': b'new-1', }
 
3089
        attribs['_id_number'] = 2
 
3090
        attribs['_non_present_ids'] = {
 
3091
            'boo': 'new-1',}
3584
3092
        return self.make_records(attribs, [])
3585
3093
 
3586
3094
    def test_serialize_missing(self):
3587
3095
        tt = self.get_preview()
3588
 
        tt.trans_id_file_id(b'boo')
 
3096
        boo_trans_id = tt.trans_id_file_id('boo')
3589
3097
        self.assertSerializesTo(self.missing_records(), tt)
3590
3098
 
3591
3099
    def test_deserialize_missing(self):
3592
3100
        tt = self.get_preview()
3593
3101
        tt.deserialize(iter(self.missing_records()))
3594
 
        self.assertEqual({b'boo': 'new-1'}, tt._non_present_ids)
 
3102
        self.assertEqual({'boo': 'new-1'}, tt._non_present_ids)
3595
3103
 
3596
3104
    def make_modification_preview(self):
3597
 
        LINES_ONE = b'aa\nbb\ncc\ndd\n'
3598
 
        LINES_TWO = b'z\nbb\nx\ndd\n'
 
3105
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
 
3106
        LINES_TWO = 'z\nbb\nx\ndd\n'
3599
3107
        tree = self.make_branch_and_tree('tree')
3600
3108
        self.build_tree_contents([('tree/file', LINES_ONE)])
3601
 
        tree.add('file', b'file-id')
3602
 
        return self.get_preview(tree), [LINES_TWO]
 
3109
        tree.add('file', 'file-id')
 
3110
        return self.get_preview(tree), LINES_TWO
3603
3111
 
3604
3112
    def modification_records(self):
3605
3113
        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')]
 
3114
        attribs['_id_number'] = 2
 
3115
        attribs['_tree_path_ids'] = {
 
3116
            'file': 'new-1',
 
3117
            '': 'new-0',}
 
3118
        attribs['_removed_contents'] = ['new-1']
 
3119
        contents = [('new-1', 'file',
 
3120
                     'i 1\nz\n\nc 0 1 1 1\ni 1\nx\n\nc 0 3 3 1\n')]
3613
3121
        return self.make_records(attribs, contents)
3614
3122
 
3615
3123
    def test_serialize_modification(self):
3616
3124
        tt, LINES = self.make_modification_preview()
3617
 
        trans_id = tt.trans_id_file_id(b'file-id')
 
3125
        trans_id = tt.trans_id_file_id('file-id')
3618
3126
        tt.delete_contents(trans_id)
3619
3127
        tt.create_file(LINES, trans_id)
3620
3128
        self.assertSerializesTo(self.modification_records(), tt)
3622
3130
    def test_deserialize_modification(self):
3623
3131
        tt, LINES = self.make_modification_preview()
3624
3132
        tt.deserialize(iter(self.modification_records()))
3625
 
        self.assertFileEqual(b''.join(LINES), tt._limbo_name('new-1'))
 
3133
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3626
3134
 
3627
3135
    def make_kind_change_preview(self):
3628
 
        LINES = b'a\nb\nc\nd\n'
 
3136
        LINES = 'a\nb\nc\nd\n'
3629
3137
        tree = self.make_branch_and_tree('tree')
3630
3138
        self.build_tree(['tree/foo/'])
3631
 
        tree.add('foo', b'foo-id')
3632
 
        return self.get_preview(tree), [LINES]
 
3139
        tree.add('foo', 'foo-id')
 
3140
        return self.get_preview(tree), LINES
3633
3141
 
3634
3142
    def kind_change_records(self):
3635
3143
        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')]
 
3144
        attribs['_id_number'] = 2
 
3145
        attribs['_tree_path_ids'] = {
 
3146
            'foo': 'new-1',
 
3147
            '': 'new-0',}
 
3148
        attribs['_removed_contents'] = ['new-1']
 
3149
        contents = [('new-1', 'file',
 
3150
                     'i 4\na\nb\nc\nd\n\n')]
3643
3151
        return self.make_records(attribs, contents)
3644
3152
 
3645
3153
    def test_serialize_kind_change(self):
3646
3154
        tt, LINES = self.make_kind_change_preview()
3647
 
        trans_id = tt.trans_id_file_id(b'foo-id')
 
3155
        trans_id = tt.trans_id_file_id('foo-id')
3648
3156
        tt.delete_contents(trans_id)
3649
3157
        tt.create_file(LINES, trans_id)
3650
3158
        self.assertSerializesTo(self.kind_change_records(), tt)
3652
3160
    def test_deserialize_kind_change(self):
3653
3161
        tt, LINES = self.make_kind_change_preview()
3654
3162
        tt.deserialize(iter(self.kind_change_records()))
3655
 
        self.assertFileEqual(b''.join(LINES), tt._limbo_name('new-1'))
 
3163
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3656
3164
 
3657
3165
    def make_add_contents_preview(self):
3658
 
        LINES = b'a\nb\nc\nd\n'
 
3166
        LINES = 'a\nb\nc\nd\n'
3659
3167
        tree = self.make_branch_and_tree('tree')
3660
3168
        self.build_tree(['tree/foo'])
3661
3169
        tree.add('foo')
3664
3172
 
3665
3173
    def add_contents_records(self):
3666
3174
        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')]
 
3175
        attribs['_id_number'] = 2
 
3176
        attribs['_tree_path_ids'] = {
 
3177
            'foo': 'new-1',
 
3178
            '': 'new-0',}
 
3179
        contents = [('new-1', 'file',
 
3180
                     'i 4\na\nb\nc\nd\n\n')]
3673
3181
        return self.make_records(attribs, contents)
3674
3182
 
3675
3183
    def test_serialize_add_contents(self):
3676
3184
        tt, LINES = self.make_add_contents_preview()
3677
3185
        trans_id = tt.trans_id_tree_path('foo')
3678
 
        tt.create_file([LINES], trans_id)
 
3186
        tt.create_file(LINES, trans_id)
3679
3187
        self.assertSerializesTo(self.add_contents_records(), tt)
3680
3188
 
3681
3189
    def test_deserialize_add_contents(self):
3684
3192
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3685
3193
 
3686
3194
    def test_get_parents_lines(self):
3687
 
        LINES_ONE = b'aa\nbb\ncc\ndd\n'
 
3195
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
 
3196
        LINES_TWO = 'z\nbb\nx\ndd\n'
3688
3197
        tree = self.make_branch_and_tree('tree')
3689
3198
        self.build_tree_contents([('tree/file', LINES_ONE)])
3690
 
        tree.add('file', b'file-id')
 
3199
        tree.add('file', 'file-id')
3691
3200
        tt = self.get_preview(tree)
3692
3201
        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))
 
3202
        self.assertEqual((['aa\n', 'bb\n', 'cc\n', 'dd\n'],),
 
3203
            tt._get_parents_lines(trans_id))
3695
3204
 
3696
3205
    def test_get_parents_texts(self):
3697
 
        LINES_ONE = b'aa\nbb\ncc\ndd\n'
 
3206
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
 
3207
        LINES_TWO = 'z\nbb\nx\ndd\n'
3698
3208
        tree = self.make_branch_and_tree('tree')
3699
3209
        self.build_tree_contents([('tree/file', LINES_ONE)])
3700
 
        tree.add('file', b'file-id')
 
3210
        tree.add('file', 'file-id')
3701
3211
        tt = self.get_preview(tree)
3702
3212
        trans_id = tt.trans_id_tree_path('file')
3703
3213
        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())
 
3214
            tt._get_parents_texts(trans_id))