python: Support the Assuan engine.
[gpgme.git] / lang / python / tests / t-protocol-assuan.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
22 with pyme.Context(protocol=pyme.constants.PROTOCOL_ASSUAN) as c:
23     # Do nothing.
24     c.assuan_transact('nop')
25     c.assuan_transact('NOP')
26     c.assuan_transact(['NOP'])
27
28     err = c.assuan_transact('idontexist')
29     assert err.getsource() == pyme.errors.SOURCE_GPGAGENT
30     assert err.getcode() == pyme.errors.ASS_UNKNOWN_CMD
31
32     # Invoke the pinentry to get a confirmation.
33     c.assuan_transact(['GET_CONFIRMATION', 'Hello there'])
34
35     data = []
36     def data_cb(line):
37         data.append(line)
38
39     err = c.assuan_transact(['GETINFO', 'version'], data_cb=data_cb)
40     assert not err
41     assert len(data) == 1
42
43     data = []
44     err = c.assuan_transact(['GETINFO', 's2k_count'], data_cb=data_cb)
45     if not err:
46         assert len(data) == 1
47         assert int(data[0]) > 0
48
49     # XXX HELP sends status lines if we could use ASSUAN_CONVEY_COMMENTS.
50
51     status = []
52     def status_cb(line, args):
53         status.append((line, args))
54
55     alphas_grip = '76F7E2B35832976B50A27A282D9B87E44577EB66'
56     err = c.assuan_transact(['KEYINFO', alphas_grip], status_cb=status_cb)
57     if not err:
58         assert len(status) == 1
59         line, args = status[0]
60         assert line.startswith('KEYINFO')
61         assert args.startswith(alphas_grip)
62
63     # XXX: test these callbacks, e.g. using PRESET_PASSPHRASE
64     # XXX: once issue2428 is resolved
65     def inq_cb(name, args):
66         print("inq_cb", name, args)