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