0b24fd5272b0493ff31cfb0840a0267f7789fe56
[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 from pyme import core, constants
22 import support
23
24 support.init_gpgme(constants.PROTOCOL_OpenPGP)
25
26 for passphrase in ("abc", b"abc"):
27     c = core.Context()
28     c.set_armor(True)
29     c.set_pinentry_mode(constants.PINENTRY_MODE_LOOPBACK)
30
31     source = core.Data("Hallo Leute\n")
32     cipher = core.Data()
33
34     passphrase_cb_called = 0
35     def passphrase_cb(hint, desc, prev_bad, hook=None):
36         global passphrase_cb_called
37         passphrase_cb_called += 1
38         return passphrase
39
40     c.set_passphrase_cb(passphrase_cb, None)
41
42     c.op_encrypt([], 0, source, cipher)
43     assert passphrase_cb_called == 1, \
44         "Callback called {} times".format(passphrase_cb_called)
45     support.print_data(cipher)
46
47     c = core.Context()
48     c.set_armor(True)
49     c.set_pinentry_mode(constants.PINENTRY_MODE_LOOPBACK)
50     c.set_passphrase_cb(passphrase_cb, None)
51     plain = core.Data()
52     cipher.seek(0, os.SEEK_SET)
53
54     c.op_decrypt(cipher, plain)
55     # Seems like the passphrase is cached.
56     #assert passphrase_cb_called == 2, \
57     #    "Callback called {} times".format(passphrase_cb_called)
58     support.print_data(plain)
59
60     plain.seek(0, os.SEEK_SET)
61     plaintext = plain.read()
62     assert plaintext == b"Hallo Leute\n", \
63         "Wrong plaintext {!r}".format(plaintext)