client/tests/iozone/iozone.py

#!/usr/bin/python
import os
from autotest_lib.client.bin import test, autotest_utils
from autotest_lib.client.common_lib import utils


class iozone(test.test):
	version = 1

	# http://www.iozone.org/src/current/iozone3_283.tar
	def setup(self, tarball = 'iozone3_283.tar'):
		tarball = utils.unmap_url(self.bindir, tarball,
		                                   self.tmpdir)
		autotest_utils.extract_tarball_to_dir(tarball, self.srcdir)
		os.chdir(os.path.join(self.srcdir, 'src/current'))

		arch = autotest_utils.get_current_kernel_arch()
		if (arch == 'ppc'):
			utils.system('make linux-powerpc')
		elif (arch == 'ppc64'):
			utils.system('make linux-powerpc64')
		elif (arch == 'x86_64'):
			utils.system('make linux-AMD64')
		else: 
			utils.system('make linux')


	def execute(self, dir = None, iterations=1, args = None):
		self.keyval = open(os.path.join(self.resultsdir, 'keyval'),
		                   'w')
		if not dir:
			dir = self.tmpdir
		os.chdir(dir)
		if not args:
			args = '-a'
		profilers = self.job.profilers
		if not profilers.only():
			for i in range(iterations):
				output = utils.system_output('%s/src/current/iozone %s' %
				                       (self.srcdir, args))
				self.__format_results(output)

		# Do a profiling run if necessary
		if profilers.present():
			profilers.start(self)
			output = utils.system_output('%s/src/current/iozone %s' %
			                       (self.srcdir, args))
			self.__format_results(output)
			profilers.stop(self)
			profilers.report(self)

		self.keyval.close()


	def __format_results(self, results):
		labels = ('write', 'rewrite', 'read', 'reread', 'randread',
			  'randwrite', 'bkwdread', 'recordrewrite',
			  'strideread', 'fwrite', 'frewrite',
			  'fread', 'freread')
		for line in results.splitlines():
			fields = line.split()
			if len(fields) != 15:
				continue
			try:
				fields = tuple([int(i) for i in fields])
			except ValueError:
				continue
			for l, v in zip(labels, fields[2:]):
				print >> self.keyval, "%d-%d-%s=%d" % (fields[0], fields[1], l, v)
		print >> self.keyval