Java Code Examples for com.alibaba.csp.sentinel.config.SentinelConfig#setConfig()

The following examples show how to use com.alibaba.csp.sentinel.config.SentinelConfig#setConfig() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: DubboUtilsTest.java From Sentinel with Apache License 2.0

6 votes

@Test
public void testGetInterfaceName() {

    URL url = URL.valueOf("dubbo://127.0.0.1:2181")
            .addParameter(CommonConstants.VERSION_KEY, "1.0.0")
            .addParameter(CommonConstants.GROUP_KEY, "grp1")
            .addParameter(CommonConstants.INTERFACE_KEY, DemoService.class.getName());
    Invoker invoker = mock(Invoker.class);
    when(invoker.getUrl()).thenReturn(url);
    when(invoker.getInterface()).thenReturn(DemoService.class);

    SentinelConfig.setConfig(DUBBO_INTERFACE_GROUP_VERSION_ENABLED, "false");
    assertEquals("com.alibaba.csp.sentinel.adapter.dubbo.provider.DemoService", DubboUtils.getInterfaceName(invoker));

    SentinelConfig.setConfig(DUBBO_INTERFACE_GROUP_VERSION_ENABLED, "true");
    assertEquals("com.alibaba.csp.sentinel.adapter.dubbo.provider.DemoService:1.0.0:grp1", DubboUtils.getInterfaceName(invoker));

}

Example 2

Source File: DubboUtilsTest.java From Sentinel with Apache License 2.0

6 votes

@Test
public void testGetResourceNameWithPrefix() throws NoSuchMethodException {
    Invoker invoker = mock(Invoker.class);
    when(invoker.getInterface()).thenReturn(DemoService.class);

    Invocation invocation = mock(Invocation.class);
    Method method = DemoService.class.getDeclaredMethod("sayHello", String.class, int.class);
    when(invocation.getMethodName()).thenReturn(method.getName());
    when(invocation.getParameterTypes()).thenReturn(method.getParameterTypes());

    //test with default prefix
    String resourceName = DubboUtils.getResourceName(invoker, invocation, DubboConfig.getDubboProviderPrefix());
    assertEquals("dubbo:provider:com.alibaba.csp.sentinel.adapter.dubbo.provider.DemoService:sayHello(java.lang.String,int)", resourceName);
    resourceName = DubboUtils.getResourceName(invoker, invocation, DubboConfig.getDubboConsumerPrefix());
    assertEquals("dubbo:consumer:com.alibaba.csp.sentinel.adapter.dubbo.provider.DemoService:sayHello(java.lang.String,int)", resourceName);


    //test with custom prefix
    SentinelConfig.setConfig(DubboConfig.DUBBO_PROVIDER_PREFIX, "my:dubbo:provider:");
    SentinelConfig.setConfig(DubboConfig.DUBBO_CONSUMER_PREFIX, "my:dubbo:consumer:");
    resourceName = DubboUtils.getResourceName(invoker, invocation, DubboConfig.getDubboProviderPrefix());
    assertEquals("my:dubbo:provider:com.alibaba.csp.sentinel.adapter.dubbo.provider.DemoService:sayHello(java.lang.String,int)", resourceName);
    resourceName = DubboUtils.getResourceName(invoker, invocation, DubboConfig.getDubboConsumerPrefix());
    assertEquals("my:dubbo:consumer:com.alibaba.csp.sentinel.adapter.dubbo.provider.DemoService:sayHello(java.lang.String,int)", resourceName);

}

Example 3

Source File: HeartbeatSenderInitFuncTest.java From Sentinel with Apache License 2.0

6 votes

@Test
public void testRetrieveInterval() {
    HeartbeatSender sender = mock(HeartbeatSender.class);

    long senderInterval = 5666;
    long configInterval = 6777;

    when(sender.intervalMs()).thenReturn(senderInterval);

    HeartbeatSenderInitFunc func = new HeartbeatSenderInitFunc();
    assertEquals(senderInterval, func.retrieveInterval(sender));

    // Invalid interval.
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_INTERVAL_MS, "-1");
    assertEquals(senderInterval, func.retrieveInterval(sender));

    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_INTERVAL_MS, String.valueOf(configInterval));
    assertEquals(configInterval, func.retrieveInterval(sender));
}

Example 4

Source File: TransportConfigTest.java From Sentinel with Apache License 2.0

6 votes

@Test
public void testGetHeartbeatApiPath() {
    // use default heartbeat api path
    assertTrue(StringUtil.isNotEmpty(TransportConfig.getHeartbeatApiPath()));
    assertEquals(TransportConfig.HEARTBEAT_DEFAULT_PATH, TransportConfig.getHeartbeatApiPath());

    // config heartbeat api path
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_API_PATH, "/demo");
    assertTrue(StringUtil.isNotEmpty(TransportConfig.getHeartbeatApiPath()));
    assertEquals("/demo", TransportConfig.getHeartbeatApiPath());

    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_API_PATH, "demo/registry");
    assertEquals("/demo/registry", TransportConfig.getHeartbeatApiPath());

    SentinelConfig.removeConfig(TransportConfig.HEARTBEAT_API_PATH);
    assertEquals(TransportConfig.HEARTBEAT_DEFAULT_PATH, TransportConfig.getHeartbeatApiPath());
}

