~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_transform.py

  • Committer: Aaron Bentley
  • Date: 2008-09-23 17:00:03 UTC
  • mto: This revision was merged to the branch mainline in revision 3728.
  • Revision ID: aaron@aaronbentley.com-20080923170003-8vh06dg6jewct4ta
Correct spelling of 'Application Data'

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2006-2011 Canonical Ltd
 
1
# Copyright (C) 2006, 2007, 2008 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
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
 
import errno
18
17
import os
 
18
import stat
19
19
from StringIO import StringIO
20
20
import sys
21
 
import time
22
21
 
23
22
from bzrlib import (
24
 
    bencode,
25
23
    errors,
26
 
    filters,
27
24
    generate_ids,
28
25
    osutils,
 
26
    progress,
29
27
    revision as _mod_revision,
30
 
    rules,
31
28
    symbol_versioning,
32
29
    tests,
33
 
    trace,
34
 
    transform,
35
30
    urlutils,
36
31
    )
37
32
from bzrlib.bzrdir import BzrDir
38
 
from bzrlib.conflicts import (
39
 
    DeletingParent,
40
 
    DuplicateEntry,
41
 
    DuplicateID,
42
 
    MissingParent,
43
 
    NonDirectoryParent,
44
 
    ParentLoop,
45
 
    UnversionedParent,
46
 
)
 
33
from bzrlib.conflicts import (DuplicateEntry, DuplicateID, MissingParent,
 
34
                              UnversionedParent, ParentLoop, DeletingParent,
 
35
                              NonDirectoryParent)
47
36
from bzrlib.diff import show_diff_trees
48
 
from bzrlib.errors import (
49
 
    DuplicateKey,
50
 
    ExistingLimbo,
51
 
    ExistingPendingDeletion,
52
 
    ImmortalLimbo,
53
 
    ImmortalPendingDeletion,
54
 
    LockError,
55
 
    MalformedTransform,
56
 
    ReusingTransform,
57
 
)
58
 
from bzrlib.osutils import (
59
 
    file_kind,
60
 
    pathjoin,
61
 
)
62
 
from bzrlib.merge import Merge3Merger, Merger
 
37
from bzrlib.errors import (DuplicateKey, MalformedTransform, NoSuchFile,
 
38
                           ReusingTransform, CantMoveRoot, 
 
39
                           PathsNotVersionedError, ExistingLimbo,
 
40
                           ExistingPendingDeletion, ImmortalLimbo,
 
41
                           ImmortalPendingDeletion, LockError)
 
42
from bzrlib.osutils import file_kind, pathjoin
 
43
from bzrlib.merge import Merge3Merger
63
44
from bzrlib.tests import (
64
 
    features,
 
45
    CaseInsensitiveFilesystemFeature,
 
46
    HardlinkFeature,
 
47
    SymlinkFeature,
 
48
    TestCase,
65
49
    TestCaseInTempDir,
66
50
    TestSkipped,
67
51
    )
68
 
from bzrlib.tests.features import (
69
 
    HardlinkFeature,
70
 
    SymlinkFeature,
71
 
    )
72
 
from bzrlib.transform import (
73
 
    build_tree,
74
 
    create_from_tree,
75
 
    cook_conflicts,
76
 
    _FileMover,
77
 
    FinalPaths,
78
 
    resolve_conflicts,
79
 
    resolve_checkout,
80
 
    ROOT_PARENT,
81
 
    TransformPreview,
82
 
    TreeTransform,
83
 
)
84
 
 
 
52
from bzrlib.transform import (TreeTransform, ROOT_PARENT, FinalPaths, 
 
53
                              resolve_conflicts, cook_conflicts, 
 
54
                              build_tree, get_backup_name,
 
55
                              _FileMover, resolve_checkout,
 
56
                              TransformPreview)
85
57
 
86
58
class TestTreeTransform(tests.TestCaseWithTransport):
87
59
 
95
67
        self.addCleanup(transform.finalize)
96
68
        return transform, transform.root
97
69
 
98
 
    def get_transform_for_sha1_test(self):
99
 
        trans, root = self.get_transform()
100
 
        self.wt.lock_tree_write()
101
 
        self.addCleanup(self.wt.unlock)
102
 
        contents = ['just some content\n']
103
 
        sha1 = osutils.sha_strings(contents)
104
 
        # Roll back the clock
105
 
        trans._creation_mtime = time.time() - 20.0
106
 
        return trans, root, contents, sha1
107
 
 
108
70
    def test_existing_limbo(self):
109
71
        transform, root = self.get_transform()
110
72
        limbo_name = transform._limbodir
137
99
        imaginary_id = transform.trans_id_tree_path('imaginary')
138
100
        imaginary_id2 = transform.trans_id_tree_path('imaginary/')
139
101
        self.assertEqual(imaginary_id, imaginary_id2)
140
 
        self.assertEqual(root, transform.get_tree_parent(imaginary_id))
141
 
        self.assertEqual('directory', transform.final_kind(root))
142
 
        self.assertEqual(self.wt.get_root_id(), transform.final_file_id(root))
 
102
        self.assertEqual(transform.get_tree_parent(imaginary_id), root)
 
103
        self.assertEqual(transform.final_kind(root), 'directory')
 
104
        self.assertEqual(transform.final_file_id(root), self.wt.get_root_id())
143
105
        trans_id = transform.create_path('name', root)
144
106
        self.assertIs(transform.final_file_id(trans_id), None)
145
 
        self.assertIs(None, transform.final_kind(trans_id))
 
107
        self.assertRaises(NoSuchFile, transform.final_kind, trans_id)
146
108
        transform.create_file('contents', trans_id)
147
109
        transform.set_executability(True, trans_id)
148
110
        transform.version_file('my_pretties', trans_id)
166
128
        self.assertIs(self.wt.is_executable('my_pretties2'), False)
167
129
        self.assertEqual('directory', file_kind(self.wt.abspath('oz')))
168
130
        self.assertEqual(len(modified_paths), 3)
169
 
        tree_mod_paths = [self.wt.id2abspath(f) for f in
 
131
        tree_mod_paths = [self.wt.id2abspath(f) for f in 
170
132
                          ('ozzie', 'my_pretties', 'my_pretties2')]
171
133
        self.assertSubset(tree_mod_paths, modified_paths)
172
134
        # is it safe to finalize repeatedly?
173
135
        transform.finalize()
174
136
        transform.finalize()
175
137
 
176
 
    def test_apply_informs_tree_of_observed_sha1(self):
177
 
        trans, root, contents, sha1 = self.get_transform_for_sha1_test()
178
 
        trans_id = trans.new_file('file1', root, contents, file_id='file1-id',
179
 
                                  sha1=sha1)
180
 
        calls = []
181
 
        orig = self.wt._observed_sha1
182
 
        def _observed_sha1(*args):
183
 
            calls.append(args)
184
 
            orig(*args)
185
 
        self.wt._observed_sha1 = _observed_sha1
186
 
        trans.apply()
187
 
        self.assertEqual([(None, 'file1', trans._observed_sha1s[trans_id])],
188
 
                         calls)
189
 
 
190
 
    def test_create_file_caches_sha1(self):
191
 
        trans, root, contents, sha1 = self.get_transform_for_sha1_test()
192
 
        trans_id = trans.create_path('file1', root)
193
 
        trans.create_file(contents, trans_id, sha1=sha1)
194
 
        st_val = osutils.lstat(trans._limbo_name(trans_id))
195
 
        o_sha1, o_st_val = trans._observed_sha1s[trans_id]
196
 
        self.assertEqual(o_sha1, sha1)
197
 
        self.assertEqualStat(o_st_val, st_val)
198
 
 
199
 
    def test__apply_insertions_updates_sha1(self):
200
 
        trans, root, contents, sha1 = self.get_transform_for_sha1_test()
201
 
        trans_id = trans.create_path('file1', root)
202
 
        trans.create_file(contents, trans_id, sha1=sha1)
203
 
        st_val = osutils.lstat(trans._limbo_name(trans_id))
204
 
        o_sha1, o_st_val = trans._observed_sha1s[trans_id]
205
 
        self.assertEqual(o_sha1, sha1)
206
 
        self.assertEqualStat(o_st_val, st_val)
207
 
        creation_mtime = trans._creation_mtime + 10.0
208
 
        # We fake a time difference from when the file was created until now it
209
 
        # is being renamed by using os.utime. Note that the change we actually
210
 
        # want to see is the real ctime change from 'os.rename()', but as long
211
 
        # as we observe a new stat value, we should be fine.
212
 
        os.utime(trans._limbo_name(trans_id), (creation_mtime, creation_mtime))
213
 
        trans.apply()
214
 
        new_st_val = osutils.lstat(self.wt.abspath('file1'))
215
 
        o_sha1, o_st_val = trans._observed_sha1s[trans_id]
216
 
        self.assertEqual(o_sha1, sha1)
217
 
        self.assertEqualStat(o_st_val, new_st_val)
218
 
        self.assertNotEqual(st_val.st_mtime, new_st_val.st_mtime)
219
 
 
220
 
    def test_new_file_caches_sha1(self):
221
 
        trans, root, contents, sha1 = self.get_transform_for_sha1_test()
222
 
        trans_id = trans.new_file('file1', root, contents, file_id='file1-id',
223
 
                                  sha1=sha1)
224
 
        st_val = osutils.lstat(trans._limbo_name(trans_id))
225
 
        o_sha1, o_st_val = trans._observed_sha1s[trans_id]
226
 
        self.assertEqual(o_sha1, sha1)
227
 
        self.assertEqualStat(o_st_val, st_val)
228
 
 
229
 
    def test_cancel_creation_removes_observed_sha1(self):
230
 
        trans, root, contents, sha1 = self.get_transform_for_sha1_test()
231
 
        trans_id = trans.new_file('file1', root, contents, file_id='file1-id',
232
 
                                  sha1=sha1)
233
 
        self.assertTrue(trans_id in trans._observed_sha1s)
234
 
        trans.cancel_creation(trans_id)
235
 
        self.assertFalse(trans_id in trans._observed_sha1s)
236
 
 
237
 
    def test_create_files_same_timestamp(self):
238
 
        transform, root = self.get_transform()
239
 
        self.wt.lock_tree_write()
240
 
        self.addCleanup(self.wt.unlock)
241
 
        # Roll back the clock, so that we know everything is being set to the
242
 
        # exact time
243
 
        transform._creation_mtime = creation_mtime = time.time() - 20.0
244
 
        transform.create_file('content-one',
245
 
                              transform.create_path('one', root))
246
 
        time.sleep(1) # *ugly*
247
 
        transform.create_file('content-two',
248
 
                              transform.create_path('two', root))
249
 
        transform.apply()
250
 
        fo, st1 = self.wt.get_file_with_stat(None, path='one', filtered=False)
251
 
        fo.close()
252
 
        fo, st2 = self.wt.get_file_with_stat(None, path='two', filtered=False)
253
 
        fo.close()
254
 
        # We only guarantee 2s resolution
255
 
        self.assertTrue(abs(creation_mtime - st1.st_mtime) < 2.0,
256
 
            "%s != %s within 2 seconds" % (creation_mtime, st1.st_mtime))
257
 
        # But if we have more than that, all files should get the same result
258
 
        self.assertEqual(st1.st_mtime, st2.st_mtime)
259
 
 
260
 
    def test_change_root_id(self):
261
 
        transform, root = self.get_transform()
262
 
        self.assertNotEqual('new-root-id', self.wt.get_root_id())
263
 
        transform.new_directory('', ROOT_PARENT, 'new-root-id')
264
 
        transform.delete_contents(root)
265
 
        transform.unversion_file(root)
266
 
        transform.fixup_new_roots()
267
 
        transform.apply()
268
 
        self.assertEqual('new-root-id', self.wt.get_root_id())
269
 
 
270
 
    def test_change_root_id_add_files(self):
271
 
        transform, root = self.get_transform()
272
 
        self.assertNotEqual('new-root-id', self.wt.get_root_id())
273
 
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'new-root-id')
274
 
        transform.new_file('file', new_trans_id, ['new-contents\n'],
275
 
                           'new-file-id')
276
 
        transform.delete_contents(root)
277
 
        transform.unversion_file(root)
278
 
        transform.fixup_new_roots()
279
 
        transform.apply()
280
 
        self.assertEqual('new-root-id', self.wt.get_root_id())
281
 
        self.assertEqual('new-file-id', self.wt.path2id('file'))
282
 
        self.assertFileEqual('new-contents\n', self.wt.abspath('file'))
283
 
 
284
 
    def test_add_two_roots(self):
285
 
        transform, root = self.get_transform()
286
 
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'new-root-id')
287
 
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'alt-root-id')
288
 
        self.assertRaises(ValueError, transform.fixup_new_roots)
289
 
 
290
 
    def test_retain_existing_root(self):
291
 
        tt, root = self.get_transform()
292
 
        with tt:
293
 
            tt.new_directory('', ROOT_PARENT, 'new-root-id')
294
 
            tt.fixup_new_roots()
295
 
            self.assertNotEqual('new-root-id', tt.final_file_id(tt.root))
296
 
 
297
 
    def test_retain_existing_root_added_file(self):
298
 
        tt, root = self.get_transform()
299
 
        new_trans_id = tt.new_directory('', ROOT_PARENT, 'new-root-id')
300
 
        child = tt.new_directory('child', new_trans_id, 'child-id')
301
 
        tt.fixup_new_roots()
302
 
        self.assertEqual(tt.root, tt.final_parent(child))
303
 
 
304
 
    def test_add_unversioned_root(self):
305
 
        transform, root = self.get_transform()
306
 
        new_trans_id = transform.new_directory('', ROOT_PARENT, None)
307
 
        transform.delete_contents(transform.root)
308
 
        transform.fixup_new_roots()
309
 
        self.assertNotIn(transform.root, transform._new_id)
310
 
 
311
 
    def test_remove_root_fixup(self):
312
 
        transform, root = self.get_transform()
313
 
        old_root_id = self.wt.get_root_id()
314
 
        self.assertNotEqual('new-root-id', old_root_id)
315
 
        transform.delete_contents(root)
316
 
        transform.unversion_file(root)
317
 
        transform.fixup_new_roots()
318
 
        transform.apply()
319
 
        self.assertEqual(old_root_id, self.wt.get_root_id())
320
 
 
321
 
        transform, root = self.get_transform()
322
 
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'new-root-id')
323
 
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'alt-root-id')
324
 
        self.assertRaises(ValueError, transform.fixup_new_roots)
325
 
 
326
 
    def test_fixup_new_roots_permits_empty_tree(self):
327
 
        transform, root = self.get_transform()
328
 
        transform.delete_contents(root)
329
 
        transform.unversion_file(root)
330
 
        transform.fixup_new_roots()
331
 
        self.assertIs(None, transform.final_kind(root))
332
 
        self.assertIs(None, transform.final_file_id(root))
333
 
 
334
 
    def test_apply_retains_root_directory(self):
