Signed-off-by: Eric Zhao <sczyh16@gmail.com>master
@@ -19,6 +19,7 @@ import com.alibaba.csp.sentinel.BaseTest; | |||
import com.alibaba.csp.sentinel.Entry; | |||
import com.alibaba.csp.sentinel.EntryType; | |||
import com.alibaba.csp.sentinel.adapter.dubbo.config.DubboConfig; | |||
import com.alibaba.csp.sentinel.adapter.dubbo.fallback.DefaultDubboFallback; | |||
import com.alibaba.csp.sentinel.adapter.dubbo.fallback.DubboFallback; | |||
import com.alibaba.csp.sentinel.adapter.dubbo.fallback.DubboFallbackRegistry; | |||
import com.alibaba.csp.sentinel.adapter.dubbo.provider.DemoService; | |||
@@ -82,6 +83,7 @@ public class SentinelDubboConsumerFilterTest extends BaseTest { | |||
@After | |||
public void cleanUp() { | |||
cleanUpAll(); | |||
DubboFallbackRegistry.setConsumerFallback(new DefaultDubboFallback()); | |||
} | |||
public void initFlowRule(String resource) { | |||
@@ -20,7 +20,9 @@ import com.alibaba.csp.sentinel.slots.block.SentinelRpcException; | |||
import com.alibaba.csp.sentinel.slots.block.flow.FlowException; | |||
import org.apache.dubbo.rpc.AsyncRpcResult; | |||
import org.apache.dubbo.rpc.Result; | |||
import org.junit.After; | |||
import org.junit.Assert; | |||
import org.junit.Before; | |||
import org.junit.Test; | |||
/** | |||
@@ -28,6 +30,16 @@ import org.junit.Test; | |||
*/ | |||
public class DubboFallbackRegistryTest { | |||
@Before | |||
public void setUp() { | |||
DubboFallbackRegistry.setConsumerFallback(new DefaultDubboFallback()); | |||
} | |||
@After | |||
public void tearDown() { | |||
DubboFallbackRegistry.setConsumerFallback(new DefaultDubboFallback()); | |||
} | |||
@Test(expected = SentinelRpcException.class) | |||
public void testDefaultFallback() { | |||
// Test for default. | |||