@corrupt_delay decorator

chaos_lib.py

def corrupt_delay(func):
    def latency(*args, **kw):
        delay, rate = get_config('delay')
        if not delay:
            return func(*args, **kw)
        print("delay: {0}, rate: {1}".format(delay, rate))
        # if delay and rate exist, delaying with that value at that rate
        start = time.time()
        if delay > 0 and rate >= 0:

            # add latency approx rate% of the time
            if random.random() <= rate:
                time.sleep(delay / 1000.0)

        result = func(*args, **kw)
        end = time.time()

        print('Added {1:.2f}ms to {0:s}'.format(
            func.__name__,
            (end - start) * 1000
        ))
        return result
return latency