335
 
        # Do not attempt to delete the physical root directory, because that
336
 
        # is impossible.
337
 
        transform, root = self.get_transform()
338
 
        with transform:
339
 
            transform.delete_contents(root)
340
 
            e = self.assertRaises(AssertionError, self.assertRaises,
341
 
                                  errors.TransformRenameFailed,
342
 
                                  transform.apply)
343
 
        self.assertContainsRe('TransformRenameFailed not raised', str(e))
344
 
 
345
 
    def test_apply_retains_file_id(self):
346
 
        transform, root = self.get_transform()
347
 
        old_root_id = transform.tree_file_id(root)
348
 
        transform.unversion_file(root)
349
 
        transform.apply()
350
 
        self.assertEqual(old_root_id, self.wt.get_root_id())
351
 
 
352
138
    def test_hardlink(self):
353
139
        self.requireFeature(HardlinkFeature)
354
140
        transform, root = self.get_transform()
359
145
        trans_id = target_transform.create_path('file1', target_transform.root)
360
146
        target_transform.create_hardlink(self.wt.abspath('file1'), trans_id)
361
147
        target_transform.apply()
362
 
        self.assertPathExists('target/file1')
 
148
        self.failUnlessExists('target/file1')
363
149
        source_stat = os.stat(self.wt.abspath('file1'))
364
150
        target_stat = os.stat('target/file1')
365
151
        self.assertEqual(source_stat, target_stat)
368
154
        transform, root = self.get_transform()
369
155
        self.wt.lock_tree_write()
370
156
        self.addCleanup(self.wt.unlock)
