1
# Copyright (C) 2005-2010 Canonical Ltd
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11
# GNU General Public License for more details.
13
# You should have received a copy of the GNU General Public License
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
17
"""Test testaments for gpg signing."""
19
# TODO: Testaments with x-bits
23
from bzrlib import osutils
24
from bzrlib.tests import TestCaseWithTransport
25
from bzrlib.testament import (
30
from bzrlib.transform import TreeTransform
31
from bzrlib.tests.features import (
36
class TestamentSetup(TestCaseWithTransport):
39
super(TestamentSetup, self).setUp()
40
self.wt = self.make_branch_and_tree('.', format='development-subtree')
41
self.wt.set_root_id('TREE_ROT')
42
b = self.b = self.wt.branch
43
b.nick = "test branch"
44
self.wt.commit(message='initial null commit',
45
committer='test@user',
46
timestamp=1129025423, # 'Tue Oct 11 20:10:23 2005'
49
self.build_tree_contents([('hello', 'contents of hello file'),
51
('src/foo.c', 'int main()\n{\n}\n')])
52
self.wt.add(['hello', 'src', 'src/foo.c'],
53
['hello-id', 'src-id', 'foo.c-id'])
54
tt = TreeTransform(self.wt)
55
trans_id = tt.trans_id_tree_path('hello')
56
tt.set_executability(True, trans_id)
58
self.wt.commit(message='add files and directories',
62
committer='test@user')
65
class TestamentTests(TestamentSetup):
67
def testament_class(self):
70
def expected(self, key):
71
return texts[self.testament_class()][key]
73
def from_revision(self, repository, revision_id):
74
return self.testament_class().from_revision(repository, revision_id)
76
def test_null_testament(self):
77
"""Testament for a revision with no contents."""
78
t = self.from_revision(self.b.repository, 'test@user-1')
81
ass(isinstance(t, Testament))
82
eq(t.revision_id, 'test@user-1')
83
eq(t.committer, 'test@user')
84
eq(t.timestamp, 1129025423)
87
def test_testment_text_form(self):
88
"""Conversion of testament to canonical text form."""
89
t = self.from_revision(self.b.repository, 'test@user-1')
90
text_form = t.as_text()
91
self.log('testament text form:\n' + text_form)
92
self.assertEqualDiff(text_form, self.expected('rev_1'))
93
short_text_form = t.as_short_text()
94
self.assertEqualDiff(short_text_form, self.expected('rev_1_short'))
96
def test_testament_with_contents(self):
97
"""Testament containing a file and a directory."""
98
t = self.from_revision(self.b.repository, 'test@user-2')
99
text_form = t.as_text()
100
self.log('testament text form:\n' + text_form)
101
self.assertEqualDiff(text_form, self.expected('rev_2'))
102
actual_short = t.as_short_text()
103
self.assertEqualDiff(actual_short, self.expected('rev_2_short'))
105
def test_testament_symlinks(self):
106
"""Testament containing symlink (where possible)"""
107
self.requireFeature(SymlinkFeature)
108
os.symlink('wibble/linktarget', 'link')
109
self.wt.add(['link'], ['link-id'])
110
self.wt.commit(message='add symlink',
111
timestamp=1129025493,
113
rev_id='test@user-3',
114
committer='test@user')
115
t = self.from_revision(self.b.repository, 'test@user-3')
116
self.assertEqualDiff(t.as_text(), self.expected('rev_3'))
118
def test_testament_revprops(self):
119
"""Testament to revision with extra properties"""
120
props = dict(flavor='sour cherry\ncream cheese',
124
self.wt.commit(message='revision with properties',
125
timestamp=1129025493,
127
rev_id='test@user-3',
128
committer='test@user',
130
t = self.from_revision(self.b.repository, 'test@user-3')
131
self.assertEqualDiff(t.as_text(), self.expected('rev_props'))
133
def test_testament_unicode_commit_message(self):
135
message=u'non-ascii commit \N{COPYRIGHT SIGN} me',
136
timestamp=1129025493,
138
rev_id='test@user-3',
139
committer=u'Erik B\xe5gfors <test@user>',
140
revprops={'uni':u'\xb5'}
142
t = self.from_revision(self.b.repository, 'test@user-3')
143
self.assertEqualDiff(
144
self.expected('sample_unicode').encode('utf-8'), t.as_text())
146
def test_from_tree(self):
147
tree = self.b.repository.revision_tree('test@user-2')
148
testament = self.testament_class().from_revision_tree(tree)
149
text_1 = testament.as_short_text()
150
text_2 = self.from_revision(self.b.repository,
151
'test@user-2').as_short_text()
152
self.assertEqual(text_1, text_2)
154
def test___init__(self):
155
revision = self.b.repository.get_revision('test@user-2')
156
tree = self.b.repository.revision_tree('test@user-2')
157
testament_1 = self.testament_class()(revision, tree)
158
text_1 = testament_1.as_short_text()
159
text_2 = self.from_revision(self.b.repository,
160
'test@user-2').as_short_text()
161
self.assertEqual(text_1, text_2)
164
class TestamentTestsStrict(TestamentTests):
166
def testament_class(self):
167
return StrictTestament
170
class TestamentTestsStrict2(TestamentTests):
172
def testament_class(self):
173
return StrictTestament3
176
REV_1_TESTAMENT = """\
177
bazaar-ng testament version 1
178
revision-id: test@user-1
180
timestamp: 1129025423
192
REV_1_STRICT_TESTAMENT = """\
193
bazaar-ng testament version 2.1
194
revision-id: test@user-1
196
timestamp: 1129025423
208
REV_1_STRICT_TESTAMENT3 = """\
209
bazaar testament version 3 strict
210
revision-id: test@user-1
212
timestamp: 1129025423
218
directory . TREE_ROT test@user-1 no
226
bazaar-ng testament short form 1
227
revision-id: test@user-1
229
""" % osutils.sha_string(REV_1_TESTAMENT)
232
REV_1_SHORT_STRICT = """\
233
bazaar-ng testament short form 2.1
234
revision-id: test@user-1
236
""" % osutils.sha_string(REV_1_STRICT_TESTAMENT)
239
REV_1_SHORT_STRICT3 = """\
240
bazaar testament short form 3 strict
241
revision-id: test@user-1
243
""" % osutils.sha_string(REV_1_STRICT_TESTAMENT3)
246
REV_2_TESTAMENT = """\
247
bazaar-ng testament version 1
248
revision-id: test@user-2
250
timestamp: 1129025483
255
add files and directories
257
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
259
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
266
REV_2_STRICT_TESTAMENT = """\
267
bazaar-ng testament version 2.1
268
revision-id: test@user-2
270
timestamp: 1129025483
275
add files and directories
277
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
278
directory src src-id test@user-2 no
279
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
286
REV_2_STRICT_TESTAMENT3 = """\
287
bazaar testament version 3 strict
288
revision-id: test@user-2
290
timestamp: 1129025483
295
add files and directories
297
directory . TREE_ROT test@user-1 no
298
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
299
directory src src-id test@user-2 no
300
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
308
bazaar-ng testament short form 1
309
revision-id: test@user-2
311
""" % osutils.sha_string(REV_2_TESTAMENT)
314
REV_2_SHORT_STRICT = """\
315
bazaar-ng testament short form 2.1
316
revision-id: test@user-2
318
""" % osutils.sha_string(REV_2_STRICT_TESTAMENT)
321
REV_2_SHORT_STRICT3 = """\
322
bazaar testament short form 3 strict
323
revision-id: test@user-2
325
""" % osutils.sha_string(REV_2_STRICT_TESTAMENT3)
328
REV_PROPS_TESTAMENT = """\
329
bazaar-ng testament version 1
330
revision-id: test@user-3
332
timestamp: 1129025493
337
revision with properties
339
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
341
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
354
REV_PROPS_TESTAMENT_STRICT = """\
355
bazaar-ng testament version 2.1
356
revision-id: test@user-3
358
timestamp: 1129025493
363
revision with properties
365
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
366
directory src src-id test@user-2 no
367
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
380
REV_PROPS_TESTAMENT_STRICT3 = """\
381
bazaar testament version 3 strict
382
revision-id: test@user-3
384
timestamp: 1129025493
389
revision with properties
391
directory . TREE_ROT test@user-1 no
392
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
393
directory src src-id test@user-2 no
394
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
407
REV_3_TESTAMENT = """\
408
bazaar-ng testament version 1
409
revision-id: test@user-3
411
timestamp: 1129025493
418
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
419
symlink link link-id wibble/linktarget
421
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
428
REV_3_TESTAMENT_STRICT = """\
429
bazaar-ng testament version 2.1
430
revision-id: test@user-3
432
timestamp: 1129025493
439
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
440
symlink link link-id wibble/linktarget test@user-3 no
441
directory src src-id test@user-2 no
442
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
449
REV_3_TESTAMENT_STRICT3 = """\
450
bazaar testament version 3 strict
451
revision-id: test@user-3
453
timestamp: 1129025493
460
directory . TREE_ROT test@user-1 no
461
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
462
symlink link link-id wibble/linktarget test@user-3 no
463
directory src src-id test@user-2 no
464
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
471
SAMPLE_UNICODE_TESTAMENT = u"""\
472
bazaar-ng testament version 1
473
revision-id: test@user-3
474
committer: Erik B\xe5gfors <test@user>
475
timestamp: 1129025493
480
non-ascii commit \N{COPYRIGHT SIGN} me
482
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
484
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
493
SAMPLE_UNICODE_TESTAMENT_STRICT = u"""\
494
bazaar-ng testament version 2.1
495
revision-id: test@user-3
496
committer: Erik B\xe5gfors <test@user>
497
timestamp: 1129025493
502
non-ascii commit \N{COPYRIGHT SIGN} me
504
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
505
directory src src-id test@user-2 no
506
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
515
SAMPLE_UNICODE_TESTAMENT_STRICT3 = u"""\
516
bazaar testament version 3 strict
517
revision-id: test@user-3
518
committer: Erik B\xe5gfors <test@user>
519
timestamp: 1129025493
524
non-ascii commit \N{COPYRIGHT SIGN} me
526
directory . TREE_ROT test@user-1 no
527
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
528
directory src src-id test@user-2 no
529
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
539
Testament: { 'rev_1': REV_1_TESTAMENT,
540
'rev_1_short': REV_1_SHORT,
541
'rev_2': REV_2_TESTAMENT,
542
'rev_2_short': REV_2_SHORT,
543
'rev_3': REV_3_TESTAMENT,
544
'rev_props': REV_PROPS_TESTAMENT,
545
'sample_unicode': SAMPLE_UNICODE_TESTAMENT,
547
StrictTestament: {'rev_1': REV_1_STRICT_TESTAMENT,
548
'rev_1_short': REV_1_SHORT_STRICT,
549
'rev_2': REV_2_STRICT_TESTAMENT,
550
'rev_2_short': REV_2_SHORT_STRICT,
551
'rev_3': REV_3_TESTAMENT_STRICT,
552
'rev_props': REV_PROPS_TESTAMENT_STRICT,
553
'sample_unicode': SAMPLE_UNICODE_TESTAMENT_STRICT,
555
StrictTestament3: {'rev_1': REV_1_STRICT_TESTAMENT3,
556
'rev_1_short': REV_1_SHORT_STRICT3,
557
'rev_2': REV_2_STRICT_TESTAMENT3,
558
'rev_2_short': REV_2_SHORT_STRICT3,
559
'rev_3': REV_3_TESTAMENT_STRICT3,
560
'rev_props': REV_PROPS_TESTAMENT_STRICT3,
561
'sample_unicode': SAMPLE_UNICODE_TESTAMENT_STRICT3,