python: Add an idiomatic interface.
[gpgme.git] / lang / python / tests / t-decrypt-verify.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 pyme
21 from pyme import core, constants, errors
22 import support
23
24 def check_verify_result(result, summary, fpr, status):
25     assert len(result.signatures) == 1, "Unexpected number of signatures"
26     sig = result.signatures[0]
27     assert sig.summary == summary, "Unexpected signature summary"
28     assert sig.fpr == fpr
29     assert errors.GPGMEError(sig.status).getcode() == status
30     assert len(sig.notations) == 0
31     assert not sig.wrong_key_usage
32     assert sig.validity == constants.VALIDITY_FULL
33     assert errors.GPGMEError(sig.validity_reason).getcode() == errors.NO_ERROR
34
35 support.init_gpgme(constants.PROTOCOL_OpenPGP)
36 c = core.Context()
37
38 source = core.Data(file=support.make_filename("cipher-2.asc"))
39 sink = core.Data()
40
41 c.op_decrypt_verify(source, sink)
42 result = c.op_decrypt_result()
43 assert not result.unsupported_algorithm, \
44     "Unsupported algorithm: {}".format(result.unsupported_algorithm)
45
46 support.print_data(sink)
47
48 verify_result = c.op_verify_result()
49 check_verify_result(verify_result,
50                     constants.SIGSUM_VALID | constants.SIGSUM_GREEN,
51                     "A0FF4590BB6122EDEF6E3C542D727CC768697734",
52                     errors.NO_ERROR)
53
54 # Idiomatic interface.
55 with pyme.Context() as c:
56     alpha = c.get_key("A0FF4590BB6122EDEF6E3C542D727CC768697734", False)
57     bob = c.get_key("D695676BDCEDCC2CDD6152BCFE180B1DA9E3B0B2", False)
58     plaintext, _, verify_result = \
59         c.decrypt(open(support.make_filename("cipher-2.asc")), verify=[alpha])
60     assert plaintext.find(b'Wenn Sie dies lesen k') >= 0, \
61         'Plaintext not found'
62     check_verify_result(verify_result,
63                         constants.SIGSUM_VALID | constants.SIGSUM_GREEN,
64                         "A0FF4590BB6122EDEF6E3C542D727CC768697734",
65                         errors.NO_ERROR)
66
67     try:
68         c.decrypt(open(support.make_filename("cipher-2.asc")),
69                   verify=[alpha, bob])
70     except errors.MissingSignatures as e:
71         assert len(e.missing) == 1
72         assert e.missing[0] == bob
73     else:
74         assert False, "Expected an error, got none"