def test_retry_failure_reduced_set(self): sleep = MagicMock() self.patch(time, "sleep", sleep) method = MagicMock() method.side_effect = [ {"FailedResourcesMap": {"arn:abc": {"ErrorCode": "ThrottlingException"}}}, {"Result": 32}, ] self.assertEqual( universal_retry(method, ["arn:abc", "arn:def"]), {"Result": 32} ) sleep.assert_called_once() self.assertTrue( method.call_args_list == [ call(ResourceARNList=["arn:abc", "arn:def"]), call(ResourceARNList=["arn:abc"]), ] )
def test_retry_failure_reduced_set(self): sleep = MagicMock() self.patch(time, 'sleep', sleep) method = MagicMock() method.side_effect = [{ 'FailedResourcesMap': { 'arn:abc': { 'ErrorCode': 'ThrottlingException' } } }, { 'Result': 32 }] self.assertEqual(universal_retry(method, ['arn:abc', 'arn:def']), {'Result': 32}) sleep.assert_called_once() self.assertTrue(method.call_args_list == [ call(ResourceARNList=['arn:abc', 'arn:def']), call(ResourceARNList=['arn:abc']) ])
def test_retry_no_error(self): mock = MagicMock() mock.side_effect = [{"Result": 42}] self.assertEqual(universal_retry(mock, []), {"Result": 42}) mock.assert_called_once()