1
# Copyright (C) 2005, 2006 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17
"""Test testaments for gpg signing."""
19
# TODO: Testaments with x-bits
24
from bzrlib.tests import TestCaseWithTransport
25
from bzrlib.testament import Testament, StrictTestament, StrictTestament3
26
from bzrlib.transform import TreeTransform
27
from bzrlib.osutils import has_symlinks
30
class TestamentSetup(TestCaseWithTransport):
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
b.nick = "test branch"
38
self.wt.commit(message='initial null commit',
39
committer='test@user',
40
timestamp=1129025423, # 'Tue Oct 11 20:10:23 2005'
43
self.build_tree_contents([('hello', 'contents of hello file'),
45
('src/foo.c', 'int main()\n{\n}\n')])
46
self.wt.add(['hello', 'src', 'src/foo.c'],
47
['hello-id', 'src-id', 'foo.c-id'])
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',
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)
70
def test_null_testament(self):
71
"""Testament for a revision with no contents."""
72
t = self.from_revision(self.b.repository, 'test@user-1')
75
ass(isinstance(t, Testament))
76
eq(t.revision_id, 'test@user-1')
77
eq(t.committer, 'test@user')
78
eq(t.timestamp, 1129025423)
81
def test_testment_text_form(self):
82
"""Conversion of testament to canonical text form."""
83
t = self.from_revision(self.b.repository, 'test@user-1')
84
text_form = t.as_text()
85
self.log('testament text form:\n' + text_form)
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'))
90
def test_testament_with_contents(self):
91
"""Testament containing a file and a directory."""
92
t = self.from_revision(self.b.repository, 'test@user-2')
93
text_form = t.as_text()
94
self.log('testament text form:\n' + text_form)
95
self.assertEqualDiff(text_form, self.expected('rev_2'))
96
actual_short = t.as_short_text()
97
self.assertEqualDiff(actual_short, self.expected('rev_2_short'))
99
def test_testament_symlinks(self):
100
"""Testament containing symlink (where possible)"""
101
if not has_symlinks():
103
os.symlink('wibble/linktarget', 'link')
104
self.wt.add(['link'], ['link-id'])
105
self.wt.commit(message='add symlink',
106
timestamp=1129025493,
108
rev_id='test@user-3',
109
committer='test@user')
110
t = self.from_revision(self.b.repository, 'test@user-3')
111
self.assertEqualDiff(t.as_text(), self.expected('rev_3'))
113
def test_testament_revprops(self):
114
"""Testament to revision with extra properties"""
115
props = dict(flavor='sour cherry\ncream cheese',
119
self.wt.commit(message='revision with properties',
120
timestamp=1129025493,
122
rev_id='test@user-3',
123
committer='test@user',
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())
141
def test___init__(self):
142
revision = self.b.repository.get_revision('test@user-2')
143
inventory = self.b.repository.get_inventory('test@user-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
163
REV_1_TESTAMENT = """\
164
bazaar-ng testament version 1
165
revision-id: test@user-1
167
timestamp: 1129025423
179
REV_1_STRICT_TESTAMENT = """\
180
bazaar-ng testament version 2.1
181
revision-id: test@user-1
183
timestamp: 1129025423
195
REV_1_STRICT_TESTAMENT3 = """\
196
bazaar testament version 3 strict
197
revision-id: test@user-1
199
timestamp: 1129025423
205
directory . TREE_ROT test@user-1 no
213
bazaar-ng testament short form 1
214
revision-id: test@user-1
216
""" % sha(REV_1_TESTAMENT).hexdigest()
219
REV_1_SHORT_STRICT = """\
220
bazaar-ng testament short form 2.1
221
revision-id: test@user-1
223
""" % sha(REV_1_STRICT_TESTAMENT).hexdigest()
226
REV_1_SHORT_STRICT3 = """\
227
bazaar testament short form 3 strict
228
revision-id: test@user-1
230
""" % sha(REV_1_STRICT_TESTAMENT3).hexdigest()
233
REV_2_TESTAMENT = """\
234
bazaar-ng testament version 1
235
revision-id: test@user-2
237
timestamp: 1129025483
242
add files and directories
244
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
246
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
253
REV_2_STRICT_TESTAMENT = """\
254
bazaar-ng testament version 2.1
255
revision-id: test@user-2
257
timestamp: 1129025483
262
add files and directories
264
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
265
directory src src-id test@user-2 no
266
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
273
REV_2_STRICT_TESTAMENT3 = """\
274
bazaar testament version 3 strict
275
revision-id: test@user-2
277
timestamp: 1129025483
282
add files and directories
284
directory . TREE_ROT test@user-1 no
285
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
286
directory src src-id test@user-2 no
287
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
295
bazaar-ng testament short form 1
296
revision-id: test@user-2
298
""" % sha(REV_2_TESTAMENT).hexdigest()
301
REV_2_SHORT_STRICT = """\
302
bazaar-ng testament short form 2.1
303
revision-id: test@user-2
305
""" % sha(REV_2_STRICT_TESTAMENT).hexdigest()
308
REV_2_SHORT_STRICT3 = """\
309
bazaar testament short form 3 strict
310
revision-id: test@user-2
312
""" % sha(REV_2_STRICT_TESTAMENT3).hexdigest()
315
REV_PROPS_TESTAMENT = """\
316
bazaar-ng testament version 1
317
revision-id: test@user-3
319
timestamp: 1129025493
324
revision with properties
326
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
328
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
341
REV_PROPS_TESTAMENT_STRICT = """\
342
bazaar-ng testament version 2.1
343
revision-id: test@user-3
345
timestamp: 1129025493
350
revision with properties
352
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
353
directory src src-id test@user-2 no
354
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
367
REV_PROPS_TESTAMENT_STRICT3 = """\
368
bazaar testament version 3 strict
369
revision-id: test@user-3
371
timestamp: 1129025493
376
revision with properties
378
directory . TREE_ROT test@user-1 no
379
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73 test@user-2 yes
380
directory src src-id test@user-2 no
381
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24 test@user-2 no
394
REV_3_TESTAMENT = """\
395
bazaar-ng testament version 1
396
revision-id: test@user-3
398
timestamp: 1129025493
405
file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
406
symlink link link-id wibble/linktarget
408
file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
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,