|
|
|
@ -16,16 +16,15 @@
|
|
|
|
|
|
|
|
|
|
package org.springframework.cloud.alibaba.sentinel;
|
|
|
|
|
|
|
|
|
|
import com.alibaba.csp.sentinel.adapter.servlet.config.WebServletConfig;
|
|
|
|
|
import com.alibaba.csp.sentinel.config.SentinelConfig;
|
|
|
|
|
import com.alibaba.csp.sentinel.log.LogBase;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.BlockException;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.degrade.DegradeRule;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.degrade.DegradeRuleManager;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager;
|
|
|
|
|
import com.alibaba.csp.sentinel.transport.config.TransportConfig;
|
|
|
|
|
import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
|
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
|
|
import static org.junit.Assert.assertNotNull;
|
|
|
|
|
import static org.mockito.Mockito.mock;
|
|
|
|
|
import static org.springframework.boot.test.context.SpringBootTest.WebEnvironment.RANDOM_PORT;
|
|
|
|
|
|
|
|
|
|
import java.util.Arrays;
|
|
|
|
|
import java.util.Map;
|
|
|
|
|
|
|
|
|
|
import org.junit.Before;
|
|
|
|
|
import org.junit.Test;
|
|
|
|
|
import org.junit.runner.RunWith;
|
|
|
|
@ -51,14 +50,16 @@ import org.springframework.test.context.junit4.SpringRunner;
|
|
|
|
|
import org.springframework.web.client.RestClientException;
|
|
|
|
|
import org.springframework.web.client.RestTemplate;
|
|
|
|
|
|
|
|
|
|
import java.util.Arrays;
|
|
|
|
|
import java.util.Map;
|
|
|
|
|
|
|
|
|
|
import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
|
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
|
|
import static org.junit.Assert.assertNotNull;
|
|
|
|
|
import static org.mockito.Mockito.mock;
|
|
|
|
|
import static org.springframework.boot.test.context.SpringBootTest.WebEnvironment.RANDOM_PORT;
|
|
|
|
|
import com.alibaba.csp.sentinel.adapter.servlet.config.WebServletConfig;
|
|
|
|
|
import com.alibaba.csp.sentinel.config.SentinelConfig;
|
|
|
|
|
import com.alibaba.csp.sentinel.log.LogBase;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.BlockException;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.degrade.DegradeRule;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.degrade.DegradeRuleManager;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
|
|
|
|
|
import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager;
|
|
|
|
|
import com.alibaba.csp.sentinel.transport.config.TransportConfig;
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* @author <a href="mailto:fangjian0423@gmail.com">Jim</a>
|
|
|
|
@ -112,7 +113,7 @@ public class SentinelAutoConfigurationTests {
|
|
|
|
|
FlowRule rule = new FlowRule();
|
|
|
|
|
rule.setGrade(RuleConstant.FLOW_GRADE_QPS);
|
|
|
|
|
rule.setCount(0);
|
|
|
|
|
rule.setResource(url);
|
|
|
|
|
rule.setResource("GET:" + url);
|
|
|
|
|
rule.setLimitApp("default");
|
|
|
|
|
rule.setControlBehavior(RuleConstant.CONTROL_BEHAVIOR_DEFAULT);
|
|
|
|
|
rule.setStrategy(RuleConstant.STRATEGY_DIRECT);
|
|
|
|
@ -120,7 +121,7 @@ public class SentinelAutoConfigurationTests {
|
|
|
|
|
|
|
|
|
|
DegradeRule degradeRule = new DegradeRule();
|
|
|
|
|
degradeRule.setGrade(RuleConstant.DEGRADE_GRADE_EXCEPTION_COUNT);
|
|
|
|
|
degradeRule.setResource(url + "/test");
|
|
|
|
|
degradeRule.setResource("GET:" + url + "/test");
|
|
|
|
|
degradeRule.setCount(0);
|
|
|
|
|
degradeRule.setTimeWindow(60);
|
|
|
|
|
DegradeRuleManager.loadRules(Arrays.asList(degradeRule));
|
|
|
|
|