| |
| # Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| |
| import os, shutil, re |
| from autotest_lib.client.bin import utils, test |
| |
| class ddtest(test.test): |
| version = 2 |
| |
| |
| def setup(self, tarball='ddtest.tar.gz'): |
| tarball = utils.unmap_url(self.bindir, tarball, self.tmpdir) |
| utils.extract_tarball_to_dir(tarball, self.srcdir) |
| os.chdir(self.srcdir) |
| utils.system('make build') |
| |
| |
| def initialize(self): |
| self.job.require_gcc() |
| self.results = [] |
| self.job.drop_caches_between_iterations = True |
| |
| |
| def run_once(self, dir=None, blocksize=1024, blocknum=262144, threads=20): |
| if not dir: |
| dir = os.path.join(self.srcdir, 'rdir') |
| shutil.rmtree(dir, True) |
| os.mkdir(dir) |
| args = '-D ' + dir |
| args += ' -b %d' % blocksize |
| args += ' -n %d' % blocknum |
| args += ' -t %d' % threads |
| self.results.append(utils.system_output(os.path.join(self.srcdir, |
| 'ddtest') + ' ' + args)) |
| |
| |
| def postprocess(self): |
| pattern = re.compile(r"throughput is (.*?) MB/sec") |
| for throughput in pattern.findall("\n".join(self.results)): |
| self.write_perf_keyval({'throughput':throughput}) |
