22
22
from sha import sha
25
from bzrlib.tests import TestCaseInTempDir
26
from bzrlib.branch import Branch
27
from bzrlib.testament import Testament
28
from bzrlib.trace import mutter
24
from bzrlib.tests import TestCaseWithTransport
25
from bzrlib.testament import Testament, StrictTestament, StrictTestament3
26
from bzrlib.transform import TreeTransform
29
27
from bzrlib.osutils import has_symlinks
32
class TestamentTests(TestCaseInTempDir):
30
class TestamentSetup(TestCaseWithTransport):
35
super(TestamentTests, self).setUp()
36
b = self.b = Branch.initialize(u'.')
33
super(TestamentSetup, self).setUp()
34
self.wt = self.make_branch_and_tree('.', format='dirstate-with-subtree')
35
self.wt.set_root_id('TREE_ROT')
36
b = self.b = self.wt.branch
37
37
b.nick = "test branch"
38
b.working_tree().commit(message='initial null commit',
38
self.wt.commit(message='initial null commit',
39
39
committer='test@user',
40
40
timestamp=1129025423, # 'Tue Oct 11 20:10:23 2005'
43
43
self.build_tree_contents([('hello', 'contents of hello file'),
45
45
('src/foo.c', 'int main()\n{\n}\n')])
46
b.working_tree().add(['hello', 'src', 'src/foo.c'],
46
self.wt.add(['hello', 'src', 'src/foo.c'],
47
47
['hello-id', 'src-id', 'foo.c-id'])
48
b.working_tree().commit(message='add files and directories',
48
tt = TreeTransform(self.wt)
49
trans_id = tt.trans_id_tree_path('hello')
50
tt.set_executability(True, trans_id)
52
self.wt.commit(message='add files and directories',
49
53
timestamp=1129025483,
51
55
rev_id='test@user-2',
52
56
committer='test@user')
59
class TestamentTests(TestamentSetup):
61
def testament_class(self):
64
def expected(self, key):
65
return texts[self.testament_class()][key]
67
def from_revision(self, repository, revision_id):
68
return self.testament_class().from_revision(repository, revision_id)
54
70
def test_null_testament(self):
55
71
"""Testament for a revision with no contents."""
56
t = Testament.from_revision(self.b.repository, 'test@user-1')
72
t = self.from_revision(self.b.repository, 'test@user-1')
57
73
ass = self.assertTrue
58
74
eq = self.assertEqual
59
75
ass(isinstance(t, Testament))
65
81
def test_testment_text_form(self):
66
82
"""Conversion of testament to canonical text form."""
67
t = Testament.from_revision(self.b.repository, 'test@user-1')
83
t = self.from_revision(self.b.repository, 'test@user-1')
68
84
text_form = t.as_text()
69
85
self.log('testament text form:\n' + text_form)
70
self.assertEqual(text_form, REV_1_TESTAMENT)
86
self.assertEqualDiff(text_form, self.expected('rev_1'))
87
short_text_form = t.as_short_text()
88
self.assertEqualDiff(short_text_form, self.expected('rev_1_short'))
72
90
def test_testament_with_contents(self):
73
91
"""Testament containing a file and a directory."""
74
t = Testament.from_revision(self.b.repository, 'test@user-2')
92
t = self.from_revision(self.b.repository, 'test@user-2')
75
93
text_form = t.as_text()
76
94
self.log('testament text form:\n' + text_form)
77
self.assertEqualDiff(text_form, REV_2_TESTAMENT)
95
self.assertEqualDiff(text_form, self.expected('rev_2'))
78
96
actual_short = t.as_short_text()
79
self.assertEqualDiff(actual_short, REV_2_SHORT)
81
def test_testament_command(self):
82
"""Testament containing a file and a directory."""
83
out, err = self.run_bzr_captured(['testament', '--long'])
84
self.assertEqualDiff(err, '')
85
self.assertEqualDiff(out, REV_2_TESTAMENT)
87
def test_testament_command_2(self):
88
"""Command getting short testament of previous version."""
89
out, err = self.run_bzr_captured(['testament', '-r1'])
90
self.assertEqualDiff(err, '')
91
self.assertEqualDiff(out, REV_1_SHORT)
97
self.assertEqualDiff(actual_short, self.expected('rev_2_short'))
93
99
def test_testament_symlinks(self):
94
100
"""Testament containing symlink (where possible)"""
95
101
if not has_symlinks():
97
103
os.symlink('wibble/linktarget', 'link')
98
self.b.working_tree().add(['link'], ['link-id'])
99
self.b.working_tree().commit(message='add symlink',
104
self.wt.add(['link'], ['link-id'])
105
self.wt.commit(message='add symlink',
100
106
timestamp=1129025493,
102
108
rev_id='test@user-3',
103
109
committer='test@user')
104
t = Testament.from_revision(self.b.repository, 'test@user-3')
105
self.assertEqualDiff(t.as_text(), REV_3_TESTAMENT)
110
t = self.from_revision(self.b.repository, 'test@user-3')
111
self.assertEqualDiff(t.as_text(), self.expected('rev_3'))
107
113
def test_testament_revprops(self):
108
114
"""Testament to revision with extra properties"""
109
115
props = dict(flavor='sour cherry\ncream cheese',
111
self.b.working_tree().commit(message='revision with properties',
119
self.wt.commit(message='revision with properties',
112
120
timestamp=1129025493,
114
122
rev_id='test@user-3',
115
123
committer='test@user',
117
t = Testament.from_revision(self.b.repository, 'test@user-3')
118
self.assertEqualDiff(t.as_text(), REV_PROPS_TESTAMENT)
125
t = self.from_revision(self.b.repository, 'test@user-3')
126
self.assertEqualDiff(t.as_text(), self.expected('rev_props'))
128
def test_testament_unicode_commit_message(self):
130
message=u'non-ascii commit \N{COPYRIGHT SIGN} me',
131
timestamp=1129025493,
133
rev_id='test@user-3',
134
committer='Erik B\xe5gfors <test@user>',
135
revprops={'uni':u'\xb5'}
137
t = self.from_revision(self.b.repository, 'test@user-3')
138
self.assertEqualDiff(
139
self.expected('sample_unicode').encode('utf-8'), t.as_text())
120
141
def test___init__(self):
121
142
revision = self.b.repository.get_revision('test@user-2')
122
143
inventory = self.b.repository.get_inventory('test@user-2')
123
testament_1 = Testament(revision, inventory).as_short_text()
124
testament_2 = Testament.from_revision(self.b.repository,
125
'test@user-2').as_short_text()
126
self.assertEqual(testament_1, testament_2)
144
testament_1 = self.testament_class()(revision, inventory)
145
text_1 = testament_1.as_short_text()
146
text_2 = self.from_revision(self.b.repository,
147
'test@user-2').as_short_text()
148
self.assertEqual(text_1, text_2)
151
class TestamentTestsStrict(TestamentTests):
153
def testament_class(self):
154
return StrictTestament
157
class TestamentTestsStrict2(TestamentTests):
159
def testament_class(self):
160
return StrictTestament3
129
163
REV_1_TESTAMENT = """\
130
164
bazaar-ng testament version 1
131
165
revision-id: test@user-1
415
REV_3_TESTAMENT_STRICT = """\
416
bazaar-ng testament version 2.1
417
revision-id: test@user-3
419
timestamp: 1129025493
426
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
427
symlink link link-id wibble/linktarget test@user-3 no
428
directory src src-id test@user-2 no
429
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
436
REV_3_TESTAMENT_STRICT3 = """\
437
bazaar testament version 3 strict
438
revision-id: test@user-3
440
timestamp: 1129025493
447
directory . TREE_ROT test@user-1 no
448
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
449
symlink link link-id wibble/linktarget test@user-3 no
450
directory src src-id test@user-2 no
451
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
458
SAMPLE_UNICODE_TESTAMENT = u"""\
459
bazaar-ng testament version 1
460
revision-id: test@user-3
461
committer: Erik B\xe5gfors <test@user>
462
timestamp: 1129025493
467
non-ascii commit \N{COPYRIGHT SIGN} me
469
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
471
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
480
SAMPLE_UNICODE_TESTAMENT_STRICT = u"""\
481
bazaar-ng testament version 2.1
482
revision-id: test@user-3
483
committer: Erik B\xe5gfors <test@user>
484
timestamp: 1129025493
489
non-ascii commit \N{COPYRIGHT SIGN} me
491
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
492
directory src src-id test@user-2 no
493
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
502
SAMPLE_UNICODE_TESTAMENT_STRICT3 = u"""\
503
bazaar testament version 3 strict
504
revision-id: test@user-3
505
committer: Erik B\xe5gfors <test@user>
506
timestamp: 1129025493
511
non-ascii commit \N{COPYRIGHT SIGN} me
513
directory . TREE_ROT test@user-1 no
514
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
515
directory src src-id test@user-2 no
516
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
526
Testament: { 'rev_1': REV_1_TESTAMENT,
527
'rev_1_short': REV_1_SHORT,
528
'rev_2': REV_2_TESTAMENT,
529
'rev_2_short': REV_2_SHORT,
530
'rev_3': REV_3_TESTAMENT,
531
'rev_props': REV_PROPS_TESTAMENT,
532
'sample_unicode': SAMPLE_UNICODE_TESTAMENT,
534
StrictTestament: {'rev_1': REV_1_STRICT_TESTAMENT,
535
'rev_1_short': REV_1_SHORT_STRICT,
536
'rev_2': REV_2_STRICT_TESTAMENT,
537
'rev_2_short': REV_2_SHORT_STRICT,
538
'rev_3': REV_3_TESTAMENT_STRICT,
539
'rev_props': REV_PROPS_TESTAMENT_STRICT,
540
'sample_unicode': SAMPLE_UNICODE_TESTAMENT_STRICT,
542
StrictTestament3: {'rev_1': REV_1_STRICT_TESTAMENT3,
543
'rev_1_short': REV_1_SHORT_STRICT3,
544
'rev_2': REV_2_STRICT_TESTAMENT3,
545
'rev_2_short': REV_2_SHORT_STRICT3,
546
'rev_3': REV_3_TESTAMENT_STRICT3,
547
'rev_props': REV_PROPS_TESTAMENT_STRICT3,
548
'sample_unicode': SAMPLE_UNICODE_TESTAMENT_STRICT3,