Synnefo » snf-ganeti

import errno

class TestIgnoreSignals(unittest.TestCase):

  """Test the IgnoreSignals decorator"""

  @staticmethod

  def _Raise(exception):

    raise exception

  @staticmethod

  def _Return(rval):

    return rval

  def testIgnoreSignals(self):

    sock_err_intr = socket.error(errno.EINTR, "Message")

    sock_err_intr.errno = errno.EINTR

    sock_err_inval = socket.error(errno.EINVAL, "Message")

    sock_err_inval.errno = errno.EINVAL

    env_err_intr = EnvironmentError(errno.EINTR, "Message")

    env_err_inval = EnvironmentError(errno.EINVAL, "Message")

    self.assertRaises(socket.error, self._Raise, sock_err_intr)

    self.assertRaises(socket.error, self._Raise, sock_err_inval)

    self.assertRaises(EnvironmentError, self._Raise, env_err_intr)

    self.assertRaises(EnvironmentError, self._Raise, env_err_inval)

    self.assertEquals(utils.IgnoreSignals(self._Raise, sock_err_intr), None)

    self.assertEquals(utils.IgnoreSignals(self._Raise, env_err_intr), None)

    self.assertRaises(socket.error, utils.IgnoreSignals, self._Raise,

                      sock_err_inval)

    self.assertRaises(EnvironmentError, utils.IgnoreSignals, self._Raise,

                      env_err_inval)

    self.assertEquals(utils.IgnoreSignals(self._Return, True), True)

    self.assertEquals(utils.IgnoreSignals(self._Return, 33), 33)