[PATCH master 4/4] jqueue: Use priority for acquiring locks

[Michael Hanselmann]

---
 lib/jqueue.py                  |    2 +-
 test/ganeti.jqueue_unittest.py |   13 +++++++------
 2 files changed, 8 insertions(+), 7 deletions(-)

diff --git a/lib/jqueue.py b/lib/jqueue.py
index 7411f46..e2b4cf4 100644
--- a/lib/jqueue.py
+++ b/lib/jqueue.py
@@ -916,7 +916,7 @@ class _JobProcessor(object):
       # Make sure not to hold queue lock while calling ExecOpCode
       result = self.opexec_fn(op.input,
                               _OpExecCallbacks(self.queue, self.job, op),
-                              timeout=timeout)
+                              timeout=timeout, priority=op.priority)
     except mcpu.LockAcquireTimeout:
       assert timeout is not None, "Received timeout for blocking acquire"
       logging.debug("Couldn't acquire locks in %0.6fs", timeout)
diff --git a/test/ganeti.jqueue_unittest.py b/test/ganeti.jqueue_unittest.py
index ba56ae5..dddf7c9 100755
--- a/test/ganeti.jqueue_unittest.py
+++ b/test/ganeti.jqueue_unittest.py
@@ -449,11 +449,11 @@ class _FakeExecOpCodeForProc:
     self._before_start = before_start
     self._after_start = after_start
 
-  def __call__(self, op, cbs, timeout=None):
+  def __call__(self, op, cbs, timeout=None, priority=None):
     assert isinstance(op, opcodes.OpTestDummy)
 
     if self._before_start:
-      self._before_start(timeout)
+      self._before_start(timeout, priority)
 
     cbs.NotifyStart()
 
@@ -506,7 +506,7 @@ class TestJobProcessor(unittest.TestCase, 
_JobProcessorTestUtils):
       # Create job
       job = self._CreateJob(queue, job_id, ops)
 
-      def _BeforeStart(_):
+      def _BeforeStart(timeout, priority):
         self.assertFalse(queue.IsAcquired())
         self.assertEqual(job.CalcStatus(), constants.JOB_STATUS_WAITLOCK)
 
@@ -660,7 +660,7 @@ class TestJobProcessor(unittest.TestCase, 
_JobProcessorTestUtils):
 
     self.assertEqual(job.CalcStatus(), constants.JOB_STATUS_QUEUED)
 
-    def _BeforeStart(_):
+    def _BeforeStart(timeout, priority):
       self.assertFalse(queue.IsAcquired())
       self.assertEqual(job.CalcStatus(), constants.JOB_STATUS_WAITLOCK)
 
@@ -849,7 +849,7 @@ class TestJobProcessor(unittest.TestCase, 
_JobProcessorTestUtils):
     # Create job
     job = self._CreateJob(queue, 29386, ops)
 
-    def _BeforeStart(_):
+    def _BeforeStart(timeout, priority):
       self.assertFalse(queue.IsAcquired())
       self.assertEqual(job.CalcStatus(), constants.JOB_STATUS_WAITLOCK)
 
@@ -955,7 +955,7 @@ class TestJobProcessorTimeouts(unittest.TestCase, 
_JobProcessorTestUtils):
     self.gave_lock = None
     self.done_lock_before_blocking = False
 
-  def _BeforeStart(self, timeout):
+  def _BeforeStart(self, timeout, priority):
     job = self.job
 
     self.assertFalse(self.queue.IsAcquired())
@@ -965,6 +965,7 @@ class TestJobProcessorTimeouts(unittest.TestCase, 
_JobProcessorTestUtils):
 
     self.assert_(timeout is None or isinstance(timeout, (int, float)))
     self.assertEqual(timeout, ts.last_timeout)
+    self.assertEqual(priority, job.ops[self.curop].priority)
 
     self.gave_lock = True
 
-- 
1.7.0.4