← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_transform.py

  • Committer: Andrew Bennetts
  • Date: 2008-10-30 01:05:59 UTC
  • mto: This revision was merged to the branch mainline in revision 3813.
  • Revision ID: andrew.bennetts@canonical.com-20081030010559-tumoefnsmhg4snxo
Add contrib/bzr_ssh_path_limiter.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_transform.py
1
 
# Copyright (C) 2006-2010 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
17
import os
 
18
import stat
18
19
from StringIO import StringIO
19
20
import sys
20
 
import time
21
21
 
22
22
from bzrlib import (
23
 
    bencode,
24
23
    errors,
25
 
    filters,
26
24
    generate_ids,
27
25
    osutils,
28
26
    progress,
29
27
    revision as _mod_revision,
30
 
    rules,
 
28
    symbol_versioning,
31
29
    tests,
32
30
    urlutils,
33
31
    )
37
35
                              NonDirectoryParent)
38
36
from bzrlib.diff import show_diff_trees
39
37
from bzrlib.errors import (DuplicateKey, MalformedTransform, NoSuchFile,
40
 
                           ReusingTransform, CantMoveRoot,
 
38
                           ReusingTransform, CantMoveRoot, 
41
39
                           PathsNotVersionedError, ExistingLimbo,
42
40
                           ExistingPendingDeletion, ImmortalLimbo,
43
41
                           ImmortalPendingDeletion, LockError)
44
42
from bzrlib.osutils import file_kind, pathjoin
45
43
from bzrlib.merge import Merge3Merger, Merger
46
44
from bzrlib.tests import (
 
45
    CaseInsensitiveFilesystemFeature,
47
46
    HardlinkFeature,
48
47
    SymlinkFeature,
49
48
    TestCase,
50
49
    TestCaseInTempDir,
51
50
    TestSkipped,
52
51
    )
53
 
from bzrlib.transform import (TreeTransform, ROOT_PARENT, FinalPaths,
54
 
                              resolve_conflicts, cook_conflicts,
 
52
from bzrlib.transform import (TreeTransform, ROOT_PARENT, FinalPaths, 
 
53
                              resolve_conflicts, cook_conflicts, 
55
54
                              build_tree, get_backup_name,
56
55
                              _FileMover, resolve_checkout,
57
56
                              TransformPreview, create_from_tree)
58
57
 
59
 
 
60
58
class TestTreeTransform(tests.TestCaseWithTransport):
61
59
 
62
60
    def setUp(self):
130
128
        self.assertIs(self.wt.is_executable('my_pretties2'), False)
131
129
        self.assertEqual('directory', file_kind(self.wt.abspath('oz')))
132
130
        self.assertEqual(len(modified_paths), 3)
133
 
        tree_mod_paths = [self.wt.id2abspath(f) for f in
 
131
        tree_mod_paths = [self.wt.id2abspath(f) for f in 
134
132
                          ('ozzie', 'my_pretties', 'my_pretties2')]
135
133
        self.assertSubset(tree_mod_paths, modified_paths)
136
134
        # is it safe to finalize repeatedly?
137
135
        transform.finalize()
138
136
        transform.finalize()
139
137
 
140
 
    def test_create_files_same_timestamp(self):
141
 
        transform, root = self.get_transform()
142
 
        self.wt.lock_tree_write()
143
 
        self.addCleanup(self.wt.unlock)
144
 
        # Roll back the clock, so that we know everything is being set to the
145
 
        # exact time
146
 
        transform._creation_mtime = creation_mtime = time.time() - 20.0
147
 
        transform.create_file('content-one',
148
 
                              transform.create_path('one', root))
149
 
        time.sleep(1) # *ugly*
150
 
        transform.create_file('content-two',
151
 
                              transform.create_path('two', root))
152
 
        transform.apply()
153
 
        fo, st1 = self.wt.get_file_with_stat(None, path='one', filtered=False)
154
 
        fo.close()
155
 
        fo, st2 = self.wt.get_file_with_stat(None, path='two', filtered=False)
156
 
        fo.close()
157
 
        # We only guarantee 2s resolution
158
 
        self.assertTrue(abs(creation_mtime - st1.st_mtime) < 2.0,
159
 
            "%s != %s within 2 seconds" % (creation_mtime, st1.st_mtime))
160
 
        # But if we have more than that, all files should get the same result
161
 
        self.assertEqual(st1.st_mtime, st2.st_mtime)
162
 
 
163
 
    def test_change_root_id(self):
164
 
        transform, root = self.get_transform()
165
 
        self.assertNotEqual('new-root-id', self.wt.get_root_id())
166
 
        transform.new_directory('', ROOT_PARENT, 'new-root-id')
167
 
        transform.delete_contents(root)
168
 
        transform.unversion_file(root)
169
 
        transform.fixup_new_roots()
170
 
        transform.apply()
171
 
        self.assertEqual('new-root-id', self.wt.get_root_id())
172
 
 
173
 
    def test_change_root_id_add_files(self):
174
 
        transform, root = self.get_transform()
175
 
        self.assertNotEqual('new-root-id', self.wt.get_root_id())
176
 
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'new-root-id')
177
 
        transform.new_file('file', new_trans_id, ['new-contents\n'],
178
 
                           'new-file-id')
179
 
        transform.delete_contents(root)
180
 
        transform.unversion_file(root)
181
 
        transform.fixup_new_roots()
182
 
        transform.apply()
183
 
        self.assertEqual('new-root-id', self.wt.get_root_id())
184
 
        self.assertEqual('new-file-id', self.wt.path2id('file'))
185
 
        self.assertFileEqual('new-contents\n', self.wt.abspath('file'))
186
 
 
187
 
    def test_add_two_roots(self):
188
 
        transform, root = self.get_transform()
189
 
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'new-root-id')
190
 
        new_trans_id = transform.new_directory('', ROOT_PARENT, 'alt-root-id')
191
 
        self.assertRaises(ValueError, transform.fixup_new_roots)
192
 
 
193
138
    def test_hardlink(self):
194
139
        self.requireFeature(HardlinkFeature)
195
140
        transform, root = self.get_transform()
209
154
        transform, root = self.get_transform()
210
155
        self.wt.lock_tree_write()
211
156
        self.addCleanup(self.wt.unlock)
