python: Add an idiomatic interface.
[gpgme.git] / lang / python / tests / support.py
1 # Copyright (C) 2016 g10 Code GmbH
2 #
3 # This file is part of GPGME.
4 #
5 # GPGME is free software; you can redistribute it and/or modify it
6 # under the terms of the GNU General Public License as published by
7 # the Free Software Foundation; either version 2 of the License, or
8 # (at your option) any later version.
9 #
10 # GPGME is distributed in the hope that it will be useful, but WITHOUT
11 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
12 # or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General
13 # Public License for more details.
14 #
15 # You should have received a copy of the GNU Lesser General Public
16 # License along with this program; if not, see <http://www.gnu.org/licenses/>.
17
18 import sys
19 import os
20 from pyme import core
21
22 # known keys
23 alpha = "A0FF4590BB6122EDEF6E3C542D727CC768697734"
24 bob = "D695676BDCEDCC2CDD6152BCFE180B1DA9E3B0B2"
25 encrypt_only = "F52770D5C4DB41408D918C9F920572769B9FE19C"
26 sign_only = "7CCA20CCDE5394CEE71C9F0BFED153F12F18F45D"
27
28 def make_filename(name):
29     return os.path.join(os.environ['top_srcdir'], 'tests', 'gpg', name)
30
31 def in_srcdir(name):
32     return os.path.join(os.environ['srcdir'], name)
33
34 def init_gpgme(proto):
35     core.engine_check_version(proto)
36
37 verbose = int(os.environ.get('verbose', 0)) > 1
38 def print_data(data):
39     if verbose:
40         try:
41             # See if it is a file-like object.
42             data.seek(0, os.SEEK_SET)
43             data = data.read()
44         except:
45             # Hope for the best.
46             pass
47         sys.stdout.buffer.write(data)
48
49 def mark_key_trusted(ctx, key):
50     class Editor(object):
51         def __init__(self):
52             self.steps = ["trust", "save"]
53         def edit(self, status, args, out):
54             if args == "keyedit.prompt":
55                 result = self.steps.pop(0)
56             elif args == "edit_ownertrust.value":
57                 result = "5"
58             elif args == "edit_ownertrust.set_ultimate.okay":
59                 result = "Y"
60             elif args == "keyedit.save.okay":
61                 result = "Y"
62             else:
63                 result = None
64             return result
65     with core.Data() as sink:
66         ctx.op_edit(key, Editor().edit, sink, sink)