← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/tests/test_gpg.py

  • Committer: Martin Pool
  • Date: 2005-05-31 03:02:45 UTC
  • Revision ID: mbp@sourcefrog.net-20050531030244-d9bbefb14ee48d88
- new bzrlib.get_bzr_revision() tells about the history of 
  bzr itself
- also display revision-id in --version output

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/tests/test_gpg.py
1
 
# Copyright (C) 2005, 2006, 2007, 2009, 2011 Canonical Ltd
2
 
#   Authors: Robert Collins <robert.collins@canonical.com>
3
 
#
4
 
# This program is free software; you can redistribute it and/or modify
5
 
# it under the terms of the GNU General Public License as published by
6
 
# the Free Software Foundation; either version 2 of the License, or
7
 
# (at your option) any later version.
8
 
#
9
 
# This program is distributed in the hope that it will be useful,
10
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
# GNU General Public License for more details.
13
 
#
14
 
# You should have received a copy of the GNU General Public License
15
 
# along with this program; if not, write to the Free Software
16
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17
 
 
18
 
"""Tests for signing and verifying blobs of data via gpg."""
19
 
 
20
 
# import system imports here
21
 
import sys
22
 
 
23
 
from bzrlib import errors, ui
24
 
import bzrlib.gpg as gpg
25
 
from bzrlib.tests import TestCase
26
 
 
27
 
class FakeConfig(object):
28
 
 
29
 
    def gpg_signing_command(self):
30
 
        return "false"
31
 
 
32
 
 
33
 
class TestCommandLine(TestCase):
34
 
 
35
 
    def test_signing_command_line(self):
36
 
        my_gpg = gpg.GPGStrategy(FakeConfig())
37
 
        self.assertEqual(['false',  '--clearsign'],
38
 
                         my_gpg._command_line())
39
 
 
40
 
    def test_checks_return_code(self):
41
 
        # This test needs a unix like platform - one with 'false' to run.
42
 
        # if you have one, please make this work :)
43
 
        my_gpg = gpg.GPGStrategy(FakeConfig())
44
 
        self.assertRaises(errors.SigningFailed, my_gpg.sign, 'content')
45
 
 
46
 
    def assertProduces(self, content):
47
 
        # This needs a 'cat' command or similar to work.
48
 
        my_gpg = gpg.GPGStrategy(FakeConfig())
49
 
        if sys.platform == 'win32':
50
 
            # Windows doesn't come with cat, and we don't require it
51
 
            # so lets try using python instead.
52
 
            # But stupid windows and line-ending conversions.
53
 
            # It is too much work to make sys.stdout be in binary mode.
54
 
            # http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/65443
55
 
            my_gpg._command_line = lambda:[sys.executable, '-c',
56
 
                    'import sys; sys.stdout.write(sys.stdin.read())']
57
 
            new_content = content.replace('\n', '\r\n')
58
 
 
59
 
            self.assertEqual(new_content, my_gpg.sign(content))
60
 
        else:
61
 
            my_gpg._command_line = lambda:['cat', '-']
62
 
            self.assertEqual(content, my_gpg.sign(content))
63
 
 
64
 
    def test_returns_output(self):
65
 
        content = "some content\nwith newlines\n"
66
 
        self.assertProduces(content)
67
 
 
68
 
    def test_clears_progress(self):
69
 
        content = "some content\nwith newlines\n"
70
 
        old_clear_term = ui.ui_factory.clear_term
71
 
        clear_term_called = []
72
 
        def clear_term():
73
 
            old_clear_term()
74
 
            clear_term_called.append(True)
75
 
        ui.ui_factory.clear_term = clear_term
76
 
        try:
77
 
            self.assertProduces(content)
78
 
        finally:
79
 
            ui.ui_factory.clear_term = old_clear_term
80
 
        self.assertEqual([True], clear_term_called)
81
 
 
82
 
    def test_aborts_on_unicode(self):
83
 
        """You can't sign Unicode text; it must be encoded first."""
84
 
        self.assertRaises(errors.BzrBadParameterUnicode,
85
 
                          self.assertProduces, u'foo')
86
 
 
87
 
 
88
 
class TestDisabled(TestCase):
89
 
 
90
 
    def test_sign(self):
91
 
        self.assertRaises(errors.SigningFailed,
92
 
                          gpg.DisabledGPGStrategy(None).sign, 'content')