Decorator for Grinder thread-rampup

rampup.py

from net.grinder.script.Grinder import grinder

def thread_rampup(interval):
    """Decorator for waking grinder threads incrementally. 
    Input argument `interval` determines time in ms between wakeup of each thread.
    
    Usage:

        class TestRunner:
            @thread_rampup(3000)
            def __call__(self):
                # do some testing
    """
    def decorator(fn):
        def wrapper(*args, **kwargs):
            if grinder.runNumber == 0: 
                sleepTime = grinder.threadNumber * interval
                grinder.sleep(sleepTime, 0)
            return fn(*args, **kwargs)
        return wrapper
    return decorator