def test_should_retry_upload_five_times(self): counter = [0] def upload_func(*args, **kwargs): counter[0] += 1 raise Exception() s3upload.upload_part(upload_func, '_', '_', '_') self.assertEqual(counter[0], 5)
def test_should_return_error_when_upload_func_raises_error(self): def upload_func(*args, **kwargs): raise Exception() with self.assertRaises(threading.ThreadError): raise s3upload.upload_part(upload_func, '_', '_', '_')