1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
# This script is used to mass register users
# ATTENTION: this script does not log
# the user in, so key creation will only
# happen on the first login
#
# You can run this with the following command:
# python create_users.py -n<number of users> -p<provider> -i<invite_code>
import argparse
import os
import tempfile
from leap.auth import SRPAuth
from leap.exceptions import SRPAuthenticationError
import requests
from requests.packages.urllib3.exceptions import InsecureRequestWarning
requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
class User(object):
def __init__(self, number, provider, certificate):
self._username = 'loadtest%d' % number
self._password = 'password_%d' % number
self._set_srp_auth(provider, certificate)
def _set_srp_auth(self, leap_provider, certificate):
leap_api_server = 'https://api.%s:4430' % leap_provider
self._srp_auth = SRPAuth(leap_api_server, certificate)
def get_or_create_user(self, invite_code=None):
try:
self.authenticate()
except SRPAuthenticationError:
self.register(invite_code)
return self._username, self._password
def authenticate(self):
self._srp_auth.authenticate(self._username, self._password)
def register(self, invite_code=None):
self._srp_auth.register(self._username, self._password, invite_code)
def mass_register(number, invite_code, provider, certificate):
for index in xrange(1, number + 1):
User(index, provider, certificate).register(invite_code)
print 'done registering loadtest%d' % index
def fetch_leap_certificate(leap_provider):
_, certificate_path = tempfile.mkstemp()
certificate = requests.get('https://%s/ca.crt' % leap_provider)
with open(certificate_path, 'w') as cert:
cert.write('%s' % certificate.content)
return certificate_path
def _parse_args():
parser = argparse.ArgumentParser()
parser.add_argument('--number', '-n', default=1, type=int, help='the number of user to be registered')
parser.add_argument('--invite-code', '-i', default=None, help='invite code')
parser.add_argument('--provider', '-p', default=None, help='leap provider e.g. unstable.pix.org')
return parser.parse_args()
if __name__ == '__main__':
args = _parse_args()
certificate_file = fetch_leap_certificate(args.provider)
mass_register(args.number, args.invite_code, args.provider, certificate_file)
|
