client/tests/tbench/tbench.py

import time, os, signal, re
from autotest_lib.client.bin import test, autotest_utils
from autotest_lib.client.common_lib import utils


class tbench(test.test):
    version = 2

    def initialize(self):
        self.job.require_gcc()


    # http://samba.org/ftp/tridge/dbench/dbench-3.04.tar.gz
    def setup(self, tarball = 'dbench-3.04.tar.gz'):
        tarball = utils.unmap_url(self.bindir, tarball, self.tmpdir)
        autotest_utils.extract_tarball_to_dir(tarball, self.srcdir)
        os.chdir(self.srcdir)

        utils.system('./configure')
        utils.system('make')


    def execute(self, iterations = 1, nprocs = None, args = ''):
        # only supports combined server+client model at the moment
        # should support separate I suppose, but nobody uses it
        if not nprocs:
            nprocs = self.job.cpu_count()
        args += ' %s' % nprocs
        results = []
        profilers = self.job.profilers
        if not profilers.only():
            for i in range(iterations):
                results.append(self.run_tbench(args))

        # Do a profiling run if necessary
        if profilers.present():
            profilers.start(self)
            results.append(self.run_tbench(args))
            profilers.stop(self)
            profilers.report(self)

        self.__format_results("\n".join(results))


    def run_tbench(self, args):
        pid = os.fork()
        if pid:                         # parent
            time.sleep(1)
            client = self.srcdir + '/client.txt'
            args = '-c ' + client + ' ' + '%s' % args
            cmd = os.path.join(self.srcdir, "tbench") + " " + args
            results = utils.system_output(cmd, retain_output=True)
            os.kill(pid, signal.SIGTERM)    # clean up the server
        else:                           # child
            server = self.srcdir + '/tbench_srv'
            os.execlp(server, server)
        return results


    def __format_results(self, results):
        out = open(self.resultsdir + '/keyval', 'w')
        pattern = re.compile(r"Throughput (.*?) MB/sec (.*?) procs")
        for result in pattern.findall(results):
            print >> out, "throughput=%s\nprocs=%s\n" % result
        out.close()