sleep_context.py

import time 

from rally import consts
from rally import exceptions
from rally.task import context


@context.configure(name="sleep_context", order=100500)
class SleepContext(context.Context):
    """Dummy context."""
    CONFIG_SCHEMA = {
        "type": "object",
        "$schema": consts.JSON_SCHEMA,
        "properties": {
            "sleep": {"type": "number"},
        },
    }

    def setup(self):
        time.sleep(self.config.get("sleep", 0))

    def cleanup(self):
        pass