22
22
from sha import sha
24
from bzrlib.tests import TestCaseWithTransport
25
from bzrlib.testament import Testament, StrictTestament
24
from bzrlib.tests import SymlinkFeature, TestCaseWithTransport
25
from bzrlib.testament import Testament, StrictTestament, StrictTestament3
26
26
from bzrlib.transform import TreeTransform
27
from bzrlib.osutils import has_symlinks
30
29
class TestamentSetup(TestCaseWithTransport):
33
32
super(TestamentSetup, self).setUp()
34
self.wt = self.make_branch_and_tree('.')
33
self.wt = self.make_branch_and_tree('.', format='dirstate-with-subtree')
34
self.wt.set_root_id('TREE_ROT')
35
35
b = self.b = self.wt.branch
36
36
b.nick = "test branch"
37
37
self.wt.commit(message='initial null commit',
58
58
class TestamentTests(TestamentSetup):
60
def testament_class(self):
63
def expected(self, key):
64
return texts[self.testament_class()][key]
66
def from_revision(self, repository, revision_id):
67
return self.testament_class().from_revision(repository, revision_id)
60
69
def test_null_testament(self):
61
70
"""Testament for a revision with no contents."""
62
t = Testament.from_revision(self.b.repository, 'test@user-1')
71
t = self.from_revision(self.b.repository, 'test@user-1')
63
72
ass = self.assertTrue
64
73
eq = self.assertEqual
65
74
ass(isinstance(t, Testament))
71
80
def test_testment_text_form(self):
72
81
"""Conversion of testament to canonical text form."""
73
t = Testament.from_revision(self.b.repository, 'test@user-1')
74
text_form = t.as_text()
75
self.log('testament text form:\n' + text_form)
76
self.assertEqual(text_form, REV_1_TESTAMENT)
78
def test_strict_testment_text_form(self):
79
"""Conversion of testament to canonical text form."""
80
t = StrictTestament.from_revision(self.b.repository, 'test@user-1')
81
text_form = t.as_text()
82
self.log('testament text form:\n' + text_form)
83
self.assertEqualDiff(text_form, REV_1_STRICT_TESTAMENT)
82
t = self.from_revision(self.b.repository, 'test@user-1')
83
text_form = t.as_text()
84
self.log('testament text form:\n' + text_form)
85
self.assertEqualDiff(text_form, self.expected('rev_1'))
86
short_text_form = t.as_short_text()
87
self.assertEqualDiff(short_text_form, self.expected('rev_1_short'))
85
89
def test_testament_with_contents(self):
86
90
"""Testament containing a file and a directory."""
87
t = Testament.from_revision(self.b.repository, 'test@user-2')
88
text_form = t.as_text()
89
self.log('testament text form:\n' + text_form)
90
self.assertEqualDiff(text_form, REV_2_TESTAMENT)
91
actual_short = t.as_short_text()
92
self.assertEqualDiff(actual_short, REV_2_SHORT)
94
def test_strict_testament_with_contents(self):
95
"""Testament containing a file and a directory."""
96
t = StrictTestament.from_revision(self.b.repository, 'test@user-2')
97
text_form = t.as_text()
98
self.log('testament text form:\n' + text_form)
99
self.assertEqualDiff(text_form, REV_2_STRICT_TESTAMENT)
100
actual_short = t.as_short_text()
101
self.assertEqualDiff(actual_short, REV_2_SHORT_STRICT)
91
t = self.from_revision(self.b.repository, 'test@user-2')
92
text_form = t.as_text()
93
self.log('testament text form:\n' + text_form)
94
self.assertEqualDiff(text_form, self.expected('rev_2'))
95
actual_short = t.as_short_text()
96
self.assertEqualDiff(actual_short, self.expected('rev_2_short'))
103
98
def test_testament_symlinks(self):
104
99
"""Testament containing symlink (where possible)"""
105
if not has_symlinks():
100
self.requireFeature(SymlinkFeature)
107
101
os.symlink('wibble/linktarget', 'link')
108
102
self.wt.add(['link'], ['link-id'])
109
103
self.wt.commit(message='add symlink',
112
106
rev_id='test@user-3',
113
107
committer='test@user')
114
t = Testament.from_revision(self.b.repository, 'test@user-3')
115
self.assertEqualDiff(t.as_text(), REV_3_TESTAMENT)
108
t = self.from_revision(self.b.repository, 'test@user-3')
109
self.assertEqualDiff(t.as_text(), self.expected('rev_3'))
117
111
def test_testament_revprops(self):
118
112
"""Testament to revision with extra properties"""
126
120
rev_id='test@user-3',
127
121
committer='test@user',
129
t = Testament.from_revision(self.b.repository, 'test@user-3')
130
self.assertEqualDiff(t.as_text(), REV_PROPS_TESTAMENT)
123
t = self.from_revision(self.b.repository, 'test@user-3')
124
self.assertEqualDiff(t.as_text(), self.expected('rev_props'))
132
126
def test_testament_unicode_commit_message(self):
138
132
committer='Erik B\xe5gfors <test@user>',
139
133
revprops={'uni':u'\xb5'}
141
t = Testament.from_revision(self.b.repository, 'test@user-3')
135
t = self.from_revision(self.b.repository, 'test@user-3')
142
136
self.assertEqualDiff(
143
SAMPLE_UNICODE_TESTAMENT.encode('utf-8'), t.as_text())
137
self.expected('sample_unicode').encode('utf-8'), t.as_text())
145
139
def test___init__(self):
146
140
revision = self.b.repository.get_revision('test@user-2')
147
141
inventory = self.b.repository.get_inventory('test@user-2')
148
testament_1 = Testament(revision, inventory).as_short_text()
149
testament_2 = Testament.from_revision(self.b.repository,
150
'test@user-2').as_short_text()
151
self.assertEqual(testament_1, testament_2)
142
testament_1 = self.testament_class()(revision, inventory)
143
text_1 = testament_1.as_short_text()
144
text_2 = self.from_revision(self.b.repository,
145
'test@user-2').as_short_text()
146
self.assertEqual(text_1, text_2)
149
class TestamentTestsStrict(TestamentTests):
151
def testament_class(self):
152
return StrictTestament
155
class TestamentTestsStrict2(TestamentTests):
157
def testament_class(self):
158
return StrictTestament3
154
161
REV_1_TESTAMENT = """\
155
162
bazaar-ng testament version 1
156
163
revision-id: test@user-1
271
REV_2_STRICT_TESTAMENT3 = """\
272
bazaar testament version 3 strict
273
revision-id: test@user-2
275
timestamp: 1129025483
280
add files and directories
282
directory . TREE_ROT test@user-1 no
283
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
284
directory src src-id test@user-2 no
285
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
239
292
REV_2_SHORT = """\
240
293
bazaar-ng testament short form 1
241
294
revision-id: test@user-2
339
REV_PROPS_TESTAMENT_STRICT = """\
340
bazaar-ng testament version 2.1
341
revision-id: test@user-3
343
timestamp: 1129025493
348
revision with properties
350
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
351
directory src src-id test@user-2 no
352
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
365
REV_PROPS_TESTAMENT_STRICT3 = """\
366
bazaar testament version 3 strict
367
revision-id: test@user-3
369
timestamp: 1129025493
374
revision with properties
376
directory . TREE_ROT test@user-1 no
377
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
378
directory src src-id test@user-2 no
379
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
279
392
REV_3_TESTAMENT = """\
280
393
bazaar-ng testament version 1
281
394
revision-id: test@user-3
413
REV_3_TESTAMENT_STRICT = """\
414
bazaar-ng testament version 2.1
415
revision-id: test@user-3
417
timestamp: 1129025493
424
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
425
symlink link link-id wibble/linktarget test@user-3 no
426
directory src src-id test@user-2 no
427
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
434
REV_3_TESTAMENT_STRICT3 = """\
435
bazaar testament version 3 strict
436
revision-id: test@user-3
438
timestamp: 1129025493
445
directory . TREE_ROT test@user-1 no
446
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
447
symlink link link-id wibble/linktarget test@user-3 no
448
directory src src-id test@user-2 no
449
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
300
456
SAMPLE_UNICODE_TESTAMENT = u"""\
301
457
bazaar-ng testament version 1
302
458
revision-id: test@user-3
478
SAMPLE_UNICODE_TESTAMENT_STRICT = u"""\
479
bazaar-ng testament version 2.1
480
revision-id: test@user-3
481
committer: Erik B\xe5gfors <test@user>
482
timestamp: 1129025493
487
non-ascii commit \N{COPYRIGHT SIGN} me
489
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
490
directory src src-id test@user-2 no
491
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
500
SAMPLE_UNICODE_TESTAMENT_STRICT3 = u"""\
501
bazaar testament version 3 strict
502
revision-id: test@user-3
503
committer: Erik B\xe5gfors <test@user>
504
timestamp: 1129025493
509
non-ascii commit \N{COPYRIGHT SIGN} me
511
directory . TREE_ROT test@user-1 no
512
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
513
directory src src-id test@user-2 no
514
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
524
Testament: { 'rev_1': REV_1_TESTAMENT,
525
'rev_1_short': REV_1_SHORT,
526
'rev_2': REV_2_TESTAMENT,
527
'rev_2_short': REV_2_SHORT,
528
'rev_3': REV_3_TESTAMENT,
529
'rev_props': REV_PROPS_TESTAMENT,
530
'sample_unicode': SAMPLE_UNICODE_TESTAMENT,
532
StrictTestament: {'rev_1': REV_1_STRICT_TESTAMENT,
533
'rev_1_short': REV_1_SHORT_STRICT,
534
'rev_2': REV_2_STRICT_TESTAMENT,
535
'rev_2_short': REV_2_SHORT_STRICT,
536
'rev_3': REV_3_TESTAMENT_STRICT,
537
'rev_props': REV_PROPS_TESTAMENT_STRICT,
538
'sample_unicode': SAMPLE_UNICODE_TESTAMENT_STRICT,
540
StrictTestament3: {'rev_1': REV_1_STRICT_TESTAMENT3,
541
'rev_1_short': REV_1_SHORT_STRICT3,
542
'rev_2': REV_2_STRICT_TESTAMENT3,
543
'rev_2_short': REV_2_SHORT_STRICT3,
544
'rev_3': REV_3_TESTAMENT_STRICT3,
545
'rev_props': REV_PROPS_TESTAMENT_STRICT3,
546
'sample_unicode': SAMPLE_UNICODE_TESTAMENT_STRICT3,