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