python: Fix skipping tests if running with GnuPG < 2.1.12.
[gpgme.git] / lang / python / tests / t-edit.py
1 #!/usr/bin/env python
2
3 # Copyright (C) 2005 Igor Belyi <belyi@users.sourceforge.net>
4 # Copyright (C) 2016 g10 Code GmbH
5 #
6 # This file is part of GPGME.
7 #
8 # GPGME is free software; you can redistribute it and/or modify it
9 # under the terms of the GNU General Public License as published by
10 # the Free Software Foundation; either version 2 of the License, or
11 # (at your option) any later version.
12 #
13 # GPGME is distributed in the hope that it will be useful, but WITHOUT
14 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
15 # or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General
16 # Public License for more details.
17 #
18 # You should have received a copy of the GNU Lesser General Public
19 # License along with this program; if not, see <http://www.gnu.org/licenses/>.
20
21 from __future__ import absolute_import, print_function, unicode_literals
22 del absolute_import, print_function, unicode_literals
23
24 import sys
25 import os
26 import gpg
27 import support
28 _ = support # to appease pyflakes.
29
30 class KeyEditor(object):
31     def __init__(self):
32         self.steps = ["fpr", "expire", "1", "primary", "quit"]
33         self.step = 0
34         self.done = False
35         self.verbose = int(os.environ.get('verbose', 0)) > 1
36
37     def edit_fnc(self, status, args, out=None):
38         if args == "keyedit.prompt":
39             result = self.steps[self.step]
40             self.step += 1
41         elif args == "keyedit.save.okay":
42             result = "Y"
43             self.done = self.step == len(self.steps)
44         elif args == "keygen.valid":
45             result = "0"
46         else:
47             result = None
48
49         if self.verbose:
50             sys.stderr.write("Code: {}, args: {!r}, Returning: {!r}\n"
51                              .format(status, args, result))
52
53         return result
54
55 c = gpg.Context()
56 c.set_pinentry_mode(gpg.constants.PINENTRY_MODE_LOOPBACK)
57 c.set_passphrase_cb(lambda *args: "abc")
58 c.set_armor(True)
59
60 # The deprecated interface.
61 editor = KeyEditor()
62 c.interact(c.get_key("A0FF4590BB6122EDEF6E3C542D727CC768697734", False),
63            editor.edit_fnc)
64 assert editor.done
65
66 # The deprecated interface.
67 sink = gpg.Data()
68 editor = KeyEditor()
69 c.op_edit(c.get_key("A0FF4590BB6122EDEF6E3C542D727CC768697734", False),
70           editor.edit_fnc, sink, sink)
71 assert editor.done