Synnefo » snf-ganeti

#   If you need more than one node, or more than one instance, acquire them at

#   the same time.

LEVEL_CLUSTER = 0

LEVELS = [LEVEL_CLUSTER,

# Lock levels which are modifiable

LEVELS_MOD = [LEVEL_NODE, LEVEL_INSTANCE]

      lu = lu_class(self, op, self.context, sstore)

      lu.ExpandNames()

      assert lu.needed_locks is not None, "needed_locks not set by LU"

    finally:

      self.context.glm.release(locking.LEVEL_CLUSTER)

      self.exclusive_BGL = False

  def testAcquireRelease(self):

    self.GL.acquire(locking.LEVEL_CLUSTER, ['BGL'], shared=1)

    self.assertEquals(self.GL._list_owned(locking.LEVEL_CLUSTER), set(['BGL']))

    self.GL.acquire(locking.LEVEL_NODE, ['n1', 'n2'], shared=1)

    self.assertEquals(self.GL._list_owned(locking.LEVEL_NODE), set(['n1']))

    self.assertEquals(self.GL._list_owned(locking.LEVEL_INSTANCE), set(['i1']))

    self.GL.release(locking.LEVEL_NODE)

    self.GL.release(locking.LEVEL_INSTANCE)

    self.assertRaises(errors.LockError, self.GL.acquire,

                      locking.LEVEL_INSTANCE, ['i5'])

  def testWrongOrder(self):

    self.GL.acquire(locking.LEVEL_CLUSTER, ['BGL'], shared=1)

    self.assertRaises(AssertionError, self.GL.acquire,

                      locking.LEVEL_NODE, ['n1'])

    self.assertRaises(AssertionError, self.GL.acquire,

    self.GL.acquire(locking.LEVEL_CLUSTER, ['BGL'], shared=1)

    Thread(target=self._doLock, args=(locking.LEVEL_INSTANCE, 'i1', 1)).start()

    self.assertEqual(self.done.get(True, 1), 'DONE')

    self.GL.acquire(locking.LEVEL_INSTANCE, ['i3'])

    Thread(target=self._doLock, args=(locking.LEVEL_INSTANCE, 'i1', 1)).start()

    self.assertEqual(self.done.get(True, 1), 'DONE')