python: Port more tests.
[gpgme.git] / lang / python / tests / t-encrypt-large.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 sys
21 import random
22 from pyme import core, constants
23 import support
24
25 if len(sys.argv) == 2:
26     nbytes = int(sys.argv[1])
27 else:
28     nbytes = 100000
29
30 support.init_gpgme(constants.PROTOCOL_OpenPGP)
31 c = core.Context()
32
33 ntoread = nbytes
34 def read_cb(amount):
35     global ntoread
36     chunk = ntoread if ntoread < amount else amount
37     ntoread -= chunk
38     assert ntoread >= 0
39     assert chunk >= 0
40     return bytes(random.randrange(256) for i in range(chunk))
41
42 nwritten = 0
43 def write_cb(data):
44     global nwritten
45     nwritten += len(data)
46     return len(data)
47
48 source = core.Data(cbs=(read_cb, None, None, lambda: None))
49 sink = core.Data(cbs=(None, write_cb, None, lambda: None))
50
51 keys = []
52 keys.append(c.get_key("A0FF4590BB6122EDEF6E3C542D727CC768697734", False))
53 keys.append(c.get_key("D695676BDCEDCC2CDD6152BCFE180B1DA9E3B0B2", False))
54
55 c.op_encrypt(keys, constants.ENCRYPT_ALWAYS_TRUST, source, sink)
56 result = c.op_encrypt_result()
57 assert not result.invalid_recipients, \
58     "Invalid recipient encountered: {}".format(result.invalid_recipients.fpr)
59 assert ntoread == 0
60
61 if support.verbose:
62     sys.stderr.write(
63         "plaintext={} bytes, ciphertext={} bytes\n".format(nbytes, nwritten))