212
 
        trans_id = transform.new_file('name', root, 'contents',
 
157
        trans_id = transform.new_file('name', root, 'contents', 
213
158
                                      'my_pretties', True)
214
159
        oz = transform.new_directory('oz', root, 'oz-id')
215
160
        dorothy = transform.new_directory('dorothy', oz, 'dorothy-id')
216
 
        toto = transform.new_file('toto', dorothy, 'toto-contents',
 
161
        toto = transform.new_file('toto', dorothy, 'toto-contents', 
217
162
                                  'toto-id', False)
218
163
 
219
164
        self.assertEqual(len(transform.find_conflicts()), 0)
243
188
 
244
189
    def test_conflicts(self):
245
190
        transform, root = self.get_transform()
246
 
        trans_id = transform.new_file('name', root, 'contents',
 
191
        trans_id = transform.new_file('name', root, 'contents', 
247
192
                                      'my_pretties')
248
193
        self.assertEqual(len(transform.find_conflicts()), 0)
249
194
        trans_id2 = transform.new_file('name', root, 'Crontents', 'toto')
250
 
        self.assertEqual(transform.find_conflicts(),
 
195
        self.assertEqual(transform.find_conflicts(), 
251
196
                         [('duplicate', trans_id, trans_id2, 'name')])
252
197
        self.assertRaises(MalformedTransform, transform.apply)
253
198
        transform.adjust_path('name', trans_id, trans_id2)
254
 
        self.assertEqual(transform.find_conflicts(),
 
199
        self.assertEqual(transform.find_conflicts(), 
255
200
                         [('non-directory parent', trans_id)])
256
201
        tinman_id = transform.trans_id_tree_path('tinman')
257
202
        transform.adjust_path('name', tinman_id, trans_id2)
258
 
        self.assertEqual(transform.find_conflicts(),
259
 
                         [('unversioned parent', tinman_id),
 
203
        self.assertEqual(transform.find_conflicts(), 
 
204
                         [('unversioned parent', tinman_id), 
260
205
                          ('missing parent', tinman_id)])
261
206
        lion_id = transform.create_path('lion', root)
262
 
        self.assertEqual(transform.find_conflicts(),
263
 
                         [('unversioned parent', tinman_id),
 
207
        self.assertEqual(transform.find_conflicts(), 
 
208
                         [('unversioned parent', tinman_id), 
264
209
                          ('missing parent', tinman_id)])
265
210
        transform.adjust_path('name', lion_id, trans_id2)
266
 
        self.assertEqual(transform.find_conflicts(),
 
211
        self.assertEqual(transform.find_conflicts(), 
267
212
                         [('unversioned parent', lion_id),
268
213
                          ('missing parent', lion_id)])
269
214
        transform.version_file("Courage", lion_id)
270
 
        self.assertEqual(transform.find_conflicts(),
271
 
                         [('missing parent', lion_id),
 
215
        self.assertEqual(transform.find_conflicts(), 
 
216
                         [('missing parent', lion_id), 
272
217
                          ('versioning no contents', lion_id)])
273
218
        transform.adjust_path('name2', root, trans_id2)
274
 
        self.assertEqual(transform.find_conflicts(),
 
219
        self.assertEqual(transform.find_conflicts(), 
275
220
                         [('versioning no contents', lion_id)])
276
221
        transform.create_file('Contents, okay?', lion_id)
277
222
        transform.adjust_path('name2', trans_id2, trans_id2)
278
 
        self.assertEqual(transform.find_conflicts(),
279
 
                         [('parent loop', trans_id2),
 
223
        self.assertEqual(transform.find_conflicts(), 
 
224
                         [('parent loop', trans_id2), 
280
225
                          ('non-directory parent', trans_id2)])
281
226
        transform.adjust_path('name2', root, trans_id2)
282
227
        oz_id = transform.new_directory('oz', root)
283
228
        transform.set_executability(True, oz_id)
284
 
        self.assertEqual(transform.find_conflicts(),
 
229
        self.assertEqual(transform.find_conflicts(), 
285
230
                         [('unversioned executability', oz_id)])
286
231
        transform.version_file('oz-id', oz_id)
287
 
        self.assertEqual(transform.find_conflicts(),
 
232
        self.assertEqual(transform.find_conflicts(), 
288
233
                         [('non-file executability', oz_id)])
289
234
        transform.set_executability(None, oz_id)
290
235
        tip_id = transform.new_file('tip', oz_id, 'ozma', 'tip-id')
299
244
        self.assert_('oz/tip' in transform2._tree_path_ids)
300
245
        self.assertEqual(fp.get_path(newtip), pathjoin('oz', 'tip'))
301
246
        self.assertEqual(len(result), 2)
302
 
        self.assertEqual((result[0][0], result[0][1]),
 
247
        self.assertEqual((result[0][0], result[0][1]), 
303
248
                         ('duplicate', newtip))
304
 
        self.assertEqual((result[1][0], result[1][2]),
 
249
        self.assertEqual((result[1][0], result[1][2]), 
305
250
                         ('duplicate id', newtip))
306
251
        transform2.finalize()
307
252
        transform3 = TreeTransform(self.wt)
308
253
        self.addCleanup(transform3.finalize)
309
254
        oz_id = transform3.trans_id_tree_file_id('oz-id')
310
255
        transform3.delete_contents(oz_id)
311
 
        self.assertEqual(transform3.find_conflicts(),
 
256
        self.assertEqual(transform3.find_conflicts(), 
312
257
                         [('missing parent', oz_id)])
313
258
        root_id = transform3.root
314
259
        tip_id = transform3.trans_id_tree_file_id('tip-id')
423
368
        self.assertContainsRe(transform._limbo_name(first), 'new-1/file')
424
369
        self.assertNotContainsRe(transform._limbo_name(second), 'new-1/FiLe')
425
370
 
426
 
    def test_adjust_path_updates_child_limbo_names(self):
427
 
        tree = self.make_branch_and_tree('tree')
428
 
        transform = TreeTransform(tree)
429
 
        self.addCleanup(transform.finalize)
430
 
        foo_id = transform.new_directory('foo', transform.root)
431
 
        bar_id = transform.new_directory('bar', foo_id)
432
 
        baz_id = transform.new_directory('baz', bar_id)
433
 
        qux_id = transform.new_directory('qux', baz_id)
434
 
        transform.adjust_path('quxx', foo_id, bar_id)
435
 
        self.assertStartsWith(transform._limbo_name(qux_id),
436
 
                              transform._limbo_name(bar_id))
437
 
 
438
371
    def test_add_del(self):
439
372
        start, root = self.get_transform()
440
373
        start.new_directory('a', root, 'a')
453
386
        self.addCleanup(unversion.finalize)
454
387
        parent = unversion.trans_id_tree_path('parent')
455
388
        unversion.unversion_file(parent)
456
 
        self.assertEqual(unversion.find_conflicts(),
 
389
        self.assertEqual(unversion.find_conflicts(), 
457
390
                         [('unversioned parent', parent_id)])
458
391
        file_id = unversion.trans_id_tree_file_id('child-id')
459
392
        unversion.unversion_file(file_id)
479
412
        mangle_tree.adjust_path('name2', root, name1)
480
413
        mangle_tree.adjust_path('name1', root, name2)
481
414
 
482
 
        #tests for deleting parent directories
 
415
        #tests for deleting parent directories 
483
416
        ddir = mangle_tree.trans_id_tree_file_id('ddir')
484
417
        mangle_tree.delete_contents(ddir)
485
418
        dfile = mangle_tree.trans_id_tree_file_id('dfile')
514
447
        create_tree,root = self.get_transform()
515
448
        newdir = create_tree.new_directory('selftest', root, 'selftest-id')
516
449
        create_tree.new_file('blackbox.py', newdir, 'hello1', 'blackbox-id')
517
 
        create_tree.apply()
 
450
        create_tree.apply()        
518
451
        mangle_tree,root = self.get_transform()
519
452
        selftest = mangle_tree.trans_id_tree_file_id('selftest-id')
520
453
        blackbox = mangle_tree.trans_id_tree_file_id('blackbox-id')
528
461
        bzrlib = create_tree.new_directory('bzrlib', root, 'bzrlib-id')
529
462
        tests = create_tree.new_directory('tests', bzrlib, 'tests-id')
530
463
        blackbox = create_tree.new_directory('blackbox', tests, 'blackbox-id')
531
 
        create_tree.new_file('test_too_much.py', blackbox, 'hello1',
 
464
        create_tree.new_file('test_too_much.py', blackbox, 'hello1', 
532
465
                             'test_too_much-id')
533
 
        create_tree.apply()
 
466
        create_tree.apply()        
534
467
        mangle_tree,root = self.get_transform()
535
468
        bzrlib = mangle_tree.trans_id_tree_file_id('bzrlib-id')
536
469
        tests = mangle_tree.trans_id_tree_file_id('tests-id')
537
470
        test_too_much = mangle_tree.trans_id_tree_file_id('test_too_much-id')
538
471
        mangle_tree.adjust_path('selftest', bzrlib, tests)
539
 
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much)
 
472
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much) 
540
473
        mangle_tree.set_executability(True, test_too_much)
541
474
        mangle_tree.apply()
542
475
 
543
476
    def test_both_rename3(self):
544
477
        create_tree,root = self.get_transform()
545
478
        tests = create_tree.new_directory('tests', root, 'tests-id')
546
 
        create_tree.new_file('test_too_much.py', tests, 'hello1',
 
479
        create_tree.new_file('test_too_much.py', tests, 'hello1', 
547
480
                             'test_too_much-id')
548
 
        create_tree.apply()
 
481
        create_tree.apply()        
549
482
        mangle_tree,root = self.get_transform()
550
483
        tests = mangle_tree.trans_id_tree_file_id('tests-id')
551
484
        test_too_much = mangle_tree.trans_id_tree_file_id('test_too_much-id')
552
485
        mangle_tree.adjust_path('selftest', root, tests)
553
 
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much)
 
486
        mangle_tree.adjust_path('blackbox.py', tests, test_too_much) 
554
487
        mangle_tree.set_executability(True, test_too_much)
555
488
        mangle_tree.apply()
556
489
 
569
502
        newdir = move_id.new_directory('dir', root, 'newdir')
570
503
        move_id.adjust_path('name2', newdir, name1)
571
504
        move_id.apply()
572
 
 
 
505
        
573
506
    def test_replace_dangling_ie(self):
574
507
        create_tree, root = self.get_transform()
575
508
        # prepare tree
591
524
        resolve_conflicts(replace)
592
525
        replace.apply()
593
526
 
594
 
    def _test_symlinks(self, link_name1,link_target1,
595
 
                       link_name2, link_target2):
596
 
 
597
 
        def ozpath(p): return 'oz/' + p
598
 
 
 
527
    def test_symlinks(self):
599
528
        self.requireFeature(SymlinkFeature)
600
 
        transform, root = self.get_transform()
 
529
        transform,root = self.get_transform()
601
530
        oz_id = transform.new_directory('oz', root, 'oz-id')
602
 
        wizard = transform.new_symlink(link_name1, oz_id, link_target1,
 
531
        wizard = transform.new_symlink('wizard', oz_id, 'wizard-target', 
603
532
                                       'wizard-id')
604
 
        wiz_id = transform.create_path(link_name2, oz_id)
605
 
        transform.create_symlink(link_target2, wiz_id)
606
 
        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)            
607
536
        transform.set_executability(True, wiz_id)
608
 
        self.assertEqual(transform.find_conflicts(),
 
537
        self.assertEqual(transform.find_conflicts(), 
609
538
                         [('non-file executability', wiz_id)])
610
539
        transform.set_executability(None, wiz_id)
611
540
        transform.apply()
612
 
        self.assertEqual(self.wt.path2id(ozpath(link_name1)), 'wizard-id')
613
 
        self.assertEqual('symlink',
614
 
                         file_kind(self.wt.abspath(ozpath(link_name1))))
615
 
        self.assertEqual(link_target2,
616
 
                         osutils.readlink(self.wt.abspath(ozpath(link_name2))))
617
 
        self.assertEqual(link_target1,
618
 
                         osutils.readlink(self.wt.abspath(ozpath(link_name1))))
619
 
 
620
 
    def test_symlinks(self):
621
 
        self._test_symlinks('wizard', 'wizard-target',
622
 
                            'wizard2', 'behind_curtain')
623
 
 
624
 
    def test_symlinks_unicode(self):
625
 
        self.requireFeature(tests.UnicodeFilenameFeature)
626
 
        self._test_symlinks(u'\N{Euro Sign}wizard',
627
 
                            u'wizard-targ\N{Euro Sign}t',
628
 
                            u'\N{Euro Sign}wizard2',
629
 
                            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')
630
547
 
631
548
    def test_unable_create_symlink(self):
632
549
        def tt_helper():
656
573
        create.apply()
657
574
        conflicts,root = self.get_transform()
658
575
        # set up duplicate entry, duplicate id
659
 
        new_dorothy = conflicts.new_file('dorothy', root, 'dorothy',
 
576
        new_dorothy = conflicts.new_file('dorothy', root, 'dorothy', 
660
577
                                         'dorothy-id')
661
578
        old_dorothy = conflicts.trans_id_tree_file_id('dorothy-id')
662
579
        oz = conflicts.trans_id_tree_file_id('oz-id')
686
603
        tt, emerald, oz, old_dorothy, new_dorothy = self.get_conflicted()
687
604
        raw_conflicts = resolve_conflicts(tt)
688
605
        cooked_conflicts = cook_conflicts(raw_conflicts, tt)
689
 
        duplicate = DuplicateEntry('Moved existing file to', 'dorothy.moved',
 
606
        duplicate = DuplicateEntry('Moved existing file to', 'dorothy.moved', 
690
607
                                   'dorothy', None, 'dorothy-id')
691
608
        self.assertEqual(cooked_conflicts[0], duplicate)
692
 
        duplicate_id = DuplicateID('Unversioned existing file',
 
609
        duplicate_id = DuplicateID('Unversioned existing file', 
693
610
                                   'dorothy.moved', 'dorothy', None,
694
611
                                   'dorothy-id')
695
612
        self.assertEqual(cooked_conflicts[1], duplicate_id)
703
620
        unversioned_parent2 = UnversionedParent('Versioned directory', 'oz',
704
621
                                               'oz-id')
705
622
        self.assertEqual(cooked_conflicts[3], unversioned_parent)
706
 
        parent_loop = ParentLoop('Cancelled move', 'oz/emeraldcity',
 
623
        parent_loop = ParentLoop('Cancelled move', 'oz/emeraldcity', 
707
624
                                 'oz/emeraldcity', 'emerald-id', 'emerald-id')
708
625
        self.assertEqual(cooked_conflicts[4], deleted_parent)
709
626
        self.assertEqual(cooked_conflicts[5], unversioned_parent2)
812
729
        create.apply()
813
730
        transform, root = self.get_transform()
814
731
        transform.adjust_root_path('oldroot', fun)
815
 
        new_root = transform.trans_id_tree_path('')
 
732
        new_root=transform.trans_id_tree_path('')
816
733
        transform.version_file('new-root', new_root)
817
734
        transform.apply()
818
735
 
832
749
 
833
750
    def test_set_executability_order(self):
834
751
        """Ensure that executability behaves the same, no matter what order.
835
 
 
 
752
        
836
753
        - create file and set executability simultaneously
837
754
        - create file and set executability afterward
838
755
        - unsetting the executability of a file whose executability has not been
1492
1409
        # textual merge
1493
1410
        self.assertEqual(this.wt.get_file('a').read(), 'y\nb\nc\nd\bz\n')
1494
1411
        # three-way text conflict
1495
 
        self.assertEqual(this.wt.get_file('b').read(),
 
1412
        self.assertEqual(this.wt.get_file('b').read(), 
1496
1413
                         conflict_text('b', 'b2'))
1497
1414
        # OTHER wins
1498
1415
        self.assertEqual(this.wt.get_file('c').read(), 'c2')
1502
1419
        self.assertEqual(this.wt.get_file('e').read(), 'e2')
1503
1420
        # No change
1504
1421
        self.assertEqual(this.wt.get_file('f').read(), 'f')
1505
 
        # Correct correct results when THIS == OTHER
 
1422
        # Correct correct results when THIS == OTHER 
1506
1423
        self.assertEqual(this.wt.get_file('g').read(), 'g')
1507
1424
        # Text conflict when THIS & OTHER are text and BASE is dir
1508
 
        self.assertEqual(this.wt.get_file('h').read(),
 
1425
        self.assertEqual(this.wt.get_file('h').read(), 
1509
1426
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
1510
1427
        self.assertEqual(this.wt.get_file_byname('h.THIS').read(),
1511
1428
                         '1\n2\n3\n4\n')
1512
1429
        self.assertEqual(this.wt.get_file_byname('h.OTHER').read(),
1513
1430
                         'h\ni\nj\nk\n')
1514
1431
        self.assertEqual(file_kind(this.wt.abspath('h.BASE')), 'directory')
1515
 
        self.assertEqual(this.wt.get_file('i').read(),
 
1432
        self.assertEqual(this.wt.get_file('i').read(), 
1516
1433
                         conflict_text('1\n2\n3\n4\n', 'h\ni\nj\nk\n'))
1517
1434
        self.assertEqual(this.wt.get_file_byname('i.THIS').read(),
1518
1435
                         '1\n2\n3\n4\n')
1542
1459
            tg.tt.new_symlink('b', tg.root, 'b', 'b')
1543
1460
            tg.tt.new_file('c', tg.root, 'c', 'c')
1544
1461
            tg.tt.new_symlink('d', tg.root, tg.name, 'd')
1545
 
        targets = ((base, 'base-e', 'base-f', None, None),
1546
 
                   (this, 'other-e', 'this-f', 'other-g', 'this-h'),
 
1462
        targets = ((base, 'base-e', 'base-f', None, None), 
 
1463
                   (this, 'other-e', 'this-f', 'other-g', 'this-h'), 
1547
1464
                   (other, 'other-e', None, 'other-g', 'other-h'))
1548
1465
        for tg, e_target, f_target, g_target, h_target in targets:
1549
 
            for link, target in (('e', e_target), ('f', f_target),
 
1466
            for link, target in (('e', e_target), ('f', f_target), 
1550
1467
                                 ('g', g_target), ('h', h_target)):
1551
1468
                if target is not None:
1552
1469
                    tg.tt.new_symlink(link, tg.root, target, link)
1578
1495
        base = TransformGroup("BASE", root_id)
1579
1496
        this = TransformGroup("THIS", root_id)
1580
1497
        other = TransformGroup("OTHER", root_id)
1581
 
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
 
1498
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a') 
1582
1499
                                   for t in [base, this, other]]
1583
 
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
 
1500
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b') 
1584
1501
                                   for t in [base, this, other]]
1585
1502
        base.tt.new_directory('c', base_a, 'c')
1586
1503
        this.tt.new_directory('c1', this_a, 'c')
1611
1528
        base = TransformGroup("BASE", root_id)
1612
1529
        this = TransformGroup("THIS", root_id)
1613
1530
        other = TransformGroup("OTHER", root_id)
1614
 
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a')
 
1531
        base_a, this_a, other_a = [t.tt.new_directory('a', t.root, 'a') 
1615
1532
                                   for t in [base, this, other]]
1616
 
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b')
 
1533
        base_b, this_b, other_b = [t.tt.new_directory('b', t.root, 'b') 
1617
1534
                                   for t in [base, this, other]]
1618
1535
 
1619
1536
        base.tt.new_file('g', base_a, 'g', 'g')
1710
1627
        os.symlink('foo', 'target2/symlink')
1711
1628
        build_tree(source.basis_tree(), target)
1712
1629
        self.assertEqual([], target.conflicts())
1713
 
 
 
1630
        
1714
1631
    def test_directory_conflict_handling(self):
1715
1632
        """Ensure that when building trees, conflict handling is done"""
1716
1633
        source = self.make_branch_and_tree('source')
1772
1689
        target = self.make_branch_and_tree('target')
1773
1690
        self.build_tree(['target/name'])
1774
1691
        target.add('name')
1775
 
        self.assertRaises(errors.WorkingTreeAlreadyPopulated,
 
1692
        self.assertRaises(errors.WorkingTreeAlreadyPopulated, 
1776
1693
            build_tree, source.basis_tree(), target)
1777
1694
 
1778
1695
    def test_build_tree_rename_count(self):
1922
1839
        self.assertEqual([], list(target.iter_changes(revision_tree)))
1923
1840
        self.assertTrue(source.is_executable('file1-id'))
1924
1841
 
1925
 
    def install_rot13_content_filter(self, pattern):
1926
 
        # We could use
1927
 
        # self.addCleanup(filters._reset_registry, filters._reset_registry())
1928
 
        # below, but that looks a bit... hard to read even if it's exactly
1929
 
        # the same thing.
1930
 
        original_registry = filters._reset_registry()
1931
 
        def restore_registry():
1932
 
            filters._reset_registry(original_registry)
1933
 
        self.addCleanup(restore_registry)
1934
 
        def rot13(chunks, context=None):
1935
 
            return [''.join(chunks).encode('rot13')]
1936
 
        rot13filter = filters.ContentFilter(rot13, rot13)
1937
 
        filters.register_filter_stack_map('rot13', {'yes': [rot13filter]}.get)
1938
 
        os.mkdir(self.test_home_dir + '/.bazaar')
1939
 
        rules_filename = self.test_home_dir + '/.bazaar/rules'
1940
 
        f = open(rules_filename, 'wb')
1941
 
        f.write('[name %s]\nrot13=yes\n' % (pattern,))
1942
 
        f.close()
1943
 
        def uninstall_rules():
1944
 
            os.remove(rules_filename)
1945
 
            rules.reset_rules()
1946
 
        self.addCleanup(uninstall_rules)
1947
 
        rules.reset_rules()
1948
 
 
1949
 
    def test_build_tree_content_filtered_files_are_not_hardlinked(self):
1950
 
        """build_tree will not hardlink files that have content filtering rules
1951
 
        applied to them (but will still hardlink other files from the same tree
1952
 
        if it can).
1953
 
        """
1954
 
        self.requireFeature(HardlinkFeature)
1955
 
        self.install_rot13_content_filter('file1')
1956
 
        source = self.create_ab_tree()
1957
 
        target = self.make_branch_and_tree('target')
1958
 
        revision_tree = source.basis_tree()
1959
 
        revision_tree.lock_read()
1960
 
        self.addCleanup(revision_tree.unlock)
1961
 
        build_tree(revision_tree, target, source, hardlink=True)
1962
 
        target.lock_read()
1963
 
        self.addCleanup(target.unlock)
1964
 
        self.assertEqual([], list(target.iter_changes(revision_tree)))
1965
 
        source_stat = os.stat('source/file1')
1966
 
        target_stat = os.stat('target/file1')
1967
 
        self.assertNotEqual(source_stat, target_stat)
1968
 
        source_stat = os.stat('source/file2')
1969
 
        target_stat = os.stat('target/file2')
1970
 
        self.assertEqualStat(source_stat, target_stat)
1971
 
 
1972
1842
    def test_case_insensitive_build_tree_inventory(self):
1973
 
        if (tests.CaseInsensitiveFilesystemFeature.available()
1974
 
            or tests.CaseInsCasePresFilenameFeature.available()):
1975
 
            raise tests.UnavailableFeature('Fully case sensitive filesystem')
1976
1843
        source = self.make_branch_and_tree('source')
1977
1844
        self.build_tree(['source/file', 'source/FILE'])
1978
1845
        source.add(['file', 'FILE'], ['lower-id', 'upper-id'])
1986
1853
        self.assertEqual('FILE', target.id2path('upper-id'))
1987
1854
 
1988
1855
 
1989
 
class TestCommitTransform(tests.TestCaseWithTransport):
1990
 
 
1991
 
    def get_branch(self):
1992
 
        tree = self.make_branch_and_tree('tree')
1993
 
        tree.lock_write()
1994
 
        self.addCleanup(tree.unlock)
1995
 
        tree.commit('empty commit')
1996
 
        return tree.branch
1997
 
 
1998
 
    def get_branch_and_transform(self):
1999
 
        branch = self.get_branch()
2000
 
        tt = TransformPreview(branch.basis_tree())
2001
 
        self.addCleanup(tt.finalize)
2002
 
        return branch, tt
2003
 
 
2004
 
    def test_commit_wrong_basis(self):
2005
 
        branch = self.get_branch()
2006
 
        basis = branch.repository.revision_tree(
2007
 
            _mod_revision.NULL_REVISION)
2008
 
        tt = TransformPreview(basis)
2009
 
        self.addCleanup(tt.finalize)
2010
 
        e = self.assertRaises(ValueError, tt.commit, branch, '')
2011
 
        self.assertEqual('TreeTransform not based on branch basis: null:',
2012
 
                         str(e))
2013
 
 
2014
 
    def test_empy_commit(self):
2015
 
        branch, tt = self.get_branch_and_transform()
2016
 
        rev = tt.commit(branch, 'my message')
2017
 
        self.assertEqual(2, branch.revno())
2018
 
        repo = branch.repository
2019
 
        self.assertEqual('my message', repo.get_revision(rev).message)
2020
 
 
2021
 
    def test_merge_parents(self):
2022
 
        branch, tt = self.get_branch_and_transform()
2023
 
        rev = tt.commit(branch, 'my message', ['rev1b', 'rev1c'])
2024
 
        self.assertEqual(['rev1b', 'rev1c'],
2025
 
                         branch.basis_tree().get_parent_ids()[1:])
2026
 
 
2027
 
    def test_first_commit(self):
2028
 
        branch = self.make_branch('branch')
2029
 
        branch.lock_write()
2030
 
        self.addCleanup(branch.unlock)
2031
 
        tt = TransformPreview(branch.basis_tree())
2032
 
        self.addCleanup(tt.finalize)
2033
 
        tt.new_directory('', ROOT_PARENT, 'TREE_ROOT')
2034
 
        rev = tt.commit(branch, 'my message')
2035
 
        self.assertEqual([], branch.basis_tree().get_parent_ids())
2036
 
        self.assertNotEqual(_mod_revision.NULL_REVISION,
2037
 
                            branch.last_revision())
2038
 
 
2039
 
    def test_first_commit_with_merge_parents(self):
2040
 
        branch = self.make_branch('branch')
2041
 
        branch.lock_write()
2042
 
        self.addCleanup(branch.unlock)
2043
 
        tt = TransformPreview(branch.basis_tree())
2044
 
        self.addCleanup(tt.finalize)
2045
 
        e = self.assertRaises(ValueError, tt.commit, branch,
2046
 
                          'my message', ['rev1b-id'])
2047
 
        self.assertEqual('Cannot supply merge parents for first commit.',
2048
 
                         str(e))
2049
 
        self.assertEqual(_mod_revision.NULL_REVISION, branch.last_revision())
2050
 
 
2051
 
    def test_add_files(self):
2052
 
        branch, tt = self.get_branch_and_transform()
2053
 
        tt.new_file('file', tt.root, 'contents', 'file-id')
2054
 
        trans_id = tt.new_directory('dir', tt.root, 'dir-id')
2055
 
        if SymlinkFeature.available():
2056
 
            tt.new_symlink('symlink', trans_id, 'target', 'symlink-id')
2057
 
        rev = tt.commit(branch, 'message')
2058
 
        tree = branch.basis_tree()
2059
 
        self.assertEqual('file', tree.id2path('file-id'))
2060
 
        self.assertEqual('contents', tree.get_file_text('file-id'))
2061
 
        self.assertEqual('dir', tree.id2path('dir-id'))
2062
 
        if SymlinkFeature.available():
2063
 
            self.assertEqual('dir/symlink', tree.id2path('symlink-id'))
2064
 
            self.assertEqual('target', tree.get_symlink_target('symlink-id'))
2065
 
 
2066
 
    def test_add_unversioned(self):
2067
 
        branch, tt = self.get_branch_and_transform()
2068
 
        tt.new_file('file', tt.root, 'contents')
2069
 
        self.assertRaises(errors.StrictCommitFailed, tt.commit, branch,
2070
 
                          'message', strict=True)
2071
 
 
2072
 
    def test_modify_strict(self):
2073
 
        branch, tt = self.get_branch_and_transform()
2074
 
        tt.new_file('file', tt.root, 'contents', 'file-id')
2075
 
        tt.commit(branch, 'message', strict=True)
2076
 
        tt = TransformPreview(branch.basis_tree())
2077
 
        self.addCleanup(tt.finalize)
2078
 
        trans_id = tt.trans_id_file_id('file-id')
2079
 
        tt.delete_contents(trans_id)
2080
 
        tt.create_file('contents', trans_id)
2081
 
        tt.commit(branch, 'message', strict=True)
2082
 
 
2083
 
    def test_commit_malformed(self):
2084
 
        """Committing a malformed transform should raise an exception.
2085
 
 
2086
 
        In this case, we are adding a file without adding its parent.
2087
 
        """
2088
 
        branch, tt = self.get_branch_and_transform()
2089
 
        parent_id = tt.trans_id_file_id('parent-id')
2090
 
        tt.new_file('file', parent_id, 'contents', 'file-id')
2091
 
        self.assertRaises(errors.MalformedTransform, tt.commit, branch,
2092
 
                          'message')
2093
 
 
2094
 
 
2095
1856
class MockTransform(object):
2096
1857
 
2097
1858
    def has_named_child(self, by_parent, parent_id, name):
2264
2025
    def create_tree(self):
2265
2026
        tree = self.make_branch_and_tree('.')
2266
2027
        self.build_tree_contents([('a', 'content 1')])
2267
 
        tree.set_root_id('TREE_ROOT')
2268
2028
        tree.add('a', 'a-id')
2269
2029
        tree.commit('rev1', rev_id='rev1')
2270
2030
        return tree.branch.repository.revision_tree('rev1')
2371
2131
    def test_ignore_pb(self):
2372
2132
        # pb could be supported, but TT.iter_changes doesn't support it.
2373
2133
        revision_tree, preview_tree = self.get_tree_and_preview_tree()
2374
 
        preview_tree.iter_changes(revision_tree)
 
2134
        preview_tree.iter_changes(revision_tree, pb=progress.DummyProgress())
2375
2135
 
2376
2136
    def test_kind(self):
2377
2137
        revision_tree = self.create_tree()
2392
2152
        self.assertEqual(os.stat(limbo_path).st_mtime,
2393
2153
                         preview_tree.get_file_mtime('file-id'))
2394
2154
 
2395
 
    def test_get_file_mtime_renamed(self):
2396
 
        work_tree = self.make_branch_and_tree('tree')
2397
 
        self.build_tree(['tree/file'])
2398
 
        work_tree.add('file', 'file-id')
2399
 
        preview = TransformPreview(work_tree)
2400
 
        self.addCleanup(preview.finalize)
2401
 
        file_trans_id = preview.trans_id_tree_file_id('file-id')
2402
 
        preview.adjust_path('renamed', preview.root, file_trans_id)
2403
 
        preview_tree = preview.get_preview_tree()
2404
 
        preview_mtime = preview_tree.get_file_mtime('file-id', 'renamed')
2405
 
        work_mtime = work_tree.get_file_mtime('file-id', 'file')
2406
 
 
2407
2155
    def test_get_file(self):
2408
2156
        preview = self.get_empty_preview()
2409
2157
        preview.new_file('file', preview.root, 'contents', 'file-id')
2557
2305
        self.assertEqual(('missing', None, None, None), summary)
2558
2306
 
2559
2307
    def test_file_content_summary_executable(self):
 
2308
        if not osutils.supports_executable():
 
2309
            raise TestNotApplicable()
2560
2310
        preview = self.get_empty_preview()
2561
2311
        path_id = preview.new_file('path', preview.root, 'contents', 'path-id')
2562
2312
        preview.set_executability(True, path_id)
2571
2321
        self.assertIs(None, summary[3])
2572
2322
 
2573
2323
    def test_change_executability(self):
 
2324
        if not osutils.supports_executable():
 
2325
            raise TestNotApplicable()
2574
2326
        tree = self.make_branch_and_tree('tree')
2575
2327
        self.build_tree(['tree/path'])
2576
2328
        tree.add('path')
2590
2342
        # size must be known
2591
2343
        self.assertEqual(len('contents'), summary[1])
2592
2344
        # not executable
2593
 
        self.assertEqual(False, summary[2])
 
2345
        if osutils.supports_executable():
 
2346
            self.assertEqual(False, summary[2])
 
2347
        else:
 
2348
            self.assertEqual(None, summary[2])
2594
2349
        # will not have hash (not cheap to determine)
2595
2350
        self.assertIs(None, summary[3])
2596
2351
 
2737
2492
 
2738
2493
    def test_walkdirs(self):
2739
2494
        preview = self.get_empty_preview()
2740
 
        root = preview.new_directory('', ROOT_PARENT, 'tree-root')
2741
 
        # FIXME: new_directory should mark root.
2742
 
        preview.fixup_new_roots()
 
2495
        preview.version_file('tree-root', preview.root)
2743
2496
        preview_tree = preview.get_preview_tree()
2744
2497
        file_trans_id = preview.new_file('a', preview.root, 'contents',
2745
2498
                                         'a-id')
2776
2529
        self.addCleanup(work_tree.unlock)
2777
2530
        preview = TransformPreview(work_tree)
2778
2531
        self.addCleanup(preview.finalize)
 
2532
        preview_tree = preview.get_preview_tree()
2779
2533
        file_trans_id = preview.trans_id_file_id('file-id')
2780
2534
        preview.delete_contents(file_trans_id)
2781
2535
        preview.create_file('a\nb\n', file_trans_id)
2782
 
        preview_tree = preview.get_preview_tree()
2783
 
        merger = Merger.from_revision_ids(None, preview_tree,
 
2536
        pb = progress.DummyProgress()
 
2537
        merger = Merger.from_revision_ids(pb, preview_tree,
2784
2538
                                          child_tree.branch.last_revision(),
2785
2539
                                          other_branch=child_tree.branch,
2786
2540
                                          tree_branch=work_tree.branch)
2792
2546
 
2793
2547
    def test_merge_preview_into_workingtree(self):
2794
2548
        tree = self.make_branch_and_tree('tree')
2795
 
        tree.set_root_id('TREE_ROOT')
2796
2549
        tt = TransformPreview(tree)
2797
2550
        self.addCleanup(tt.finalize)
2798
2551
        tt.new_file('name', tt.root, 'content', 'file-id')
2799
2552
        tree2 = self.make_branch_and_tree('tree2')
2800
 
        tree2.set_root_id('TREE_ROOT')
 
2553
        pb = progress.DummyProgress()
2801
2554
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
2802
 
                                         None, tree.basis_tree())
 
2555
                                         pb, tree.basis_tree())
2803
2556
        merger.merge_type = Merge3Merger
2804
2557
        merger.do_merge()
2805
2558
 
2815
2568
        tt.create_file('baz', trans_id)
2816
2569
        tree2 = tree.bzrdir.sprout('tree2').open_workingtree()
2817
2570
        self.build_tree_contents([('tree2/foo', 'qux')])
2818
 
        pb = None
 
2571
        pb = progress.DummyProgress()
2819
2572
        merger = Merger.from_uncommitted(tree2, tt.get_preview_tree(),
2820
2573
                                         pb, tree.basis_tree())
2821
2574
        merger.merge_type = Merge3Merger
2830
2583
        self.assertEqual(False, preview_tree.is_executable('baz-id',
2831
2584
                                                           'tree/foo'))
2832
2585
        self.assertEqual(False, preview_tree.is_executable('baz-id'))
2833
 
 
2834
 
    def test_commit_preview_tree(self):
2835
 
        tree = self.make_branch_and_tree('tree')
2836
 
        rev_id = tree.commit('rev1')
2837
 
        tree.branch.lock_write()
2838
 
        self.addCleanup(tree.branch.unlock)
2839
 
        tt = TransformPreview(tree)
2840
 
        tt.new_file('file', tt.root, 'contents', 'file_id')
2841
 
        self.addCleanup(tt.finalize)
2842
 
        preview = tt.get_preview_tree()
2843
 
        preview.set_parent_ids([rev_id])
2844
 
        builder = tree.branch.get_commit_builder([rev_id])
2845
 
        list(builder.record_iter_changes(preview, rev_id, tt.iter_changes()))
2846
 
        builder.finish_inventory()
2847
 
        rev2_id = builder.commit('rev2')
2848
 
        rev2_tree = tree.branch.repository.revision_tree(rev2_id)
2849
 
        self.assertEqual('contents', rev2_tree.get_file_text('file_id'))
2850
 
 
2851
 
    def test_ascii_limbo_paths(self):
2852
 
        self.requireFeature(tests.UnicodeFilenameFeature)
2853
 
        branch = self.make_branch('any')
2854
 
        tree = branch.repository.revision_tree(_mod_revision.NULL_REVISION)
2855
 
        tt = TransformPreview(tree)
2856
 
        self.addCleanup(tt.finalize)
2857
 
        foo_id = tt.new_directory('', ROOT_PARENT)
2858
 
        bar_id = tt.new_file(u'\u1234bar', foo_id, 'contents')
2859
 
        limbo_path = tt._limbo_name(bar_id)
2860
 
        self.assertEqual(limbo_path.encode('ascii', 'replace'), limbo_path)
2861
 
 
2862
 
 
2863
 
class FakeSerializer(object):
2864
 
    """Serializer implementation that simply returns the input.
2865
 
 
2866
 
    The input is returned in the order used by pack.ContainerPushParser.
2867
 
    """
2868
 
    @staticmethod
2869
 
    def bytes_record(bytes, names):
2870
 
        return names, bytes
2871
 
 
2872
 
 
2873
 
class TestSerializeTransform(tests.TestCaseWithTransport):
2874
 
 
2875
 
    _test_needs_features = [tests.UnicodeFilenameFeature]
2876
 
 
2877
 
    def get_preview(self, tree=None):
2878
 
        if tree is None:
2879
 
            tree = self.make_branch_and_tree('tree')
2880
 
        tt = TransformPreview(tree)
2881
 
        self.addCleanup(tt.finalize)
2882
 
        return tt
2883
 
 
2884
 
    def assertSerializesTo(self, expected, tt):
2885
 
        records = list(tt.serialize(FakeSerializer()))
2886
 
        self.assertEqual(expected, records)
2887
 
 
2888
 
    @staticmethod
2889
 
    def default_attribs():
2890
 
        return {
2891
 
            '_id_number': 1,
2892
 
            '_new_name': {},
2893
 
            '_new_parent': {},
2894
 
            '_new_executability': {},
2895
 
            '_new_id': {},
2896
 
            '_tree_path_ids': {'': 'new-0'},
2897
 
            '_removed_id': [],
2898
 
            '_removed_contents': [],
2899
 
            '_non_present_ids': {},
2900
 
            }
2901
 
 
2902
 
    def make_records(self, attribs, contents):
2903
 
        records = [
2904
 
            (((('attribs'),),), bencode.bencode(attribs))]
2905
 
        records.extend([(((n, k),), c) for n, k, c in contents])
2906
 
        return records
2907
 
 
2908
 
    def creation_records(self):
2909
 
        attribs = self.default_attribs()
2910
 
        attribs['_id_number'] = 3
2911
 
        attribs['_new_name'] = {
2912
 
            'new-1': u'foo\u1234'.encode('utf-8'), 'new-2': 'qux'}
2913
 
        attribs['_new_id'] = {'new-1': 'baz', 'new-2': 'quxx'}
2914
 
        attribs['_new_parent'] = {'new-1': 'new-0', 'new-2': 'new-0'}
2915
 
        attribs['_new_executability'] = {'new-1': 1}
2916
 
        contents = [
2917
 
            ('new-1', 'file', 'i 1\nbar\n'),
2918
 
            ('new-2', 'directory', ''),
2919
 
            ]
2920
 
        return self.make_records(attribs, contents)
2921
 
 
2922
 
    def test_serialize_creation(self):
2923
 
        tt = self.get_preview()
2924
 
        tt.new_file(u'foo\u1234', tt.root, 'bar', 'baz', True)
2925
 
        tt.new_directory('qux', tt.root, 'quxx')
2926
 
        self.assertSerializesTo(self.creation_records(), tt)
2927
 
 
2928
 
    def test_deserialize_creation(self):
2929
 
        tt = self.get_preview()
2930
 
        tt.deserialize(iter(self.creation_records()))
2931
 
        self.assertEqual(3, tt._id_number)
2932
 
        self.assertEqual({'new-1': u'foo\u1234',
2933
 
                          'new-2': 'qux'}, tt._new_name)
2934
 
        self.assertEqual({'new-1': 'baz', 'new-2': 'quxx'}, tt._new_id)
2935
 
        self.assertEqual({'new-1': tt.root, 'new-2': tt.root}, tt._new_parent)
2936
 
        self.assertEqual({'baz': 'new-1', 'quxx': 'new-2'}, tt._r_new_id)
2937
 
        self.assertEqual({'new-1': True}, tt._new_executability)
2938
 
        self.assertEqual({'new-1': 'file',
2939
 
                          'new-2': 'directory'}, tt._new_contents)
2940
 
        foo_limbo = open(tt._limbo_name('new-1'), 'rb')
2941
 
        try:
2942
 
            foo_content = foo_limbo.read()
2943
 
        finally:
2944
 
            foo_limbo.close()
2945
 
        self.assertEqual('bar', foo_content)
2946
 
 
2947
 
    def symlink_creation_records(self):
2948
 
        attribs = self.default_attribs()
2949
 
        attribs['_id_number'] = 2
2950
 
        attribs['_new_name'] = {'new-1': u'foo\u1234'.encode('utf-8')}
2951
 
        attribs['_new_parent'] = {'new-1': 'new-0'}
2952
 
        contents = [('new-1', 'symlink', u'bar\u1234'.encode('utf-8'))]
2953
 
        return self.make_records(attribs, contents)
2954
 
 
2955
 
    def test_serialize_symlink_creation(self):
2956
 
        self.requireFeature(tests.SymlinkFeature)
2957
 
        tt = self.get_preview()
2958
 
        tt.new_symlink(u'foo\u1234', tt.root, u'bar\u1234')
2959
 
        self.assertSerializesTo(self.symlink_creation_records(), tt)
2960
 
 
2961
 
    def test_deserialize_symlink_creation(self):
2962
 
        self.requireFeature(tests.SymlinkFeature)
2963
 
        tt = self.get_preview()
2964
 
        tt.deserialize(iter(self.symlink_creation_records()))
2965
 
        abspath = tt._limbo_name('new-1')
2966
 
        foo_content = osutils.readlink(abspath)
2967
 
        self.assertEqual(u'bar\u1234', foo_content)
2968
 
 
2969
 
    def make_destruction_preview(self):
2970
 
        tree = self.make_branch_and_tree('.')
2971
 
        self.build_tree([u'foo\u1234', 'bar'])
2972
 
        tree.add([u'foo\u1234', 'bar'], ['foo-id', 'bar-id'])
2973
 
        return self.get_preview(tree)
2974
 
 
2975
 
    def destruction_records(self):
2976
 
        attribs = self.default_attribs()
2977
 
        attribs['_id_number'] = 3
2978
 
        attribs['_removed_id'] = ['new-1']
2979
 
        attribs['_removed_contents'] = ['new-2']
2980
 
        attribs['_tree_path_ids'] = {
2981
 
            '': 'new-0',
2982
 
            u'foo\u1234'.encode('utf-8'): 'new-1',
2983
 
            'bar': 'new-2',
2984
 
            }
2985
 
        return self.make_records(attribs, [])
2986
 
 
2987
 
    def test_serialize_destruction(self):
2988
 
        tt = self.make_destruction_preview()
2989
 
        foo_trans_id = tt.trans_id_tree_file_id('foo-id')
2990
 
        tt.unversion_file(foo_trans_id)
2991
 
        bar_trans_id = tt.trans_id_tree_file_id('bar-id')
2992
 
        tt.delete_contents(bar_trans_id)
2993
 
        self.assertSerializesTo(self.destruction_records(), tt)
2994
 
 
2995
 
    def test_deserialize_destruction(self):
2996
 
        tt = self.make_destruction_preview()
2997
 
        tt.deserialize(iter(self.destruction_records()))
2998
 
        self.assertEqual({u'foo\u1234': 'new-1',
2999
 
                          'bar': 'new-2',
3000
 
                          '': tt.root}, tt._tree_path_ids)
3001
 
        self.assertEqual({'new-1': u'foo\u1234',
3002
 
                          'new-2': 'bar',
3003
 
                          tt.root: ''}, tt._tree_id_paths)
3004
 
        self.assertEqual(set(['new-1']), tt._removed_id)
3005
 
        self.assertEqual(set(['new-2']), tt._removed_contents)
3006
 
 
3007
 
    def missing_records(self):
3008
 
        attribs = self.default_attribs()
3009
 
        attribs['_id_number'] = 2
3010
 
        attribs['_non_present_ids'] = {
3011
 
            'boo': 'new-1',}
3012
 
        return self.make_records(attribs, [])
3013
 
 
3014
 
    def test_serialize_missing(self):
3015
 
        tt = self.get_preview()
3016
 
        boo_trans_id = tt.trans_id_file_id('boo')
3017
 
        self.assertSerializesTo(self.missing_records(), tt)
3018
 
 
3019
 
    def test_deserialize_missing(self):
3020
 
        tt = self.get_preview()
3021
 
        tt.deserialize(iter(self.missing_records()))
3022
 
        self.assertEqual({'boo': 'new-1'}, tt._non_present_ids)
3023
 
 
3024
 
    def make_modification_preview(self):
3025
 
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
3026
 
        LINES_TWO = 'z\nbb\nx\ndd\n'
3027
 
        tree = self.make_branch_and_tree('tree')
3028
 
        self.build_tree_contents([('tree/file', LINES_ONE)])
3029
 
        tree.add('file', 'file-id')
3030
 
        return self.get_preview(tree), LINES_TWO
3031
 
 
3032
 
    def modification_records(self):
3033
 
        attribs = self.default_attribs()
3034
 
        attribs['_id_number'] = 2
3035
 
        attribs['_tree_path_ids'] = {
3036
 
            'file': 'new-1',
3037
 
            '': 'new-0',}
3038
 
        attribs['_removed_contents'] = ['new-1']
3039
 
        contents = [('new-1', 'file',
3040
 
                     'i 1\nz\n\nc 0 1 1 1\ni 1\nx\n\nc 0 3 3 1\n')]
3041
 
        return self.make_records(attribs, contents)
3042
 
 
3043
 
    def test_serialize_modification(self):
3044
 
        tt, LINES = self.make_modification_preview()
3045
 
        trans_id = tt.trans_id_file_id('file-id')
3046
 
        tt.delete_contents(trans_id)
3047
 
        tt.create_file(LINES, trans_id)
3048
 
        self.assertSerializesTo(self.modification_records(), tt)
3049
 
 
3050
 
    def test_deserialize_modification(self):
3051
 
        tt, LINES = self.make_modification_preview()
3052
 
        tt.deserialize(iter(self.modification_records()))
3053
 
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3054
 
 
3055
 
    def make_kind_change_preview(self):
3056
 
        LINES = 'a\nb\nc\nd\n'
3057
 
        tree = self.make_branch_and_tree('tree')
3058
 
        self.build_tree(['tree/foo/'])
3059
 
        tree.add('foo', 'foo-id')
3060
 
        return self.get_preview(tree), LINES
3061
 
 
3062
 
    def kind_change_records(self):
3063
 
        attribs = self.default_attribs()
3064
 
        attribs['_id_number'] = 2
3065
 
        attribs['_tree_path_ids'] = {
3066
 
            'foo': 'new-1',
3067
 
            '': 'new-0',}
3068
 
        attribs['_removed_contents'] = ['new-1']
3069
 
        contents = [('new-1', 'file',
3070
 
                     'i 4\na\nb\nc\nd\n\n')]
3071
 
        return self.make_records(attribs, contents)
3072
 
 
3073
 
    def test_serialize_kind_change(self):
3074
 
        tt, LINES = self.make_kind_change_preview()
3075
 
        trans_id = tt.trans_id_file_id('foo-id')
3076
 
        tt.delete_contents(trans_id)
3077
 
        tt.create_file(LINES, trans_id)
3078
 
        self.assertSerializesTo(self.kind_change_records(), tt)
3079
 
 
3080
 
    def test_deserialize_kind_change(self):
3081
 
        tt, LINES = self.make_kind_change_preview()
3082
 
        tt.deserialize(iter(self.kind_change_records()))
3083
 
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3084
 
 
3085
 
    def make_add_contents_preview(self):
3086
 
        LINES = 'a\nb\nc\nd\n'
3087
 
        tree = self.make_branch_and_tree('tree')
3088
 
        self.build_tree(['tree/foo'])
3089
 
        tree.add('foo')
3090
 
        os.unlink('tree/foo')
3091
 
        return self.get_preview(tree), LINES
3092
 
 
3093
 
    def add_contents_records(self):
3094
 
        attribs = self.default_attribs()
3095
 
        attribs['_id_number'] = 2
3096
 
        attribs['_tree_path_ids'] = {
3097
 
            'foo': 'new-1',
3098
 
            '': 'new-0',}
3099
 
        contents = [('new-1', 'file',
3100
 
                     'i 4\na\nb\nc\nd\n\n')]
3101
 
        return self.make_records(attribs, contents)
3102
 
 
3103
 
    def test_serialize_add_contents(self):
3104
 
        tt, LINES = self.make_add_contents_preview()
3105
 
        trans_id = tt.trans_id_tree_path('foo')
3106
 
        tt.create_file(LINES, trans_id)
3107
 
        self.assertSerializesTo(self.add_contents_records(), tt)
3108
 
 
3109
 
    def test_deserialize_add_contents(self):
3110
 
        tt, LINES = self.make_add_contents_preview()
3111
 
        tt.deserialize(iter(self.add_contents_records()))
3112
 
        self.assertFileEqual(LINES, tt._limbo_name('new-1'))
3113
 
 
3114
 
    def test_get_parents_lines(self):
3115
 
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
3116
 
        LINES_TWO = 'z\nbb\nx\ndd\n'
3117
 
        tree = self.make_branch_and_tree('tree')
3118
 
        self.build_tree_contents([('tree/file', LINES_ONE)])
3119
 
        tree.add('file', 'file-id')
3120
 
        tt = self.get_preview(tree)
3121
 
        trans_id = tt.trans_id_tree_path('file')
3122
 
        self.assertEqual((['aa\n', 'bb\n', 'cc\n', 'dd\n'],),
3123
 
            tt._get_parents_lines(trans_id))
3124
 
 
3125
 
    def test_get_parents_texts(self):
3126
 
        LINES_ONE = 'aa\nbb\ncc\ndd\n'
3127
 
        LINES_TWO = 'z\nbb\nx\ndd\n'
3128
 
        tree = self.make_branch_and_tree('tree')
3129
 
        self.build_tree_contents([('tree/file', LINES_ONE)])
3130
 
        tree.add('file', 'file-id')
3131
 
        tt = self.get_preview(tree)
3132
 
        trans_id = tt.trans_id_tree_path('file')
3133
 
        self.assertEqual((LINES_ONE,),
3134
 
            tt._get_parents_texts(trans_id))