com.alibaba.csp.sentinel.slots.block.flow.FlowException Java Examples

@Test
public void testTraceException() {
    ClusterNode clusterNode = new ClusterNode();

    Exception exception = new RuntimeException("test");

    // test count<=0, no exceptionQps added
    clusterNode.trace(exception, 0);
    clusterNode.trace(exception, -1);
    assertEquals(0, clusterNode.exceptionQps(), 0.01);
    assertEquals(0, clusterNode.totalException());

    // test count=1, not BlockException, 1 exceptionQps added
    clusterNode.trace(exception, 1);
    assertEquals(1, clusterNode.exceptionQps(), 0.01);
    assertEquals(1, clusterNode.totalException());

    // test count=1, BlockException, no exceptionQps added
    FlowException flowException = new FlowException("flow");
    clusterNode.trace(flowException, 1);
    assertEquals(1, clusterNode.exceptionQps(), 0.01);
    assertEquals(1, clusterNode.totalException());
}
 

@Test
public void testFlowControlFallbackResponse() throws Exception {
    ZuulBlockFallbackProvider fallbackProvider = ZuulBlockFallbackManager.getFallbackProvider(ALL_ROUTE);
    BlockResponse clientHttpResponse = fallbackProvider.fallbackResponse(ALL_ROUTE,
        new FlowException("flow exception"));
    Assert.assertEquals(clientHttpResponse.getCode(), 429);
}
 

@Test
public void onBlocked() throws Exception {
    FakeMetricExtension extension = new FakeMetricExtension();
    MetricExtensionProvider.addMetricExtension(extension);

    MetricEntryCallback entryCallback = new MetricEntryCallback();
    StringResourceWrapper resourceWrapper = new StringResourceWrapper("resource", EntryType.OUT);
    Context context = mock(Context.class);
    when(context.getOrigin()).thenReturn("origin1");
    int count = 2;
    Object[] args = {"args1", "args2"};
    entryCallback.onBlocked(new FlowException("xx"), context, resourceWrapper, null, count, args);
    Assert.assertEquals(extension.block, count);
}
 

@Test(expected = FlowException.class)
public void testBlockHandlerNotFound() {
    assertThat(fooService.baz("Sentinel")).isEqualTo("cheers, Sentinel");
    String resourceName = "apiBaz";
    ClusterNode cn = ClusterBuilderSlot.getClusterNode(resourceName);
    assertThat(cn).isNotNull();
    assertThat(cn.passQps()).isPositive();

    FlowRuleManager.loadRules(Collections.singletonList(
            new FlowRule(resourceName).setCount(0)
    ));
    fooService.baz("Sentinel");
}
 

@Test
public void testCustomFallback() {
    BlockException ex = new FlowException("xxx");
    DubboFallbackRegistry.setConsumerFallback(new DubboFallback() {
        @Override
        public Result handle(Invoker<?> invoker, Invocation invocation, BlockException e) {
            return new RpcResult("Error: " + e.getClass().getName());
        }
    });
    Result result = DubboFallbackRegistry.getConsumerFallback()
        .handle(null, null, ex);
    Assert.assertFalse("The invocation should not fail", result.hasException());
    Assert.assertEquals("Error: " + ex.getClass().getName(), result.getValue());
}
 

@Test(expected = SentinelRpcException.class)
public void testDefaultFallback() {
    // Test for default.
    BlockException ex = new FlowException("xxx");
    DubboFallbackRegistry.getConsumerFallback()
        .handle(null, null, ex);
}
 

@Test
public void testRegisterProvider() throws Exception {
    MyNullResponseFallBackProvider myNullResponseFallBackProvider = new MyNullResponseFallBackProvider();
    ZuulBlockFallbackManager.registerProvider(myNullResponseFallBackProvider);
    Assert.assertEquals(myNullResponseFallBackProvider.getRoute(), ROUTE);
    Assert.assertNull(myNullResponseFallBackProvider.fallbackResponse(ROUTE, new FlowException("flow ex")));
}
 

@Test
public void testFlowControlFallbackResponse() throws Exception {
    ZuulBlockFallbackProvider fallbackProvider = ZuulBlockFallbackManager.getFallbackProvider(ALL_ROUTE);
    BlockResponse clientHttpResponse = fallbackProvider.fallbackResponse(ALL_ROUTE,
        new FlowException("flow exception"));
    Assert.assertEquals(clientHttpResponse.getCode(), 429);
}
 

@Test
public void testCustomFallback() {
    BlockException ex = new FlowException("xxx");
    DubboFallbackRegistry.setConsumerFallback(
            (invoker, invocation, e) -> AsyncRpcResult.newDefaultAsyncResult("Error: " + e.getClass().getName(), invocation));
    Result result = DubboFallbackRegistry.getConsumerFallback()
        .handle(null, null, ex);
    Assert.assertFalse("The invocation should not fail", result.hasException());
    Assert.assertEquals("Error: " + ex.getClass().getName(), result.getValue());
}
 

