def test_internal(core_retry):
error = mock.Mock(
code=mock.Mock(return_value=grpc.StatusCode.INTERNAL)
)
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is True
core_retry.if_transient_error.assert_called_once_with(error)
def test_unauthenticated(core_retry):
error = mock.Mock(
code=mock.Mock(return_value=grpc.StatusCode.UNAUTHENTICATED)
)
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is False
core_retry.if_transient_error.assert_called_once_with(error)
def test_unavailable(core_retry):
error = mock.Mock(
code=mock.Mock(return_value=grpc.StatusCode.UNAVAILABLE)
)
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is True
core_retry.if_transient_error.assert_called_once_with(error)
def test_aborted(core_retry):
error = mock.Mock(
spec=grpc.Call,
code=mock.Mock(return_value=grpc.StatusCode.ABORTED),
)
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is True
core_retry.if_transient_error.assert_called_once_with(error)
def test_core_says_no_we_say_no(core_retry):
error = object()
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is False
core_retry.if_transient_error.assert_called_once_with(error)
def test_core_says_yes(core_retry):
error = object()
core_retry.if_transient_error.return_value = True
assert _retry.is_transient_error(error) is True
core_retry.if_transient_error.assert_called_once_with(error)
def test_not_a_grpc_call(core_retry):
error = object()
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is False
core_retry.if_transient_error.assert_called_once_with(error)
def test_code_is_not_transient(core_retry):
error = mock.Mock(spec=grpc.Call, code=mock.Mock(return_value=42))
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is False
core_retry.if_transient_error.assert_called_once_with(error)
def test_unknown(core_retry):
error = core_exceptions.Unknown("testing")
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is True
core_retry.if_transient_error.assert_called_once_with(error)
def test_unauthenticated(core_retry):
error = core_exceptions.Unauthenticated("testing")
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is False
core_retry.if_transient_error.assert_called_once_with(error)
def test_internal(core_retry):
error = core_exceptions.InternalServerError("testing")
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is True
core_retry.if_transient_error.assert_called_once_with(error)
def test_unavailable(core_retry):
error = core_exceptions.ServiceUnavailable("testing")
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is True
core_retry.if_transient_error.assert_called_once_with(error)
def test_error_is_not_transient(core_retry):
error = Exception("whatever")
core_retry.if_transient_error.return_value = False
assert _retry.is_transient_error(error) is False
core_retry.if_transient_error.assert_called_once_with(error)