Example 5

Source File: SentinelHealthIndicatorTests.java From spring-cloud-alibaba with Apache License 2.0

5 votes

@Test
public void testSentinelDashboardConfiguredSuccess() throws Exception {
	when(sentinelProperties.isEnabled()).thenReturn(true);
	SentinelConfig.setConfig(TransportConfig.CONSOLE_SERVER, "localhost:8080");
	when(heartbeatSender.sendHeartbeat()).thenReturn(true);

	Health health = sentinelHealthIndicator.health();

	assertThat(health.getStatus()).isEqualTo(Status.UP);
}

Example 6

Source File: SentinelHealthIndicatorTests.java From spring-cloud-alibaba with Apache License 2.0

5 votes

@Before
public void setUp() {
	beanFactory = mock(DefaultListableBeanFactory.class);
	sentinelProperties = mock(SentinelProperties.class);
	sentinelHealthIndicator = new SentinelHealthIndicator(beanFactory,
			sentinelProperties);

	SentinelConfig.setConfig(TransportConfig.CONSOLE_SERVER, "");

	heartbeatSender = mock(HeartbeatSender.class);
	Field heartbeatSenderField = ReflectionUtils
			.findField(HeartbeatSenderProvider.class, "heartbeatSender");
	heartbeatSenderField.setAccessible(true);
	ReflectionUtils.setField(heartbeatSenderField, null, heartbeatSender);
}

Example 7

Source File: BaseTest.java From Sentinel with Apache License 2.0

5 votes

private void clearDubboContext() {
    SentinelConfig.setConfig("csp.sentinel.dubbo.resource.use.prefix", "false");
    SentinelConfig.setConfig(DubboConfig.DUBBO_PROVIDER_PREFIX, "");
    SentinelConfig.setConfig(DubboConfig.DUBBO_CONSUMER_PREFIX, "");
    SentinelConfig.setConfig(DubboConfig.DUBBO_INTERFACE_GROUP_VERSION_ENABLED, "false");
    DubboFallbackRegistry.setConsumerFallback(new DefaultDubboFallback());
    RpcContext.removeContext();

}

Example 8

Source File: DubboUtilsTest.java From Sentinel with Apache License 2.0

5 votes

@After
public void tearDown() {
    SentinelConfig.setConfig("csp.sentinel.dubbo.resource.use.prefix", "false");
    SentinelConfig.setConfig(DubboConfig.DUBBO_PROVIDER_PREFIX, "");
    SentinelConfig.setConfig(DubboConfig.DUBBO_CONSUMER_PREFIX, "");
    SentinelConfig.setConfig(DUBBO_INTERFACE_GROUP_VERSION_ENABLED, "false");
}

Example 9

Source File: DubboUtilsTest.java From Sentinel with Apache License 2.0

5 votes

@Before
public void setUp() {
    SentinelConfig.setConfig("csp.sentinel.dubbo.resource.use.prefix", "true");
    SentinelConfig.setConfig(DubboConfig.DUBBO_PROVIDER_PREFIX, "");
    SentinelConfig.setConfig(DubboConfig.DUBBO_CONSUMER_PREFIX, "");
    SentinelConfig.setConfig(DUBBO_INTERFACE_GROUP_VERSION_ENABLED, "false");
}

Example 10

Source File: SentinelHealthIndicatorTests.java From spring-cloud-alibaba with Apache License 2.0

5 votes

@Test
public void testSentinelDashboardConfiguredFailed() throws Exception {
	when(sentinelProperties.isEnabled()).thenReturn(true);
	SentinelConfig.setConfig(TransportConfig.CONSOLE_SERVER, "localhost:8080");
	when(heartbeatSender.sendHeartbeat()).thenReturn(false);

	Health health = sentinelHealthIndicator.health();

	assertThat(health.getStatus()).isEqualTo(Status.DOWN);
	assertThat(health.getDetails().get("dashboard")).isEqualTo(
			new Status(Status.DOWN.getCode(), "localhost:8080 can't be connected"));
}

Example 11

Source File: SentinelHealthIndicatorTests.java From spring-cloud-alibaba with Apache License 2.0

5 votes

