1
# Copyright (C) 2005 by 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
 
 
25
from bzrlib.selftest import TestCaseInTempDir
 
 
26
from bzrlib.selftest.treeshape import build_tree_contents
 
 
27
from bzrlib.branch import Branch
 
 
28
from bzrlib.testament import Testament
 
 
29
from bzrlib.trace import mutter
 
 
30
from bzrlib.osutils import has_symlinks
 
 
33
class TestamentTests(TestCaseInTempDir):
 
 
36
        super(TestamentTests, self).setUp()
 
 
37
        b = self.b = Branch.initialize('.')
 
 
38
        b.commit(message='initial null commit',
 
 
39
                 committer='test@user',
 
 
40
                 timestamp=1129025423, # 'Tue Oct 11 20:10:23 2005'
 
 
43
        build_tree_contents([('hello', 'contents of hello file'),
 
 
45
                             ('src/foo.c', 'int main()\n{\n}\n')])
 
 
46
        b.add(['hello', 'src', 'src/foo.c'],
 
 
47
              ['hello-id', 'src-id', 'foo.c-id'])
 
 
48
        b.commit(message='add files and directories',
 
 
52
                 committer='test@user')
 
 
54
    def test_null_testament(self):
 
 
55
        """Testament for a revision with no contents."""
 
 
56
        t = Testament.from_revision(self.b, 'test@user-1')
 
 
59
        ass(isinstance(t, Testament))
 
 
60
        eq(t.revision_id, 'test@user-1')
 
 
61
        eq(t.committer, 'test@user')
 
 
62
        eq(t.timestamp, 1129025423)
 
 
65
    def test_testment_text_form(self):
 
 
66
        """Conversion of testament to canonical text form."""
 
 
67
        t = Testament.from_revision(self.b, 'test@user-1')
 
 
68
        text_form = t.as_text()
 
 
69
        self.log('testament text form:\n' + text_form)
 
 
70
        self.assertEqual(text_form, REV_1_TESTAMENT)
 
 
72
    def test_testament_with_contents(self):
 
 
73
        """Testament containing a file and a directory."""
 
 
74
        t = Testament.from_revision(self.b, 'test@user-2')
 
 
75
        text_form = t.as_text()
 
 
76
        self.log('testament text form:\n' + text_form)
 
 
77
        self.assertEqualDiff(text_form, REV_2_TESTAMENT)
 
 
78
        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)
 
 
93
    def test_testament_symlinks(self):
 
 
94
        """Testament containing symlink (where possible)"""
 
 
95
        if not has_symlinks():
 
 
97
        os.symlink('wibble/linktarget', 'link')
 
 
98
        self.b.add(['link'], ['link-id'])
 
 
99
        self.b.commit(message='add symlink',
 
 
100
                 timestamp=1129025493,
 
 
102
                 rev_id='test@user-3',
 
 
103
                 committer='test@user')
 
 
104
        t = Testament.from_revision(self.b, 'test@user-3')
 
 
105
        self.assertEqualDiff(t.as_text(), REV_3_TESTAMENT)
 
 
107
    def test_testament_revprops(self):
 
 
108
        """Testament to revision with extra properties"""
 
 
109
        props = dict(flavor='sour cherry\ncream cheese',
 
 
111
        self.b.commit(message='revision with properties',
 
 
112
                      timestamp=1129025493,
 
 
114
                      rev_id='test@user-3',
 
 
115
                      committer='test@user',
 
 
117
        t = Testament.from_revision(self.b, 'test@user-3')
 
 
118
        self.assertEqualDiff(t.as_text(), REV_PROPS_TESTAMENT)
 
 
120
    def test___init__(self):
 
 
121
        revision = self.b.get_revision('test@user-2')
 
 
122
        inventory = self.b.get_inventory('test@user-2')
 
 
123
        testament_1 = Testament(revision, inventory).as_short_text()
 
 
124
        testament_2 = Testament.from_revision(self.b, 
 
 
125
                                              'test@user-2').as_short_text()
 
 
126
        self.assertEqual(testament_1, testament_2)
 
 
129
REV_1_TESTAMENT = """\
 
 
130
bazaar-ng testament version 1
 
 
131
revision-id: test@user-1
 
 
133
timestamp: 1129025423
 
 
142
bazaar-ng testament short form 1
 
 
143
revision-id: test@user-1
 
 
145
""" % sha(REV_1_TESTAMENT).hexdigest()
 
 
148
REV_2_TESTAMENT = """\
 
 
149
bazaar-ng testament version 1
 
 
150
revision-id: test@user-2
 
 
152
timestamp: 1129025483
 
 
157
  add files and directories
 
 
159
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
 
 
161
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
 
 
166
bazaar-ng testament short form 1
 
 
167
revision-id: test@user-2
 
 
169
""" % sha(REV_2_TESTAMENT).hexdigest()
 
 
172
REV_PROPS_TESTAMENT = """\
 
 
173
bazaar-ng testament version 1
 
 
174
revision-id: test@user-3
 
 
176
timestamp: 1129025493
 
 
181
  revision with properties
 
 
183
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
 
 
185
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24
 
 
195
REV_3_TESTAMENT = """\
 
 
196
bazaar-ng testament version 1
 
 
197
revision-id: test@user-3
 
 
199
timestamp: 1129025493
 
 
206
  file hello hello-id 34dd0ac19a24bf80c4d33b5c8960196e8d8d1f73
 
 
207
  symlink link link-id wibble/linktarget
 
 
209
  file src/foo.c foo.c-id a2a049c20f908ae31b231d98779eb63c66448f24