python: Add an idiomatic interface.
[gpgme.git] / lang / python / tests / t-decrypt.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
27 source = core.Data(file=support.make_filename("cipher-1.asc"))
28 sink = core.Data()
29
30 c.op_decrypt(source, sink)
31 result = c.op_decrypt_result()
32 assert not result.unsupported_algorithm, \
33     "Unsupported algorithm: {}".format(result.unsupported_algorithm)
34
35 support.print_data(sink)
36
37 # Idiomatic interface.
38 with pyme.Context() as c:
39     plaintext, _, _ = c.decrypt(open(support.make_filename("cipher-1.asc")))
40     assert len(plaintext) > 0
41     assert plaintext.find(b'Wenn Sie dies lesen k') >= 0, \
42         'Plaintext not found'