34
class FakeConfig(object):
36
def gpg_signing_key(self):
37
return "amy@example.com"
39
def gpg_signing_command(self):
42
def acceptable_keys(self):
46
class TestCommandLine(TestCase):
37
class FakeConfig(config.Stack):
39
def __init__(self, content=None):
40
store = config.IniFileStore()
43
gpg_signing_key=amy@example.com
44
gpg_signing_command=false'''
45
store._load_from_string(content)
46
super(FakeConfig, self).__init__([store.get_sections])
49
class TestCommandLine(tests.TestCase):
52
super(TestCommandLine, self).setUp()
53
self.my_gpg = gpg.GPGStrategy(FakeConfig())
48
55
def test_signing_command_line(self):
49
my_gpg = gpg.GPGStrategy(FakeConfig())
56
self.assertEqual(['false', '--clearsign', '-u', 'amy@example.com'],
57
self.my_gpg._command_line())
59
def test_signing_command_line_from_default(self):
60
# Using 'default' for gpg_signing_key will use the mail part of 'email'
61
my_gpg = gpg.GPGStrategy(FakeConfig('''
62
email=Amy <amy@example.com>
63
gpg_signing_key=default
64
gpg_signing_command=false'''))
65
self.assertEqual(['false', '--clearsign', '-u', 'amy@example.com'],
66
my_gpg._command_line())
68
def test_signing_command_line_from_email(self):
69
# Not setting gpg_signing_key will use the mail part of 'email'
70
my_gpg = gpg.GPGStrategy(FakeConfig('''
71
email=Amy <amy@example.com>
72
gpg_signing_command=false'''))
50
73
self.assertEqual(['false', '--clearsign', '-u', 'amy@example.com'],
51
74
my_gpg._command_line())
53
76
def test_checks_return_code(self):
54
77
# This test needs a unix like platform - one with 'false' to run.
55
78
# if you have one, please make this work :)
56
my_gpg = gpg.GPGStrategy(FakeConfig())
57
self.assertRaises(errors.SigningFailed, my_gpg.sign, 'content')
79
self.assertRaises(errors.SigningFailed, self.my_gpg.sign, 'content')
59
81
def assertProduces(self, content):
60
82
# This needs a 'cat' command or similar to work.
61
my_gpg = gpg.GPGStrategy(FakeConfig())
62
83
if sys.platform == 'win32':
63
84
# Windows doesn't come with cat, and we don't require it
64
85
# so lets try using python instead.
65
86
# But stupid windows and line-ending conversions.
66
87
# It is too much work to make sys.stdout be in binary mode.
67
88
# http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/65443
68
my_gpg._command_line = lambda:[sys.executable, '-c',
89
self.my_gpg._command_line = lambda:[sys.executable, '-c',
69
90
'import sys; sys.stdout.write(sys.stdin.read())']
70
91
new_content = content.replace('\n', '\r\n')
72
self.assertEqual(new_content, my_gpg.sign(content))
93
self.assertEqual(new_content, self.my_gpg.sign(content))
74
my_gpg._command_line = lambda:['cat', '-']
75
self.assertEqual(content, my_gpg.sign(content))
95
self.my_gpg._command_line = lambda:['cat', '-']
96
self.assertEqual(content, self.my_gpg.sign(content))
77
98
def test_returns_output(self):
78
99
content = "some content\nwith newlines\n"
added
removed
bzrlib/tests/test_gpg.py