@Test(expected = SentinelRpcException.class)
public void testDefaultFallback() {
    // Test for default.
    BlockException ex = new FlowException("xxx");
    DubboFallbackRegistry.getConsumerFallback()
        .handle(null, null, ex);
}
 

@Test
public void testRegisterProvider() throws Exception {
    MyNullResponseFallBackProvider myNullResponseFallBackProvider = new MyNullResponseFallBackProvider();
    ZuulBlockFallbackManager.registerProvider(myNullResponseFallBackProvider);
    Assert.assertEquals(myNullResponseFallBackProvider.getRoute(), ROUTE);
    Assert.assertNull(myNullResponseFallBackProvider.fallbackResponse(ROUTE, new FlowException("flow ex")));
}
 

@Test
public void testFlowControlFallbackResponse() throws Exception {
    ZuulBlockFallbackProvider fallbackProvider = ZuulBlockFallbackManager.getFallbackProvider(ALL_ROUTE);
    BlockResponse clientHttpResponse = fallbackProvider.fallbackResponse(ALL_ROUTE,
        new FlowException("flow exception"));
    Assert.assertEquals(clientHttpResponse.getCode(), 429);
}
 

@Test
public void testCustomFallback() {
    BlockException ex = new FlowException("xxx");
    DubboFallbackRegistry.setConsumerFallback(
        (invoker, invocation, e) -> new RpcResult("Error: " + e.getClass().getName()));
    Result result = DubboFallbackRegistry.getConsumerFallback()
        .handle(null, null, ex);
    Assert.assertFalse("The invocation should not fail", result.hasException());
    Assert.assertEquals("Error: " + ex.getClass().getName(), result.getValue());
}
 

@Test(expected = SentinelRpcException.class)
public void testDefaultFallback() {
    // Test for default.
    BlockException ex = new FlowException("xxx");
    DubboFallbackRegistry.getConsumerFallback()
        .handle(null, null, ex);
}
 

@Test
public void onBlocked() throws Exception {
    FakeMetricExtension extension = new FakeMetricExtension();
    MetricExtensionProvider.addMetricExtension(extension);

    MetricEntryCallback entryCallback = new MetricEntryCallback();
    StringResourceWrapper resourceWrapper = new StringResourceWrapper("resource", EntryType.OUT);
    Context context = mock(Context.class);
    when(context.getOrigin()).thenReturn("origin1");
    int count = 2;
    Object[] args = {"args1", "args2"};
    entryCallback.onBlocked(new FlowException("xx"), context, resourceWrapper, null, count, args);
    Assert.assertEquals(extension.block, count);
}
 

@Test
public void testCustomFallback() {
    BlockException ex = new FlowException("xxx");
    DubboFallbackRegistry.setConsumerFallback(new DubboFallback() {
        @Override
        public Result handle(Invoker<?> invoker, Invocation invocation, BlockException e) {
            return new RpcResult("Error: " + e.getClass().getName());
        }
    });
    Result result = DubboFallbackRegistry.getConsumerFallback()
        .handle(null, null, ex);
    Assert.assertFalse("The invocation should not fail", result.hasException());
    Assert.assertEquals("Error: " + ex.getClass().getName(), result.getValue());
}
 

@Test(expected = SentinelRpcException.class)
public void testDefaultFallback() {
    // Test for default.
    BlockException ex = new FlowException("xxx");
    DubboFallbackRegistry.getConsumerFallback()
        .handle(null, null, ex);
}
 

@Test
public void testCustomFallback() {
    BlockException ex = new FlowException("xxx");
    DubboFallbackRegistry.setConsumerFallback(
        (invoker, invocation, e) -> new RpcResult("Error: " + e.getClass().getName()));
    Result result = DubboFallbackRegistry.getConsumerFallback()
        .handle(null, null, ex);
    Assert.assertFalse("The invocation should not fail", result.hasException());
    Assert.assertEquals("Error: " + ex.getClass().getName(), result.getValue());
}
 

@Test(expected = SentinelRpcException.class)
public void testDefaultFallback() {
    // Test for default.
    BlockException ex = new FlowException("xxx");
    DubboFallbackRegistry.getConsumerFallback()
        .handle(null, null, ex);
}
 

@Test
public void testRegisterProvider() throws Exception {
    MyNullResponseFallBackProvider myNullResponseFallBackProvider = new MyNullResponseFallBackProvider();
    ZuulBlockFallbackManager.registerProvider(myNullResponseFallBackProvider);
    Assert.assertEquals(myNullResponseFallBackProvider.getRoute(), ROUTE);
    Assert.assertNull(myNullResponseFallBackProvider.fallbackResponse(ROUTE, new FlowException("flow ex")));
}
 

@Test(expected = SentinelRpcException.class)
public void testDefaultOkHttpFallback() {
    BlockException e = new FlowException("xxx");
    OkHttpFallback fallback = new DefaultOkHttpFallback();
    fallback.handle(null, null, e);
}
 

@GetMapping("/error")
public Mono<?> apiError() {
    return Mono.error(new FlowException("testWebFluxError"));
}
 

@GetMapping("/error")
public Mono<?> apiError() {
    return Mono.error(new FlowException("testWebFluxError"));
}