371
 
        trans_id = transform.new_file('name', root, 'contents',
 
157
        trans_id = transform.new_file('name', root, 'contents', 
372
158
                                      'my_pretties', True)
373
159
        oz = transform.new_directory('oz', root, 'oz-id')
374
160
        dorothy = transform.new_directory('dorothy', oz, 'dorothy-id')
375
 
        toto = transform.new_file('toto', dorothy, 'toto-contents',
 
161
        toto = transform.new_file('toto', dorothy, 'toto-contents', 
376
162
                                  'toto-id', False)
377
163
 
378
164
        self.assertEqual(len(transform.find_conflicts()), 0)
402
188
 
403
189
    def test_conflicts(self):
404
190
        transform, root = self.get_transform()
405
 
        trans_id = transform.new_file('name', root, 'contents',
 
191
        trans_id = transform.new_file('name', root, 'contents', 
406
192
                                      'my_pretties')
407
193
        self.assertEqual(len(transform.find_conflicts()), 0)
408
194
        trans_id2 = transform.new_file('name', root, 'Crontents', 'toto')
409
 
        self.assertEqual(transform.find_conflicts(),
 
195
        self.assertEqual(transform.find_conflicts(), 
410
196
                         [('duplicate', trans_id, trans_id2, 'name')])
411
197
        self.assertRaises(MalformedTransform, transform.apply)
412
198
        transform.adjust_path('name', trans_id, trans_id2)
413
 
        self.assertEqual(transform.find_conflicts(),
 
199
        self.assertEqual(transform.find_conflicts(), 
414
200
                         [('non-directory parent', trans_id)])
415
201
        tinman_id = transform.trans_id_tree_path('tinman')
416
202
        transform.adjust_path('name', tinman_id, trans_id2)
417
 
        self.assertEqual(transform.find_conflicts(),
418
 
                         [('unversioned parent', tinman_id),
 
203
        self.assertEqual(transform.find_conflicts(), 
 
204
                         [('unversioned parent', tinman_id), 
419
205
                          ('missing parent', tinman_id)])
420
206
        lion_id = transform.create_path('lion', root)
421
 
        self.assertEqual(transform.find_conflicts(),
422
 
                         [('unversioned parent', tinman_id),
 
207
        self.assertEqual(transform.find_conflicts(), 
 
208
                         [('unversioned parent', tinman_id), 
423
209
                          ('missing parent', tinman_id)])
424
210
        transform.adjust_path('name', lion_id, trans_id2)
425
 
        self.assertEqual(transform.find_conflicts(),
 
211
        self.assertEqual(transform.find_conflicts(), 
426
212
                         [('unversioned parent', lion_id),
427
213
                          ('missing parent', lion_id)])
428
214
        transform.version_file("Courage", lion_id)
429
 
        self.assertEqual(transform.find_conflicts(),
430
 
                         [('missing parent', lion_id),
 
215
        self.assertEqual(transform.find_conflicts(), 
 
216
                         [('missing parent', lion_id), 
431
217
                          ('versioning no contents', lion_id)])
432
218
        transform.adjust_path('name2', root, trans_id2)
433
 
        self.assertEqual(transform.find_conflicts(),
 
219
        self.assertEqual(transform.find_conflicts(), 
434
220
                         [('versioning no contents', lion_id)])
435
221
        transform.create_file('Contents, okay?', lion_id)
436
222
        transform.adjust_path('name2', trans_id2, trans_id2)
437
 
        self.assertEqual(transform.find_conflicts(),
438
 
                         [('parent loop', trans_id2),
 
223
        self.assertEqual(transform.find_conflicts(), 
 
224
                         [('parent loop', trans_id2), 
439
225
                          ('non-directory parent', trans_id2)])
440
226
        transform.adjust_path('name2', root, trans_id2)
441
227
        oz_id = transform.new_directory('oz', root)
442
228
        transform.set_executability(True, oz_id)
443
 
        self.assertEqual(transform.find_conflicts(),
 
229
        self.assertEqual(transform.find_conflicts(), 
444
230
                         [('unversioned executability', oz_id)])
445
231
        transform.version_file('oz-id', oz_id)
446
 
        self.assertEqual(transform.find_conflicts(),
 
232
        self.assertEqual(transform.find_conflicts(), 
447
233
                         [('non-file executability', oz_id)])
448
234
        transform.set_executability(None, oz_id)
449
235
        tip_id = transform.new_file('tip', oz_id, 'ozma', 'tip-id')
458
244
        self.assert_('oz/tip' in transform2._tree_path_ids)
459
245
        self.assertEqual(fp.get_path(newtip), pathjoin('oz', 'tip'))
460
246
        self.assertEqual(len(result), 2)
461
 
        self.assertEqual((result[0][0], result[0][1]),
 
247
        self.assertEqual((result[0][0], result[0][1]), 
462
248
                         ('duplicate', newtip))
463
 
        self.assertEqual((result[1][0], result[1][2]),
 
249
        self.assertEqual((result[1][0], result[1][2]), 
464
250
                         ('duplicate id', newtip))
465
251
        transform2.finalize()
466
252
        transform3 = TreeTransform(self.wt)
467
253
        self.addCleanup(transform3.finalize)
468
254
        oz_id = transform3.trans_id_tree_file_id('oz-id')
469
255
        transform3.delete_contents(oz_id)
470
 
        self.assertEqual(transform3.find_conflicts(),
 
256
        self.assertEqual(transform3.find_conflicts(), 
471
257
                         [('missing parent', oz_id)])
472
258
        root_id = transform3.root
473
259
        tip_id = transform3.trans_id_tree_file_id('tip-id')
531
317
        transform.new_file('FiLe', transform.root, 'content')
532
318
        resolve_conflicts(transform)
533
319
        transform.apply()
534
 
        self.assertPathExists('tree/file')
535
 
        self.assertPathExists('tree/FiLe.moved')
 
320
        self.failUnlessExists('tree/file')
 
321
        self.failUnlessExists('tree/FiLe.moved')
536
322
 
537
323
    def test_resolve_checkout_case_conflict(self):
538
324
        tree = self.make_branch_and_tree('tree')
547
333
        resolve_conflicts(transform,
548
334
                          pass_func=lambda t, c: resolve_checkout(t, c, []))
549
335
        transform.apply()
550
 
        self.assertPathExists('tree/file')
551
 
        self.assertPathExists('tree/FiLe.moved')
 
336
        self.failUnlessExists('tree/file')
 
337
        self.failUnlessExists('tree/FiLe.moved')
552
338
 
553
339
    def test_apply_case_conflict(self):
554
340
        """Ensure that a transform with case conflicts can always be applied"""
562
348
        transform.new_file('dirFiLe', dir, 'content')
563
349
        resolve_conflicts(transform)
564
350
        transform.apply()
565
 
        self.assertPathExists('tree/file')
 
351
        self.failUnlessExists('tree/file')
566
352
        if not os.path.exists('tree/FiLe.moved'):
567
 
            self.assertPathExists('tree/FiLe')
568
 
        self.assertPathExists('tree/dir/dirfile')
 
353
            self.failUnlessExists('tree/FiLe')
 
354
        self.failUnlessExists('tree/dir/dirfile')
569
355
        if not os.path.exists('tree/dir/dirFiLe.moved'):
570
 
            self.assertPathExists('tree/dir/dirFiLe')
 
356
            self.failUnlessExists('tree/dir/dirFiLe')
571
357
 
572
358
    def test_case_insensitive_limbo(self):
573
359
        tree = self.make_branch_and_tree('tree')
582
368
        self.assertContainsRe(transform._limbo_name(first), 'new-1/file')
583
369
        self.assertNotContainsRe(transform._limbo_name(second), 'new-1/FiLe')
584
370
 
585
 
    def test_adjust_path_updates_child_limbo_names(self):
586
 
        tree = self.make_branch_and_tree('tree')
587
 
        transform = TreeTransform(tree)
588
 
        self.addCleanup(transform.finalize)
589
 
        foo_id = transform.new_directory('foo', transform.root)
590
 
        bar_id = transform.new_directory('bar', foo_id)
591
 
        baz_id = transform.new_directory('baz', bar_id)
592
 
        qux_id = transform.new_directory('qux', baz_id)
593
 
        transform.adjust_path('quxx', foo_id, bar_id)
594
 
        self.assertStartsWith(transform._limbo_name(qux_id),
595
 
                              transform._limbo_name(bar_id))
596
 
 
597
371
    def test_add_del(self):
598
372
        start, root = self.get_transform()
599
373
        start.new_directory('a', root, 'a')
612
386
        self.addCleanup(unversion.finalize)
613
387
        parent = unversion.trans_id_tree_path('parent')
614
388
        unversion.unversion_file(parent)
615
 
        self.assertEqual(unversion.find_conflicts(),
 
389
        self.assertEqual(unversion.find_conflicts(), 
616
390
                         [('unversioned parent', parent_id)])
617
391
        file_id = unversion.trans_id_tree_file_id('child-id')
618
392
        unversion.unversion_file(file_id)
638
412
        mangle_tree.adjust_path('name2', root, name1)
639
413
        mangle_tree.adjust_path('name1', root, name2)
640
414
 
641
 
        #tests for deleting parent directories
 
415
        #tests for deleting parent directories 
642
416
        ddir = mangle_tree.trans_id_tree_file_id('ddir')
643
417
        mangle_tree.delete_contents(ddir)
644
418
        dfile = mangle_tree.trans_id_tree_file_id('dfile')
673
447
        create_tree,root = self.get_transform()
674
448
        newdir = create_tree.new_directory('selftest', root, 'selftest-id')
675
449
        create_tree.new_file('blackbox.py', newdir, 'hello1', 'blackbox-id')
676
 
        create_tree.apply()
 
450
        create_tree.apply()        
677
451
        mangle_tree,root = self.get_transform()
678
452
        selftest = mangle_tree.trans_id_tree_file_id('selftest-id')
679
453
        blackbox = mangle_tree.trans_id_tree_file_id('blackbox-id')
687
461
        bzrlib = create_tree.new_directory('bzrlib', root, 'bzrlib-id')
688
462
        tests = create_tree.new_directory('tests', bzrlib, 'tests-id')
689
463
        blackbox = create_tree.new_directory('blackbox', tests, 'blackbox-id')
690
 
        create_tree.new_file('test_too_much.py', blackbox, 'hello1',
 
464
        create_tree.new_file('test_too_much.py', blackbox, 'hello1', 
691
465
                             'test_too_much-id')
692
 
        create_tree.apply()
 
466
        create_tree.apply()        
693
467
        mangle_tree,root = self.get_transform()
694
468
        bzrlib = mangle_tree.trans_id_tree_file_id('bzrlib-id')
695
469
        tests = mangle_tree.trans_id_tree_file_id('tests-id')
696
470
        test_too_much = mangle_tree.trans_id_tree_file_id('test_too_much-id')
697
471
        mangle_tree.adjust_path('selftest', bzrlib, tests)
698
 
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much)
 
472
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much) 
699
473
        mangle_tree.set_executability(True, test_too_much)
700
474
        mangle_tree.apply()
701
475
 
702
476
    def test_both_rename3(self):
703
477
        create_tree,root = self.get_transform()
704
478
        tests = create_tree.new_directory('tests', root, 'tests-id')
705
 
        create_tree.new_file('test_too_much.py', tests, 'hello1',
 
479
        create_tree.new_file('test_too_much.py', tests, 'hello1', 
706
480
                             'test_too_much-id')
707
 
        create_tree.apply()
 
481
        create_tree.apply()        
708
482
        mangle_tree,root = self.get_transform()
709
483
        tests = mangle_tree.trans_id_tree_file_id('tests-id')
710
484
        test_too_much = mangle_tree.trans_id_tree_file_id('test_too_much-id')
711
485
        mangle_tree.adjust_path('selftest', root, tests)
712
 
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much)
 
486
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much) 
713
487
        mangle_tree.set_executability(True, test_too_much)
714
488
        mangle_tree.apply()
715
489
 
728
502
        newdir = move_id.new_directory('dir', root, 'newdir')
729
503
        move_id.adjust_path('name2', newdir, name1)
730
504
        move_id.apply()
731
 
 
 
505
        
732
506
    def test_replace_dangling_ie(self):
733
507
        create_tree, root = self.get_transform()
734
508
        # prepare tree
750
524
        resolve_conflicts(replace)
751
525
        replace.apply()
752
526
 
753
 
    def _test_symlinks(self, link_name1,link_target1,
754
 
                       link_name2, link_target2):
755
 
 
756
 
        def ozpath(p): return 'oz/' + p
757
 
 
 
527
    def test_symlinks(self):
758
528
        self.requireFeature(SymlinkFeature)
759
 
        transform, root = self.get_transform()
 
529
        transform,root = self.get_transform()
760
530
        oz_id = transform.new_directory('oz', root, 'oz-id')
761
 
        wizard = transform.new_symlink(link_name1, oz_id, link_target1,
 
531
        wizard = transform.new_symlink('wizard', oz_id, 'wizard-target', 
762
532
                                       'wizard-id')
763
 
        wiz_id = transform.create_path(link_name2, oz_id)
764
 
        transform.create_symlink(link_target2, wiz_id)
765
 
        transform.version_file('wiz-id2', wiz_id)
 
533
        wiz_id = transform.create_path('wizard2', oz_id)
 
534
        transform.create_symlink('behind_curtain', wiz_id)
 
535
        transform.version_file('wiz-id2', wiz_id)            
766
536
        transform.set_executability(True, wiz_id)
767
 
        self.assertEqual(transform.find_conflicts(),
 
537
        self.assertEqual(transform.find_conflicts(), 
768
538
                         [('non-file executability', wiz_id)])
769
539
        transform.set_executability(None, wiz_id)
770
540
        transform.apply()
771
 
        self.assertEqual(self.wt.path2id(ozpath(link_name1)), 'wizard-id')
772
 
        self.assertEqual('symlink',
773
 
                         file_kind(self.wt.abspath(ozpath(link_name1))))
774
 
        self.assertEqual(link_target2,
775
 
                         osutils.readlink(self.wt.abspath(ozpath(link_name2))))
776
 
        self.assertEqual(link_target1,
777
 
                         osutils.readlink(self.wt.abspath(ozpath(link_name1))))
778
 
 
779
 
    def test_symlinks(self):
780
 
        self._test_symlinks('wizard', 'wizard-target',
781
 
                            'wizard2', 'behind_curtain')
782
 
 
783
 
    def test_symlinks_unicode(self):
784
 
        self.requireFeature(features.UnicodeFilenameFeature)
785
 
        self._test_symlinks(u'\N{Euro Sign}wizard',
786
 
                            u'wizard-targ\N{Euro Sign}t',
787
 
                            u'\N{Euro Sign}wizard2',
788
 
                            u'b\N{Euro Sign}hind_curtain')
 
541
        self.assertEqual(self.wt.path2id('oz/wizard'), 'wizard-id')
 
542
        self.assertEqual(file_kind(self.wt.abspath('oz/wizard')), 'symlink')
 
543
        self.assertEqual(os.readlink(self.wt.abspath('oz/wizard2')), 
 
544
                         'behind_curtain')
 
545
        self.assertEqual(os.readlink(self.wt.abspath('oz/wizard')),
 
546
                         'wizard-target')
789
547
 
790
548
    def test_unable_create_symlink(self):
791
549
        def tt_helper():
815
573
        create.apply()
816
574
        conflicts,root = self.get_transform()
817
575
        # set up duplicate entry, duplicate id
818
 
        new_dorothy = conflicts.new_file('dorothy', root, 'dorothy',
 
576
        new_dorothy = conflicts.new_file('dorothy', root, 'dorothy', 
819
577
                                         'dorothy-id')
820
578
        old_dorothy = conflicts.trans_id_tree_file_id('dorothy-id')
821
579
        oz = conflicts.trans_id_tree_file_id('oz-id')
845
603
        tt, emerald, oz, old_dorothy, new_dorothy = self.get_conflicted()
846
604
        raw_conflicts = resolve_conflicts(tt)
847
605
        cooked_conflicts = cook_conflicts(raw_conflicts, tt)
848
 
        duplicate = DuplicateEntry('Moved existing file to', 'dorothy.moved',
 
606
        duplicate = DuplicateEntry('Moved existing file to', 'dorothy.moved', 
849
607
                                   'dorothy', None, 'dorothy-id')
850
608
        self.assertEqual(cooked_conflicts[0], duplicate)
851
 
        duplicate_id = DuplicateID('Unversioned existing file',
 
609
        duplicate_id = DuplicateID('Unversioned existing file', 
852
610
                                   'dorothy.moved', 'dorothy', None,
853
611
                                   'dorothy-id')
854
612
        self.assertEqual(cooked_conflicts[1], duplicate_id)
862
620
        unversioned_parent2 = UnversionedParent('Versioned directory', 'oz',
863
621
                                               'oz-id')
864
622
        self.assertEqual(cooked_conflicts[3], unversioned_parent)
865
 
        parent_loop = ParentLoop('Cancelled move', 'oz/emeraldcity',
 
623
        parent_loop = ParentLoop('Cancelled move', 'oz/emeraldcity', 
866
624
                                 'oz/emeraldcity', 'emerald-id', 'emerald-id')
867
625
        self.assertEqual(cooked_conflicts[4], deleted_parent)
868
626
        self.assertEqual(cooked_conflicts[5], unversioned_parent2)
875
633
        raw_conflicts = resolve_conflicts(tt)
876
634
        cooked_conflicts = cook_conflicts(raw_conflicts, tt)
877
635
        tt.finalize()
878
 
        conflicts_s = [unicode(c) for c in cooked_conflicts]
 
636
        conflicts_s = [str(c) for c in cooked_conflicts]
879
637
        self.assertEqual(len(cooked_conflicts), len(conflicts_s))
880
638
        self.assertEqual(conflicts_s[0], 'Conflict adding file dorothy.  '
881
639
                                         'Moved existing file to '
894
652
                                         ' versioned, but has versioned'
895
653
                                         ' children.  Versioned directory.')
896
654
        self.assertEqual(conflicts_s[6], 'Conflict moving oz/emeraldcity into'
897
 
                                         ' oz/emeraldcity. Cancelled move.')
 
655
                                         ' oz/emeraldcity.  Cancelled move.')
898
656
 
899
657
    def prepare_wrong_parent_kind(self):
900
658
        tt, root = self.get_transform()
951
709
        self.assertIs(None, self.wt.path2id('parent'))
952
710
        self.assertIs(None, self.wt.path2id('parent.new'))
953
711
 
954
 
    def test_resolve_conflicts_missing_parent(self):
955
 
        wt = self.make_branch_and_tree('.')
956
 
        tt = TreeTransform(wt)
957
 
        self.addCleanup(tt.finalize)
958
 
        parent = tt.trans_id_file_id('parent-id')
959
 
        tt.new_file('file', parent, 'Contents')
960
 
        raw_conflicts = resolve_conflicts(tt)
961
 
        # Since the directory doesn't exist it's seen as 'missing'.  So
962
 
        # 'resolve_conflicts' create a conflict asking for it to be created.
963
 
        self.assertLength(1, raw_conflicts)
964
 
        self.assertEqual(('missing parent', 'Created directory', 'new-1'),
965
 
                         raw_conflicts.pop())
966
 
        # apply fail since the missing directory doesn't exist
967
 
        self.assertRaises(errors.NoFinalPath, tt.apply)
968
 
 
969
712
    def test_moving_versioned_directories(self):
970
713
        create, root = self.get_transform()
971
714
        kansas = create.new_directory('kansas', root, 'kansas-id')
986
729
        create.apply()
987
730
        transform, root = self.get_transform()
988
731
        transform.adjust_root_path('oldroot', fun)
989
 
        new_root = transform.trans_id_tree_path('')
 
732
        new_root=transform.trans_id_tree_path('')
990
733
        transform.version_file('new-root', new_root)
991
734
        transform.apply()
992
735
 
1004
747
        rename.set_executability(True, myfile)
1005
748
        rename.apply()
1006
749
 
1007
 
    def test_rename_fails(self):
1008
 
        self.requireFeature(features.not_running_as_root)
1009
 
        # see https://bugs.launchpad.net/bzr/+bug/491763
1010
 
        create, root_id = self.get_transform()
1011
 
        first_dir = create.new_directory('first-dir', root_id, 'first-id')
1012
 
        myfile = create.new_file('myfile', root_id, 'myfile-text',
1013
 
                                 'myfile-id')
1014
 
        create.apply()
1015
 
        if os.name == "posix" and sys.platform != "cygwin":
1016
 
            # posix filesystems fail on renaming if the readonly bit is set
1017
 
            osutils.make_readonly(self.wt.abspath('first-dir'))
1018
 
        elif os.name == "nt":
1019
 
            # windows filesystems fail on renaming open files
1020
 
            self.addCleanup(file(self.wt.abspath('myfile')).close)
1021
 
        else:
1022
 
            self.skip("Don't know how to force a permissions error on rename")
1023
 
        # now transform to rename
1024
 
        rename_transform, root_id = self.get_transform()
1025
 
        file_trans_id = rename_transform.trans_id_file_id('myfile-id')
1026
 
        dir_id = rename_transform.trans_id_file_id('first-id')
1027
 
        rename_transform.adjust_path('newname', dir_id, file_trans_id)
1028
 
        e = self.assertRaises(errors.TransformRenameFailed,
1029
 
            rename_transform.apply)
1030
 
        # On nix looks like: 
1031
 
        # "Failed to rename .../work/.bzr/checkout/limbo/new-1
1032
 
        # to .../first-dir/newname: [Errno 13] Permission denied"
1033
 
        # On windows looks like:
1034
 
        # "Failed to rename .../work/myfile to 
1035
 
        # .../work/.bzr/checkout/limbo/new-1: [Errno 13] Permission denied"
1036
 
        # This test isn't concerned with exactly what the error looks like,
1037
 
        # and the strerror will vary across OS and locales, but the assert
1038
 
        # that the exeception attributes are what we expect
1039
 
        self.assertEqual(e.errno, errno.EACCES)
1040
 
        if os.name == "posix":
1041
 
            self.assertEndsWith(e.to_path, "/first-dir/newname")
1042
 
        else:
1043
 
            self.assertEqual(os.path.basename(e.from_path), "myfile")
1044
 
 
1045
750
    def test_set_executability_order(self):
1046
751
        """Ensure that executability behaves the same, no matter what order.
1047
 
 
 
752
        
1048
753
        - create file and set executability simultaneously
1049
754
        - create file and set executability afterward
1050
755
        - unsetting the executability of a file whose executability has not been
1319
1024
        parent2 = transform.new_directory('parent2', root)
1320
1025
        transform.adjust_path('child1', parent2, child1)
1321
1026
        transform.apply()
1322
 
        self.assertPathDoesNotExist(self.wt.abspath('parent1/child1'))
1323
 
        self.assertPathExists(self.wt.abspath('parent2/child1'))
 
1027
        self.failIfExists(self.wt.abspath('parent1/child1'))
 
1028
        self.failUnlessExists(self.wt.abspath('parent2/child1'))
1324
1029
        # rename limbo/new-1 => parent1, rename limbo/new-3 => parent2
1325
1030
        # no rename for child1 (counting only renames during apply)
1326
 
        self.assertEqual(2, transform.rename_count)
 
1031
        self.failUnlessEqual(2, transform.rename_count)
1327
1032
 
1328
1033
    def test_cancel_parent(self):
1329
1034
        """Cancelling a parent doesn't cause deletion of a non-empty directory
1352
1057
        parent2 = transform.new_directory('parent2', root)
1353
1058
        transform.adjust_path('child1', parent2, child1)
1354
1059
        transform.apply()
1355
 
        self.assertPathDoesNotExist(self.wt.abspath('parent1'))
1356
 
        self.assertPathExists(self.wt.abspath('parent2/child1'))
 
1060
        self.failIfExists(self.wt.abspath('parent1'))
 
1061
        self.failUnlessExists(self.wt.abspath('parent2/child1'))
1357
1062
        # rename limbo/new-3 => parent2, rename limbo/new-2 => child1
1358
 
        self.assertEqual(2, transform.rename_count)
 
1063
        self.failUnlessEqual(2, transform.rename_count)
1359
1064
 
1360
1065
    def test_adjust_and_cancel(self):
1361
1066
        """Make sure adjust_path keeps track of limbo children properly"""
1394
1099
        child = transform.new_directory('child', parent)
1395
1100
        transform.adjust_path('parent', root, parent)
1396
1101
        transform.apply()
1397
 
        self.assertPathExists(self.wt.abspath('parent/child'))
 
1102
        self.failUnlessExists(self.wt.abspath('parent/child'))
1398
1103
        self.assertEqual(1, transform.rename_count)
1399
1104
 
1400
1105
    def test_reuse_name(self):
1531
1236
        tt.create_file(["aa\n"], bar_trans_id)
1532
1237
        tt.version_file("bar-1", bar_trans_id)
1533
1238
        tt.apply()
1534
 
        self.assertPathExists("foo/bar")
 
1239
        self.failUnlessExists("foo/bar")
1535
1240
        wt.lock_read()
1536
1241
        try:
1537
1242
            self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
1554
1259
        tt.delete_contents(foo_trans_id)
1555
1260
        tt.create_symlink("bar", foo_trans_id)
1556
1261
        tt.apply()
1557
 
        self.assertPathExists("foo")
 
1262
        self.failUnlessExists("foo")
1558
1263
        wt.lock_read()
1559
1264
        self.addCleanup(wt.unlock)
1560
1265
        self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
1573
1278
        tt.delete_versioned(bar_trans_id)
1574
1279
        tt.create_file(["aa\n"], foo_trans_id)
1575
1280
        tt.apply()
1576
 
        self.assertPathExists("foo")
 
1281
        self.failUnlessExists("foo")
1577
1282
        wt.lock_read()
1578
1283
        self.addCleanup(wt.unlock)
1579
1284
        self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
1594
1299
        self.build_tree(['baz'])
1595
1300
        tt.create_hardlink("baz", foo_trans_id)
1596
1301
        tt.apply()
1597
 
        self.assertPathExists("foo")
1598
 
        self.assertPathExists("baz")
 
1302
        self.failUnlessExists("foo")
 
1303
        self.failUnlessExists("baz")
1599
1304
        wt.lock_read()
1600
1305
        self.addCleanup(wt.unlock)
1601
1306
        self.assertEqual(wt.inventory.get_file_kind(wt.path2id("foo")),
1608
1313
        transform.cancel_creation(trans_id)
1609
1314
        transform.apply()
1610
1315
 
1611
 
    def test_create_from_tree(self):
1612
 
        tree1 = self.make_branch_and_tree('tree1')
1613
 
        self.build_tree_contents([('tree1/foo/',), ('tree1/bar', 'baz')])
1614
 
        tree1.add(['foo', 'bar'], ['foo-id', 'bar-id'])
1615
 
        tree2 = self.make_branch_and_tree('tree2')
1616
 
        tt = TreeTransform(tree2)
1617
 
        foo_trans_id = tt.create_path('foo', tt.root)
1618
 
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id')
1619
 
        bar_trans_id = tt.create_path('bar', tt.root)
1620
 
        create_from_tree(tt, bar_trans_id, tree1, 'bar-id')
1621
 
        tt.apply()
1622
 
        self.assertEqual('directory', osutils.file_kind('tree2/foo'))
1623
 
        self.assertFileEqual('baz', 'tree2/bar')
1624
 
 
1625
 
    def test_create_from_tree_bytes(self):
1626
 
        """Provided lines are used instead of tree content."""
1627
 
        tree1 = self.make_branch_and_tree('tree1')
1628
 
        self.build_tree_contents([('tree1/foo', 'bar'),])
1629
 
        tree1.add('foo', 'foo-id')
1630
 
        tree2 = self.make_branch_and_tree('tree2')
1631
 
        tt = TreeTransform(tree2)
1632
 
        foo_trans_id = tt.create_path('foo', tt.root)
1633
 
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id', bytes='qux')
1634
 
        tt.apply()
1635
 
        self.assertFileEqual('qux', 'tree2/foo')
1636
 
 
1637
 
    def test_create_from_tree_symlink(self):
1638
 
        self.requireFeature(SymlinkFeature)
1639
 
        tree1 = self.make_branch_and_tree('tree1')
1640
 
        os.symlink('bar', 'tree1/foo')
1641
 
        tree1.add('foo', 'foo-id')
1642
 
        tt = TreeTransform(self.make_branch_and_tree('tree2'))
1643
 
        foo_trans_id = tt.create_path('foo', tt.root)
1644
 
        create_from_tree(tt, foo_trans_id, tree1, 'foo-id')
1645
 
        tt.apply()
1646
 
        self.assertEqual('bar', os.readlink('tree2/foo'))
1647
 
 
1648
1316
 
1649
1317
class TransformGroup(object):
1650
1318
 
1663
1331
    return template % ('<' * 7, tree, '=' * 7, merge, '>' * 7)
1664
1332
 
1665
1333
 
1666
 
class TestInventoryAltered(tests.TestCaseWithTransport):
1667
 
 
1668
 
    def test_inventory_altered_unchanged(self):
1669
 
        tree = self.make_branch_and_tree('tree')
1670
 
        self.build_tree(['tree/foo'])
1671
 
        tree.add('foo', 'foo-id')
1672
 
        with TransformPreview(tree) as tt:
1673
 
            self.assertEqual([], tt._inventory_altered())
1674
 
 
1675
 
    def test_inventory_altered_changed_parent_id(self):
1676
 
        tree = self.make_branch_and_tree('tree')
1677
 
        self.build_tree(['tree/foo'])
1678
 
        tree.add('foo', 'foo-id')
1679
 
        with TransformPreview(tree) as tt:
1680
 
            tt.unversion_file(tt.root)
1681
 
            tt.version_file('new-id', tt.root)
1682
 
            foo_trans_id = tt.trans_id_tree_file_id('foo-id')
1683
 
            foo_tuple = ('foo', foo_trans_id)
1684
 
            root_tuple = ('', tt.root)
1685
 
            self.assertEqual([root_tuple, foo_tuple], tt._inventory_altered())
1686
 
 
1687
 
    def test_inventory_altered_noop_changed_parent_id(self):
1688
 
        tree = self.make_branch_and_tree('tree')
1689
 
        self.build_tree(['tree/foo'])
1690
 
        tree.add('foo', 'foo-id')
1691
 
        with TransformPreview(tree) as tt:
1692
 
            tt.unversion_file(tt.root)
1693
 
            tt.version_file(tree.get_root_id(), tt.root)
1694
 
            foo_trans_id = tt.trans_id_tree_file_id('foo-id')
1695
 
            self.assertEqual([], tt._inventory_altered())
1696
 
 
1697
 
 
1698
1334
class TestTransformMerge(TestCaseInTempDir):
1699
1335
 
1700
1336
    def test_text_merge(self):
1736
1372
        # textual merge
1737
1373
        self.assertEqual(this.wt.get_file('a').read(), 'y\nb\nc\nd\bz\n')
1738
1374
        # three-way text conflict
1739
 
        self.assertEqual(this.wt.get_file('b').read(),
 
1375
        self.assertEqual(this.wt.get_file('b').read(), 
1740
1376
                         conflict_text('b', 'b2'))
1741
1377
        # OTHER wins
1742
1378
        self.assertEqual(this.wt.get_file('c').read(), 'c2')
1746
1382
        self.assertEqual(this.wt.get_file('e').read(), 'e2')
1747
1383
        # No change
1748
1384
        self.assertEqual(this.wt.get_file('f').read(), 'f')
1749
 
        # Correct correct results when THIS == OTHER
 
1385
        # Correct correct results when THIS == OTHER 
1750
1386
        self.assertEqual(this.wt.get_file('g').read(), 'g')
1751
1387
        # Text conflict when THIS & OTHER are text and BASE is dir
1752
 
        self.assertEqual(this.wt.get_file('h').read(),
 
1388
        self.assertEqual(this.wt.get_file('h').read(), 
1753
1389
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
1754
1390
        self.assertEqual(this.wt.get_file_byname('h.THIS').read(),
1755
1391
                         '1\n2\n3\n4\n')
1756
1392
        self.assertEqual(this.wt.get_file_byname('h.OTHER').read(),
1757
1393
                         'h\ni\nj\nk\n')
1758
1394
        self.assertEqual(file_kind(this.wt.abspath('h.BASE')), 'directory')
1759
 
        self.assertEqual(this.wt.get_file('i').read(),
 
1395
        self.assertEqual(this.wt.get_file('i').read(), 
1760
1396
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
1761
1397
        self.assertEqual(this.wt.get_file_byname('i.THIS').read(),
1762
1398
                         '1\n2\n3\n4\n')
1786
1422
            tg.tt.new_symlink('b', tg.root, 'b', 'b')
1787
1423
            tg.tt.new_file('c', tg.root, 'c', 'c')
1788
1424
            tg.tt.new_symlink('d', tg.root, tg.name, 'd')
1789
 
        targets = ((base, 'base-e', 'base-f', None, None),
1790
 
                   (this, 'other-e', 'this-f', 'other-g', 'this-h'),
 
1425
        targets = ((base, 'base-e', 'base-f', None, None), 
 
1426
                   (this, 'other-e', 'this-f', 'other-g', 'this-h'), 
1791
1427
                   (other, 'other-e', None, 'other-g', 'other-h'))
1792
1428
        for tg, e_target, f_target, g_target, h_target in targets:
1793
 
            for link, target in (('e', e_target), ('f', f_target),
 
1429
            for link, target in (('e', e_target), ('f', f_target), 
1794
1430
                                 ('g', g_target), ('h', h_target)):
1795
1431
                if target is not None:
1796
1432
                    tg.tt.new_symlink(link, tg.root, target, link)
1822
1458
        base = TransformGroup("BASE", root_id)
1823
1459
        this = TransformGroup("THIS", root_id)
1824
1460
        other = TransformGroup("OTHER", root_id)
1825
 
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
 
1461
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a') 
1826
1462
                                   for t in [base, this, other]]
1827
 
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
 
1463
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b') 
1828
1464
                                   for t in [base, this, other]]
1829
1465
        base.tt.new_directory('c', base_a, 'c')
1830
1466
        this.tt.new_directory('c1', this_a, 'c')
1855
1491
        base = TransformGroup("BASE", root_id)
1856
1492
        this = TransformGroup("THIS", root_id)
1857
1493
        other = TransformGroup("OTHER", root_id)
1858
 
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
 
1494
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a') 
1859
1495
                                   for t in [base, this, other]]
1860
 
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
 
1496
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b') 
1861
1497
                                   for t in [base, this, other]]
1862
1498
 
1863
1499
        base.tt.new_file('g', base_a, 'g', 'g')
1910
1546
        tree.add_reference(subtree)
1911
1547
        tree.commit('a revision')
1912
1548
        tree.branch.create_checkout('target')
1913
 
        self.assertPathExists('target')
1914
 
        self.assertPathExists('target/subtree')
 
1549
        self.failUnlessExists('target')
 
1550
        self.failUnlessExists('target/subtree')
1915
1551
 
1916
1552
    def test_file_conflict_handling(self):
1917
1553
        """Ensure that when building trees, conflict handling is done"""
1954
1590
        os.symlink('foo', 'target2/symlink')
1955
1591
        build_tree(source.basis_tree(), target)
1956
1592
        self.assertEqual([], target.conflicts())
1957
 
 
 
1593
        
1958
1594
    def test_directory_conflict_handling(self):
1959
1595
        """Ensure that when building trees, conflict handling is done"""
1960
1596
        source = self.make_branch_and_tree('source')
1964
1600
        source.commit('added file')
1965
1601
        build_tree(source.basis_tree(), target)
1966
1602
        self.assertEqual([], target.conflicts())
1967
 
        self.assertPathExists('target/dir1/file')
 
1603
        self.failUnlessExists('target/dir1/file')
1968
1604
 
1969
1605
        # Ensure contents are merged
1970
1606
        target = self.make_branch_and_tree('target2')
1971
1607
        self.build_tree(['target2/dir1/', 'target2/dir1/file2'])
1972
1608
        build_tree(source.basis_tree(), target)
1973
1609
        self.assertEqual([], target.conflicts())
1974
 
        self.assertPathExists('target2/dir1/file2')
1975
 
        self.assertPathExists('target2/dir1/file')
 
1610
        self.failUnlessExists('target2/dir1/file2')
 
1611
        self.failUnlessExists('target2/dir1/file')
1976
1612
 
1977
1613
        # Ensure new contents are suppressed for existing branches
1978
1614
        target = self.make_branch_and_tree('target3')
1979
1615
        self.make_branch('target3/dir1')
1980
1616
        self.build_tree(['target3/dir1/file2'])
1981
1617
        build_tree(source.basis_tree(), target)
1982
 
        self.assertPathDoesNotExist('target3/dir1/file')
1983
 
        self.assertPathExists('target3/dir1/file2')
1984
 
        self.assertPathExists('target3/dir1.diverted/file')
 
1618
        self.failIfExists('target3/dir1/file')
 
1619
        self.failUnlessExists('target3/dir1/file2')
 
1620
        self.failUnlessExists('target3/dir1.diverted/file')
1985
1621
        self.assertEqual([DuplicateEntry('Diverted to',
1986
1622
            'dir1.diverted', 'dir1', 'new-dir1', None)],
1987
1623
            target.conflicts())
1990
1626
        self.build_tree(['target4/dir1/'])
1991
1627
        self.make_branch('target4/dir1/file')
1992
1628
        build_tree(source.basis_tree(), target)
1993
 
        self.assertPathExists('target4/dir1/file')
 
1629
        self.failUnlessExists('target4/dir1/file')
1994
1630
        self.assertEqual('directory', file_kind('target4/dir1/file'))
1995
 
        self.assertPathExists('target4/dir1/file.diverted')
 
1631
        self.failUnlessExists('target4/dir1/file.diverted')
1996
1632
        self.assertEqual([DuplicateEntry('Diverted to',
1997
1633
            'dir1/file.diverted', 'dir1/file', 'new-file', None)],
1998
1634
            target.conflicts())
2016
1652
        target = self.make_branch_and_tree('target')
2017
1653
        self.build_tree(['target/name'])
2018
1654
        target.add('name')
2019
 
        self.assertRaises(errors.WorkingTreeAlreadyPopulated,
 
1655
        self.assertRaises(errors.WorkingTreeAlreadyPopulated, 
2020
1656
            build_tree, source.basis_tree(), target)
2021
1657
 
2022
1658
    def test_build_tree_rename_count(self):
2066
1702
        self.addCleanup(target.unlock)
2067
1703
        self.assertEqual([], list(target.iter_changes(revision_tree)))
2068
1704
 
2069
 
    def test_build_tree_accelerator_tree_observes_sha1(self):
2070
 
        source = self.create_ab_tree()
2071
 
        sha1 = osutils.sha_string('A')
2072
 
        target = self.make_branch_and_tree('target')
2073
 
        target.lock_write()
2074
 
        self.addCleanup(target.unlock)
2075
 
        state = target.current_dirstate()
2076
 
        state._cutoff_time = time.time() + 60
2077
 
        build_tree(source.basis_tree(), target, source)
2078
 
        entry = state._get_entry(0, path_utf8='file1')
2079
 
        self.assertEqual(sha1, entry[1][0][1])
2080
 
 
2081
1705
    def test_build_tree_accelerator_tree_missing_file(self):
2082
1706
        source = self.create_ab_tree()
2083
1707
        os.unlink('source/file1')
2178
1802
        self.assertEqual([], list(target.iter_changes(revision_tree)))
2179
1803
        self.assertTrue(source.is_executable('file1-id'))
2180
1804
 
2181
 
    def install_rot13_content_filter(self, pattern):
2182
 
        # We could use
2183
 
        # self.addCleanup(filters._reset_registry, filters._reset_registry())
2184
 
        # below, but that looks a bit... hard to read even if it's exactly
2185
 
        # the same thing.
2186
 
        original_registry = filters._reset_registry()
2187
 
        def restore_registry():
2188
 
            filters._reset_registry(original_registry)
2189
 
        self.addCleanup(restore_registry)
2190
 
        def rot13(chunks, context=None):
2191
 
            return [''.join(chunks).encode('rot13')]
2192
 
        rot13filter = filters.ContentFilter(rot13, rot13)
2193
 
        filters.register_filter_stack_map('rot13', {'yes': [rot13filter]}.get)
2194
 
        os.mkdir(self.test_home_dir + '/.bazaar')
2195
 
        rules_filename = self.test_home_dir + '/.bazaar/rules'
2196
 
        f = open(rules_filename, 'wb')
2197
 
        f.write('[name %s]\nrot13=yes\n' % (pattern,))
2198
 
        f.close()
2199
 
        def uninstall_rules():
2200
 
            os.remove(rules_filename)
2201
 
            rules.reset_rules()
2202
 
        self.addCleanup(uninstall_rules)
2203
 
        rules.reset_rules()
2204
 
 
2205
 
    def test_build_tree_content_filtered_files_are_not_hardlinked(self):
2206
 
        """build_tree will not hardlink files that have content filtering rules
2207
 
        applied to them (but will still hardlink other files from the same tree
2208
 
        if it can).
2209
 
        """
2210
 
        self.requireFeature(HardlinkFeature)
2211
 
        self.install_rot13_content_filter('file1')
2212
 
        source = self.create_ab_tree()
2213
 
        target = self.make_branch_and_tree('target')
2214
 
        revision_tree = source.basis_tree()
2215
 
        revision_tree.lock_read()
2216
 
        self.addCleanup(revision_tree.unlock)
2217
 
        build_tree(revision_tree, target, source, hardlink=True)
2218
 
        target.lock_read()
2219
 
        self.addCleanup(target.unlock)
2220
 
        self.assertEqual([], list(target.iter_changes(revision_tree)))
2221
 
        source_stat = os.stat('source/file1')
2222
 
        target_stat = os.stat('target/file1')
2223
 
        self.assertNotEqual(source_stat, target_stat)
2224
 
        source_stat = os.stat('source/file2')
2225
 
        target_stat = os.stat('target/file2')
2226
 
        self.assertEqualStat(source_stat, target_stat)
2227
 
 
2228
1805
    def test_case_insensitive_build_tree_inventory(self):
2229
 
        if (features.CaseInsensitiveFilesystemFeature.available()
2230
 
            or features.CaseInsCasePresFilenameFeature.available()):
2231
 
            raise tests.UnavailableFeature('Fully case sensitive filesystem')
2232
1806
        source = self.make_branch_and_tree('source')
2233
1807
        self.build_tree(['source/file', 'source/FILE'])
2234
1808
        source.add(['file', 'FILE'], ['lower-id', 'upper-id'])
2241
1815
        self.assertEqual('file.moved', target.id2path('lower-id'))
2242
1816
        self.assertEqual('FILE', target.id2path('upper-id'))
2243
1817
 
2244
 
    def test_build_tree_observes_sha(self):
2245
 
        source = self.make_branch_and_tree('source')
2246
 
        self.build_tree(['source/file1', 'source/dir/', 'source/dir/file2'])
2247
 
        source.add(['file1', 'dir', 'dir/file2'],
2248
 
                   ['file1-id', 'dir-id', 'file2-id'])
2249
 
        source.commit('new files')
2250
 
        target = self.make_branch_and_tree('target')
2251
 
        target.lock_write()
2252
 
        self.addCleanup(target.unlock)
2253
 
        # We make use of the fact that DirState caches its cutoff time. So we
2254
 
        # set the 'safe' time to one minute in the future.
2255
 
        state = target.current_dirstate()
2256
 
        state._cutoff_time = time.time() + 60
2257
 
        build_tree(source.basis_tree(), target)
2258
 
        entry1_sha = osutils.sha_file_by_name('source/file1')
2259
 
        entry2_sha = osutils.sha_file_by_name('source/dir/file2')
2260
 
        # entry[1] is the state information, entry[1][0] is the state of the
2261
 
        # working tree, entry[1][0][1] is the sha value for the current working
2262
 
        # tree
2263
 
        entry1 = state._get_entry(0, path_utf8='file1')
2264
 
        self.assertEqual(entry1_sha, entry1[1][0][1])
2265
 
        # The 'size' field must also be set.
2266
 
        self.assertEqual(25, entry1[1][0][2])
2267
 
        entry1_state = entry1[1][0]
2268
 
        entry2 = state._get_entry(0, path_utf8='dir/file2')
2269
 
        self.assertEqual(entry2_sha, entry2[1][0][1])
2270
 
        self.assertEqual(29, entry2[1][0][2])
2271
 
        entry2_state = entry2[1][0]
2272
 
        # Now, make sure that we don't have to re-read the content. The
2273
 
        # packed_stat should match exactly.
2274
 
        self.assertEqual(entry1_sha, target.get_file_sha1('file1-id', 'file1'))
2275
 
        self.assertEqual(entry2_sha,
2276
 
                         target.get_file_sha1('file2-id', 'dir/file2'))
2277
 
        self.assertEqual(entry1_state, entry1[1][0])
2278
 
        self.assertEqual(entry2_state, entry2[1][0])
2279
 
 
2280
 
 
2281
 
class TestCommitTransform(tests.TestCaseWithTransport):
2282
 
 
2283
 
    def get_branch(self):
2284
 
        tree = self.make_branch_and_tree('tree')
2285
 
        tree.lock_write()
2286
 
        self.addCleanup(tree.unlock)
2287
 
        tree.commit('empty commit')
2288
 
        return tree.branch
2289
 
 
2290
 
    def get_branch_and_transform(self):
2291
 
        branch = self.get_branch()
2292
 
        tt = TransformPreview(branch.basis_tree())
2293
 
        self.addCleanup(tt.finalize)
2294
 
        return branch, tt
2295
 
 
2296
 
    def test_commit_wrong_basis(self):
2297
 
        branch = self.get_branch()
2298
 
        basis = branch.repository.revision_tree(
2299
 
            _mod_revision.NULL_REVISION)
2300
 
        tt = TransformPreview(basis)
2301
 
        self.addCleanup(tt.finalize)
2302
 
        e = self.assertRaises(ValueError, tt.commit, branch, '')
2303
 
        self.assertEqual('TreeTransform not based on branch basis: null:',
2304
 
                         str(e))
2305
 
 
2306
 
    def test_empy_commit(self):
2307
 
        branch, tt = self.get_branch_and_transform()
2308
 
        rev = tt.commit(branch, 'my message')
2309
 
        self.assertEqual(2, branch.revno())
2310
 
        repo = branch.repository
2311
 
        self.assertEqual('my message', repo.get_revision(rev).message)
2312
 
 
2313
 
    def test_merge_parents(self):
2314
 
        branch, tt = self.get_branch_and_transform()
2315
 
        rev = tt.commit(branch, 'my message', ['rev1b', 'rev1c'])
2316
 
        self.assertEqual(['rev1b', 'rev1c'],
2317
 
                         branch.basis_tree().get_parent_ids()[1:])
2318
 
 
2319
 
    def test_first_commit(self):
2320
 
        branch = self.make_branch('branch')
2321
 
        branch.lock_write()
2322
 
        self.addCleanup(branch.unlock)
2323
 
        tt = TransformPreview(branch.basis_tree())
2324
 
        self.addCleanup(tt.finalize)
2325
 
        tt.new_directory('', ROOT_PARENT, 'TREE_ROOT')
2326
 
        rev = tt.commit(branch, 'my message')
2327
 
        self.assertEqual([], branch.basis_tree().get_parent_ids())
2328
 
        self.assertNotEqual(_mod_revision.NULL_REVISION,
2329
 
                            branch.last_revision())
2330
 
 
2331
 
    def test_first_commit_with_merge_parents(self):
2332
 
        branch = self.make_branch('branch')
2333
 
        branch.lock_write()
2334
 
        self.addCleanup(branch.unlock)
2335
 
        tt = TransformPreview(branch.basis_tree())
2336
 
        self.addCleanup(tt.finalize)
2337
 
        e = self.assertRaises(ValueError, tt.commit, branch,
2338
 
                          'my message', ['rev1b-id'])
2339
 
        self.assertEqual('Cannot supply merge parents for first commit.',
2340
 
                         str(e))
2341
 
        self.assertEqual(_mod_revision.NULL_REVISION, branch.last_revision())
2342
 
 
2343
 
    def test_add_files(self):
2344
 
        branch, tt = self.get_branch_and_transform()
2345
 
        tt.new_file('file', tt.root, 'contents', 'file-id')
2346
 
        trans_id = tt.new_directory('dir', tt.root, 'dir-id')
2347
 
        if SymlinkFeature.available():
2348
 
            tt.new_symlink('symlink', trans_id, 'target', 'symlink-id')
2349
 
        rev = tt.commit(branch, 'message')
2350
 
        tree = branch.basis_tree()
2351
 
        self.assertEqual('file', tree.id2path('file-id'))
2352
 
        self.assertEqual('contents', tree.get_file_text('file-id'))
2353
 
        self.assertEqual('dir', tree.id2path('dir-id'))
2354
 
        if SymlinkFeature.available():
2355
 
            self.assertEqual('dir/symlink', tree.id2path('symlink-id'))
2356
 
            self.assertEqual('target', tree.get_symlink_target('symlink-id'))
2357
 
 
2358
 
    def test_add_unversioned(self):
2359
 
        branch, tt = self.get_branch_and_transform()
2360
 
        tt.new_file('file', tt.root, 'contents')
2361
 
        self.assertRaises(errors.StrictCommitFailed, tt.commit, branch,
2362
 
                          'message', strict=True)
2363
 
 
2364
 
    def test_modify_strict(self):
2365
 
        branch, tt = self.get_branch_and_transform()
2366
 
        tt.new_file('file', tt.root, 'contents', 'file-id')
2367
 
        tt.commit(branch, 'message', strict=True)
2368
 
        tt = TransformPreview(branch.basis_tree())
2369
 
        self.addCleanup(tt.finalize)
2370
 
        trans_id = tt.trans_id_file_id('file-id')
2371
 
        tt.delete_contents(trans_id)
2372
 
        tt.create_file('contents', trans_id)
2373
 
        tt.commit(branch, 'message', strict=True)
2374
 
 
2375
 
    def test_commit_malformed(self):
2376
 
        """Committing a malformed transform should raise an exception.
2377
 
 
2378
 
        In this case, we are adding a file without adding its parent.
2379
 
        """
2380
 
        branch, tt = self.get_branch_and_transform()
2381
 
        parent_id = tt.trans_id_file_id('parent-id')
2382
 
        tt.new_file('file', parent_id, 'contents', 'file-id')
2383
 
        self.assertRaises(errors.MalformedTransform, tt.commit, branch,
2384
 
                          'message')
2385
 
 
2386
 
    def test_commit_rich_revision_data(self):
2387
 
        branch, tt = self.get_branch_and_transform()
2388
 
        rev_id = tt.commit(branch, 'message', timestamp=1, timezone=43201,
2389
 
                           committer='me <me@example.com>',
2390
 
                           revprops={'foo': 'bar'}, revision_id='revid-1',
2391
 
                           authors=['Author1 <author1@example.com>',
2392
 
                              'Author2 <author2@example.com>',
2393
 
                               ])
2394
 
        self.assertEqual('revid-1', rev_id)
2395
 
        revision = branch.repository.get_revision(rev_id)
2396
 
        self.assertEqual(1, revision.timestamp)
2397
 
        self.assertEqual(43201, revision.timezone)
2398
 
        self.assertEqual('me <me@example.com>', revision.committer)
2399
 
        self.assertEqual(['Author1 <author1@example.com>',
2400
 
                          'Author2 <author2@example.com>'],
2401
 
                         revision.get_apparent_authors())
2402
 
        del revision.properties['authors']
2403
 
        self.assertEqual({'foo': 'bar',
2404
 
                          'branch-nick': 'tree'},
2405
 
                         revision.properties)
2406
 
 
2407
 
    def test_no_explicit_revprops(self):
2408
 
        branch, tt = self.get_branch_and_transform()
2409
 
        rev_id = tt.commit(branch, 'message', authors=[
2410
 
            'Author1 <author1@example.com>',
2411
 
            'Author2 <author2@example.com>', ])
2412
 
        revision = branch.repository.get_revision(rev_id)
2413
 
        self.assertEqual(['Author1 <author1@example.com>',
2414
 
                          'Author2 <author2@example.com>'],
2415
 
                         revision.get_apparent_authors())
2416
 
        self.assertEqual('tree', revision.properties['branch-nick'])
2417
 
 
2418
 
 
2419
 
class TestBackupName(tests.TestCase):
2420
 
 
2421
 
    def test_deprecations(self):
2422
 
        class MockTransform(object):
2423
 
 
2424
 
            def has_named_child(self, by_parent, parent_id, name):
2425
 
                return name in by_parent.get(parent_id, [])
2426
 
 
2427
 
        class MockEntry(object):
2428
 
 
2429
 
            def __init__(self):
2430
 
                object.__init__(self)
2431
 
                self.name = "name"
2432
 
 
 
1818
 
 
1819
class MockTransform(object):
 
1820
 
 
1821
    def has_named_child(self, by_parent, parent_id, name):
 
1822
        for child_id in by_parent[parent_id]:
 
1823
            if child_id == '0':
 
1824
                if name == "name~":
 
1825
                    return True
 
1826
            elif name == "name.~%s~" % child_id:
 
1827
                return True
 
1828
        return False
 
1829
 
 
1830
 
 
1831
class MockEntry(object):
 
1832
    def __init__(self):
 
1833
        object.__init__(self)
 
1834
        self.name = "name"
 
1835
 
 
1836
 
 
1837
class TestGetBackupName(TestCase):
 
1838
    def test_get_backup_name(self):
2433
1839
        tt = MockTransform()
2434
 
        name1 = self.applyDeprecated(
2435
 
            symbol_versioning.deprecated_in((2, 3, 0)),
2436
 
            transform.get_backup_name, MockEntry(), {'a':[]}, 'a', tt)
2437
 
        self.assertEqual('name.~1~', name1)
2438
 
        name2 = self.applyDeprecated(
2439
 
            symbol_versioning.deprecated_in((2, 3, 0)),
2440
 
            transform._get_backup_name, 'name', {'a':['name.~1~']}, 'a', tt)
2441
 
        self.assertEqual('name.~2~', name2)
 
1840
        name = get_backup_name(MockEntry(), {'a':[]}, 'a', tt)
 
1841
        self.assertEqual(name, 'name.~1~')
 
1842
        name = get_backup_name(MockEntry(), {'a':['1']}, 'a', tt)
 
1843
        self.assertEqual(name, 'name.~2~')
 
1844
        name = get_backup_name(MockEntry(), {'a':['2']}, 'a', tt)
 
1845
        self.assertEqual(name, 'name.~1~')
 
1846
        name = get_backup_name(MockEntry(), {'a':['2'], 'b':[]}, 'b', tt)
 
1847
        self.assertEqual(name, 'name.~1~')
 
1848
        name = get_backup_name(MockEntry(), {'a':['1', '2', '3']}, 'a', tt)
 
1849
        self.assertEqual(name, 'name.~4~')
2442
1850
 
2443
1851
 
2444
1852
class TestFileMover(tests.TestCaseWithTransport):
2447
1855
        self.build_tree(['a/', 'a/b', 'c/', 'c/d'])
2448
1856
        mover = _FileMover()
2449
1857
        mover.rename('a', 'q')
2450
 
        self.assertPathExists('q')
2451
 
        self.assertPathDoesNotExist('a')
2452
 
        self.assertPathExists('q/b')
2453
 
        self.assertPathExists('c')
2454
 
        self.assertPathExists('c/d')
 
1858
        self.failUnlessExists('q')
 
1859
        self.failIfExists('a')
 
1860
        self.failUnlessExists('q/b')
 
1861
        self.failUnlessExists('c')
 
1862
        self.failUnlessExists('c/d')
2455
1863
 
2456
1864
    def test_pre_delete_rollback(self):
2457
1865
        self.build_tree(['a/'])
2458
1866
        mover = _FileMover()
2459
1867
        mover.pre_delete('a', 'q')
2460
 
        self.assertPathExists('q')
2461
 
        self.assertPathDoesNotExist('a')
 
1868
        self.failUnlessExists('q')
 
1869
        self.failIfExists('a')
2462
1870
        mover.rollback()
2463
 
        self.assertPathDoesNotExist('q')
2464
 
        self.assertPathExists('a')
 
1871
        self.failIfExists('q')
 
1872
        self.failUnlessExists('a')
2465
1873
 
2466
1874
    def test_apply_deletions(self):
2467
1875
        self.build_tree(['a/', 'b/'])
2468
1876
        mover = _FileMover()
2469
1877
        mover.pre_delete('a', 'q')
2470
1878
        mover.pre_delete('b', 'r')
2471
 
        self.assertPathExists('q')
2472
 
        self.assertPathExists('r')
2473
 
        self.assertPathDoesNotExist('a')
2474
 
        self.assertPathDoesNotExist('b')
 
1879
        self.failUnlessExists('q')
 
1880
        self.failUnlessExists('r')
 
1881
        self.failIfExists('a')
 
1882
        self.failIfExists('b')
2475
1883
        mover.apply_deletions()
2476
 
        self.assertPathDoesNotExist('q')
2477
 
        self.assertPathDoesNotExist('r')
2478
 
        self.assertPathDoesNotExist('a')
2479
 
        self.assertPathDoesNotExist('b')
 
1884
        self.failIfExists('q')
 
1885
        self.failIfExists('r')
 
1886
        self.failIfExists('a')
 
1887
        self.failIfExists('b')
2480
1888
 
2481
1889
    def test_file_mover_rollback(self):
2482
1890
        self.build_tree(['a/', 'a/b', 'c/', 'c/d/', 'c/e/'])
2487
1895
            mover.rename('a', 'c')
2488
1896
        except errors.FileExists, e:
2489
1897
            mover.rollback()
2490
 
        self.assertPathExists('a')
2491
 
        self.assertPathExists('c/d')
 
1898
        self.failUnlessExists('a')
 
1899
        self.failUnlessExists('c/d')
2492
1900
 
2493
1901
 
2494
1902
class Bogus(Exception):
2524
1932
        tt.adjust_path('d', a_id, tt.trans_id_tree_path('a/b'))
2525
1933
        self.assertRaises(Bogus, tt.apply,
2526
1934
                          _mover=self.ExceptionFileMover(bad_source='a'))
2527
 
        self.assertPathExists('a')
2528
 
        self.assertPathExists('a/b')
 
1935
        self.failUnlessExists('a')
 
1936
        self.failUnlessExists('a/b')
2529
1937
        tt.apply()
2530
 
        self.assertPathExists('c')
2531
 
        self.assertPathExists('c/d')
 
1938
        self.failUnlessExists('c')
 
1939
        self.failUnlessExists('c/d')
2532
1940
 
2533
1941
    def test_rollback_rename_into_place(self):
2534
1942
        tree = self.make_branch_and_tree('.')
2540
1948
        tt.adjust_path('d', a_id, tt.trans_id_tree_path('a/b'))
2541
1949
        self.assertRaises(Bogus, tt.apply,
2542
1950
                          _mover=self.ExceptionFileMover(bad_target='c/d'))
2543
 
        self.assertPathExists('a')
2544
 
        self.assertPathExists('a/b')
 
1951
        self.failUnlessExists('a')
 
1952
        self.failUnlessExists('a/b')
2545
1953
        tt.apply()
2546
 
        self.assertPathExists('c')
2547
 
        self.assertPathExists('c/d')
 
1954
        self.failUnlessExists('c')
 
1955
        self.failUnlessExists('c/d')
2548
1956
 
2549
1957
    def test_rollback_deletion(self):
2550
1958
        tree = self.make_branch_and_tree('.')
2556
1964
        tt.adjust_path('d', tt.root, tt.trans_id_tree_path('a/b'))
2557
1965
        self.assertRaises(Bogus, tt.apply,
2558
1966
                          _mover=self.ExceptionFileMover(bad_target='d'))
2559
 
        self.assertPathExists('a')
2560
 
        self.assertPathExists('a/b')
2561
 
 
2562
 
 
2563
 
class TestFinalizeRobustness(tests.TestCaseWithTransport):
2564
 
    """Ensure treetransform creation errors can be safely cleaned up after"""
2565
 
 
2566
 
    def _override_globals_in_method(self, instance, method_name, globals):
2567
 
        """Replace method on instance with one with updated globals"""
2568
 
        import types
2569
 
        func = getattr(instance, method_name).im_func
2570
 
        new_globals = dict(func.func_globals)
2571
 
        new_globals.update(globals)
2572
 
        new_func = types.FunctionType(func.func_code, new_globals,
2573
 
            func.func_name, func.func_defaults)
2574
 
        setattr(instance, method_name,
2575
 
            types.MethodType(new_func, instance, instance.__class__))
2576
 
        self.addCleanup(delattr, instance, method_name)
2577
 
 
2578
 
    @staticmethod
2579
 
    def _fake_open_raises_before(name, mode):
2580
 
        """Like open() but raises before doing anything"""
2581
 
        raise RuntimeError
2582
 
 
2583
 
    @staticmethod
2584
 
    def _fake_open_raises_after(name, mode):
2585
 
        """Like open() but raises after creating file without returning"""
2586
 
        open(name, mode).close()
2587
 
        raise RuntimeError
2588
 
 
2589
 
    def create_transform_and_root_trans_id(self):
2590
 
        """Setup a transform creating a file in limbo"""
2591
 
        tree = self.make_branch_and_tree('.')
2592
 
        tt = TreeTransform(tree)
2593
 
        return tt, tt.create_path("a", tt.root)
2594
 
 
2595
 
    def create_transform_and_subdir_trans_id(self):
2596
 
        """Setup a transform creating a directory containing a file in limbo"""
2597
 
        tree = self.make_branch_and_tree('.')
2598
 
        tt = TreeTransform(tree)
2599
 
        d_trans_id = tt.create_path("d", tt.root)
2600
 
        tt.create_directory(d_trans_id)
2601
 
        f_trans_id = tt.create_path("a", d_trans_id)
2602
 
        tt.adjust_path("a", d_trans_id, f_trans_id)
2603
 
        return tt, f_trans_id
2604
 
 
2605
 
    def test_root_create_file_open_raises_before_creation(self):
2606
 
        tt, trans_id = self.create_transform_and_root_trans_id()
2607
 
        self._override_globals_in_method(tt, "create_file",
2608
 
            {"open": self._fake_open_raises_before})
2609
 
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
2610
 
        path = tt._limbo_name(trans_id)
2611
 
        self.assertPathDoesNotExist(path)
2612
 
        tt.finalize()
2613
 
        self.assertPathDoesNotExist(tt._limbodir)
2614
 
 
2615
 
    def test_root_create_file_open_raises_after_creation(self):
2616
 
        tt, trans_id = self.create_transform_and_root_trans_id()
2617
 
        self._override_globals_in_method(tt, "create_file",
2618
 
            {"open": self._fake_open_raises_after})
2619
 
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
2620
 
        path = tt._limbo_name(trans_id)
2621
 
        self.assertPathExists(path)
2622
 
        tt.finalize()
2623
 
        self.assertPathDoesNotExist(path)
2624
 
        self.assertPathDoesNotExist(tt._limbodir)
2625
 
 
2626
 
    def test_subdir_create_file_open_raises_before_creation(self):
2627
 
        tt, trans_id = self.create_transform_and_subdir_trans_id()
2628
 
        self._override_globals_in_method(tt, "create_file",
2629
 
            {"open": self._fake_open_raises_before})
2630
 
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
2631
 
        path = tt._limbo_name(trans_id)
2632
 
        self.assertPathDoesNotExist(path)
2633
 
        tt.finalize()
2634
 
        self.assertPathDoesNotExist(tt._limbodir)
2635
 
 
2636
 
    def test_subdir_create_file_open_raises_after_creation(self):
2637
 
        tt, trans_id = self.create_transform_and_subdir_trans_id()
2638
 
        self._override_globals_in_method(tt, "create_file",
2639
 
            {"open": self._fake_open_raises_after})
2640
 
        self.assertRaises(RuntimeError, tt.create_file, ["contents"], trans_id)
2641
 
        path = tt._limbo_name(trans_id)
2642
 
        self.assertPathExists(path)
2643
 
        tt.finalize()
2644
 
        self.assertPathDoesNotExist(path)
2645
 
        self.assertPathDoesNotExist(tt._limbodir)
2646
 
 
2647
 
    def test_rename_in_limbo_rename_raises_after_rename(self):
2648
 
        tt, trans_id = self.create_transform_and_root_trans_id()
2649
 
        parent1 = tt.new_directory('parent1', tt.root)
2650
 
        child1 = tt.new_file('child1', parent1, 'contents')
2651
 
        parent2 = tt.new_directory('parent2', tt.root)
2652
 
 
2653
 
        class FakeOSModule(object):
2654
 
            def rename(self, old, new):
2655
 
                os.rename(old, new)
2656
 
                raise RuntimeError
2657
 
        self._override_globals_in_method(tt, "_rename_in_limbo",
2658
 
            {"os": FakeOSModule()})
2659
 
        self.assertRaises(
2660
 
            RuntimeError, tt.adjust_path, "child1", parent2, child1)
2661
 
        path = osutils.pathjoin(tt._limbo_name(parent2), "child1")
2662
 
        self.assertPathExists(path)
2663
 
        tt.finalize()
2664
 
        self.assertPathDoesNotExist(path)
2665
 
        self.assertPathDoesNotExist(tt._limbodir)
2666
 
 
2667
 
    def test_rename_in_limbo_rename_raises_before_rename(self):
2668
 
        tt, trans_id = self.create_transform_and_root_trans_id()
2669
 
        parent1 = tt.new_directory('parent1', tt.root)
2670
 
        child1 = tt.new_file('child1', parent1, 'contents')
2671
 
        parent2 = tt.new_directory('parent2', tt.root)
2672
 
 
2673
 
        class FakeOSModule(object):
2674
 
            def rename(self, old, new):
2675
 
                raise RuntimeError
2676
 
        self._override_globals_in_method(tt, "_rename_in_limbo",
2677
 
            {"os": FakeOSModule()})
2678
 
        self.assertRaises(
2679
 
            RuntimeError, tt.adjust_path, "child1", parent2, child1)
2680
 
        path = osutils.pathjoin(tt._limbo_name(parent1), "child1")
2681
 
        self.assertPathExists(path)
2682
 
        tt.finalize()
2683
 
        self.assertPathDoesNotExist(path)
2684
 
        self.assertPathDoesNotExist(tt._limbodir)
2685
 
 
2686
 
 
2687
 
class TestTransformMissingParent(tests.TestCaseWithTransport):
2688
 
 
2689
 
    def make_tt_with_versioned_dir(self):
 
1967
        self.failUnlessExists('a')
 
1968
        self.failUnlessExists('a/b')
 
1969
 
 
1970
    def test_resolve_no_parent(self):
2690
1971
        wt = self.make_branch_and_tree('.')
2691
 
        self.build_tree(['dir/',])
2692
 
        wt.add(['dir'], ['dir-id'])
2693
 
        wt.commit('Create dir')
2694
1972
        tt = TreeTransform(wt)
2695
1973
        self.addCleanup(tt.finalize)
2696
 
        return wt, tt
2697
 
 
2698
 
    def test_resolve_create_parent_for_versioned_file(self):
2699
 
        wt, tt = self.make_tt_with_versioned_dir()
2700
 
        dir_tid = tt.trans_id_tree_file_id('dir-id')
2701
 
        file_tid = tt.new_file('file', dir_tid, 'Contents', file_id='file-id')
2702
 
        tt.delete_contents(dir_tid)
2703
 
        tt.unversion_file(dir_tid)
2704
 
        conflicts = resolve_conflicts(tt)
2705
 
        # one conflict for the missing directory, one for the unversioned
2706
 
        # parent
2707
 
        self.assertLength(2, conflicts)
2708
 
 
2709
 
    def test_non_versioned_file_create_conflict(self):
2710
 
        wt, tt = self.make_tt_with_versioned_dir()
2711
 
        dir_tid = tt.trans_id_tree_file_id('dir-id')
2712
 
        tt.new_file('file', dir_tid, 'Contents')
2713
 
        tt.delete_contents(dir_tid)
2714
 
        tt.unversion_file(dir_tid)
2715
 
        conflicts = resolve_conflicts(tt)
2716
 
        # no conflicts or rather: orphaning 'file' resolve the 'dir' conflict
2717
 
        self.assertLength(1, conflicts)
2718
 
        self.assertEqual(('deleting parent', 'Not deleting', 'new-1'),
2719
 
                         conflicts.pop())
2720
 
 
2721
 
 
2722
 
A_ENTRY = ('a-id', ('a', 'a'), True, (True, True),
2723
 
                  ('TREE_ROOT', 'TREE_ROOT'), ('a', 'a'), ('file', 'file'),
2724
 
                  (False, False))
2725
 
ROOT_ENTRY = ('TREE_ROOT', ('', ''), False, (True, True), (None, None),
2726
 
              ('', ''), ('directory', 'directory'), (False, False))
 
1974
        parent = tt.trans_id_file_id('parent-id')
 
1975
        tt.new_file('file', parent, 'Contents')
 
1976
        resolve_conflicts(tt)
2727
1977
 
2728
1978
 
2729
1979
class TestTransformPreview(tests.TestCaseWithTransport):
2731
1981
    def create_tree(self):
2732
1982
        tree = self.make_branch_and_tree('.')
2733
1983
        self.build_tree_contents([('a', 'content 1')])
2734
 
        tree.set_root_id('TREE_ROOT')
2735
1984
        tree.add('a', 'a-id')
2736
1985
        tree.commit('rev1', rev_id='rev1')
2737
1986
        return tree.branch.repository.revision_tree('rev1')
2804
2053
                          (False, False))],
2805
2054
                          list(preview_tree.iter_changes(revision_tree)))
2806
2055
 
2807
 
    def test_include_unchanged_succeeds(self):
 
2056
    def test_wrong_tree_value_error(self):
2808
2057
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
2809
 
        changes = preview_tree.iter_changes(revision_tree,
2810
 
                                            include_unchanged=True)
2811
 
        root = revision_tree.inventory.root.file_id
 
2058
        e = self.assertRaises(ValueError, preview_tree.iter_changes,
 
2059
                              preview_tree)
 
2060
        self.assertEqual('from_tree must be transform source tree.', str(e))
2812
2061
 
2813
 
        self.assertEqual([ROOT_ENTRY, A_ENTRY], list(changes))
 
2062
    def test_include_unchanged_value_error(self):
 
2063
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
 
2064
        e = self.assertRaises(ValueError, preview_tree.iter_changes,
 
2065
                              revision_tree, include_unchanged=True)
 
2066
        self.assertEqual('include_unchanged is not supported', str(e))
2814
2067
 
2815
2068
    def test_specific_files(self):
2816
2069
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
2817
 
        changes = preview_tree.iter_changes(revision_tree,
2818
 
                                            specific_files=[''])
2819
 
        self.assertEqual([A_ENTRY], list(changes))
 
2070
        e = self.assertRaises(ValueError, preview_tree.iter_changes,
 
2071
                              revision_tree, specific_files=['pete'])
 
2072
        self.assertEqual('specific_files is not supported', str(e))
2820
2073
 
2821
 
    def test_want_unversioned(self):
 
2074
    def test_want_unversioned_value_error(self):
2822
2075
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
2823
 
        changes = preview_tree.iter_changes(revision_tree,
2824
 
                                            want_unversioned=True)
2825
 
        self.assertEqual([A_ENTRY], list(changes))
 
2076
        e = self.assertRaises(ValueError, preview_tree.iter_changes,
 
2077
                              revision_tree, want_unversioned=True)
 
2078
        self.assertEqual('want_unversioned is not supported', str(e))
2826
2079
 
2827
2080
    def test_ignore_extra_trees_no_specific_files(self):
2828
2081
        # extra_trees is harmless without specific_files, so we'll silently
2838
2091
    def test_ignore_pb(self):
2839
2092
        # pb could be supported, but TT.iter_changes doesn't support it.
2840
2093
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
2841
 
        preview_tree.iter_changes(revision_tree)
 
2094
        preview_tree.iter_changes(revision_tree, pb=progress.DummyProgress())
2842
2095
 
2843
2096
    def test_kind(self):
2844
2097
        revision_tree = self.create_tree()
2859
2112
        self.assertEqual(os.stat(limbo_path).st_mtime,
2860
2113
                         preview_tree.get_file_mtime('file-id'))
2861
2114
 
2862
 
    def test_get_file_mtime_renamed(self):
2863
 
        work_tree = self.make_branch_and_tree('tree')
2864
 
        self.build_tree(['tree/file'])
2865
 
        work_tree.add('file', 'file-id')
2866
 
        preview = TransformPreview(work_tree)
2867
 
        self.addCleanup(preview.finalize)
2868
 
        file_trans_id = preview.trans_id_tree_file_id('file-id')
2869
 
        preview.adjust_path('renamed', preview.root, file_trans_id)
2870
 
        preview_tree = preview.get_preview_tree()
2871
 
        preview_mtime = preview_tree.get_file_mtime('file-id', 'renamed')
2872
 
        work_mtime = work_tree.get_file_mtime('file-id', 'file')
2873
 
 
2874
 
    def test_get_file_size(self):
2875
 
        work_tree = self.make_branch_and_tree('tree')
2876
 
        self.build_tree_contents([('tree/old', 'old')])
2877
 
        work_tree.add('old', 'old-id')
2878
 
        preview = TransformPreview(work_tree)
2879
 
        self.addCleanup(preview.finalize)
2880
 
        new_id = preview.new_file('name', preview.root, 'contents', 'new-id',
2881
 
                                  'executable')
2882
 
        tree = preview.get_preview_tree()
2883
 
        self.assertEqual(len('old'), tree.get_file_size('old-id'))
2884
 
        self.assertEqual(len('contents'), tree.get_file_size('new-id'))
2885
 
 
2886
2115
    def test_get_file(self):
2887
2116
        preview = self.get_empty_preview()
2888
2117
        preview.new_file('file', preview.root, 'contents', 'file-id')
3036
2265
        self.assertEqual(('missing', None, None, None), summary)
3037
2266
 
3038
2267
    def test_file_content_summary_executable(self):
 
2268
        if not osutils.supports_executable():
 
2269
            raise TestNotApplicable()
3039
2270
        preview = self.get_empty_preview()
3040
2271
        path_id = preview.new_file('path', preview.root, 'contents', 'path-id')
3041
2272
        preview.set_executability(True, path_id)
3050
2281
        self.assertIs(None, summary[3])
3051
2282
 
3052
2283
    def test_change_executability(self):
 
2284
        if not osutils.supports_executable():
 
2285
            raise TestNotApplicable()
3053
2286
        tree = self.make_branch_and_tree('tree')
3054
2287
        self.build_tree(['tree/path'])
3055
2288
        tree.add('path')
3069
2302
        # size must be known
3070
2303
        self.assertEqual(len('contents'), summary[1])
3071
2304
        # not executable
3072
 
        self.assertEqual(False, summary[2])
 
2305
        if osutils.supports_executable():
 
2306
            self.assertEqual(False, summary[2])
 
2307
        else:
 
2308
            self.assertEqual(None, summary[2])
3073
2309
        # will not have hash (not cheap to determine)
3074
2310
        self.assertIs(None, summary[3])
3075
2311
 
3216
2452
 
3217
2453
    def test_walkdirs(self):
3218
2454
        preview = self.get_empty_preview()
3219
 
        root = preview.new_directory('', ROOT_PARENT, 'tree-root')
3220
 
        # FIXME: new_directory should mark root.
3221
 
        preview.fixup_new_roots()
 
2455
        preview.version_file('tree-root', preview.root)
3222
2456
        preview_tree = preview.get_preview_tree()
3223
2457
        file_trans_id = preview.new_file('a', preview.root, 'contents',
3224
2458
                                         'a-id')
3240
2474
        preview.unversion_file(preview.trans_id_tree_path('removed-file'))
3241
2475
        self.assertEqual(set(['new-file', 'removed-file', 'existing-file']),
3242
2476
                         set(tree.extras()))
3243
 
 
3244
 
    def test_merge_into_preview(self):
3245
 
        work_tree = self.make_branch_and_tree('tree')
3246
 
        self.build_tree_contents([('tree/file','b\n')])
3247
 
        work_tree.add('file', 'file-id')
3248
 
        work_tree.commit('first commit')
3249
 
        child_tree = work_tree.bzrdir.sprout('child').open_workingtree()
3250
 
        self.build_tree_contents([('child/file','b\nc\n')])
3251
 
        child_tree.commit('child commit')
3252
 
        child_tree.lock_write()
3253
 
        self.addCleanup(child_tree.unlock)
3254
 
        work_tree.lock_write()
3255
 
        self.addCleanup(work_tree.unlock)
3256
 
        preview = TransformPreview(work_tree)
3257
 
        self.addCleanup(preview.finalize)
3258
 
        file_trans_id = preview.trans_id_file_id('file-id')
3259
 
        preview.delete_contents(file_trans_id)
3260
 
        preview.create_file('a\nb\n', file_trans_id)
3261
 
        preview_tree = preview.get_preview_tree()
3262
 
        merger = Merger.from_revision_ids(None, preview_tree,
3263
 
                                          child_tree.branch.last_revision(),
3264
 
                                          other_branch=child_tree.branch,
3265
 
                                          tree_branch=work_tree.branch)
3266
 
        merger.merge_type = Merge3Merger
3267
 
        tt = merger.make_merger().make_preview_transform()
3268
 
        self.addCleanup(tt.finalize)
3269
 
        final_tree = tt.get_preview_tree()
3270
 
        self.assertEqual('a\nb\nc\n', final_tree.get_file_text('file-id'))
3271
 
 
3272
 
    def test_merge_preview_into_workingtree(self):
3273
 
        tree = self.make_branch_and_tree('tree')
3274
 
        tree.set_root_id('TREE_ROOT')
3275
 
        tt = TransformPreview(tree)
3276
 
        self.addCleanup(tt.finalize)
3277
 
        tt.new_file('name', tt.root, 'content', 'file-id')
3278
 
        tree2 = self.make_branch_and_tree('tree2')
3279
 
        tree2.set_root_id('TREE_ROOT')
3280
 
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
3281
 
                                         None, tree.basis_tree())
3282
 
        merger.merge_type = Merge3Merger
3283
 
        merger.do_merge()
3284
 
 
3285
 
    def test_merge_preview_into_workingtree_handles_conflicts(self):
3286
 
        tree = self.make_branch_and_tree('tree')
3287
 
        self.build_tree_contents([('tree/foo', 'bar')])
3288
 
        tree.add('foo', 'foo-id')
3289
 
        tree.commit('foo')
3290
 
        tt = TransformPreview(tree)
3291
 
        self.addCleanup(tt.finalize)
3292
 
        trans_id = tt.trans_id_file_id('foo-id')
3293
 
        tt.delete_contents(trans_id)
3294
 
        tt.create_file('baz', trans_id)
3295
 
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
3296
 
        self.build_tree_contents([('tree2/foo', 'qux')])
3297
 
        pb = None
3298
 
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
3299
 
                                         pb, tree.basis_tree())
3300
 
        merger.merge_type = Merge3Merger
3301
 
        merger.do_merge()
3302
 
 
3303
 
    def test_has_filename(self):
3304
 
        wt = self.make_branch_and_tree('tree')
3305
 
        self.build_tree(['tree/unmodified', 'tree/removed', 'tree/modified'])
3306
 
        tt = TransformPreview(wt)
3307
 
        removed_id = tt.trans_id_tree_path('removed')
3308
 
        tt.delete_contents(removed_id)
3309
 
        tt.new_file('new', tt.root, 'contents')
3310
 
        modified_id = tt.trans_id_tree_path('modified')
3311
 
        tt.delete_contents(modified_id)
3312
 
        tt.create_file('modified-contents', modified_id)
3313
 
        self.addCleanup(tt.finalize)
3314
 
        tree = tt.get_preview_tree()
3315
 
        self.assertTrue(tree.has_filename('unmodified'))
3316
 
        self.assertFalse(tree.has_filename('not-present'))
3317
 
        self.assertFalse(tree.has_filename('removed'))
3318
 
        self.assertTrue(tree.has_filename('new'))
3319
 
        self.assertTrue(tree.has_filename('modified'))
3320
 
 
3321
 
    def test_is_executable(self):
3322
 
        tree = self.make_branch_and_tree('tree')
3323
 
        preview = TransformPreview(tree)
3324
 
        self.addCleanup(preview.finalize)
3325
 
        preview.new_file('foo', preview.root, 'bar', 'baz-id')
3326
 
        preview_tree = preview.get_preview_tree()
3327
 
        self.assertEqual(False, preview_tree.is_executable('baz-id',
3328
 
                                                           'tree/foo'))
3329
 
        self.assertEqual(False, preview_tree.is_executable('baz-id'))
3330
 
 
3331
 
    def test_commit_preview_tree(self):
3332
 
        tree = self.make_branch_and_tree('tree')
3333
 
        rev_id = tree.commit('rev1')
3334
 
        tree.branch.lock_write()
3335
 
        self.addCleanup(tree.branch.unlock)
3336
 
        tt = TransformPreview(tree)
3337
 
        tt.new_file('file', tt.root, 'contents', 'file_id')
3338
 
        self.addCleanup(tt.finalize)
3339
 
        preview = tt.get_preview_tree()
3340
 
        preview.set_parent_ids([rev_id])
3341
 
        builder = tree.branch.get_commit_builder([rev_id])
3342
 
        list(builder.record_iter_changes(preview, rev_id, tt.iter_changes()))
3343
 
        builder.finish_inventory()
3344
 
        rev2_id = builder.commit('rev2')
3345
 
        rev2_tree = tree.branch.repository.revision_tree(rev2_id)
3346
 
        self.assertEqual('contents', rev2_tree.get_file_text('file_id'))
3347
 
 
3348
 
    def test_ascii_limbo_paths(self):
3349
 
        self.requireFeature(features.UnicodeFilenameFeature)
3350
 
        branch = self.make_branch('any')
3351
 
        tree = branch.repository.revision_tree(_mod_revision.NULL_REVISION)
3352
 
        tt = TransformPreview(tree)
3353
 
        self.addCleanup(tt.finalize)
3354
 
        foo_id = tt.new_directory('', ROOT_PARENT)
3355
 
        bar_id = tt.new_file(u'\u1234bar', foo_id, 'contents')
3356
 
        limbo_path = tt._limbo_name(bar_id)
3357
 
        self.assertEqual(limbo_path.encode('ascii', 'replace'), limbo_path)
3358
 
 
3359
 
 
3360
 
class FakeSerializer(object):
3361
 
    """Serializer implementation that simply returns the input.
3362
 
 
3363
 
    The input is returned in the order used by pack.ContainerPushParser.
3364
 
    """
3365
 
    @staticmethod
3366
 
    def bytes_record(bytes, names):
3367
 
        return names, bytes
3368
 
 
3369
 
 
3370
 
class TestSerializeTransform(tests.TestCaseWithTransport):
3371
 
 
3372
 
    _test_needs_features = [features.UnicodeFilenameFeature]
3373
 
 
3374
 
    def get_preview(self, tree=None):
3375
 
        if tree is None:
3376
 
            tree = self.make_branch_and_tree('tree')
3377
 
        tt = TransformPreview(tree)
3378
 
        self.addCleanup(tt.finalize)
3379
 
        return tt
3380
 
 
3381
 
    def assertSerializesTo(self, expected, tt):
3382
 
        records = list(tt.serialize(FakeSerializer()))
3383
 
        self.assertEqual(expected, records)
3384
 
 
3385
 
    @staticmethod
3386
 
    def default_attribs():
3387
 
        return {
3388
 
            '_id_number': 1,
3389
 
            '_new_name': {},
3390
 
            '_new_parent': {},
3391
 
            '_new_executability': {},
3392
 
            '_new_id': {},
3393
 
            '_tree_path_ids': {'': 'new-0'},
3394
 
            '_removed_id': [],
3395
 
            '_removed_contents': [],
3396
 
            '_non_present_ids': {},
3397
 
            }
3398
 
 
3399
 
    def make_records(self, attribs, contents):
3400
 
        records = [
3401
 
            (((('attribs'),),), bencode.bencode(attribs))]
3402
 
        records.extend([(((n, k),), c) for n, k, c in contents])
3403
 
        return records
3404
 
 
3405
 
    def creation_records(self):
3406
 
        attribs = self.default_attribs()
3407
 
        attribs['_id_number'] = 3
3408
 
        attribs['_new_name'] = {
3409
 
            'new-1': u'foo\u1234'.encode('utf-8'), 'new-2': 'qux'}
3410
 
        attribs['_new_id'] = {'new-1': 'baz', 'new-2': 'quxx'}
3411
 
        attribs['_new_parent'] = {'new-1': 'new-0', 'new-2': 'new-0'}
3412
 
        attribs['_new_executability'] = {'new-1': 1}
3413
 
        contents = [
3414
 
            ('new-1', 'file', 'i 1\nbar\n'),
3415
 
            ('new-2', 'directory', ''),
3416
 
            ]
3417
 
        return self.make_records(attribs, contents)
3418
 
 
3419
 
    def test_serialize_creation(self):
3420
 
        tt = self.get_preview()
3421
 
        tt.new_file(u'foo\u1234', tt.root, 'bar', 'baz', True)
3422
 
        tt.new_directory('qux', tt.root, 'quxx')
3423
 
        self.assertSerializesTo(self.creation_records(), tt)
3424
 
 
3425
 
    def test_deserialize_creation(self):
3426
 
        tt = self.get_preview()
3427
 
        tt.deserialize(iter(self.creation_records()))
3428
 
        self.assertEqual(3, tt._id_number)
3429
 
        self.assertEqual({'new-1': u'foo\u1234',
3430
 
                          'new-2': 'qux'}, tt._new_name)
3431
 
        self.assertEqual({'new-1': 'baz', 'new-2': 'quxx'}, tt._new_id)
3432
 
        self.assertEqual({'new-1': tt.root, 'new-2': tt.root}, tt._new_parent)
3433
 
        self.assertEqual({'baz': 'new-1', 'quxx': 'new-2'}, tt._r_new_id)
3434
 
        self.assertEqual({'new-1': True}, tt._new_executability)
3435
 
        self.assertEqual({'new-1': 'file',
3436
 
                          'new-2': 'directory'}, tt._new_contents)
3437
 
        foo_limbo = open(tt._limbo_name('new-1'), 'rb')
3438
 
        try:
3439
 
            foo_content = foo_limbo.read()
3440
 
        finally:
3441
 
            foo_limbo.close()
3442
 
        self.assertEqual('bar', foo_content)
3443
 
 
3444
 
    def symlink_creation_records(self):
3445
 
        attribs = self.default_attribs()
3446
 
        attribs['_id_number'] = 2
3447
 
        attribs['_new_name'] = {'new-1': u'foo\u1234'.encode('utf-8')}
3448
 
        attribs['_new_parent'] = {'new-1': 'new-0'}
3449
 
        contents = [('new-1', 'symlink', u'bar\u1234'.encode('utf-8'))]
3450
 
        return self.make_records(attribs, contents)
3451
 
 
3452
 
    def test_serialize_symlink_creation(self):
3453
 
        self.requireFeature(features.SymlinkFeature)
3454
 
        tt = self.get_preview()
3455
 
        tt.new_symlink(u'foo\u1234', tt.root, u'bar\u1234')
3456
 
        self.assertSerializesTo(self.symlink_creation_records(), tt)
3457
 
 
3458
 
    def test_deserialize_symlink_creation(self):
3459
 
        self.requireFeature(features.SymlinkFeature)
3460
 
        tt = self.get_preview()
3461
 
        tt.deserialize(iter(self.symlink_creation_records()))
3462
 
        abspath = tt._limbo_name('new-1')
3463
 
        foo_content = osutils.readlink(abspath)
3464
 
        self.assertEqual(u'bar\u1234', foo_content)
3465
 
 
3466
 
    def make_destruction_preview(self):
3467
 
        tree = self.make_branch_and_tree('.')
3468
 
        self.build_tree([u'foo\u1234', 'bar'])
3469
 
        tree.add([u'foo\u1234', 'bar'], ['foo-id', 'bar-id'])
3470
 
        return self.get_preview(tree)
3471
 
 
3472
 
    def destruction_records(self):
3473
 
        attribs = self.default_attribs()
3474
 
        attribs['_id_number'] = 3
3475
 
        attribs['_removed_id'] = ['new-1']
3476
 
        attribs['_removed_contents'] = ['new-2']
3477
 
        attribs['_tree_path_ids'] = {
3478
 
            '': 'new-0',
3479
 
            u'foo\u1234'.encode('utf-8'): 'new-1',
3480
 
            'bar': 'new-2',
3481
 
            }
3482
 
        return self.make_records(attribs, [])
3483
 
 
3484
 
    def test_serialize_destruction(self):
3485
 
        tt = self.make_destruction_preview()
3486
 
        foo_trans_id = tt.trans_id_tree_file_id('foo-id')
3487
 
        tt.unversion_file(foo_trans_id)
3488
 
        bar_trans_id = tt.trans_id_tree_file_id('bar-id')
3489
 
        tt.delete_contents(bar_trans_id)
3490
 
        self.assertSerializesTo(self.destruction_records(), tt)
3491
 
 
3492
 
    def test_deserialize_destruction(self):
3493
 
        tt = self.make_destruction_preview()
3494
 
        tt.deserialize(iter(self.destruction_records()))
3495
 
        self.assertEqual({u'foo\u1234': 'new-1',
3496
 
                          'bar': 'new-2',
3497
 
                          '': tt.root}, tt._tree_path_ids)
3498
 
        self.assertEqual({'new-1': u'foo\u1234',
3499
 
                          'new-2': 'bar',
3500
 
                          tt.root: ''}, tt._tree_id_paths)
3501
 
        self.assertEqual(set(['new-1']), tt._removed_id)
3502
 
        self.assertEqual(set(['new-2']), tt._removed_contents)
3503
 
 
3504
 
    def missing_records(self):
3505
 
        attribs = self.default_attribs()
3506
 
        attribs['_id_number'] = 2
3507
 
        attribs['_non_present_ids'] = {
3508
 
            'boo': 'new-1',}
3509
 
        return self.make_records(attribs, [])
3510
 
 
3511
 
    def test_serialize_missing(self):
3512
 
        tt = self.get_preview()
3513
 
        boo_trans_id = tt.trans_id_file_id('boo')
3514
 
        self.assertSerializesTo(self.missing_records(), tt)
3515
 
 
3516
 
    def test_deserialize_missing(self):
3517
 
        tt = self.get_preview()
3518
 
        tt.deserialize(iter(self.missing_records()))
3519
 
        self.assertEqual({'boo': 'new-1'}, tt._non_present_ids)
3520
 
 
3521
 
    def make_modification_preview(self):
3522
 
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
3523
 
        LINES_TWO = 'z\nbb\nx\ndd\n'
3524
 
        tree = self.make_branch_and_tree('tree')
3525
 
        self.build_tree_contents([('tree/file', LINES_ONE)])
3526
 
        tree.add('file', 'file-id')
3527
 
        return self.get_preview(tree), LINES_TWO
3528
 
 
3529
 
    def modification_records(self):
3530
 
        attribs = self.default_attribs()
3531
 
        attribs['_id_number'] = 2
3532
 
        attribs['_tree_path_ids'] = {
3533
 
            'file': 'new-1',
3534
 
            '': 'new-0',}
3535
 
        attribs['_removed_contents'] = ['new-1']
3536
 
        contents = [('new-1', 'file',
3537
 
                     'i 1\nz\n\nc 0 1 1 1\ni 1\nx\n\nc 0 3 3 1\n')]
3538
 
        return self.make_records(attribs, contents)
3539
 
 
3540
 
    def test_serialize_modification(self):
3541
 
        tt, LINES = self.make_modification_preview()
3542
 
        trans_id = tt.trans_id_file_id('file-id')
3543
 
        tt.delete_contents(trans_id)
3544
 
        tt.create_file(LINES, trans_id)
3545
 
        self.assertSerializesTo(self.modification_records(), tt)
3546
 
 
3547
 
    def test_deserialize_modification(self):
3548
 
        tt, LINES = self.make_modification_preview()
3549
 
        tt.deserialize(iter(self.modification_records()))
3550
 
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3551
 
 
3552
 
    def make_kind_change_preview(self):
3553
 
        LINES = 'a\nb\nc\nd\n'
3554
 
        tree = self.make_branch_and_tree('tree')
3555
 
        self.build_tree(['tree/foo/'])
3556
 
        tree.add('foo', 'foo-id')
3557
 
        return self.get_preview(tree), LINES
3558
 
 
3559
 
    def kind_change_records(self):
3560
 
        attribs = self.default_attribs()
3561
 
        attribs['_id_number'] = 2
3562
 
        attribs['_tree_path_ids'] = {
3563
 
            'foo': 'new-1',
3564
 
            '': 'new-0',}
3565
 
        attribs['_removed_contents'] = ['new-1']
3566
 
        contents = [('new-1', 'file',
3567
 
                     'i 4\na\nb\nc\nd\n\n')]
3568
 
        return self.make_records(attribs, contents)
3569
 
 
3570
 
    def test_serialize_kind_change(self):
3571
 
        tt, LINES = self.make_kind_change_preview()
3572
 
        trans_id = tt.trans_id_file_id('foo-id')
3573
 
        tt.delete_contents(trans_id)
3574
 
        tt.create_file(LINES, trans_id)
3575
 
        self.assertSerializesTo(self.kind_change_records(), tt)
3576
 
 
3577
 
    def test_deserialize_kind_change(self):
3578
 
        tt, LINES = self.make_kind_change_preview()
3579
 
        tt.deserialize(iter(self.kind_change_records()))
3580
 
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3581
 
 
3582
 
    def make_add_contents_preview(self):
3583
 
        LINES = 'a\nb\nc\nd\n'
3584
 
        tree = self.make_branch_and_tree('tree')
3585
 
        self.build_tree(['tree/foo'])
3586
 
        tree.add('foo')
3587
 
        os.unlink('tree/foo')
3588
 
        return self.get_preview(tree), LINES
3589
 
 
3590
 
    def add_contents_records(self):
3591
 
        attribs = self.default_attribs()
3592
 
        attribs['_id_number'] = 2
3593
 
        attribs['_tree_path_ids'] = {
3594
 
            'foo': 'new-1',
3595
 
            '': 'new-0',}
3596
 
        contents = [('new-1', 'file',
3597
 
                     'i 4\na\nb\nc\nd\n\n')]
3598
 
        return self.make_records(attribs, contents)
3599
 
 
3600
 
    def test_serialize_add_contents(self):
3601
 
        tt, LINES = self.make_add_contents_preview()
3602
 
        trans_id = tt.trans_id_tree_path('foo')
3603
 
        tt.create_file(LINES, trans_id)
3604
 
        self.assertSerializesTo(self.add_contents_records(), tt)
3605
 
 
3606
 
    def test_deserialize_add_contents(self):
3607
 
        tt, LINES = self.make_add_contents_preview()
3608
 
        tt.deserialize(iter(self.add_contents_records()))
3609
 
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3610
 
 
3611
 
    def test_get_parents_lines(self):
3612
 
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
3613
 
        LINES_TWO = 'z\nbb\nx\ndd\n'
3614
 
        tree = self.make_branch_and_tree('tree')
3615
 
        self.build_tree_contents([('tree/file', LINES_ONE)])
3616
 
        tree.add('file', 'file-id')
3617
 
        tt = self.get_preview(tree)
3618
 
        trans_id = tt.trans_id_tree_path('file')
3619
 
        self.assertEqual((['aa\n', 'bb\n', 'cc\n', 'dd\n'],),
3620
 
            tt._get_parents_lines(trans_id))
3621
 
 
3622
 
    def test_get_parents_texts(self):
3623
 
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
3624
 
        LINES_TWO = 'z\nbb\nx\ndd\n'
3625
 
        tree = self.make_branch_and_tree('tree')
3626
 
        self.build_tree_contents([('tree/file', LINES_ONE)])
3627
 
        tree.add('file', 'file-id')
3628
 
        tt = self.get_preview(tree)
3629
 
        trans_id = tt.trans_id_tree_path('file')
3630
 
        self.assertEqual((LINES_ONE,),
3631
 
            tt._get_parents_texts(trans_id))
3632
 
 
3633
 
 
3634
 
class TestOrphan(tests.TestCaseWithTransport):
3635
 
 
3636
 
    def test_no_orphan_for_transform_preview(self):
3637
 
        tree = self.make_branch_and_tree('tree')
3638
 
        tt = transform.TransformPreview(tree)
3639
 
        self.addCleanup(tt.finalize)
3640
 
        self.assertRaises(NotImplementedError, tt.new_orphan, 'foo', 'bar')
3641
 
 
3642
 
    def _set_orphan_policy(self, wt, policy):
3643
 
        wt.branch.get_config().set_user_option('bzr.transform.orphan_policy',
3644
 
                                               policy)
3645
 
 
3646
 
    def _prepare_orphan(self, wt):
3647
 
        self.build_tree(['dir/', 'dir/file', 'dir/foo'])
3648
 
        wt.add(['dir', 'dir/file'], ['dir-id', 'file-id'])
3649
 
        wt.commit('add dir and file ignoring foo')
3650
 
        tt = transform.TreeTransform(wt)
3651
 
        self.addCleanup(tt.finalize)
3652
 
        # dir and bar are deleted
3653
 
        dir_tid = tt.trans_id_tree_path('dir')
3654
 
        file_tid = tt.trans_id_tree_path('dir/file')
3655
 
        orphan_tid = tt.trans_id_tree_path('dir/foo')
3656
 
        tt.delete_contents(file_tid)
3657
 
        tt.unversion_file(file_tid)
3658
 
        tt.delete_contents(dir_tid)
3659
 
        tt.unversion_file(dir_tid)
3660
 
        # There should be a conflict because dir still contain foo
3661
 
        raw_conflicts = tt.find_conflicts()
3662
 
        self.assertLength(1, raw_conflicts)
3663
 
        self.assertEqual(('missing parent', 'new-1'), raw_conflicts[0])
3664
 
        return tt, orphan_tid
3665
 
 
3666
 
    def test_new_orphan_created(self):
3667
 
        wt = self.make_branch_and_tree('.')
3668
 
        self._set_orphan_policy(wt, 'move')
3669
 
        tt, orphan_tid = self._prepare_orphan(wt)
3670
 
        warnings = []
3671
 
        def warning(*args):
3672
 
            warnings.append(args[0] % args[1:])
3673
 
        self.overrideAttr(trace, 'warning', warning)
3674
 
        remaining_conflicts = resolve_conflicts(tt)
3675
 
        self.assertEquals(['dir/foo has been orphaned in bzr-orphans'],
3676
 
                          warnings)
3677
 
        # Yeah for resolved conflicts !
3678
 
        self.assertLength(0, remaining_conflicts)
3679
 
        # We have a new orphan
3680
 
        self.assertEquals('foo.~1~', tt.final_name(orphan_tid))
3681
 
        self.assertEquals('bzr-orphans',
3682
 
                          tt.final_name(tt.final_parent(orphan_tid)))
3683
 
 
3684
 
    def test_never_orphan(self):
3685
 
        wt = self.make_branch_and_tree('.')
3686
 
        self._set_orphan_policy(wt, 'conflict')
3687
 
        tt, orphan_tid = self._prepare_orphan(wt)
3688
 
        remaining_conflicts = resolve_conflicts(tt)
3689
 
        self.assertLength(1, remaining_conflicts)
3690
 
        self.assertEqual(('deleting parent', 'Not deleting', 'new-1'),
3691
 
                         remaining_conflicts.pop())
3692
 
 
3693
 
    def test_orphan_error(self):
3694
 
        def bogus_orphan(tt, orphan_id, parent_id):
3695
 
            raise transform.OrphaningError(tt.final_name(orphan_id),
3696
 
                                           tt.final_name(parent_id))
3697
 
        transform.orphaning_registry.register('bogus', bogus_orphan,
3698
 
                                              'Raise an error when orphaning')
3699
 
        wt = self.make_branch_and_tree('.')
3700
 
        self._set_orphan_policy(wt, 'bogus')
3701
 
        tt, orphan_tid = self._prepare_orphan(wt)
3702
 
        remaining_conflicts = resolve_conflicts(tt)
3703
 
        self.assertLength(1, remaining_conflicts)
3704
 
        self.assertEqual(('deleting parent', 'Not deleting', 'new-1'),
3705
 
                         remaining_conflicts.pop())
3706
 
 
3707
 
    def test_unknown_orphan_policy(self):
3708
 
        wt = self.make_branch_and_tree('.')
3709
 
        # Set a fictional policy nobody ever implemented
3710
 
        self._set_orphan_policy(wt, 'donttouchmypreciouuus')
3711
 
        tt, orphan_tid = self._prepare_orphan(wt)
3712
 
        warnings = []
3713
 
        def warning(*args):
3714
 
            warnings.append(args[0] % args[1:])
3715
 
        self.overrideAttr(trace, 'warning', warning)
3716
 
        remaining_conflicts = resolve_conflicts(tt)
3717
 
        # We fallback to the default policy which create a conflict
3718
 
        self.assertLength(1, remaining_conflicts)
3719
 
        self.assertEqual(('deleting parent', 'Not deleting', 'new-1'),
3720
 
                         remaining_conflicts.pop())
3721
 
        self.assertLength(1, warnings)
3722
 
        self.assertStartsWith(warnings[0], 'donttouchmypreciouuus')