client/tests/selftest/selftest.py

import os
from autotest_lib.client.bin import test
from autotest_lib.client.common_lib import error


class selftest(test.test):
	version = 1

	def setup(self):
		name = self.job.resultdir + '/sequence'
		if (not os.path.exists(name)):
			fd = file(name, 'w')
			fd.write('0')
			fd.close()
	
	def __mark(self, checkpoint):
		name = self.job.resultdir + '/sequence'
		fd = file(name, 'r')
		current = int(fd.readline())
		fd.close()

		current += 1
		fd = file(name + '.new', 'w')
		fd.write('%d' % current)
		fd.close()

		os.rename(name + '.new', name)

		print "checkpoint %d %d" % (current, checkpoint)

		if (current != checkpoint):
			raise error.JobError("selftest: sequence was " +
				"%d when %d expected" % (current, checkpoint))

	def __throw(self):
		__does_not_exist = __does_not_exist_either

	def __print(self, msg):
		sys.stdout.write(msg)

	def __warn(self, msg):
		sys.stderr.write(msg)

	def execute(self, cmd, *args):
		if cmd == 'mark':
			self.__mark(*args)
		elif cmd == 'throw':
			self.__throw(*args)
		elif cmd == 'print':
			self.__print(*args)
		elif cmd == 'warn':
			self.__warn(*args)