python: Add an idiomatic interface.
[gpgme.git] / lang / python / tests / t-encrypt-sym.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 import pyme
22 from pyme import core, constants
23 import support
24
25 support.init_gpgme(constants.PROTOCOL_OpenPGP)
26
27 for passphrase in ("abc", b"abc"):
28     c = core.Context()
29     c.set_armor(True)
30     c.set_pinentry_mode(constants.PINENTRY_MODE_LOOPBACK)
31
32     source = core.Data("Hallo Leute\n")
33     cipher = core.Data()
34
35     passphrase_cb_called = 0
36     def passphrase_cb(hint, desc, prev_bad, hook=None):
37         global passphrase_cb_called
38         passphrase_cb_called += 1
39         return passphrase
40
41     c.set_passphrase_cb(passphrase_cb, None)
42
43     c.op_encrypt([], 0, source, cipher)
44     assert passphrase_cb_called == 1, \
45         "Callback called {} times".format(passphrase_cb_called)
46     support.print_data(cipher)
47
48     c = core.Context()
49     c.set_armor(True)
50     c.set_pinentry_mode(constants.PINENTRY_MODE_LOOPBACK)
51     c.set_passphrase_cb(passphrase_cb, None)
52     plain = core.Data()
53     cipher.seek(0, os.SEEK_SET)
54
55     c.op_decrypt(cipher, plain)
56     # Seems like the passphrase is cached.
57     #assert passphrase_cb_called == 2, \
58     #    "Callback called {} times".format(passphrase_cb_called)
59     support.print_data(plain)
60
61     plain.seek(0, os.SEEK_SET)
62     plaintext = plain.read()
63     assert plaintext == b"Hallo Leute\n", \
64         "Wrong plaintext {!r}".format(plaintext)
65
66 # Idiomatic interface.
67 for passphrase in ("abc", b"abc"):
68     with pyme.Context(armor=True) as c:
69         # Check that the passphrase callback is not altered.
70         def f(*args):
71             assert False
72         c.set_passphrase_cb(f)
73
74         message = "Hallo Leute\n".encode()
75         ciphertext, _, _ = c.encrypt(message,
76                                      passphrase=passphrase,
77                                      sign=False)
78         assert ciphertext.find(b'BEGIN PGP MESSAGE') > 0, 'Marker not found'
79
80         plaintext, _, _ = c.decrypt(ciphertext, passphrase=passphrase)
81         assert plaintext == message, 'Message body not recovered'
82
83         assert c._passphrase_cb[1] == f, "Passphrase callback not restored"