Synnefo » snf-ganeti » ganeti-local

from ganeti import workerpool

class TestEvaluateJobProcessorResult(unittest.TestCase):

  def testFinished(self):

    depmgr = _FakeDependencyManager()

    job = _IdOnlyFakeJob(30953)

    jqueue._EvaluateJobProcessorResult(depmgr, job,

                                       jqueue._JobProcessor.FINISHED)

    self.assertEqual(depmgr.GetNextNotification(), job.id)

    self.assertRaises(IndexError, depmgr.GetNextNotification)

  def testDefer(self):

    depmgr = _FakeDependencyManager()

    job = _IdOnlyFakeJob(11326, priority=5463)

    try:

      jqueue._EvaluateJobProcessorResult(depmgr, job,

                                         jqueue._JobProcessor.DEFER)

    except workerpool.DeferTask, err:

      self.assertEqual(err.priority, 5463)

    else:

      self.fail("Didn't raise exception")

    self.assertRaises(IndexError, depmgr.GetNextNotification)

  def testWaitdep(self):

    depmgr = _FakeDependencyManager()

    job = _IdOnlyFakeJob(21317)

    jqueue._EvaluateJobProcessorResult(depmgr, job,

                                       jqueue._JobProcessor.WAITDEP)

    self.assertRaises(IndexError, depmgr.GetNextNotification)

  def testOther(self):

    depmgr = _FakeDependencyManager()

    job = _IdOnlyFakeJob(5813)

    self.assertRaises(errors.ProgrammerError,

                      jqueue._EvaluateJobProcessorResult,

                      depmgr, job, "Other result")

    self.assertRaises(IndexError, depmgr.GetNextNotification)