-
Notifications
You must be signed in to change notification settings - Fork 0
/
locustfile.py
92 lines (75 loc) · 2.98 KB
/
locustfile.py
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Author: Igor Barinov <igorbarinov@me.com>
from locust import HttpLocust, TaskSet, task
import requests
from ConfigParser import SafeConfigParser
import datetime
import pickle
import uuid
import hashlib
import json
import base64
from Crypto.Hash import SHA256
import gzip
# setup loggging
import logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
# load configuration
parser = SafeConfigParser()
parser.read('load.ini')
AL_JOURNAL = parser.get('general', 'journal')
AL_HOST = parser.get('general', 'host')
AL_AUTH = parser.get('general','auth')
SIGNED_VOTES = 'data/signed_votes.p.gz'
"""
Run this script in standalone mode
locust --host=http://167.114.247.67:8080
or use it as master node. You have to
locust --master-host=167.114.247.67:8080
or use it as slave and connect to a master
locust --slave --master-host=167.114.247.67:8080
"""
def load_zipped_pickle(filename):
with gzip.open(filename, 'rb') as f:
loaded_object = pickle.load(f)
return loaded_object
signed_votes = load_zipped_pickle(SIGNED_VOTES)
class UserBehavior(TaskSet):
def on_start(self):
self.i = 0
logger.info('Loading signed votes..')
self.max = len(signed_votes)
""" on_start is called when a Locust start before any task is scheduled """
self.client.headers = {"Authorization":AL_AUTH,'Content-Type':'application/json'}
self.s = requests.Session()
self.s.headers.update({'Authorization': AL_AUTH, 'Content-Type':'application/json'})
# weight for task in ()
@task(1)
def timestamp(self):
self.s.post(AL_HOST+'/api/journals/' + AL_JOURNAL +'/timestamp')
print "timestamp journal " + str(datetime.datetime.now())
@task(100)
def create_record(self):
logger.info('Creating a new record..')
r = self.client.post("/api/records", '{}')
if self.i > self.max:
self.i = 0
self.vote = signed_votes[self.i]
# add fingerprint
record_id = r.json()['id']
logger.info('Adding a new fingerprint..')
self.s.post(AL_HOST + "/api/records/" + record_id + "/fingerprints", json.dumps({"metadata": base64.b64encode(json.dumps(self.vote)),
"metadataContentType":"application/json;enc=v1",
"metadataHash": hashlib.sha256(json.dumps(self.vote)).hexdigest(),
"nonce": base64.b64encode(str(uuid.uuid4()))}))
# commit the record to a journal
logger.info('Commiting the record..')
self.s.post(AL_HOST+'/api/journals/' + AL_JOURNAL + '/commit/' + record_id, '{}')
logger.info('Finished adding a record')
self.i += 1
class WebsiteUser(HttpLocust):
task_set = UserBehavior
min_wait=5000
max_wait=9000