Browse Source

added test

pull/15302/head
dashevchenko 6 months ago
parent
commit
5deeb2ab22
  1. 25
      dao/src/test/java/org/thingsboard/server/dao/service/AlarmServiceTest.java

25
dao/src/test/java/org/thingsboard/server/dao/service/AlarmServiceTest.java

@ -18,6 +18,7 @@ package org.thingsboard.server.dao.service;
import com.datastax.oss.driver.api.core.uuid.Uuids;
import org.junit.Assert;
import org.junit.Test;
import org.junit.jupiter.api.Assertions;
import org.springframework.beans.factory.annotation.Autowired;
import org.thingsboard.common.util.JacksonUtil;
import org.thingsboard.server.common.data.Customer;
@ -57,6 +58,7 @@ import org.thingsboard.server.dao.alarm.AlarmService;
import org.thingsboard.server.dao.asset.AssetService;
import org.thingsboard.server.dao.customer.CustomerService;
import org.thingsboard.server.dao.device.DeviceService;
import org.thingsboard.server.dao.exception.DataValidationException;
import org.thingsboard.server.dao.relation.RelationService;
import org.thingsboard.server.dao.user.UserService;
@ -64,6 +66,8 @@ import java.util.Collections;
import java.util.List;
import java.util.concurrent.ExecutionException;
import static org.assertj.core.api.Assertions.assertThat;
@DaoSqlTest
public class AlarmServiceTest extends AbstractServiceTest {
@ -987,4 +991,25 @@ public class AlarmServiceTest extends AbstractServiceTest {
Assert.assertEquals(1, alarmsCount);
}
@Test
public void testShouldFailToCreateAlarmWithBadType() {
AssetId originatorId = new AssetId(Uuids.timeBased());
long ts = System.currentTimeMillis();
AlarmCreateOrUpdateActiveRequest request = AlarmCreateOrUpdateActiveRequest.builder()
.tenantId(tenantId)
.originator(originatorId)
.type("<img src=1 onerror=alert()>")
.severity(AlarmSeverity.CRITICAL)
.startTs(ts).build();
Assertions.assertThrows(DataValidationException.class, () -> {
alarmService.createAlarm(request);
});
request.setType(TEST_ALARM);
AlarmApiCallResult result = alarmService.createAlarm(request);
assertThat(result.getAlarm().getId()).isNotNull();
}
}

Loading…
Cancel
Save