Synnefo » snf-ganeti » ganeti-local

class TestLocking(unittest.TestCase):

  """Testing case for the Lock/Unlock functions"""

  def setUp(self):

    lock_dir = tempfile.mkdtemp(prefix="ganeti.unittest.",

                                suffix=".locking")

    self.old_lock_dir = constants.LOCK_DIR

    constants.LOCK_DIR = lock_dir

  def tearDown(self):

    try:

      ganeti.utils.Unlock("unittest")

    except LockError:

      pass

    shutil.rmtree(constants.LOCK_DIR, ignore_errors=True)

    constants.LOCK_DIR = self.old_lock_dir

  def clean_lock(self, name):

    try:

      ganeti.utils.Unlock("unittest")

    except LockError:

      pass

  def testLock(self):

    self.clean_lock("unittest")

    self.assertEqual(None, Lock("unittest"))

  def testUnlock(self):

    self.clean_lock("unittest")

    ganeti.utils.Lock("unittest")

    self.assertEqual(None, Unlock("unittest"))

  def testDoubleLock(self):

    self.clean_lock("unittest")

    ganeti.utils.Lock("unittest")

    self.assertRaises(LockError, Lock, "unittest")