Revision 4e338533 test/ganeti.cli_unittest.py
b/test/ganeti.cli_unittest.py | ||
---|---|---|
29 | 29 |
|
30 | 30 |
from ganeti import constants |
31 | 31 |
from ganeti import cli |
32 |
from ganeti import errors |
|
33 |
from ganeti import utils |
|
32 | 34 |
from ganeti.errors import OpPrereqError, ParameterError |
33 | 35 |
|
34 | 36 |
|
... | ... | |
100 | 102 |
|
101 | 103 |
|
102 | 104 |
class TestToStream(unittest.TestCase): |
103 |
"""Thes the ToStream functions"""
|
|
105 |
"""Test the ToStream functions"""
|
|
104 | 106 |
|
105 | 107 |
def testBasic(self): |
106 | 108 |
for data in ["foo", |
... | ... | |
246 | 248 |
None, None, "m", exp) |
247 | 249 |
|
248 | 250 |
|
251 |
class _MockJobPollCb(cli.JobPollCbBase, cli.JobPollReportCbBase): |
|
252 |
def __init__(self, tc, job_id): |
|
253 |
self.tc = tc |
|
254 |
self.job_id = job_id |
|
255 |
self._wfjcr = [] |
|
256 |
self._jobstatus = [] |
|
257 |
self._expect_notchanged = False |
|
258 |
self._expect_log = [] |
|
259 |
|
|
260 |
def CheckEmpty(self): |
|
261 |
self.tc.assertFalse(self._wfjcr) |
|
262 |
self.tc.assertFalse(self._jobstatus) |
|
263 |
self.tc.assertFalse(self._expect_notchanged) |
|
264 |
self.tc.assertFalse(self._expect_log) |
|
265 |
|
|
266 |
def AddWfjcResult(self, *args): |
|
267 |
self._wfjcr.append(args) |
|
268 |
|
|
269 |
def AddQueryJobsResult(self, *args): |
|
270 |
self._jobstatus.append(args) |
|
271 |
|
|
272 |
def WaitForJobChangeOnce(self, job_id, fields, |
|
273 |
prev_job_info, prev_log_serial): |
|
274 |
self.tc.assertEqual(job_id, self.job_id) |
|
275 |
self.tc.assertEqualValues(fields, ["status"]) |
|
276 |
self.tc.assertFalse(self._expect_notchanged) |
|
277 |
self.tc.assertFalse(self._expect_log) |
|
278 |
|
|
279 |
(exp_prev_job_info, exp_prev_log_serial, result) = self._wfjcr.pop(0) |
|
280 |
self.tc.assertEqualValues(prev_job_info, exp_prev_job_info) |
|
281 |
self.tc.assertEqual(prev_log_serial, exp_prev_log_serial) |
|
282 |
|
|
283 |
if result == constants.JOB_NOTCHANGED: |
|
284 |
self._expect_notchanged = True |
|
285 |
elif result: |
|
286 |
(_, logmsgs) = result |
|
287 |
if logmsgs: |
|
288 |
self._expect_log.extend(logmsgs) |
|
289 |
|
|
290 |
return result |
|
291 |
|
|
292 |
def QueryJobs(self, job_ids, fields): |
|
293 |
self.tc.assertEqual(job_ids, [self.job_id]) |
|
294 |
self.tc.assertEqualValues(fields, ["status", "opstatus", "opresult"]) |
|
295 |
self.tc.assertFalse(self._expect_notchanged) |
|
296 |
self.tc.assertFalse(self._expect_log) |
|
297 |
|
|
298 |
result = self._jobstatus.pop(0) |
|
299 |
self.tc.assertEqual(len(fields), len(result)) |
|
300 |
return [result] |
|
301 |
|
|
302 |
def ReportLogMessage(self, job_id, serial, timestamp, log_type, log_msg): |
|
303 |
self.tc.assertEqual(job_id, self.job_id) |
|
304 |
self.tc.assertEqualValues((serial, timestamp, log_type, log_msg), |
|
305 |
self._expect_log.pop(0)) |
|
306 |
|
|
307 |
def ReportNotChanged(self, job_id, status): |
|
308 |
self.tc.assertEqual(job_id, self.job_id) |
|
309 |
self.tc.assert_(self._expect_notchanged) |
|
310 |
self._expect_notchanged = False |
|
311 |
|
|
312 |
|
|
313 |
class TestGenericPollJob(testutils.GanetiTestCase): |
|
314 |
def testSuccessWithLog(self): |
|
315 |
job_id = 29609 |
|
316 |
cbs = _MockJobPollCb(self, job_id) |
|
317 |
|
|
318 |
cbs.AddWfjcResult(None, None, constants.JOB_NOTCHANGED) |
|
319 |
|
|
320 |
cbs.AddWfjcResult(None, None, |
|
321 |
((constants.JOB_STATUS_QUEUED, ), None)) |
|
322 |
|
|
323 |
cbs.AddWfjcResult((constants.JOB_STATUS_QUEUED, ), None, |
|
324 |
constants.JOB_NOTCHANGED) |
|
325 |
|
|
326 |
cbs.AddWfjcResult((constants.JOB_STATUS_QUEUED, ), None, |
|
327 |
((constants.JOB_STATUS_RUNNING, ), |
|
328 |
[(1, utils.SplitTime(1273491611.0), |
|
329 |
constants.ELOG_MESSAGE, "Step 1"), |
|
330 |
(2, utils.SplitTime(1273491615.9), |
|
331 |
constants.ELOG_MESSAGE, "Step 2"), |
|
332 |
(3, utils.SplitTime(1273491625.02), |
|
333 |
constants.ELOG_MESSAGE, "Step 3"), |
|
334 |
(4, utils.SplitTime(1273491635.05), |
|
335 |
constants.ELOG_MESSAGE, "Step 4"), |
|
336 |
(37, utils.SplitTime(1273491645.0), |
|
337 |
constants.ELOG_MESSAGE, "Step 5"), |
|
338 |
(203, utils.SplitTime(127349155.0), |
|
339 |
constants.ELOG_MESSAGE, "Step 6")])) |
|
340 |
|
|
341 |
cbs.AddWfjcResult((constants.JOB_STATUS_RUNNING, ), 203, |
|
342 |
((constants.JOB_STATUS_RUNNING, ), |
|
343 |
[(300, utils.SplitTime(1273491711.01), |
|
344 |
constants.ELOG_MESSAGE, "Step X"), |
|
345 |
(302, utils.SplitTime(1273491815.8), |
|
346 |
constants.ELOG_MESSAGE, "Step Y"), |
|
347 |
(303, utils.SplitTime(1273491925.32), |
|
348 |
constants.ELOG_MESSAGE, "Step Z")])) |
|
349 |
|
|
350 |
cbs.AddWfjcResult((constants.JOB_STATUS_RUNNING, ), 303, |
|
351 |
((constants.JOB_STATUS_SUCCESS, ), None)) |
|
352 |
|
|
353 |
cbs.AddQueryJobsResult(constants.JOB_STATUS_SUCCESS, |
|
354 |
[constants.OP_STATUS_SUCCESS, |
|
355 |
constants.OP_STATUS_SUCCESS], |
|
356 |
["Hello World", "Foo man bar"]) |
|
357 |
|
|
358 |
self.assertEqual(["Hello World", "Foo man bar"], |
|
359 |
cli.GenericPollJob(job_id, cbs, cbs)) |
|
360 |
cbs.CheckEmpty() |
|
361 |
|
|
362 |
def testJobLost(self): |
|
363 |
job_id = 13746 |
|
364 |
|
|
365 |
cbs = _MockJobPollCb(self, job_id) |
|
366 |
cbs.AddWfjcResult(None, None, constants.JOB_NOTCHANGED) |
|
367 |
cbs.AddWfjcResult(None, None, None) |
|
368 |
self.assertRaises(errors.JobLost, cli.GenericPollJob, job_id, cbs, cbs) |
|
369 |
cbs.CheckEmpty() |
|
370 |
|
|
371 |
def testError(self): |
|
372 |
job_id = 31088 |
|
373 |
|
|
374 |
cbs = _MockJobPollCb(self, job_id) |
|
375 |
cbs.AddWfjcResult(None, None, constants.JOB_NOTCHANGED) |
|
376 |
cbs.AddWfjcResult(None, None, ((constants.JOB_STATUS_ERROR, ), None)) |
|
377 |
cbs.AddQueryJobsResult(constants.JOB_STATUS_ERROR, |
|
378 |
[constants.OP_STATUS_SUCCESS, |
|
379 |
constants.OP_STATUS_ERROR], |
|
380 |
["Hello World", "Error code 123"]) |
|
381 |
self.assertRaises(errors.OpExecError, cli.GenericPollJob, job_id, cbs, cbs) |
|
382 |
cbs.CheckEmpty() |
|
383 |
|
|
384 |
def testError2(self): |
|
385 |
job_id = 22235 |
|
386 |
|
|
387 |
cbs = _MockJobPollCb(self, job_id) |
|
388 |
cbs.AddWfjcResult(None, None, ((constants.JOB_STATUS_ERROR, ), None)) |
|
389 |
encexc = errors.EncodeException(errors.LockError("problem")) |
|
390 |
cbs.AddQueryJobsResult(constants.JOB_STATUS_ERROR, |
|
391 |
[constants.OP_STATUS_ERROR], [encexc]) |
|
392 |
self.assertRaises(errors.LockError, cli.GenericPollJob, job_id, cbs, cbs) |
|
393 |
cbs.CheckEmpty() |
|
394 |
|
|
395 |
def testWeirdError(self): |
|
396 |
job_id = 28847 |
|
397 |
|
|
398 |
cbs = _MockJobPollCb(self, job_id) |
|
399 |
cbs.AddWfjcResult(None, None, ((constants.JOB_STATUS_ERROR, ), None)) |
|
400 |
cbs.AddQueryJobsResult(constants.JOB_STATUS_ERROR, |
|
401 |
[constants.OP_STATUS_RUNNING, |
|
402 |
constants.OP_STATUS_RUNNING], |
|
403 |
[None, None]) |
|
404 |
self.assertRaises(errors.OpExecError, cli.GenericPollJob, job_id, cbs, cbs) |
|
405 |
cbs.CheckEmpty() |
|
406 |
|
|
407 |
def testCancel(self): |
|
408 |
job_id = 4275 |
|
409 |
|
|
410 |
cbs = _MockJobPollCb(self, job_id) |
|
411 |
cbs.AddWfjcResult(None, None, constants.JOB_NOTCHANGED) |
|
412 |
cbs.AddWfjcResult(None, None, ((constants.JOB_STATUS_CANCELING, ), None)) |
|
413 |
cbs.AddQueryJobsResult(constants.JOB_STATUS_CANCELING, |
|
414 |
[constants.OP_STATUS_CANCELING, |
|
415 |
constants.OP_STATUS_CANCELING], |
|
416 |
[None, None]) |
|
417 |
self.assertRaises(errors.OpExecError, cli.GenericPollJob, job_id, cbs, cbs) |
|
418 |
cbs.CheckEmpty() |
|
419 |
|
|
420 |
|
|
249 | 421 |
if __name__ == '__main__': |
250 | 422 |
testutils.GanetiTestProgram() |
Also available in: Unified diff