client/tests/disktest/disktest.py

import test
from autotest_utils import *
import os, sys
from subprocess import *

class disktest(test.test):
	version = 1

	def setup(self):
		os.mkdir(self.srcdir)
		os.chdir(self.bindir)
		system('cp disktest.c src/')
		os.chdir(self.srcdir)
		cflags = '-D_FILE_OFFSET_BITS=64 -D _GNU_SOURCE -static -Wall'
		system('cc disktest.c ' + cflags + ' -o disktest')


	def test_one_disk_chunk(self, disk, chunk):
		print "testing %d MB files on %s in %d MB memory" % \
					(self.chunk_mb, disk, self.memory_mb)
		cmd = "%s/disktest -m %d -f %s/testfile.%d -i -S" % \
				(self.srcdir, self.chunk_mb, disk, chunk)
		p = Popen(cmd, shell=True)
		return(p.pid)


	def execute(self, disks = None, gigabytes = None,
						chunk_mb = memtotal() / 1024):
		os.chdir(self.srcdir)

		if not disks:
			disks = [self.tmpdir]
		if not gigabytes:
			free = 100       # cap it at 100GB by default
			for disk in disks:
				free = min(freespace(disk) / 1024**3, free)
			gigabytes = free
			print "resizing to %s GB" % gigabytes
			sys.stdout.flush()

		self.chunk_mb = chunk_mb
		self.memory_mb = memtotal()/1024
		if self.memory_mb > chunk_mb:
			e_msg = "Too much RAM (%dMB) for this test to work" % self.memory_mb
			raise TestError(e_msg)

		chunks = (1024 * gigabytes) / chunk_mb

		for i in range(chunks):
			pids = []
			for disk in disks:
				pid = self.test_one_disk_chunk(disk, i)
				pids.append(pid)
			errors = []
			for pid in pids:
				(junk, retval) = os.waitpid(pid, 0)
				if (retval != 0):
					errors.append(retval)
			if errors:
				raise TestError("Errors from children: %s" % errors)