python: Robust exception handling in callbacks.
[gpgme.git] / lang / python / tests / t-callbacks.py
1 #!/usr/bin/env python3
2
3 # Copyright (C) 2016 g10 Code GmbH
4 #
5 # This file is part of GPGME.
6 #
7 # GPGME is free software; you can redistribute it and/or modify it
8 # under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 2 of the License, or
10 # (at your option) any later version.
11 #
12 # GPGME is distributed in the hope that it will be useful, but WITHOUT
13 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
14 # or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General
15 # Public License for more details.
16 #
17 # You should have received a copy of the GNU Lesser General Public
18 # License along with this program; if not, see <http://www.gnu.org/licenses/>.
19
20 import os
21 from pyme import core, constants
22 import support
23
24 support.init_gpgme(constants.PROTOCOL_OpenPGP)
25
26 c = core.Context()
27 c.set_pinentry_mode(constants.PINENTRY_MODE_LOOPBACK)
28
29 source = core.Data("Hallo Leute\n")
30 sink = core.Data()
31
32 # Valid passphrases, both as string and bytes.
33 for passphrase in ('foo', b'foo'):
34     def passphrase_cb(hint, desc, prev_bad, hook=None):
35         assert hook == passphrase
36         return hook
37
38     c.set_passphrase_cb(passphrase_cb, passphrase)
39     c.op_encrypt([], 0, source, sink)
40
41 # Returning an invalid type.
42 def passphrase_cb(hint, desc, prev_bad, hook=None):
43     return 0
44
45 c.set_passphrase_cb(passphrase_cb, None)
46 try:
47     c.op_encrypt([], 0, source, sink)
48 except Exception as e:
49     assert type(e) == TypeError
50     assert str(e) == "expected str or bytes from passphrase callback, got int"
51 else:
52     assert False, "Expected an error, got none"
53
54 # Raising an exception inside callback.
55 myException = Exception()
56 def passphrase_cb(hint, desc, prev_bad, hook=None):
57     raise myException
58
59 c.set_passphrase_cb(passphrase_cb, None)
60 try:
61     c.op_encrypt([], 0, source, sink)
62 except Exception as e:
63     assert e == myException
64 else:
65     assert False, "Expected an error, got none"
66
67 # Wrong kind of callback function.
68 def bad_passphrase_cb():
69     pass
70
71 c.set_passphrase_cb(bad_passphrase_cb, None)
72 try:
73     c.op_encrypt([], 0, source, sink)
74 except Exception as e:
75     assert type(e) == TypeError
76 else:
77     assert False, "Expected an error, got none"