blob: 4c5aaf19c2cfa0640c32a62930e0ec3462abf3cd [file] [log] [blame]
mblighc86b0b42006-07-28 17:35:28 +00001"""The main job wrapper
mbligha2508052006-05-28 21:29:53 +00002
mblighc86b0b42006-07-28 17:35:28 +00003This is the core infrastructure.
4"""
5
6__author__ = """Copyright Andy Whitcroft, Martin J. Bligh 2006"""
mbligha2508052006-05-28 21:29:53 +00007
mbligh8f243ec2006-10-10 05:55:49 +00008# standard stuff
mbligh366ff1b2008-04-25 16:07:56 +00009import os, sys, re, pickle, shutil, time, traceback, types, copy
mbligh302482e2008-05-01 20:06:16 +000010
mbligh8f243ec2006-10-10 05:55:49 +000011# autotest stuff
mbligh302482e2008-05-01 20:06:16 +000012from autotest_lib.client.bin import autotest_utils
13from autotest_lib.client.common_lib import error, barrier, logging
14
15import parallel, kernel, xen, test, profilers, filesystem, fd_stack, boottool
16import harness, config, sysinfo, cpuset
17
mblighf4c35322006-03-13 01:01:10 +000018
mbligh12a04cb2008-04-25 16:07:20 +000019
mbligh366ff1b2008-04-25 16:07:56 +000020JOB_PREAMBLE = """
21from common.error import *
22from autotest_utils import *
23"""
24
25
mbligh302482e2008-05-01 20:06:16 +000026class StepError(error.AutotestError):
mbligh12a04cb2008-04-25 16:07:20 +000027 pass
28
29
mblighcaa62c22008-04-07 21:51:17 +000030class base_job:
mblighc86b0b42006-07-28 17:35:28 +000031 """The actual job against which we do everything.
32
33 Properties:
mbligh72b88fc2006-12-16 18:41:35 +000034 autodir
mblighc86b0b42006-07-28 17:35:28 +000035 The top level autotest directory (/usr/local/autotest).
36 Comes from os.environ['AUTODIR'].
mbligh72b88fc2006-12-16 18:41:35 +000037 bindir
mblighc86b0b42006-07-28 17:35:28 +000038 <autodir>/bin/
mblighd5a38832008-01-25 18:15:39 +000039 libdir
40 <autodir>/lib/
mbligh72b88fc2006-12-16 18:41:35 +000041 testdir
mblighc86b0b42006-07-28 17:35:28 +000042 <autodir>/tests/
mbligh84bafdb2008-01-26 19:43:34 +000043 site_testdir
44 <autodir>/site_tests/
mblighc86b0b42006-07-28 17:35:28 +000045 profdir
46 <autodir>/profilers/
47 tmpdir
48 <autodir>/tmp/
49 resultdir
50 <autodir>/results/<jobtag>
51 stdout
52 fd_stack object for stdout
53 stderr
54 fd_stack object for stderr
55 profilers
56 the profilers object for this job
apw504a7dd2006-10-12 17:18:37 +000057 harness
58 the server harness object for this job
apw059e1b12006-10-12 17:18:26 +000059 config
60 the job configuration for this job
mblighc86b0b42006-07-28 17:35:28 +000061 """
62
mblighd528d302007-12-19 16:19:05 +000063 DEFAULT_LOG_FILENAME = "status"
64
mblighcaa62c22008-04-07 21:51:17 +000065 def __init__(self, control, jobtag, cont, harness_type=None,
66 use_external_logging = False):
mblighc86b0b42006-07-28 17:35:28 +000067 """
68 control
69 The control file (pathname of)
70 jobtag
71 The job tag string (eg "default")
apw96da1a42006-11-02 00:23:18 +000072 cont
73 If this is the continuation of this job
apwe68a7132006-12-01 11:21:37 +000074 harness_type
75 An alternative server harness
mblighc86b0b42006-07-28 17:35:28 +000076 """
mblighf4c35322006-03-13 01:01:10 +000077 self.autodir = os.environ['AUTODIR']
apw870988b2007-09-25 16:50:53 +000078 self.bindir = os.path.join(self.autodir, 'bin')
mblighd5a38832008-01-25 18:15:39 +000079 self.libdir = os.path.join(self.autodir, 'lib')
apw870988b2007-09-25 16:50:53 +000080 self.testdir = os.path.join(self.autodir, 'tests')
mbligh84bafdb2008-01-26 19:43:34 +000081 self.site_testdir = os.path.join(self.autodir, 'site_tests')
apw870988b2007-09-25 16:50:53 +000082 self.profdir = os.path.join(self.autodir, 'profilers')
83 self.tmpdir = os.path.join(self.autodir, 'tmp')
84 self.resultdir = os.path.join(self.autodir, 'results', jobtag)
mbligh0fb83972008-01-10 16:30:02 +000085 self.sysinfodir = os.path.join(self.resultdir, 'sysinfo')
mbligh8d83cdc2007-12-03 18:09:18 +000086 self.control = os.path.abspath(control)
mbligh366ff1b2008-04-25 16:07:56 +000087 self.state_file = self.control + '.state'
jadmanskia9c75c42008-05-01 22:05:31 +000088 self.__load_state()
mbligha2508052006-05-28 21:29:53 +000089
apw96da1a42006-11-02 00:23:18 +000090 if not cont:
91 if os.path.exists(self.tmpdir):
mbligh302482e2008-05-01 20:06:16 +000092 cmd = ('umount -f %s > /dev/null 2> /dev/null'
93 % (self.tmpdir))
mbligh8ea61e22008-05-09 18:09:37 +000094 autotest_utils.system(cmd, ignore_status=True)
mbligh302482e2008-05-01 20:06:16 +000095 autotest_utils.system('rm -rf ' + self.tmpdir)
apw96da1a42006-11-02 00:23:18 +000096 os.mkdir(self.tmpdir)
97
apw870988b2007-09-25 16:50:53 +000098 results = os.path.join(self.autodir, 'results')
99 if not os.path.exists(results):
100 os.mkdir(results)
mblighfbfb77d2007-02-15 18:54:03 +0000101
apwf3d28622007-09-25 16:49:17 +0000102 download = os.path.join(self.testdir, 'download')
103 if os.path.exists(download):
mbligh302482e2008-05-01 20:06:16 +0000104 autotest_utils.system('rm -rf ' + download)
apwf3d28622007-09-25 16:49:17 +0000105 os.mkdir(download)
106
apw96da1a42006-11-02 00:23:18 +0000107 if os.path.exists(self.resultdir):
mbligh302482e2008-05-01 20:06:16 +0000108 autotest_utils.system('rm -rf '
109 + self.resultdir)
apw96da1a42006-11-02 00:23:18 +0000110 os.mkdir(self.resultdir)
mbligh0fb83972008-01-10 16:30:02 +0000111 os.mkdir(self.sysinfodir)
apw96da1a42006-11-02 00:23:18 +0000112
apw870988b2007-09-25 16:50:53 +0000113 os.mkdir(os.path.join(self.resultdir, 'debug'))
114 os.mkdir(os.path.join(self.resultdir, 'analysis'))
apw870988b2007-09-25 16:50:53 +0000115
mbligh8d83cdc2007-12-03 18:09:18 +0000116 shutil.copyfile(self.control,
117 os.path.join(self.resultdir, 'control'))
mblighf4ca14f2008-03-03 16:03:05 +0000118
mbligh4b089662006-06-14 22:34:58 +0000119
apwecf41b72006-03-31 14:00:55 +0000120 self.control = control
mbligh27113602007-10-31 21:07:51 +0000121 self.jobtag = jobtag
mblighd528d302007-12-19 16:19:05 +0000122 self.log_filename = self.DEFAULT_LOG_FILENAME
mbligh68119582008-01-25 18:16:41 +0000123 self.container = None
mblighf4c35322006-03-13 01:01:10 +0000124
mbligh56f1fbb2006-10-01 15:10:56 +0000125 self.stdout = fd_stack.fd_stack(1, sys.stdout)
126 self.stderr = fd_stack.fd_stack(2, sys.stderr)
jadmanskia9c75c42008-05-01 22:05:31 +0000127
128 self._init_group_level()
mblighf4c35322006-03-13 01:01:10 +0000129
apw059e1b12006-10-12 17:18:26 +0000130 self.config = config.config(self)
131
apwd27e55f2006-12-01 11:22:08 +0000132 self.harness = harness.select(harness_type, self)
133
mbligha35553b2006-04-23 15:52:25 +0000134 self.profilers = profilers.profilers(self)
mbligh72905562006-05-25 01:30:49 +0000135
mblighcaa605c2006-10-02 00:37:35 +0000136 try:
apw90154af2006-12-01 11:23:36 +0000137 tool = self.config_get('boottool.executable')
138 self.bootloader = boottool.boottool(tool)
mblighcaa605c2006-10-02 00:37:35 +0000139 except:
140 pass
141
mbligh0fb83972008-01-10 16:30:02 +0000142 sysinfo.log_per_reboot_data(self.sysinfodir)
mbligh3a6d6ca2006-04-23 15:50:24 +0000143
mbligh30270302007-11-05 20:33:52 +0000144 if not cont:
mblighc3430162007-11-14 23:57:19 +0000145 self.record('START', None, None)
jadmanskia9c75c42008-05-01 22:05:31 +0000146 self._increment_group_level()
apw357f50f2006-12-01 11:22:39 +0000147
apwf91efaf2007-11-24 17:32:13 +0000148 self.harness.run_start()
mblighcaa62c22008-04-07 21:51:17 +0000149
150 if use_external_logging:
151 self.enable_external_logging()
apwf91efaf2007-11-24 17:32:13 +0000152
jadmanski8415f962008-05-06 20:38:53 +0000153 # load the max disk usage rate - default to no monitoring
154 self.max_disk_usage_rate = self.get_state('__monitor_disk',
155 default=0.0)
156
157
158 def monitor_disk_usage(self, max_rate):
159 """\
160 Signal that the job should monitor disk space usage on /
161 and generate a warning if a test uses up disk space at a
162 rate exceeding 'max_rate'.
163
164 Parameters:
165 max_rate - the maximium allowed rate of disk consumption
166 during a test, in MB/hour, or 0 to indicate
167 no limit.
168 """
169 self.set_state('__monitor_disk', max_rate)
170 self.max_disk_usage_rate = max_rate
171
mbligh0692e472007-08-30 16:07:53 +0000172
173 def relative_path(self, path):
174 """\
175 Return a patch relative to the job results directory
176 """
mbligh1c250ca2007-08-30 16:31:38 +0000177 head = len(self.resultdir) + 1 # remove the / inbetween
178 return path[head:]
mbligh0692e472007-08-30 16:07:53 +0000179
180
mbligh362ab3d2007-08-30 11:24:04 +0000181 def control_get(self):
182 return self.control
183
mblighcaa605c2006-10-02 00:37:35 +0000184
mbligh8d83cdc2007-12-03 18:09:18 +0000185 def control_set(self, control):
186 self.control = os.path.abspath(control)
187
188
apwde1503a2006-10-10 08:34:21 +0000189 def harness_select(self, which):
190 self.harness = harness.select(which, self)
191
192
apw059e1b12006-10-12 17:18:26 +0000193 def config_set(self, name, value):
194 self.config.set(name, value)
195
196
197 def config_get(self, name):
198 return self.config.get(name)
199
mbligh8baa2ea2006-12-17 23:01:24 +0000200 def setup_dirs(self, results_dir, tmp_dir):
mbligh1e8858e2006-11-24 22:18:35 +0000201 if not tmp_dir:
apw870988b2007-09-25 16:50:53 +0000202 tmp_dir = os.path.join(self.tmpdir, 'build')
mbligh1e8858e2006-11-24 22:18:35 +0000203 if not os.path.exists(tmp_dir):
204 os.mkdir(tmp_dir)
205 if not os.path.isdir(tmp_dir):
mbligh642b03e2008-01-14 16:53:15 +0000206 e_msg = "Temp dir (%s) is not a dir - args backwards?" % self.tmpdir
207 raise ValueError(e_msg)
mbligh1e8858e2006-11-24 22:18:35 +0000208
209 # We label the first build "build" and then subsequent ones
210 # as "build.2", "build.3", etc. Whilst this is a little bit
211 # inconsistent, 99.9% of jobs will only have one build
212 # (that's not done as kernbench, sparse, or buildtest),
213 # so it works out much cleaner. One of life's comprimises.
214 if not results_dir:
215 results_dir = os.path.join(self.resultdir, 'build')
216 i = 2
217 while os.path.exists(results_dir):
218 results_dir = os.path.join(self.resultdir, 'build.%d' % i)
mblighd9223fc2006-11-26 17:19:54 +0000219 i += 1
mbligh1e8858e2006-11-24 22:18:35 +0000220 if not os.path.exists(results_dir):
221 os.mkdir(results_dir)
mbligh72b88fc2006-12-16 18:41:35 +0000222
mbligh8baa2ea2006-12-17 23:01:24 +0000223 return (results_dir, tmp_dir)
224
225
226 def xen(self, base_tree, results_dir = '', tmp_dir = '', leave = False, \
227 kjob = None ):
228 """Summon a xen object"""
229 (results_dir, tmp_dir) = self.setup_dirs(results_dir, tmp_dir)
230 build_dir = 'xen'
231 return xen.xen(self, base_tree, results_dir, tmp_dir, build_dir, leave, kjob)
232
233
234 def kernel(self, base_tree, results_dir = '', tmp_dir = '', leave = False):
235 """Summon a kernel object"""
mbligh669caa12007-11-05 18:32:13 +0000236 (results_dir, tmp_dir) = self.setup_dirs(results_dir, tmp_dir)
mbligh8baa2ea2006-12-17 23:01:24 +0000237 build_dir = 'linux'
mbligh6ee7ee02007-11-13 23:49:05 +0000238 return kernel.auto_kernel(self, base_tree, results_dir,
239 tmp_dir, build_dir, leave)
mblighf4c35322006-03-13 01:01:10 +0000240
mblighcaa605c2006-10-02 00:37:35 +0000241
mbligh6b504ff2007-12-12 21:03:49 +0000242 def barrier(self, *args, **kwds):
mblighfadca202006-09-23 04:40:01 +0000243 """Create a barrier object"""
mbligh6b504ff2007-12-12 21:03:49 +0000244 return barrier.barrier(*args, **kwds)
mblighfadca202006-09-23 04:40:01 +0000245
mblighcaa605c2006-10-02 00:37:35 +0000246
mbligh4b089662006-06-14 22:34:58 +0000247 def setup_dep(self, deps):
mblighc86b0b42006-07-28 17:35:28 +0000248 """Set up the dependencies for this test.
249
250 deps is a list of libraries required for this test.
251 """
mbligh4b089662006-06-14 22:34:58 +0000252 for dep in deps:
253 try:
apw870988b2007-09-25 16:50:53 +0000254 os.chdir(os.path.join(self.autodir, 'deps', dep))
mbligh302482e2008-05-01 20:06:16 +0000255 autotest_utils.system('./' + dep + '.py')
mbligh4b089662006-06-14 22:34:58 +0000256 except:
mbligh302482e2008-05-01 20:06:16 +0000257 err = "setting up dependency " + dep + "\n"
258 raise error.UnhandledError(err)
mbligh4b089662006-06-14 22:34:58 +0000259
260
mbligh72b88fc2006-12-16 18:41:35 +0000261 def __runtest(self, url, tag, args, dargs):
262 try:
mbligh53c41502007-10-23 20:45:04 +0000263 l = lambda : test.runtest(self, url, tag, args, dargs)
mbligh302482e2008-05-01 20:06:16 +0000264 pid = parallel.fork_start(self.resultdir, l)
265 parallel.fork_waitfor(self.resultdir, pid)
266 except error.AutotestError:
mbligh72b88fc2006-12-16 18:41:35 +0000267 raise
268 except:
mbligh302482e2008-05-01 20:06:16 +0000269 raise error.UnhandledError('running test ' + \
mbligh72b88fc2006-12-16 18:41:35 +0000270 self.__class__.__name__ + "\n")
apwf1a81162006-04-25 10:10:29 +0000271
mblighcaa605c2006-10-02 00:37:35 +0000272
mblighd016ecc2006-11-25 21:41:07 +0000273 def run_test(self, url, *args, **dargs):
mblighc86b0b42006-07-28 17:35:28 +0000274 """Summon a test object and run it.
275
276 tag
277 tag to add to testname
mbligh12a7df72006-10-06 03:54:33 +0000278 url
279 url of the test to run
mblighc86b0b42006-07-28 17:35:28 +0000280 """
mbligh12a7df72006-10-06 03:54:33 +0000281
mblighd016ecc2006-11-25 21:41:07 +0000282 if not url:
mbligh302482e2008-05-01 20:06:16 +0000283 raise TypeError("Test name is invalid. "
284 "Switched arguments?")
mbligh09f288a2007-09-18 21:34:57 +0000285 (group, testname) = test.testname(url)
mbligh7dd510c2007-11-13 17:11:22 +0000286 tag = dargs.pop('tag', None)
mbligh65938a22007-12-10 16:58:52 +0000287 container = dargs.pop('container', None)
mbligh09f288a2007-09-18 21:34:57 +0000288 subdir = testname
mbligh7dd510c2007-11-13 17:11:22 +0000289 if tag:
290 subdir += '.' + tag
291
mbligh65938a22007-12-10 16:58:52 +0000292 if container:
mbligh68119582008-01-25 18:16:41 +0000293 cname = container.get('name', None)
294 if not cname: # get old name
295 cname = container.get('container_name', None)
296 mbytes = container.get('mbytes', None)
297 if not mbytes: # get old name
298 mbytes = container.get('mem', None)
299 cpus = container.get('cpus', None)
300 if not cpus: # get old name
301 cpus = container.get('cpu', None)
mbligh9ea52602008-04-02 00:15:35 +0000302 root = container.get('root', '')
mbligh68119582008-01-25 18:16:41 +0000303 self.new_container(mbytes=mbytes, cpus=cpus,
304 root=root, name=cname)
mbligh65938a22007-12-10 16:58:52 +0000305 # We are running in a container now...
306
jadmanski8415f962008-05-06 20:38:53 +0000307 def log_warning(reason):
308 self.record("WARN", subdir, testname, reason)
309 @disk_usage_monitor.watch(log_warning, "/",
310 self.max_disk_usage_rate)
mbligh7dd510c2007-11-13 17:11:22 +0000311 def group_func():
apwf1a81162006-04-25 10:10:29 +0000312 try:
mblighd016ecc2006-11-25 21:41:07 +0000313 self.__runtest(url, tag, args, dargs)
mbligh302482e2008-05-01 20:06:16 +0000314 except error.TestNAError, detail:
315 self.record('TEST_NA', subdir, testname,
316 str(detail))
317 raise
apwf1a81162006-04-25 10:10:29 +0000318 except Exception, detail:
mbligh7dd510c2007-11-13 17:11:22 +0000319 self.record('FAIL', subdir, testname,
320 str(detail))
apwf1a81162006-04-25 10:10:29 +0000321 raise
322 else:
mbligh7dd510c2007-11-13 17:11:22 +0000323 self.record('GOOD', subdir, testname,
324 'completed successfully')
jadmanski8415f962008-05-06 20:38:53 +0000325
mblighcfc6dd32007-11-20 00:44:35 +0000326 result, exc_info = self.__rungroup(subdir, group_func)
mbligh68119582008-01-25 18:16:41 +0000327 if container:
328 self.release_container()
mbligh302482e2008-05-01 20:06:16 +0000329 if exc_info and isinstance(exc_info[1], error.TestError):
mbligh7dd510c2007-11-13 17:11:22 +0000330 return False
331 elif exc_info:
mbligh71ea2492008-01-15 20:35:52 +0000332 raise exc_info[0], exc_info[1], exc_info[2]
apwf1a81162006-04-25 10:10:29 +0000333 else:
mbligh7dd510c2007-11-13 17:11:22 +0000334 return True
335
336
337 def __rungroup(self, name, function, *args, **dargs):
338 """\
339 name:
340 name of the group
341 function:
342 subroutine to run
343 *args:
344 arguments for the function
345
346 Returns a 2-tuple (result, exc_info) where result
347 is the return value of function, and exc_info is
348 the sys.exc_info() of the exception thrown by the
349 function (which may be None).
350 """
351
352 result, exc_info = None, None
353 try:
354 self.record('START', None, name)
jadmanskia9c75c42008-05-01 22:05:31 +0000355 self._increment_group_level()
mbligh7dd510c2007-11-13 17:11:22 +0000356 result = function(*args, **dargs)
jadmanskia9c75c42008-05-01 22:05:31 +0000357 self._decrement_group_level()
mbligh7dd510c2007-11-13 17:11:22 +0000358 self.record('END GOOD', None, name)
mbligh302482e2008-05-01 20:06:16 +0000359 except error.TestNAError, e:
jadmanskia9c75c42008-05-01 22:05:31 +0000360 self._decrement_group_level()
mbligh302482e2008-05-01 20:06:16 +0000361 self.record('END TEST_NA', None, name, str(e))
mbligh7dd510c2007-11-13 17:11:22 +0000362 except Exception, e:
363 exc_info = sys.exc_info()
jadmanskia9c75c42008-05-01 22:05:31 +0000364 self._decrement_group_level()
mbligh302482e2008-05-01 20:06:16 +0000365 err_msg = str(e) + '\n' + traceback.format_exc()
mbligh51144e02007-11-20 20:38:18 +0000366 self.record('END FAIL', None, name, err_msg)
mbligh7dd510c2007-11-13 17:11:22 +0000367
368 return result, exc_info
apw0865f482006-03-30 18:50:19 +0000369
mblighd7fb4a62006-10-01 00:57:53 +0000370
apw1da244b2007-09-27 17:18:01 +0000371 def run_group(self, function, *args, **dargs):
mbligh88ab90f2007-08-29 15:52:49 +0000372 """\
373 function:
374 subroutine to run
375 *args:
376 arguments for the function
377 """
378
mbligh7dd510c2007-11-13 17:11:22 +0000379 # Allow the tag for the group to be specified
mbligh88ab90f2007-08-29 15:52:49 +0000380 name = function.__name__
mbligh7dd510c2007-11-13 17:11:22 +0000381 tag = dargs.pop('tag', None)
382 if tag:
383 name = tag
apw1da244b2007-09-27 17:18:01 +0000384
mbligh7dd510c2007-11-13 17:11:22 +0000385 result, exc_info = self.__rungroup(name, function,
386 *args, **dargs)
apw1da244b2007-09-27 17:18:01 +0000387
mbligh7dd510c2007-11-13 17:11:22 +0000388 # if there was a non-TestError exception, raise it
mbligh302482e2008-05-01 20:06:16 +0000389 if exc_info and not isinstance(exc_info[1], error.TestError):
mbligh7dd510c2007-11-13 17:11:22 +0000390 err = ''.join(traceback.format_exception(*exc_info))
mbligh302482e2008-05-01 20:06:16 +0000391 raise error.TestError(name + ' failed\n' + err)
mbligh88ab90f2007-08-29 15:52:49 +0000392
mbligh7dd510c2007-11-13 17:11:22 +0000393 # pass back the actual return value from the function
apw08403ca2007-09-27 17:17:22 +0000394 return result
395
mbligh88ab90f2007-08-29 15:52:49 +0000396
mbligh1fc7ba12008-03-31 17:50:53 +0000397 def new_container(self, mbytes=None, cpus=None, root='', name=None):
mbligh8ea61e22008-05-09 18:09:37 +0000398 if not autotest_utils.grep('cpuset', '/proc/filesystems'):
mbligh68119582008-01-25 18:16:41 +0000399 print "Containers not enabled by latest reboot"
400 return # containers weren't enabled in this kernel boot
401 pid = os.getpid()
mbligh68119582008-01-25 18:16:41 +0000402 if not name:
403 name = 'test%d' % pid # make arbitrary unique name
404 self.container = cpuset.cpuset(name, job_size=mbytes,
mbligh337bb762008-04-16 21:23:10 +0000405 job_pid=pid, cpus=cpus, root=root)
mbligh68119582008-01-25 18:16:41 +0000406 # This job's python shell is now running in the new container
407 # and all forked test processes will inherit that container
408
409
410 def release_container(self):
411 if self.container:
mbligh337bb762008-04-16 21:23:10 +0000412 self.container.release()
mbligh68119582008-01-25 18:16:41 +0000413 self.container = None
414
415
416 def cpu_count(self):
417 if self.container:
418 return len(self.container.cpus)
jadmanskia9c75c42008-05-01 22:05:31 +0000419 return autotest_utils.count_cpus() # use total system count
mbligh68119582008-01-25 18:16:41 +0000420
421
apwce73d892007-09-25 16:53:05 +0000422 # Check the passed kernel identifier against the command line
423 # and the running kernel, abort the job on missmatch.
mbligh38a4a112008-03-19 13:11:34 +0000424 def kernel_check_ident(self, expected_when, expected_id, subdir,
jadmanskia9c75c42008-05-01 22:05:31 +0000425 type = 'src', patches=[]):
mbligh38a4a112008-03-19 13:11:34 +0000426 print (("POST BOOT: checking booted kernel " +
427 "mark=%d identity='%s' type='%s'") %
428 (expected_when, expected_id, type))
apwce73d892007-09-25 16:53:05 +0000429
jadmanskia9c75c42008-05-01 22:05:31 +0000430 running_id = autotest_utils.running_os_ident()
apwce73d892007-09-25 16:53:05 +0000431
jadmanskia9c75c42008-05-01 22:05:31 +0000432 cmdline = autotest_utils.read_one_line("/proc/cmdline")
apwce73d892007-09-25 16:53:05 +0000433
434 find_sum = re.compile(r'.*IDENT=(\d+)')
435 m = find_sum.match(cmdline)
436 cmdline_when = -1
437 if m:
438 cmdline_when = int(m.groups()[0])
439
440 # We have all the facts, see if they indicate we
441 # booted the requested kernel or not.
442 bad = False
mblighda0311e2007-10-25 16:03:33 +0000443 if (type == 'src' and expected_id != running_id or
jadmanskia9c75c42008-05-01 22:05:31 +0000444 type == 'rpm' and
445 not running_id.startswith(expected_id + '::')):
apwce73d892007-09-25 16:53:05 +0000446 print "check_kernel_ident: kernel identifier mismatch"
447 bad = True
448 if expected_when != cmdline_when:
449 print "check_kernel_ident: kernel command line mismatch"
450 bad = True
451
452 if bad:
453 print " Expected Ident: " + expected_id
454 print " Running Ident: " + running_id
455 print " Expected Mark: %d" % (expected_when)
456 print "Command Line Mark: %d" % (cmdline_when)
457 print " Command Line: " + cmdline
458
mbligh302482e2008-05-01 20:06:16 +0000459 raise error.JobError("boot failure", "reboot.verify")
apwce73d892007-09-25 16:53:05 +0000460
jadmanskia9c75c42008-05-01 22:05:31 +0000461 kernel_info = {'kernel': expected_id}
462 for i, patch in enumerate(patches):
463 kernel_info["patch%d" % i] = patch
mblighb7fd2702008-03-25 14:57:08 +0000464 self.record('GOOD', subdir, 'reboot.verify', expected_id)
jadmanskia9c75c42008-05-01 22:05:31 +0000465 self._decrement_group_level()
466 self.record('END GOOD', subdir, 'reboot',
467 optional_fields=kernel_info)
apwce73d892007-09-25 16:53:05 +0000468
469
mblighc2359852007-08-28 18:11:48 +0000470 def filesystem(self, device, mountpoint = None, loop_size = 0):
mblighd7fb4a62006-10-01 00:57:53 +0000471 if not mountpoint:
472 mountpoint = self.tmpdir
mblighc2359852007-08-28 18:11:48 +0000473 return filesystem.filesystem(self, device, mountpoint,loop_size)
mblighd7fb4a62006-10-01 00:57:53 +0000474
mblighcaa62c22008-04-07 21:51:17 +0000475
476 def enable_external_logging(self):
477 pass
478
479
480 def disable_external_logging(self):
481 pass
482
483
484 def reboot_setup(self):
485 pass
486
mblighcaa605c2006-10-02 00:37:35 +0000487
488 def reboot(self, tag='autotest'):
mblighcaa62c22008-04-07 21:51:17 +0000489 self.reboot_setup()
jadmanskia9c75c42008-05-01 22:05:31 +0000490 self.record('START', None, 'reboot')
491 self._increment_group_level()
mbligh30270302007-11-05 20:33:52 +0000492 self.record('GOOD', None, 'reboot.start')
apwde1503a2006-10-10 08:34:21 +0000493 self.harness.run_reboot()
apw11985b72007-10-04 15:44:47 +0000494 default = self.config_get('boot.set_default')
495 if default:
496 self.bootloader.set_default(tag)
497 else:
498 self.bootloader.boot_once(tag)
mbligh302482e2008-05-01 20:06:16 +0000499 cmd = "(sleep 5; reboot) </dev/null >/dev/null 2>&1 &"
500 autotest_utils.system(cmd)
apw0778a2f2006-10-06 03:11:40 +0000501 self.quit()
mblighcaa605c2006-10-02 00:37:35 +0000502
503
apw0865f482006-03-30 18:50:19 +0000504 def noop(self, text):
505 print "job: noop: " + text
506
mblighcaa605c2006-10-02 00:37:35 +0000507
mblighc86b0b42006-07-28 17:35:28 +0000508 def parallel(self, *tasklist):
509 """Run tasks in parallel"""
apw8fef4ac2006-10-10 22:53:37 +0000510
511 pids = []
mblighd528d302007-12-19 16:19:05 +0000512 old_log_filename = self.log_filename
513 for i, task in enumerate(tasklist):
514 self.log_filename = old_log_filename + (".%d" % i)
515 task_func = lambda: task[0](*task[1:])
mbligh302482e2008-05-01 20:06:16 +0000516 pids.append(parallel.fork_start(self.resultdir,
517 task_func))
mblighd528d302007-12-19 16:19:05 +0000518
519 old_log_path = os.path.join(self.resultdir, old_log_filename)
520 old_log = open(old_log_path, "a")
mblighd509b712008-01-14 17:41:25 +0000521 exceptions = []
mblighd528d302007-12-19 16:19:05 +0000522 for i, pid in enumerate(pids):
523 # wait for the task to finish
mblighd509b712008-01-14 17:41:25 +0000524 try:
mbligh302482e2008-05-01 20:06:16 +0000525 parallel.fork_waitfor(self.resultdir, pid)
mblighd509b712008-01-14 17:41:25 +0000526 except Exception, e:
527 exceptions.append(e)
mblighd528d302007-12-19 16:19:05 +0000528 # copy the logs from the subtask into the main log
529 new_log_path = old_log_path + (".%d" % i)
530 if os.path.exists(new_log_path):
531 new_log = open(new_log_path)
532 old_log.write(new_log.read())
533 new_log.close()
534 old_log.flush()
535 os.remove(new_log_path)
536 old_log.close()
537
538 self.log_filename = old_log_filename
apw0865f482006-03-30 18:50:19 +0000539
mblighd509b712008-01-14 17:41:25 +0000540 # handle any exceptions raised by the parallel tasks
541 if exceptions:
542 msg = "%d task(s) failed" % len(exceptions)
mbligh302482e2008-05-01 20:06:16 +0000543 raise error.JobError(msg, str(exceptions), exceptions)
mblighd509b712008-01-14 17:41:25 +0000544
mblighcaa605c2006-10-02 00:37:35 +0000545
apw0865f482006-03-30 18:50:19 +0000546 def quit(self):
mblighc86b0b42006-07-28 17:35:28 +0000547 # XXX: should have a better name.
apwde1503a2006-10-10 08:34:21 +0000548 self.harness.run_pause()
mbligh302482e2008-05-01 20:06:16 +0000549 raise error.JobContinue("more to come")
apw0865f482006-03-30 18:50:19 +0000550
mblighcaa605c2006-10-02 00:37:35 +0000551
apw0865f482006-03-30 18:50:19 +0000552 def complete(self, status):
mblighc86b0b42006-07-28 17:35:28 +0000553 """Clean up and exit"""
apw0865f482006-03-30 18:50:19 +0000554 # We are about to exit 'complete' so clean up the control file.
555 try:
mbligh366ff1b2008-04-25 16:07:56 +0000556 os.unlink(self.state_file)
apw0865f482006-03-30 18:50:19 +0000557 except:
558 pass
mblighc0b10d32008-03-03 16:03:28 +0000559
mbligh61a6c1a2006-12-25 01:26:38 +0000560 self.harness.run_complete()
mblighcaa62c22008-04-07 21:51:17 +0000561 self.disable_external_logging()
apw1b021902006-04-03 17:02:56 +0000562 sys.exit(status)
apw0865f482006-03-30 18:50:19 +0000563
mblighcaa605c2006-10-02 00:37:35 +0000564
mbligh366ff1b2008-04-25 16:07:56 +0000565 def set_state(self, var, val):
566 # Deep copies make sure that the state can't be altered
567 # without it being re-written. Perf wise, deep copies
568 # are overshadowed by pickling/loading.
569 self.state[var] = copy.deepcopy(val)
570 pickle.dump(self.state, open(self.state_file, 'w'))
571
572
573 def __load_state(self):
jadmanskia9c75c42008-05-01 22:05:31 +0000574 assert not hasattr(self, "state")
mbligh366ff1b2008-04-25 16:07:56 +0000575 try:
576 self.state = pickle.load(open(self.state_file, 'r'))
jadmanskia9c75c42008-05-01 22:05:31 +0000577 self.state_existed = True
mbligh366ff1b2008-04-25 16:07:56 +0000578 except Exception:
579 print "Initializing the state engine."
580 self.state = {}
mblighf1ae0a42008-04-25 16:09:20 +0000581 self.set_state('__steps', []) # writes pickle file
jadmanskia9c75c42008-05-01 22:05:31 +0000582 self.state_existed = False
mbligh366ff1b2008-04-25 16:07:56 +0000583
584
585 def get_state(self, var, default=None):
586 if var in self.state or default == None:
587 val = self.state[var]
588 else:
589 val = default
590 return copy.deepcopy(val)
591
592
mbligh12a04cb2008-04-25 16:07:20 +0000593 def __create_step_tuple(self, fn, args, dargs):
594 # Legacy code passes in an array where the first arg is
595 # the function or its name.
596 if isinstance(fn, list):
597 assert(len(args) == 0)
598 assert(len(dargs) == 0)
599 args = fn[1:]
600 fn = fn[0]
601 # Pickling actual functions is harry, thus we have to call
602 # them by name. Unfortunately, this means only functions
603 # defined globally can be used as a next step.
604 if isinstance(fn, types.FunctionType):
605 fn = fn.__name__
606 if not isinstance(fn, types.StringTypes):
607 raise StepError("Next steps must be functions or "
608 "strings containing the function name")
609 return (fn, args, dargs)
610
611
mbligh12a04cb2008-04-25 16:07:20 +0000612 def next_step(self, fn, *args, **dargs):
mblighc86b0b42006-07-28 17:35:28 +0000613 """Define the next step"""
mblighf1ae0a42008-04-25 16:09:20 +0000614 steps = self.get_state('__steps')
mbligh366ff1b2008-04-25 16:07:56 +0000615 steps.append(self.__create_step_tuple(fn, args, dargs))
mblighf1ae0a42008-04-25 16:09:20 +0000616 self.set_state('__steps', steps)
apw0865f482006-03-30 18:50:19 +0000617
mblighcaa605c2006-10-02 00:37:35 +0000618
mbligh12a04cb2008-04-25 16:07:20 +0000619 def next_step_prepend(self, fn, *args, **dargs):
mbligh237bed32007-09-05 13:05:57 +0000620 """Insert a new step, executing first"""
mblighf1ae0a42008-04-25 16:09:20 +0000621 steps = self.get_state('__steps')
mbligh366ff1b2008-04-25 16:07:56 +0000622 steps.insert(0, self.__create_step_tuple(fn, args, dargs))
mblighf1ae0a42008-04-25 16:09:20 +0000623 self.set_state('__steps', steps)
mbligh237bed32007-09-05 13:05:57 +0000624
625
apw83f8d772006-04-27 14:12:56 +0000626 def step_engine(self):
mblighc86b0b42006-07-28 17:35:28 +0000627 """the stepping engine -- if the control file defines
628 step_init we will be using this engine to drive multiple runs.
629 """
630 """Do the next step"""
apw83f8d772006-04-27 14:12:56 +0000631
mbligh366ff1b2008-04-25 16:07:56 +0000632 # Set up the environment and then interpret the control file.
633 # Some control files will have code outside of functions,
634 # which means we need to have our state engine initialized
635 # before reading in the file.
mbligh366ff1b2008-04-25 16:07:56 +0000636 lcl = {'job': self}
637 exec(JOB_PREAMBLE, lcl, lcl)
apw83f8d772006-04-27 14:12:56 +0000638 execfile(self.control, lcl, lcl)
639
mbligh366ff1b2008-04-25 16:07:56 +0000640 # If we loaded in a mid-job state file, then we presumably
641 # know what steps we have yet to run.
jadmanskia9c75c42008-05-01 22:05:31 +0000642 if not self.state_existed:
apw83f8d772006-04-27 14:12:56 +0000643 if lcl.has_key('step_init'):
644 self.next_step([lcl['step_init']])
apw0865f482006-03-30 18:50:19 +0000645
mbligh366ff1b2008-04-25 16:07:56 +0000646 # Iterate through the steps. If we reboot, we'll simply
647 # continue iterating on the next step.
mblighf1ae0a42008-04-25 16:09:20 +0000648 while len(self.get_state('__steps')) > 0:
649 steps = self.get_state('__steps')
mbligh366ff1b2008-04-25 16:07:56 +0000650 (fn, args, dargs) = steps.pop(0)
mblighf1ae0a42008-04-25 16:09:20 +0000651 self.set_state('__steps', steps)
apw0865f482006-03-30 18:50:19 +0000652
mbligh12a04cb2008-04-25 16:07:20 +0000653 lcl['__args'] = args
654 lcl['__dargs'] = dargs
655 exec(fn + "(*__args, **__dargs)", lcl, lcl)
apw0865f482006-03-30 18:50:19 +0000656
mblighcaa605c2006-10-02 00:37:35 +0000657
jadmanskia9c75c42008-05-01 22:05:31 +0000658 def _init_group_level(self):
659 self.group_level = self.get_state("__group_level", default=0)
660
661
662 def _increment_group_level(self):
663 self.group_level += 1
664 self.set_state("__group_level", self.group_level)
665
666
667 def _decrement_group_level(self):
668 self.group_level -= 1
669 self.set_state("__group_level", self.group_level)
670
671
672 def record(self, status_code, subdir, operation, status = '',
673 optional_fields=None):
mbligh09f288a2007-09-18 21:34:57 +0000674 """
675 Record job-level status
apw7db8d0b2006-10-09 08:10:25 +0000676
mbligh09f288a2007-09-18 21:34:57 +0000677 The intent is to make this file both machine parseable and
678 human readable. That involves a little more complexity, but
679 really isn't all that bad ;-)
680
681 Format is <status code>\t<subdir>\t<operation>\t<status>
682
683 status code: (GOOD|WARN|FAIL|ABORT)
684 or START
685 or END (GOOD|WARN|FAIL|ABORT)
686
687 subdir: MUST be a relevant subdirectory in the results,
688 or None, which will be represented as '----'
689
690 operation: description of what you ran (e.g. "dbench", or
691 "mkfs -t foobar /dev/sda9")
692
693 status: error message or "completed sucessfully"
694
695 ------------------------------------------------------------
696
697 Initial tabs indicate indent levels for grouping, and is
mbligh7dd510c2007-11-13 17:11:22 +0000698 governed by self.group_level
mbligh09f288a2007-09-18 21:34:57 +0000699
700 multiline messages have secondary lines prefaced by a double
701 space (' ')
702 """
703
mblighb0570ad2007-09-19 18:18:11 +0000704 if subdir:
705 if re.match(r'[\n\t]', subdir):
jadmanskia9c75c42008-05-01 22:05:31 +0000706 raise ValueError("Invalid character in "
707 "subdir string")
mblighb0570ad2007-09-19 18:18:11 +0000708 substr = subdir
709 else:
710 substr = '----'
mbligh09f288a2007-09-18 21:34:57 +0000711
mbligh302482e2008-05-01 20:06:16 +0000712 if not logging.is_valid_status(status_code):
jadmanskia9c75c42008-05-01 22:05:31 +0000713 raise ValueError("Invalid status code supplied: %s" %
714 status_code)
mbligh9c5ac322007-10-31 18:01:59 +0000715 if not operation:
716 operation = '----'
jadmanskia9c75c42008-05-01 22:05:31 +0000717
mbligh09f288a2007-09-18 21:34:57 +0000718 if re.match(r'[\n\t]', operation):
jadmanskia9c75c42008-05-01 22:05:31 +0000719 raise ValueError("Invalid character in "
720 "operation string")
mbligh09f288a2007-09-18 21:34:57 +0000721 operation = operation.rstrip()
jadmanskia9c75c42008-05-01 22:05:31 +0000722
723 if not optional_fields:
724 optional_fields = {}
725
mbligh09f288a2007-09-18 21:34:57 +0000726 status = status.rstrip()
727 status = re.sub(r"\t", " ", status)
apw7db8d0b2006-10-09 08:10:25 +0000728 # Ensure any continuation lines are marked so we can
729 # detect them in the status file to ensure it is parsable.
jadmanskia9c75c42008-05-01 22:05:31 +0000730 status = re.sub(r"\n", "\n" + "\t" * self.group_level + " ",
731 status)
mbligh09f288a2007-09-18 21:34:57 +0000732
mbligh30270302007-11-05 20:33:52 +0000733 # Generate timestamps for inclusion in the logs
734 epoch_time = int(time.time()) # seconds since epoch, in UTC
735 local_time = time.localtime(epoch_time)
jadmanskia9c75c42008-05-01 22:05:31 +0000736 optional_fields["timestamp"] = str(epoch_time)
737 optional_fields["localtime"] = time.strftime("%b %d %H:%M:%S",
738 local_time)
mbligh30270302007-11-05 20:33:52 +0000739
jadmanskia9c75c42008-05-01 22:05:31 +0000740 fields = [status_code, substr, operation]
741 fields += ["%s=%s" % x for x in optional_fields.iteritems()]
742 fields.append(status)
743
744 msg = '\t'.join(str(x) for x in fields)
mbligh7dd510c2007-11-13 17:11:22 +0000745 msg = '\t' * self.group_level + msg
apw7db8d0b2006-10-09 08:10:25 +0000746
mblighd528d302007-12-19 16:19:05 +0000747 msg_tag = ""
748 if "." in self.log_filename:
749 msg_tag = self.log_filename.split(".", 1)[1]
750
jadmanskia9c75c42008-05-01 22:05:31 +0000751 self.harness.test_status_detail(status_code, substr,
752 operation, status, msg_tag)
mblighd528d302007-12-19 16:19:05 +0000753 self.harness.test_status(msg, msg_tag)
754
755 # log to stdout (if enabled)
756 #if self.log_filename == self.DEFAULT_LOG_FILENAME:
apwf1a81162006-04-25 10:10:29 +0000757 print msg
mblighd528d302007-12-19 16:19:05 +0000758
759 # log to the "root" status log
760 status_file = os.path.join(self.resultdir, self.log_filename)
mbligh7dd510c2007-11-13 17:11:22 +0000761 open(status_file, "a").write(msg + "\n")
mblighd528d302007-12-19 16:19:05 +0000762
763 # log to the subdir status log (if subdir is set)
mblighb0570ad2007-09-19 18:18:11 +0000764 if subdir:
mblighadff6ca2008-01-22 16:38:25 +0000765 dir = os.path.join(self.resultdir, subdir)
766 if not os.path.exists(dir):
767 os.mkdir(dir)
768
769 status_file = os.path.join(dir,
mblighd528d302007-12-19 16:19:05 +0000770 self.DEFAULT_LOG_FILENAME)
mblighb0570ad2007-09-19 18:18:11 +0000771 open(status_file, "a").write(msg + "\n")
apwce9abe92006-04-27 14:14:04 +0000772
773
jadmanski8415f962008-05-06 20:38:53 +0000774class disk_usage_monitor:
775 def __init__(self, logging_func, device, max_mb_per_hour):
776 self.func = logging_func
777 self.device = device
778 self.max_mb_per_hour = max_mb_per_hour
779
780
781 def start(self):
782 self.initial_space = autotest_utils.freespace(self.device)
783 self.start_time = time.time()
784
785
786 def stop(self):
787 # if no maximum usage rate was set, we don't need to
788 # generate any warnings
789 if not self.max_mb_per_hour:
790 return
791
792 final_space = autotest_utils.freespace(self.device)
793 used_space = self.initial_space - final_space
794 stop_time = time.time()
795 total_time = stop_time - self.start_time
796 # round up the time to one minute, to keep extremely short
797 # tests from generating false positives due to short, badly
798 # timed bursts of activity
799 total_time = max(total_time, 60.0)
800
801 # determine the usage rate
802 bytes_per_sec = used_space / total_time
803 mb_per_sec = bytes_per_sec / 1024**2
804 mb_per_hour = mb_per_sec * 60 * 60
805
806 if mb_per_hour > self.max_mb_per_hour:
807 msg = ("disk space on %s was consumed at a rate of "
808 "%.2f MB/hour")
809 msg %= (self.device, mb_per_hour)
810 self.func(msg)
811
812
813 @classmethod
814 def watch(cls, *monitor_args, **monitor_dargs):
815 """ Generic decorator to wrap a function call with the
816 standard create-monitor -> start -> call -> stop idiom."""
817 def decorator(func):
818 def watched_func(*args, **dargs):
819 monitor = cls(*monitor_args, **monitor_dargs)
820 monitor.start()
821 try:
822 func(*args, **dargs)
823 finally:
824 monitor.stop()
825 return watched_func
826 return decorator
827
828
mblighcaa62c22008-04-07 21:51:17 +0000829def runjob(control, cont = False, tag = "default", harness_type = '',
830 use_external_logging = False):
mblighc86b0b42006-07-28 17:35:28 +0000831 """The main interface to this module
832
mbligh72b88fc2006-12-16 18:41:35 +0000833 control
mblighc86b0b42006-07-28 17:35:28 +0000834 The control file to use for this job.
835 cont
836 Whether this is the continuation of a previously started job
837 """
mblighb4eef242007-07-23 18:22:49 +0000838 control = os.path.abspath(control)
apwce9abe92006-04-27 14:14:04 +0000839 state = control + '.state'
840
841 # instantiate the job object ready for the control file.
842 myjob = None
843 try:
844 # Check that the control file is valid
845 if not os.path.exists(control):
mbligh302482e2008-05-01 20:06:16 +0000846 raise error.JobError(control +
847 ": control file not found")
apwce9abe92006-04-27 14:14:04 +0000848
849 # When continuing, the job is complete when there is no
850 # state file, ensure we don't try and continue.
mblighf3fef462006-09-13 16:05:05 +0000851 if cont and not os.path.exists(state):
mbligh302482e2008-05-01 20:06:16 +0000852 raise error.JobComplete("all done")
mblighf3fef462006-09-13 16:05:05 +0000853 if cont == False and os.path.exists(state):
apwce9abe92006-04-27 14:14:04 +0000854 os.unlink(state)
855
mblighcaa62c22008-04-07 21:51:17 +0000856 myjob = job(control, tag, cont, harness_type,
857 use_external_logging)
apwce9abe92006-04-27 14:14:04 +0000858
859 # Load in the users control file, may do any one of:
860 # 1) execute in toto
861 # 2) define steps, and select the first via next_step()
862 myjob.step_engine()
863
mbligh302482e2008-05-01 20:06:16 +0000864 except error.JobContinue:
apwce9abe92006-04-27 14:14:04 +0000865 sys.exit(5)
866
mbligh302482e2008-05-01 20:06:16 +0000867 except error.JobComplete:
apwb832e1b2007-11-24 20:24:38 +0000868 sys.exit(1)
869
mbligh302482e2008-05-01 20:06:16 +0000870 except error.JobError, instance:
apwce9abe92006-04-27 14:14:04 +0000871 print "JOB ERROR: " + instance.args[0]
mbligh9c5ac322007-10-31 18:01:59 +0000872 if myjob:
mbligh30270302007-11-05 20:33:52 +0000873 command = None
874 if len(instance.args) > 1:
875 command = instance.args[1]
876 myjob.record('ABORT', None, command, instance.args[0])
jadmanskia9c75c42008-05-01 22:05:31 +0000877 myjob._decrement_group_level()
mblighc3430162007-11-14 23:57:19 +0000878 myjob.record('END ABORT', None, None)
jadmanskia9c75c42008-05-01 22:05:31 +0000879 assert(myjob.group_level == 0)
apwce9abe92006-04-27 14:14:04 +0000880 myjob.complete(1)
apwb832e1b2007-11-24 20:24:38 +0000881 else:
882 sys.exit(1)
apwce9abe92006-04-27 14:14:04 +0000883
mblighc3430162007-11-14 23:57:19 +0000884 except Exception, e:
mbligh302482e2008-05-01 20:06:16 +0000885 msg = str(e) + '\n' + traceback.format_exc()
mblighc3430162007-11-14 23:57:19 +0000886 print "JOB ERROR: " + msg
mblighfbfb77d2007-02-15 18:54:03 +0000887 if myjob:
mblighc3430162007-11-14 23:57:19 +0000888 myjob.record('ABORT', None, None, msg)
jadmanskia9c75c42008-05-01 22:05:31 +0000889 myjob._decrement_group_level()
mblighc3430162007-11-14 23:57:19 +0000890 myjob.record('END ABORT', None, None)
jadmanskia9c75c42008-05-01 22:05:31 +0000891 assert(myjob.group_level == 0)
mbligh9c5ac322007-10-31 18:01:59 +0000892 myjob.complete(1)
apwb832e1b2007-11-24 20:24:38 +0000893 else:
894 sys.exit(1)
mbligh892d37f2007-03-01 17:03:25 +0000895
mbligh0144e5a2008-03-07 18:17:53 +0000896 # If we get here, then we assume the job is complete and good.
jadmanskia9c75c42008-05-01 22:05:31 +0000897 myjob._decrement_group_level()
mbligh0144e5a2008-03-07 18:17:53 +0000898 myjob.record('END GOOD', None, None)
jadmanskia9c75c42008-05-01 22:05:31 +0000899 assert(myjob.group_level == 0)
mbligh0144e5a2008-03-07 18:17:53 +0000900
mbligh892d37f2007-03-01 17:03:25 +0000901 myjob.complete(0)
mblighcaa62c22008-04-07 21:51:17 +0000902
903
904# site_job.py may be non-existant or empty, make sure that an appropriate
905# site_job class is created nevertheless
906try:
907 from site_job import site_job
908except ImportError:
909 class site_job(base_job):
910 pass
911
912class job(site_job):
913 pass