httpbenchmark 0.5

Installation

Installation is easily accomplished using pip or easy_install.

pip install httpbenchmark

Basic Usage

A script named pb is provided and is intended to be a basic analog to the Apache ab script.

pb -n 1000 -c 25 http://my-host.tld/endpoint/

Worker code

By default HTTPBenchmark does nothing more than send a GET request and expect a 200 OK return code. To fully utilize the system you should create your own script that subclasses HTTPBenchmark.

Here’s an incomplete and untested example that is load testing some application that deals with users and friends. It aims to illustrate the fundamentals of using HTTPBenchmark

from httpbenchmark import HTTPBenchmark
from tornado import gen
import random

USER_IDS = [1111, 2222, 3333, 4444]

_url = lambda x: ''.join(['http://my-host.tld/', x])

class MyBenchmark(HTTPBenchmark):
    @gen.coroutine
    def worker(self):
        '''
        get_worker should return a callable that will be called by the async http client
        '''
        if random.choice([True, False]):
            yield self.new_user()
        else:
            yield self.returning_user()

    @gen.coroutine
    def new_user(self):
        user_id = random.choice(USER_IDS)
        self.log.debug("New user: %s" % user_id)

        friends = yield self.open_json(_url("register?uid=%d" % user_id))
        # ... handle registration response ...
        if failure:
            self.log.error("Indicate reason for failure")
            self.log.debug("Show debugging info if you want")
            self.finish_request(False)
        else:
            yield self.next_step(user_id, friends['friendList'][0])

    @gen.coroutine
    def returning_user(self):
        user_id = random.choice(USER_IDS)
        self.log.debug("Returning user: %s" % user_id)

        def handle_login(response, friends):
        friends = yield self.open_json(_url("login?uid=%d" % user_id))
        # ... handle login response ...
        if failure:
            self.log.error("Indicate reason for failure")
            self.log.debug("Show debugging info if you want")
            self.finish_request(False)
        else:
            yield self.next_step(user_id, friends['friendList'][0])

    @gen.coroutine
    def next_step(self, user_id, friend_id):
        # ... do something else ...
        if failure:
            self.log.error("Indicate reason for failure")
            self.log.debug("Show debugging info if you want")
            self.finish_request(False)
        else:
            # success!
            self.finish_request()

if __name__ == '__main__':
    MyBenchmark().main()

TODO

• Add some working examples