def test_wrap_retry_w_login_unsuccessful(self): func = mock.Mock() def side_effect(*args, **kwargs): exception = vmware_exceptions.VMwareDriverException() exception.message = dvs_const.LOGIN_PROBLEM_TEXT raise exception func.side_effect = side_effect def double(*args, **kwargs): return func(*args, **kwargs) self.assertRaises(vmware_exceptions.VMwareDriverException, dvs_util.wrap_retry(double)) self.assertEqual(3, func.call_count)
def test_wrap_retry_w_login_unsuccessful(self): func = mock.Mock() def side_effect(*args, **kwargs): exception = vmware_exceptions.VMwareDriverException() exception.message = dvs_const.LOGIN_PROBLEM_TEXT raise exception func.side_effect = side_effect def double(*args, **kwargs): return func(*args, **kwargs) self.assertRaises( vmware_exceptions.VMwareDriverException, dvs_util.wrap_retry(double)) self.assertEqual(3, func.call_count)
def test_wrap_retry_w_concurrent_modification(self): func = mock.Mock() func.side_effect = [ exceptions.VMWareDVSException( message=dvs_const.CONCURRENT_MODIFICATION_TEXT, type='TestException', cause='Test cause'), exceptions.VMWareDVSException(message='Some exception text', type='TestException', cause='Test cause') ] def double(*args, **kwargs): return func(*args, **kwargs) self.assertRaises(exceptions.VMWareDVSException, dvs_util.wrap_retry(double)) self.assertEqual(2, func.call_count)
def test_wrap_retry_w_concurrent_modification(self): func = mock.Mock() func.side_effect = [ exceptions.VMWareDVSException( message=dvs_const.CONCURRENT_MODIFICATION_TEXT, type='TestException', cause='Test cause' ), exceptions.VMWareDVSException( message='Some exception text', type='TestException', cause='Test cause' ) ] def double(*args, **kwargs): return func(*args, **kwargs) self.assertRaises( exceptions.VMWareDVSException, dvs_util.wrap_retry(double)) self.assertEqual(2, func.call_count)