/brz/remove-bazaar

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

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_transform.py

  • Committer: Robert Collins
  • Date: 2010-05-06 23:41:35 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506234135-yivbzczw1sejxnxc
Lock methods on ``Tree``, ``Branch`` and ``Repository`` are now
expected to return an object which can be used to unlock them. This reduces
duplicate code when using cleanups. The previous 'tokens's returned by
``Branch.lock_write`` and ``Repository.lock_write`` are now attributes
on the result of the lock_write. ``repository.RepositoryWriteLockResult``
and ``branch.BranchWriteLockResult`` document this. (Robert Collins)

``log._get_info_for_log_files`` now takes an add_cleanup callable.
(Robert Collins)

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