Andrew Shvayka
4 years ago
committed by
GitHub
GitHub
20 changed files with 463 additions and 112 deletions
@ -0,0 +1,40 @@ |
|||
/** |
|||
* Copyright © 2016-2023 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.service.notification; |
|||
|
|||
import org.springframework.context.annotation.Primary; |
|||
import org.springframework.stereotype.Service; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.dao.notification.DefaultNotificationSettingsService; |
|||
import org.thingsboard.server.dao.notification.NotificationRuleService; |
|||
import org.thingsboard.server.dao.notification.NotificationTargetService; |
|||
import org.thingsboard.server.dao.notification.NotificationTemplateService; |
|||
import org.thingsboard.server.dao.settings.AdminSettingsService; |
|||
|
|||
@Service |
|||
@Primary |
|||
public class MockNotificationSettingsService extends DefaultNotificationSettingsService { |
|||
|
|||
public MockNotificationSettingsService(AdminSettingsService adminSettingsService, NotificationTargetService notificationTargetService, NotificationTemplateService notificationTemplateService, NotificationRuleService notificationRuleService) { |
|||
super(adminSettingsService, notificationTargetService, notificationTemplateService, notificationRuleService); |
|||
} |
|||
|
|||
@Override |
|||
public void createDefaultNotificationConfigs(TenantId tenantId) { |
|||
// do nothing
|
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,110 @@ |
|||
/** |
|||
* Copyright © 2016-2023 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.service.stats; |
|||
|
|||
import lombok.SneakyThrows; |
|||
import org.junit.Before; |
|||
import org.junit.Test; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.springframework.test.context.TestPropertySource; |
|||
import org.thingsboard.server.common.data.ApiUsageRecordKey; |
|||
import org.thingsboard.server.common.data.Device; |
|||
import org.thingsboard.server.common.data.id.ApiUsageStateId; |
|||
import org.thingsboard.server.common.data.kv.KvEntry; |
|||
import org.thingsboard.server.controller.AbstractControllerTest; |
|||
import org.thingsboard.server.dao.service.DaoSqlTest; |
|||
import org.thingsboard.server.dao.timeseries.TimeseriesService; |
|||
import org.thingsboard.server.service.apiusage.TbApiUsageStateService; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.List; |
|||
import java.util.concurrent.TimeUnit; |
|||
|
|||
import static org.assertj.core.api.Assertions.assertThat; |
|||
import static org.awaitility.Awaitility.await; |
|||
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status; |
|||
|
|||
@DaoSqlTest |
|||
@TestPropertySource(properties = { |
|||
"usage.stats.report.enabled=true", |
|||
"transport.http.enabled=true", |
|||
"usage.stats.report.interval=2", |
|||
"usage.stats.gauge_report_interval=1", |
|||
"state.defaultStateCheckIntervalInSec=3", |
|||
"state.defaultInactivityTimeoutInSec=10" |
|||
|
|||
}) |
|||
public class DevicesStatisticsTest extends AbstractControllerTest { |
|||
|
|||
@Autowired |
|||
private TbApiUsageStateService apiUsageStateService; |
|||
@Autowired |
|||
private TimeseriesService timeseriesService; |
|||
|
|||
private ApiUsageStateId apiUsageStateId; |
|||
|
|||
@Before |
|||
public void beforeEach() throws Exception { |
|||
loginTenantAdmin(); |
|||
apiUsageStateId = apiUsageStateService.getApiUsageState(tenantId).getId(); |
|||
} |
|||
|
|||
@Test |
|||
public void testDevicesActivityStats() throws Exception { |
|||
int activeDevicesCount = 5; |
|||
List<Device> activeDevices = new ArrayList<>(); |
|||
for (int i = 1; i <= activeDevicesCount; i++) { |
|||
String name = "active_device_" + i; |
|||
Device device = createDevice(name, name); |
|||
activeDevices.add(device); |
|||
} |
|||
int inactiveDevicesCount = 10; |
|||
List<Device> inactiveDevices = new ArrayList<>(); |
|||
for (int i = 1; i <= inactiveDevicesCount; i++) { |
|||
String name = "inactive_device_" + i; |
|||
Device device = createDevice(name, name); |
|||
inactiveDevices.add(device); |
|||
} |
|||
|
|||
await().atMost(15, TimeUnit.SECONDS) |
|||
.untilAsserted(() -> { |
|||
assertThat(getLatestStats(ApiUsageRecordKey.ACTIVE_DEVICES, false)).isZero(); |
|||
assertThat(getLatestStats(ApiUsageRecordKey.INACTIVE_DEVICES, false)).isEqualTo(activeDevicesCount + inactiveDevicesCount); |
|||
}); |
|||
|
|||
for (Device device : activeDevices) { |
|||
postTelemetry(device.getName(), "{\"dp\":1}"); |
|||
} |
|||
|
|||
await().atMost(40, TimeUnit.SECONDS) |
|||
.untilAsserted(() -> { |
|||
assertThat(getLatestStats(ApiUsageRecordKey.ACTIVE_DEVICES, false)).isEqualTo(activeDevicesCount); |
|||
assertThat(getLatestStats(ApiUsageRecordKey.INACTIVE_DEVICES, false)).isEqualTo(inactiveDevicesCount); |
|||
}); |
|||
} |
|||
|
|||
@SneakyThrows |
|||
private Long getLatestStats(ApiUsageRecordKey key, boolean hourly) { |
|||
return timeseriesService.findLatest(tenantId, apiUsageStateId, List.of(key.getApiCountKey() + (hourly ? "Hourly" : ""))) |
|||
.get().stream().findFirst().flatMap(KvEntry::getLongValue).orElse(null); |
|||
} |
|||
|
|||
@SneakyThrows |
|||
private void postTelemetry(String accessToken, String json) { |
|||
doPost("/api/v1/" + accessToken + "/telemetry", json, new String[0]).andExpect(status().isOk()); |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue