Viacheslav Klimov
11 months ago
committed by
GitHub
GitHub
2 changed files with 393 additions and 6 deletions
@ -0,0 +1,128 @@ |
|||
/** |
|||
* Copyright © 2016-2025 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.server.msa.connectivity; |
|||
|
|||
import org.apache.hc.client5.http.impl.classic.CloseableHttpClient; |
|||
import org.apache.hc.client5.http.impl.classic.HttpClients; |
|||
import org.apache.hc.client5.http.io.HttpClientConnectionManager; |
|||
import org.apache.hc.client5.http.impl.io.PoolingHttpClientConnectionManagerBuilder; |
|||
import org.apache.hc.client5.http.ssl.DefaultClientTlsStrategy; |
|||
import org.apache.hc.client5.http.ssl.HostnameVerificationPolicy; |
|||
import org.apache.hc.client5.http.ssl.NoopHostnameVerifier; |
|||
import org.apache.hc.core5.ssl.SSLContexts; |
|||
import org.springframework.http.client.HttpComponentsClientHttpRequestFactory; |
|||
import org.springframework.web.client.RestTemplate; |
|||
import org.testcontainers.shaded.org.apache.commons.lang3.RandomStringUtils; |
|||
import org.testng.annotations.AfterMethod; |
|||
import org.testng.annotations.BeforeClass; |
|||
import org.testng.annotations.BeforeMethod; |
|||
import org.testng.annotations.Test; |
|||
import org.thingsboard.rest.client.RestClient; |
|||
import org.thingsboard.server.common.data.Device; |
|||
import org.thingsboard.server.common.data.alarm.Alarm; |
|||
import org.thingsboard.server.common.data.alarm.AlarmInfo; |
|||
import org.thingsboard.server.common.data.alarm.AlarmSearchStatus; |
|||
import org.thingsboard.server.common.data.alarm.AlarmSeverity; |
|||
import org.thingsboard.server.common.data.page.PageData; |
|||
import org.thingsboard.server.common.data.page.TimePageLink; |
|||
import org.thingsboard.server.msa.AbstractContainerTest; |
|||
import org.thingsboard.server.msa.TestProperties; |
|||
|
|||
import javax.net.ssl.SSLContext; |
|||
import java.util.List; |
|||
|
|||
import static org.assertj.core.api.Assertions.assertThat; |
|||
import static org.thingsboard.server.msa.prototypes.DevicePrototypes.defaultDevicePrototype; |
|||
|
|||
public class JavaRestClientTest extends AbstractContainerTest { |
|||
|
|||
private RestClient restClient; |
|||
|
|||
@BeforeClass |
|||
public void beforeClass() throws Exception { |
|||
SSLContext ssl = SSLContexts.custom() |
|||
.loadTrustMaterial((chain, authType) -> true) |
|||
.build(); |
|||
|
|||
var tls = new DefaultClientTlsStrategy( |
|||
ssl, |
|||
HostnameVerificationPolicy.CLIENT, |
|||
NoopHostnameVerifier.INSTANCE |
|||
); |
|||
|
|||
HttpClientConnectionManager cm = PoolingHttpClientConnectionManagerBuilder.create() |
|||
.setTlsSocketStrategy(tls) |
|||
.build(); |
|||
|
|||
CloseableHttpClient httpClient = HttpClients.custom() |
|||
.setConnectionManager(cm) |
|||
.build(); |
|||
|
|||
RestTemplate rt = new RestTemplate(new HttpComponentsClientHttpRequestFactory(httpClient)); |
|||
restClient = new RestClient(rt, TestProperties.getBaseUrl()); |
|||
} |
|||
|
|||
@BeforeMethod |
|||
public void setUp() throws Exception { |
|||
restClient.login("tenant@thingsboard.org", "tenant"); |
|||
} |
|||
|
|||
@AfterMethod |
|||
public void tearDown() { |
|||
} |
|||
|
|||
@Test |
|||
public void testGetAlarmsV2() { |
|||
Device device = restClient.saveDevice(defaultDevicePrototype(RandomStringUtils.randomAlphabetic(5))); |
|||
assertThat(device).isNotNull(); |
|||
|
|||
String type = "High temp" + RandomStringUtils.randomAlphabetic(5); |
|||
Alarm alarm = Alarm.builder() |
|||
.originator(device.getId()) |
|||
.severity(AlarmSeverity.CRITICAL) |
|||
.type(type) |
|||
.build(); |
|||
restClient.saveAlarm(alarm); |
|||
|
|||
// get /api/v2/alarm
|
|||
PageData<AlarmInfo> alarmsV2 = restClient.getAlarmsV2(device.getId(), null, null, List.of(type), null, new TimePageLink(10, 0)); |
|||
assertThat(alarmsV2.getData()).hasSize(1); |
|||
|
|||
PageData<AlarmInfo> activeAlarms = restClient.getAlarmsV2(device.getId(), List.of(AlarmSearchStatus.ACTIVE), null, List.of(type), null, new TimePageLink(10, 0)); |
|||
assertThat(activeAlarms.getData()).hasSize(1); |
|||
|
|||
PageData<AlarmInfo> cleared = restClient.getAlarmsV2(device.getId(), List.of(AlarmSearchStatus.CLEARED), null, List.of(type), null, new TimePageLink(10, 0)); |
|||
assertThat(cleared.getData()).hasSize(0); |
|||
|
|||
PageData<AlarmInfo> activeAndClearedAlarms = restClient.getAlarmsV2(device.getId(), List.of(AlarmSearchStatus.CLEARED, AlarmSearchStatus.ACTIVE), null, null, null, new TimePageLink(10, 0)); |
|||
assertThat(activeAndClearedAlarms.getData()).hasSize(1); |
|||
|
|||
// get /api/v2/alarms
|
|||
PageData<AlarmInfo> allAlarmsV2 = restClient.getAllAlarmsV2(List.of(AlarmSearchStatus.ACTIVE), null, List.of(type), null, new TimePageLink(10, 0)); |
|||
assertThat(allAlarmsV2.getData()).hasSize(1); |
|||
|
|||
PageData<AlarmInfo> allClearedAlarmsV2 = restClient.getAllAlarmsV2(List.of(AlarmSearchStatus.CLEARED), null, List.of(type), null, new TimePageLink(10, 0)); |
|||
assertThat(allClearedAlarmsV2.getData()).hasSize(0); |
|||
|
|||
// get /api/alarms
|
|||
PageData<AlarmInfo> allAlarms = restClient.getAllAlarms(AlarmSearchStatus.ACTIVE, null, new TimePageLink(10, 0), null); |
|||
assertThat(allAlarms.getData()).hasSize(1); |
|||
|
|||
PageData<AlarmInfo> allClearedAlarms = restClient.getAllAlarms(AlarmSearchStatus.CLEARED, null, new TimePageLink(10, 0), null); |
|||
assertThat(allClearedAlarms.getData()).hasSize(0); |
|||
|
|||
} |
|||
} |
|||
Loading…
Reference in new issue