httpbenchmark 0.3.1

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
import random

USER_IDS = [1111, 2222, 3333, 4444]

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

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

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

        def handle_register(response, friends):
            # ... handle registration response ...

            if failure:
                self.log.error("Indicate reason for failure")
                self.log.debug("Show debugging info if you want")
                return self.finish_request(False)

            self.next_step(user_id, friends['friendList'][0])

        self.open_json(_url("register?uid=%d" % user_id), handle_register)

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

        def handle_login(response, friends):
            # ... handle login response ...

            if failure:
                self.log.error("Indicate reason for failure")
                self.log.debug("Show debugging info if you want")
                return self.finish_request(False)

            self.next_step(user_id, friends['friendList'][0])

       self.open_json(_url("login?uid=%d" % user_id), handle_login)

    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")
            return self.finish_request(False)

        # success!
        return self.finish_request(True)

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

TODO

• Add some working examples

• Upgrade Tornado