test: Improve test cases in sentinel-apache-dubbo-adapter module

Signed-off-by: Eric Zhao <sczyh16@gmail.com>
This commit is contained in:
Eric Zhao 2019-12-25 11:54:19 +08:00
parent 87142ce512
commit fe23d5c11f
2 changed files with 14 additions and 0 deletions

View File

@ -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) {

View File

@ -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.