@Test
public void testSentinelDataSourceFailed() throws Exception {
	when(sentinelProperties.isEnabled()).thenReturn(true);
	SentinelConfig.setConfig(TransportConfig.CONSOLE_SERVER, "localhost:8080");
	when(heartbeatSender.sendHeartbeat()).thenReturn(true);

	Map<String, AbstractDataSource> dataSourceMap = new HashMap<>();

	FileRefreshableDataSource fileDataSource1 = mock(FileRefreshableDataSource.class);
	dataSourceMap.put("ds1-sentinel-file-datasource", fileDataSource1);

	FileRefreshableDataSource fileDataSource2 = mock(FileRefreshableDataSource.class);
	when(fileDataSource2.loadConfig())
			.thenThrow(new RuntimeException("fileDataSource2 error"));
	dataSourceMap.put("ds2-sentinel-file-datasource", fileDataSource2);

	when(beanFactory.getBeansOfType(AbstractDataSource.class))
			.thenReturn(dataSourceMap);

	Health health = sentinelHealthIndicator.health();

	assertThat(health.getStatus()).isEqualTo(Status.DOWN);
	Map<String, Status> dataSourceDetailMap = (Map<String, Status>) health
			.getDetails().get("dataSource");
	assertThat(dataSourceDetailMap.get("ds1-sentinel-file-datasource"))
			.isEqualTo(Status.UP);
	assertThat(dataSourceDetailMap.get("ds2-sentinel-file-datasource"))
			.isEqualTo(new Status(Status.DOWN.getCode(), "fileDataSource2 error"));
}

Example 12

Source File: TransportConfigTest.java From Sentinel with Apache License 2.0

5 votes

@Test
public void testGetHeartbeatClientIp() {
    String clientIp = "10.10.10.10";
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_CLIENT_IP, clientIp);
    // Set heartbeat client ip to system property.
    String ip = TransportConfig.getHeartbeatClientIp();

    assertNotNull(ip);
    assertEquals(clientIp, ip);

    // Set no heartbeat client ip.
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_CLIENT_IP, "");
    assertTrue(StringUtil.isNotEmpty(TransportConfig.getHeartbeatClientIp()));
}

Example 13

Source File: TransportConfigTest.java From Sentinel with Apache License 2.0

5 votes

@Test
public void testGetHeartbeatInterval() {
    long interval = 20000;
    assertNull(TransportConfig.getHeartbeatIntervalMs());
    // Set valid interval.
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_INTERVAL_MS, String.valueOf(interval));
    assertEquals(new Long(interval), TransportConfig.getHeartbeatIntervalMs());
    // Set invalid interval.
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_INTERVAL_MS, "Sentinel");
    assertNull(TransportConfig.getHeartbeatIntervalMs());
}

Example 14

Source File: TransportConfigTest.java From Sentinel-Dashboard-Nacos with Apache License 2.0

5 votes

@Test
public void testGetHeartbeatClientIp() {
    String clientIp = "10.10.10.10";
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_CLIENT_IP, clientIp);
    // Set heartbeat client ip to system property.
    String ip = TransportConfig.getHeartbeatClientIp();

    assertNotNull(ip);
    assertEquals(clientIp, ip);

    // Set no heartbeat client ip.
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_CLIENT_IP, "");
    assertTrue(StringUtil.isNotEmpty(TransportConfig.getHeartbeatClientIp()));
}

Example 15

Source File: TransportConfigTest.java From Sentinel-Dashboard-Nacos with Apache License 2.0

5 votes

@Test
public void testGetHeartbeatInterval() {
    long interval = 20000;
    assertNull(TransportConfig.getHeartbeatIntervalMs());
    // Set valid interval.
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_INTERVAL_MS, String.valueOf(interval));
    assertEquals(new Long(interval), TransportConfig.getHeartbeatIntervalMs());
    // Set invalid interval.
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_INTERVAL_MS, "Sentinel");
    assertNull(TransportConfig.getHeartbeatIntervalMs());
}

Example 16

Source File: HeartbeatSenderInitFunc.java From Sentinel with Apache License 2.0

4 votes

private void setIntervalIfNotExists(long interval) {
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_INTERVAL_MS, String.valueOf(interval));
}

Example 17

Source File: WebServletConfig.java From Sentinel-Dashboard-Nacos with Apache License 2.0

4 votes

public static void setBlockPage(String blockPage) {
    SentinelConfig.setConfig(BLOCK_PAGE, blockPage);
}

Example 18

Source File: HttpHeartbeatSenderTest.java From Sentinel-Dashboard-Nacos with Apache License 2.0

4 votes

private void setAddr(String serverList) {
    SentinelConfig.setConfig(TransportConfig.CONSOLE_SERVER, serverList);
}

Example 19

Source File: AbstractDubboFilterTest.java From Sentinel with Apache License 2.0

4 votes

@Before
public void setUp() {
    SentinelConfig.setConfig("csp.sentinel.dubbo.resource.use.prefix", "true");
    SentinelConfig.setConfig(DubboConfig.DUBBO_PROVIDER_PREFIX, "");
    SentinelConfig.setConfig(DubboConfig.DUBBO_CONSUMER_PREFIX, "");
}

Example 20

Source File: HeartbeatSenderInitFunc.java From Sentinel-Dashboard-Nacos with Apache License 2.0

4 votes

private void setIntervalIfNotExists(long interval) {
    SentinelConfig.setConfig(TransportConfig.HEARTBEAT_INTERVAL_MS, String.valueOf(interval));
}