from random import randint
from random import uniform
from locust import HttpLocust, TaskSet, task
from requestsCoinfit import coinfitParamPost
class Query(TaskSet):
# @task()
# def get_samples(self):
# response = self.client.get("/api/v2/depth.json?market=ltceth")
# print response.status_code
# print response.json()
# with self.client.get("/api/v2/depth.json?market=ltceth", catch_response=True) as response:
# if response.status_code == 404:
# response.success()
# response = self.client.post("/login", {"username": "testuser", "password": "secret"})
def on_start(self):
self.orders()
def user_key(self):
userKey = {
"user1":['axPhr8dI034nMEjVKxScjlgRmT1gSU8KTDOO7xu1','zoZRok8Mi03CiJmnxLHLxbVVICiLuvRRkxVlJ31p'],#harstaber163
"user2":['5HOzHz2Sg283ojvJoe6GZ78i1UxW4eBZHpXJnFVX','ffwW2qFEqzxVVemV4xCqa8MEZrRvGxZ2Bkq86kSB'], #harstanber126
"user3":['TxxaOw24S15OuhpzW9GN0BLG1usdayftmUXk8YGn','YPazdD9xHjktnZ8OYuIOlTxUccaUB6glu1tCVjWU'] #1079650300
}
data = randint(1,3)
username = 'user' + str(data)
print("###############username############:",username)
access_key = userKey[username][0]
secret_key = userKey[username][1]
return access_key,secret_key
def price_volume(self):
side = ['sell','buy']
data = randint(0,1)
volumn = round(uniform(0.01,1),2)
price = round(uniform(1,10),5)
return side[data], str(price), str(volumn)
@task()
def orders(self):
access_key, secret_key = self.user_key()
side, price, volume = self.price_volume()
parm = {
"market":"eoseth",
"side":side,
"volume":volume,
"price":price
}
parms = coinfitParamPost(access_key, secret_key, "/api/v2/orders.json", parm)
response = self.client.post("/api/v2/orders.json",parms)
print("######status_code:",response.status_code)
print("######status_code:",response.content)
class CeilometerQuery(HttpLocust):
task_set = Query
host = 'https://xxxx.xxxx.xxx'
min_wait = 2000
max_wait = 3000