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)