python: Add an idiomatic interface.
[gpgme.git] / lang / python / tests / t-encrypt.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
22 import support
23
24 support.init_gpgme(constants.PROTOCOL_OpenPGP)
25 c = core.Context()
26 c.set_armor(True)
27
28 source = core.Data("Hallo Leute\n")
29 sink = core.Data()
30
31 keys = []
32 keys.append(c.get_key("A0FF4590BB6122EDEF6E3C542D727CC768697734", False))
33 keys.append(c.get_key("D695676BDCEDCC2CDD6152BCFE180B1DA9E3B0B2", False))
34
35 c.op_encrypt(keys, constants.ENCRYPT_ALWAYS_TRUST, source, sink)
36 result = c.op_encrypt_result()
37 assert not result.invalid_recipients, \
38   "Invalid recipients: {}".format(", ".join(r.fpr for r in result.recipients))
39 support.print_data(sink)
40
41 # Idiomatic interface.
42 with pyme.Context(armor=True) as c:
43     ciphertext, _, _ = c.encrypt("Hallo Leute\n".encode(),
44                                  recipients=keys,
45                                  sign=False,
46                                  always_trust=True)
47     assert len(ciphertext) > 0
48     assert ciphertext.find(b'BEGIN PGP MESSAGE') > 0, 'Marker not found'
49
50     c.encrypt("Hallo Leute\n".encode(),
51               recipients=[c.get_key(support.encrypt_only, False)],
52               sign=False, always_trust=True)
53
54     try:
55         c.encrypt("Hallo Leute\n".encode(),
56                   recipients=[c.get_key(support.sign_only, False)],
57                   sign=False, always_trust=True)
58     except pyme.errors.InvalidRecipients as e:
59         assert len(e.recipients) == 1
60         assert support.sign_only.endswith(e.recipients[0].fpr)
61     else:
62         assert False, "Expected an InvalidRecipients error, got none"