tools/bench.py

import json, requests
from multiprocessing import Pool

clients = []


BASE_URL = "http://127.0.0.1:8000/hiccup/"


def create_client():
    ret_client = {}
    register_url = "api/v1/devices/register/"
    params = {"board_date": "2017-01-01", "chipset": "HICCUPBENCH"}
    resp = requests.post(BASE_URL + register_url, params)
    data = json.loads(resp.text)
    ret_client["token"] = data["token"]
    ret_client["uuid"] = data["uuid"]
    return ret_client


def send_heartbeat(client):
    heartbeat_url = "api/v1/heartbeats/"
    params = {
        "uuid": client["uuid"],
        "build_fingerprint": "HICCUPBENCH",
        "uptime": "string",
        "date": "1984-06-02T19:05:00.000Z",
        "app_version": 2000000,
    }
    resp = requests.post(
        BASE_URL + heartbeat_url,
        params,
        headers={"Authorization": "Token " + client["token"]},
    )
    return resp


import time


def bench(client):
    start_time = time.time()
    for i in range(50):
        send_heartbeat(client)
    end_time = time.time()
    print(end_time - start_time)


clients = []
for i in range(20):
    clients = clients + [create_client()]

with Pool(20) as p:
    p.map(bench, clients)