Merge remote-tracking branch 'upstream/master' into improvement/mqtt-reason-codes
This commit is contained in:
		
						commit
						224a22d093
					
				@ -22,6 +22,7 @@ import lombok.extern.slf4j.Slf4j;
 | 
			
		||||
import org.springframework.beans.factory.annotation.Autowired;
 | 
			
		||||
import org.springframework.beans.factory.annotation.Value;
 | 
			
		||||
import org.springframework.context.annotation.Lazy;
 | 
			
		||||
import org.springframework.core.env.Environment;
 | 
			
		||||
import org.thingsboard.server.cluster.TbClusterService;
 | 
			
		||||
import org.thingsboard.server.common.data.EntityType;
 | 
			
		||||
import org.thingsboard.server.common.data.User;
 | 
			
		||||
@ -39,11 +40,15 @@ import org.thingsboard.server.service.telemetry.AlarmSubscriptionService;
 | 
			
		||||
 | 
			
		||||
import java.util.List;
 | 
			
		||||
import java.util.Optional;
 | 
			
		||||
import java.util.Set;
 | 
			
		||||
import java.util.UUID;
 | 
			
		||||
 | 
			
		||||
@Slf4j
 | 
			
		||||
public abstract class AbstractTbEntityService {
 | 
			
		||||
 | 
			
		||||
    @Autowired
 | 
			
		||||
    private Environment env;
 | 
			
		||||
 | 
			
		||||
    @Value("${server.log_controller_error_stack_trace}")
 | 
			
		||||
    @Getter
 | 
			
		||||
    private boolean logControllerErrorStackTrace;
 | 
			
		||||
@ -67,6 +72,10 @@ public abstract class AbstractTbEntityService {
 | 
			
		||||
    @Lazy
 | 
			
		||||
    private EntitiesVersionControlService vcService;
 | 
			
		||||
 | 
			
		||||
    protected boolean isTestProfile() {
 | 
			
		||||
        return Set.of(this.env.getActiveProfiles()).contains("test");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    protected <T> T checkNotNull(T reference) throws ThingsboardException {
 | 
			
		||||
        return checkNotNull(reference, "Requested item wasn't found!");
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
@ -17,6 +17,7 @@ package org.thingsboard.server.service.entitiy.tenant;
 | 
			
		||||
 | 
			
		||||
import lombok.RequiredArgsConstructor;
 | 
			
		||||
import org.springframework.stereotype.Service;
 | 
			
		||||
import org.springframework.beans.factory.annotation.Value;
 | 
			
		||||
import org.thingsboard.server.common.data.Tenant;
 | 
			
		||||
import org.thingsboard.server.common.data.TenantProfile;
 | 
			
		||||
import org.thingsboard.server.common.data.id.TenantId;
 | 
			
		||||
@ -52,7 +53,9 @@ public class DefaultTbTenantService extends AbstractTbEntityService implements T
 | 
			
		||||
        Tenant savedTenant = tenantService.saveTenant(tenant, tenantId -> {
 | 
			
		||||
            installScripts.createDefaultRuleChains(tenantId);
 | 
			
		||||
            installScripts.createDefaultEdgeRuleChains(tenantId);
 | 
			
		||||
            installScripts.createDefaultTenantDashboards(tenantId, null);
 | 
			
		||||
            if (!isTestProfile()) {
 | 
			
		||||
                installScripts.createDefaultTenantDashboards(tenantId, null);
 | 
			
		||||
            }
 | 
			
		||||
        });
 | 
			
		||||
        tenantProfileCache.evict(savedTenant.getId());
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -317,7 +317,8 @@ public class InstallScripts {
 | 
			
		||||
    @SneakyThrows
 | 
			
		||||
    public void loadSystemImages() {
 | 
			
		||||
        log.info("Loading system images...");
 | 
			
		||||
        Stream<Path> dashboardsFiles = Files.list(Paths.get(getDataDir(), JSON_DIR, DEMO_DIR, DASHBOARDS_DIR));
 | 
			
		||||
        Stream<Path> dashboardsFiles = Stream.concat(Files.list(Paths.get(getDataDir(), JSON_DIR, DEMO_DIR, DASHBOARDS_DIR)),
 | 
			
		||||
                                                     Files.list(Paths.get(getDataDir(), JSON_DIR, TENANT_DIR, DASHBOARDS_DIR)));
 | 
			
		||||
        try (dashboardsFiles) {
 | 
			
		||||
            dashboardsFiles.forEach(file -> {
 | 
			
		||||
                try {
 | 
			
		||||
 | 
			
		||||
@ -210,11 +210,12 @@ public class JwtTokenFactory {
 | 
			
		||||
 | 
			
		||||
        ZonedDateTime currentTime = ZonedDateTime.now();
 | 
			
		||||
 | 
			
		||||
        claimsBuilder.expiration(Date.from(currentTime.plusSeconds(expirationTime).toInstant()));
 | 
			
		||||
 | 
			
		||||
        return Jwts.builder()
 | 
			
		||||
                .claims(claimsBuilder.build())
 | 
			
		||||
                .issuer(jwtSettingsService.getJwtSettings().getTokenIssuer())
 | 
			
		||||
                .issuedAt(Date.from(currentTime.toInstant()))
 | 
			
		||||
                .expiration(Date.from(currentTime.plusSeconds(expirationTime).toInstant()))
 | 
			
		||||
                .signWith(getSecretKey(false), Jwts.SIG.HS512);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -92,7 +92,7 @@ public class HomePageApiTest extends AbstractControllerTest {
 | 
			
		||||
    @MockBean
 | 
			
		||||
    private SmsService smsService;
 | 
			
		||||
 | 
			
		||||
    private static final int DEFAULT_DASHBOARDS_COUNT = 1;
 | 
			
		||||
    private static final int DEFAULT_DASHBOARDS_COUNT = 0;
 | 
			
		||||
 | 
			
		||||
    //For system administrator
 | 
			
		||||
    @Test
 | 
			
		||||
 | 
			
		||||
@ -20,12 +20,12 @@ import lombok.SneakyThrows;
 | 
			
		||||
import lombok.extern.slf4j.Slf4j;
 | 
			
		||||
import org.apache.commons.collections4.ListUtils;
 | 
			
		||||
import org.apache.commons.lang3.RandomStringUtils;
 | 
			
		||||
import org.junit.Assert;
 | 
			
		||||
import org.junit.Before;
 | 
			
		||||
import org.junit.Test;
 | 
			
		||||
import org.junit.runner.RunWith;
 | 
			
		||||
import org.junit.jupiter.api.Assertions;
 | 
			
		||||
import org.junit.jupiter.api.Test;
 | 
			
		||||
import org.junit.jupiter.api.BeforeEach;
 | 
			
		||||
import org.mockito.Mockito;
 | 
			
		||||
import org.mockito.junit.MockitoJUnitRunner;
 | 
			
		||||
import org.mockito.junit.jupiter.MockitoSettings;
 | 
			
		||||
import org.mockito.quality.Strictness;
 | 
			
		||||
import org.springframework.context.ApplicationEventPublisher;
 | 
			
		||||
import org.springframework.test.util.ReflectionTestUtils;
 | 
			
		||||
import org.thingsboard.server.common.data.DataConstants;
 | 
			
		||||
@ -67,7 +67,7 @@ import static org.mockito.Mockito.verify;
 | 
			
		||||
import static org.mockito.Mockito.when;
 | 
			
		||||
 | 
			
		||||
@Slf4j
 | 
			
		||||
@RunWith(MockitoJUnitRunner.class)
 | 
			
		||||
@MockitoSettings(strictness = Strictness.LENIENT)
 | 
			
		||||
public class HashPartitionServiceTest {
 | 
			
		||||
 | 
			
		||||
    public static final int ITERATIONS = 1000000;
 | 
			
		||||
@ -82,7 +82,7 @@ public class HashPartitionServiceTest {
 | 
			
		||||
 | 
			
		||||
    private String hashFunctionName = "murmur3_128";
 | 
			
		||||
 | 
			
		||||
    @Before
 | 
			
		||||
    @BeforeEach
 | 
			
		||||
    public void setup() throws Exception {
 | 
			
		||||
        serviceInfoProvider = mock(TbServiceInfoProvider.class);
 | 
			
		||||
        applicationEventPublisher = mock(ApplicationEventPublisher.class);
 | 
			
		||||
@ -173,7 +173,7 @@ public class HashPartitionServiceTest {
 | 
			
		||||
        for (Map.Entry<T, Integer> entry : data) {
 | 
			
		||||
            System.out.println(entry.getKey() + ": " + entry.getValue());
 | 
			
		||||
        }
 | 
			
		||||
        Assert.assertTrue(diffPercent < maxDiffPercent);
 | 
			
		||||
        Assertions.assertTrue(diffPercent < maxDiffPercent);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @Test
 | 
			
		||||
 | 
			
		||||
@ -15,24 +15,16 @@
 | 
			
		||||
 */
 | 
			
		||||
package org.thingsboard.server.service.apiusage;
 | 
			
		||||
 | 
			
		||||
import org.junit.Before;
 | 
			
		||||
import org.junit.Test;
 | 
			
		||||
import org.junit.runner.RunWith;
 | 
			
		||||
import org.junit.jupiter.api.BeforeEach;
 | 
			
		||||
import org.junit.jupiter.api.Test;
 | 
			
		||||
import org.junit.jupiter.api.extension.ExtendWith;
 | 
			
		||||
import org.mockito.InjectMocks;
 | 
			
		||||
import org.mockito.Mock;
 | 
			
		||||
import org.mockito.Mockito;
 | 
			
		||||
import org.mockito.Spy;
 | 
			
		||||
import org.mockito.junit.MockitoJUnitRunner;
 | 
			
		||||
import org.thingsboard.rule.engine.api.MailService;
 | 
			
		||||
import org.thingsboard.server.cluster.TbClusterService;
 | 
			
		||||
import org.mockito.junit.jupiter.MockitoExtension;
 | 
			
		||||
import org.thingsboard.server.common.data.ApiUsageState;
 | 
			
		||||
import org.thingsboard.server.common.data.id.TenantId;
 | 
			
		||||
import org.thingsboard.server.dao.tenant.TbTenantProfileCache;
 | 
			
		||||
import org.thingsboard.server.dao.tenant.TenantService;
 | 
			
		||||
import org.thingsboard.server.dao.timeseries.TimeseriesService;
 | 
			
		||||
import org.thingsboard.server.dao.usagerecord.ApiUsageStateService;
 | 
			
		||||
import org.thingsboard.server.queue.discovery.PartitionService;
 | 
			
		||||
import org.thingsboard.server.service.executors.DbCallbackExecutorService;
 | 
			
		||||
 | 
			
		||||
import java.util.UUID;
 | 
			
		||||
 | 
			
		||||
@ -40,27 +32,11 @@ import static org.hamcrest.CoreMatchers.is;
 | 
			
		||||
import static org.hamcrest.MatcherAssert.assertThat;
 | 
			
		||||
import static org.mockito.Mockito.never;
 | 
			
		||||
 | 
			
		||||
@RunWith(MockitoJUnitRunner.class)
 | 
			
		||||
@ExtendWith(MockitoExtension.class)
 | 
			
		||||
public class DefaultTbApiUsageStateServiceTest {
 | 
			
		||||
 | 
			
		||||
    @Mock
 | 
			
		||||
    TenantService tenantService;
 | 
			
		||||
    @Mock
 | 
			
		||||
    TimeseriesService tsService;
 | 
			
		||||
    @Mock
 | 
			
		||||
    TbClusterService clusterService;
 | 
			
		||||
    @Mock
 | 
			
		||||
    PartitionService partitionService;
 | 
			
		||||
    @Mock
 | 
			
		||||
    TenantApiUsageState tenantUsageStateMock;
 | 
			
		||||
    @Mock
 | 
			
		||||
    ApiUsageStateService apiUsageStateService;
 | 
			
		||||
    @Mock
 | 
			
		||||
    TbTenantProfileCache tenantProfileCache;
 | 
			
		||||
    @Mock
 | 
			
		||||
    MailService mailService;
 | 
			
		||||
    @Mock
 | 
			
		||||
    DbCallbackExecutorService dbExecutor;
 | 
			
		||||
 | 
			
		||||
    TenantId tenantId = TenantId.fromUUID(UUID.fromString("00797a3b-7aeb-4b5b-b57a-c2a810d0f112"));
 | 
			
		||||
 | 
			
		||||
@ -68,7 +44,7 @@ public class DefaultTbApiUsageStateServiceTest {
 | 
			
		||||
    @InjectMocks
 | 
			
		||||
    DefaultTbApiUsageStateService service;
 | 
			
		||||
 | 
			
		||||
    @Before
 | 
			
		||||
    @BeforeEach
 | 
			
		||||
    public void setUp() {
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -17,11 +17,11 @@ package org.thingsboard.server.service.edge.rpc.constructor;
 | 
			
		||||
 | 
			
		||||
import com.fasterxml.jackson.databind.JsonNode;
 | 
			
		||||
import lombok.extern.slf4j.Slf4j;
 | 
			
		||||
import org.junit.Assert;
 | 
			
		||||
import org.junit.Before;
 | 
			
		||||
import org.junit.Test;
 | 
			
		||||
import org.junit.runner.RunWith;
 | 
			
		||||
import org.mockito.junit.MockitoJUnitRunner;
 | 
			
		||||
import org.junit.jupiter.api.Assertions;
 | 
			
		||||
import org.junit.jupiter.api.BeforeEach;
 | 
			
		||||
import org.junit.jupiter.api.Test;
 | 
			
		||||
import org.junit.jupiter.api.extension.ExtendWith;
 | 
			
		||||
import org.mockito.junit.jupiter.MockitoExtension;
 | 
			
		||||
import org.thingsboard.common.util.JacksonUtil;
 | 
			
		||||
import org.thingsboard.server.common.data.id.RuleChainId;
 | 
			
		||||
import org.thingsboard.server.common.data.id.RuleNodeId;
 | 
			
		||||
@ -44,7 +44,7 @@ import java.util.Optional;
 | 
			
		||||
import java.util.UUID;
 | 
			
		||||
 | 
			
		||||
@Slf4j
 | 
			
		||||
@RunWith(MockitoJUnitRunner.class)
 | 
			
		||||
@ExtendWith(MockitoExtension.class)
 | 
			
		||||
public class RuleChainMsgConstructorTest {
 | 
			
		||||
 | 
			
		||||
    private static final String RPC_CONNECTION_TYPE = "RPC";
 | 
			
		||||
@ -53,7 +53,7 @@ public class RuleChainMsgConstructorTest {
 | 
			
		||||
 | 
			
		||||
    private TenantId tenantId;
 | 
			
		||||
 | 
			
		||||
    @Before
 | 
			
		||||
    @BeforeEach
 | 
			
		||||
    public void setup() {
 | 
			
		||||
        ruleChainMsgConstructorV1 = new RuleChainMsgConstructorV1();
 | 
			
		||||
        tenantId = new TenantId(UUID.randomUUID());
 | 
			
		||||
@ -98,10 +98,10 @@ public class RuleChainMsgConstructorTest {
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    private void assetV_3_3_3_and_V_3_4_0(RuleChainMetadataUpdateMsg ruleChainMetadataUpdateMsg) {
 | 
			
		||||
        Assert.assertEquals("First rule node index incorrect!", 3, ruleChainMetadataUpdateMsg.getFirstNodeIndex());
 | 
			
		||||
        Assert.assertEquals("Nodes count incorrect!", 12, ruleChainMetadataUpdateMsg.getNodesCount());
 | 
			
		||||
        Assert.assertEquals("Connections count incorrect!", 13, ruleChainMetadataUpdateMsg.getConnectionsCount());
 | 
			
		||||
        Assert.assertEquals("Rule chain connections count incorrect!", 0, ruleChainMetadataUpdateMsg.getRuleChainConnectionsCount());
 | 
			
		||||
        Assertions.assertEquals(3, ruleChainMetadataUpdateMsg.getFirstNodeIndex(), "First rule node index incorrect!");
 | 
			
		||||
        Assertions.assertEquals(12, ruleChainMetadataUpdateMsg.getNodesCount(), "Nodes count incorrect!");
 | 
			
		||||
        Assertions.assertEquals(13, ruleChainMetadataUpdateMsg.getConnectionsCount(), "Connections count incorrect!");
 | 
			
		||||
        Assertions.assertEquals(0, ruleChainMetadataUpdateMsg.getRuleChainConnectionsCount(), "Rule chain connections count incorrect!");
 | 
			
		||||
 | 
			
		||||
        compareNodeConnectionInfoAndProto(createNodeConnectionInfo(3, 6, TbNodeConnectionType.SUCCESS), ruleChainMetadataUpdateMsg.getConnections(0));
 | 
			
		||||
        compareNodeConnectionInfoAndProto(createNodeConnectionInfo(3, 10, TbNodeConnectionType.SUCCESS), ruleChainMetadataUpdateMsg.getConnections(1));
 | 
			
		||||
@ -129,10 +129,10 @@ public class RuleChainMsgConstructorTest {
 | 
			
		||||
                        ruleChainMetaData,
 | 
			
		||||
                        EdgeVersion.V_3_3_0);
 | 
			
		||||
 | 
			
		||||
        Assert.assertEquals("First rule node index incorrect!", 2, ruleChainMetadataUpdateMsg.getFirstNodeIndex());
 | 
			
		||||
        Assert.assertEquals("Nodes count incorrect!", 10, ruleChainMetadataUpdateMsg.getNodesCount());
 | 
			
		||||
        Assert.assertEquals("Connections count incorrect!", 10, ruleChainMetadataUpdateMsg.getConnectionsCount());
 | 
			
		||||
        Assert.assertEquals("Rule chain connections count incorrect!", 1, ruleChainMetadataUpdateMsg.getRuleChainConnectionsCount());
 | 
			
		||||
        Assertions.assertEquals(2, ruleChainMetadataUpdateMsg.getFirstNodeIndex(),"First rule node index incorrect!");
 | 
			
		||||
        Assertions.assertEquals(10, ruleChainMetadataUpdateMsg.getNodesCount(), "Nodes count incorrect!");
 | 
			
		||||
        Assertions.assertEquals(10, ruleChainMetadataUpdateMsg.getConnectionsCount(),"Connections count incorrect!");
 | 
			
		||||
        Assertions.assertEquals(1, ruleChainMetadataUpdateMsg.getRuleChainConnectionsCount(), "Rule chain connections count incorrect!");
 | 
			
		||||
 | 
			
		||||
        compareNodeConnectionInfoAndProto(createNodeConnectionInfo(2, 5, TbNodeConnectionType.SUCCESS), ruleChainMetadataUpdateMsg.getConnections(0));
 | 
			
		||||
        compareNodeConnectionInfoAndProto(createNodeConnectionInfo(3, 9, TbNodeConnectionType.SUCCESS), ruleChainMetadataUpdateMsg.getConnections(1));
 | 
			
		||||
@ -146,13 +146,12 @@ public class RuleChainMsgConstructorTest {
 | 
			
		||||
        compareNodeConnectionInfoAndProto(createNodeConnectionInfo(6, 9, TbNodeConnectionType.SUCCESS), ruleChainMetadataUpdateMsg.getConnections(9));
 | 
			
		||||
 | 
			
		||||
        RuleChainConnectionInfoProto ruleChainConnection = ruleChainMetadataUpdateMsg.getRuleChainConnections(0);
 | 
			
		||||
        Assert.assertEquals("From index incorrect!", 2, ruleChainConnection.getFromIndex());
 | 
			
		||||
        Assert.assertEquals("Type index incorrect!", TbNodeConnectionType.SUCCESS, ruleChainConnection.getType());
 | 
			
		||||
        Assert.assertEquals("Additional info incorrect!",
 | 
			
		||||
                "{\"description\":\"\",\"layoutX\":477,\"layoutY\":560,\"ruleChainNodeId\":\"rule-chain-node-UNDEFINED\"}",
 | 
			
		||||
                ruleChainConnection.getAdditionalInfo());
 | 
			
		||||
        Assert.assertTrue("Target rule chain id MSB incorrect!", ruleChainConnection.getTargetRuleChainIdMSB() != 0);
 | 
			
		||||
        Assert.assertTrue("Target rule chain id LSB incorrect!", ruleChainConnection.getTargetRuleChainIdLSB() != 0);
 | 
			
		||||
        Assertions.assertEquals(2, ruleChainConnection.getFromIndex(), "From index incorrect!");
 | 
			
		||||
        Assertions.assertEquals(TbNodeConnectionType.SUCCESS, ruleChainConnection.getType(), "Type index incorrect!");
 | 
			
		||||
        Assertions.assertEquals("{\"description\":\"\",\"layoutX\":477,\"layoutY\":560,\"ruleChainNodeId\":\"rule-chain-node-UNDEFINED\"}",
 | 
			
		||||
                ruleChainConnection.getAdditionalInfo(),"Additional info incorrect!");
 | 
			
		||||
        Assertions.assertTrue(ruleChainConnection.getTargetRuleChainIdMSB() != 0,"Target rule chain id MSB incorrect!");
 | 
			
		||||
        Assertions.assertTrue(ruleChainConnection.getTargetRuleChainIdLSB() != 0,"Target rule chain id LSB incorrect!");
 | 
			
		||||
 | 
			
		||||
        assertCheckpointRuleNodeConfiguration(
 | 
			
		||||
                ruleChainMetadataUpdateMsg.getNodesList(),
 | 
			
		||||
@ -171,10 +170,10 @@ public class RuleChainMsgConstructorTest {
 | 
			
		||||
                        ruleChainMetaData1, 
 | 
			
		||||
                        EdgeVersion.V_3_3_0);
 | 
			
		||||
 | 
			
		||||
        Assert.assertEquals("First rule node index incorrect!", 7, ruleChainMetadataUpdateMsg.getFirstNodeIndex());
 | 
			
		||||
        Assert.assertEquals("Nodes count incorrect!", 10, ruleChainMetadataUpdateMsg.getNodesCount());
 | 
			
		||||
        Assert.assertEquals("Connections count incorrect!", 10, ruleChainMetadataUpdateMsg.getConnectionsCount());
 | 
			
		||||
        Assert.assertEquals("Rule chain connections count incorrect!", 1, ruleChainMetadataUpdateMsg.getRuleChainConnectionsCount());
 | 
			
		||||
        Assertions.assertEquals(7, ruleChainMetadataUpdateMsg.getFirstNodeIndex(), "First rule node index incorrect!");
 | 
			
		||||
        Assertions.assertEquals(10, ruleChainMetadataUpdateMsg.getNodesCount(),"Nodes count incorrect!");
 | 
			
		||||
        Assertions.assertEquals(10, ruleChainMetadataUpdateMsg.getConnectionsCount(),"Connections count incorrect!");
 | 
			
		||||
        Assertions.assertEquals(1, ruleChainMetadataUpdateMsg.getRuleChainConnectionsCount(),"Rule chain connections count incorrect!");
 | 
			
		||||
 | 
			
		||||
        compareNodeConnectionInfoAndProto(createNodeConnectionInfo(3, 0, TbNodeConnectionType.SUCCESS), ruleChainMetadataUpdateMsg.getConnections(0));
 | 
			
		||||
        compareNodeConnectionInfoAndProto(createNodeConnectionInfo(4, 0, TbMsgType.ATTRIBUTES_UPDATED.getRuleNodeConnection()), ruleChainMetadataUpdateMsg.getConnections(1));
 | 
			
		||||
@ -188,13 +187,12 @@ public class RuleChainMsgConstructorTest {
 | 
			
		||||
        compareNodeConnectionInfoAndProto(createNodeConnectionInfo(7, 4, TbNodeConnectionType.SUCCESS), ruleChainMetadataUpdateMsg.getConnections(9));
 | 
			
		||||
 | 
			
		||||
        RuleChainConnectionInfoProto ruleChainConnection = ruleChainMetadataUpdateMsg.getRuleChainConnections(0);
 | 
			
		||||
        Assert.assertEquals("From index incorrect!", 7, ruleChainConnection.getFromIndex());
 | 
			
		||||
        Assert.assertEquals("Type index incorrect!", TbNodeConnectionType.SUCCESS, ruleChainConnection.getType());
 | 
			
		||||
        Assert.assertEquals("Additional info incorrect!",
 | 
			
		||||
                "{\"description\":\"\",\"layoutX\":477,\"layoutY\":560,\"ruleChainNodeId\":\"rule-chain-node-UNDEFINED\"}",
 | 
			
		||||
                ruleChainConnection.getAdditionalInfo());
 | 
			
		||||
        Assert.assertTrue("Target rule chain id MSB incorrect!", ruleChainConnection.getTargetRuleChainIdMSB() != 0);
 | 
			
		||||
        Assert.assertTrue("Target rule chain id LSB incorrect!", ruleChainConnection.getTargetRuleChainIdLSB() != 0);
 | 
			
		||||
        Assertions.assertEquals(7, ruleChainConnection.getFromIndex(),"From index incorrect!");
 | 
			
		||||
        Assertions.assertEquals(TbNodeConnectionType.SUCCESS, ruleChainConnection.getType(), "Type index incorrect!");
 | 
			
		||||
        Assertions.assertEquals( "{\"description\":\"\",\"layoutX\":477,\"layoutY\":560,\"ruleChainNodeId\":\"rule-chain-node-UNDEFINED\"}",
 | 
			
		||||
                ruleChainConnection.getAdditionalInfo(), "Additional info incorrect!");
 | 
			
		||||
        Assertions.assertTrue(ruleChainConnection.getTargetRuleChainIdMSB() != 0, "Target rule chain id MSB incorrect!");
 | 
			
		||||
        Assertions.assertTrue(ruleChainConnection.getTargetRuleChainIdLSB() != 0, "Target rule chain id LSB incorrect!");
 | 
			
		||||
 | 
			
		||||
        assertCheckpointRuleNodeConfiguration(
 | 
			
		||||
                ruleChainMetadataUpdateMsg.getNodesList(),
 | 
			
		||||
@ -206,15 +204,15 @@ public class RuleChainMsgConstructorTest {
 | 
			
		||||
        Optional<RuleNodeProto> checkpointRuleNodeOpt = nodesList.stream()
 | 
			
		||||
                .filter(rn -> "org.thingsboard.rule.engine.flow.TbCheckpointNode".equals(rn.getType()))
 | 
			
		||||
                .findFirst();
 | 
			
		||||
        Assert.assertTrue(checkpointRuleNodeOpt.isPresent());
 | 
			
		||||
        Assertions.assertTrue(checkpointRuleNodeOpt.isPresent());
 | 
			
		||||
        RuleNodeProto checkpointRuleNode = checkpointRuleNodeOpt.get();
 | 
			
		||||
        Assert.assertEquals(expectedConfiguration, checkpointRuleNode.getConfiguration());
 | 
			
		||||
        Assertions.assertEquals(expectedConfiguration, checkpointRuleNode.getConfiguration());
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    private void compareNodeConnectionInfoAndProto(NodeConnectionInfo expected, org.thingsboard.server.gen.edge.v1.NodeConnectionInfoProto actual) {
 | 
			
		||||
        Assert.assertEquals(expected.getFromIndex(), actual.getFromIndex());
 | 
			
		||||
        Assert.assertEquals(expected.getToIndex(), actual.getToIndex());
 | 
			
		||||
        Assert.assertEquals(expected.getType(), actual.getType());
 | 
			
		||||
        Assertions.assertEquals(expected.getFromIndex(), actual.getFromIndex());
 | 
			
		||||
        Assertions.assertEquals(expected.getToIndex(), actual.getToIndex());
 | 
			
		||||
        Assertions.assertEquals(expected.getType(), actual.getType());
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    private RuleChainMetaData createRuleChainMetaData(RuleChainId ruleChainId, Integer firstNodeIndex, List<RuleNode> nodes, List<NodeConnectionInfo> connections) {
 | 
			
		||||
 | 
			
		||||
@ -15,11 +15,12 @@
 | 
			
		||||
 */
 | 
			
		||||
package org.thingsboard.server.service.limits;
 | 
			
		||||
 | 
			
		||||
import org.junit.Before;
 | 
			
		||||
import org.junit.Test;
 | 
			
		||||
import org.junit.runner.RunWith;
 | 
			
		||||
import org.junit.jupiter.api.Assertions;
 | 
			
		||||
import org.junit.jupiter.api.BeforeEach;
 | 
			
		||||
import org.junit.jupiter.api.Test;
 | 
			
		||||
import org.junit.jupiter.api.extension.ExtendWith;
 | 
			
		||||
import org.mockito.Mockito;
 | 
			
		||||
import org.mockito.junit.MockitoJUnitRunner;
 | 
			
		||||
import org.mockito.junit.jupiter.MockitoExtension;
 | 
			
		||||
import org.thingsboard.server.cache.limits.DefaultRateLimitService;
 | 
			
		||||
import org.thingsboard.server.cache.limits.RateLimitService;
 | 
			
		||||
import org.thingsboard.server.common.data.TenantProfile;
 | 
			
		||||
@ -35,21 +36,19 @@ import org.thingsboard.server.dao.tenant.DefaultTbTenantProfileCache;
 | 
			
		||||
import java.util.List;
 | 
			
		||||
import java.util.UUID;
 | 
			
		||||
 | 
			
		||||
import static org.junit.Assert.assertFalse;
 | 
			
		||||
import static org.junit.Assert.assertTrue;
 | 
			
		||||
import static org.mockito.ArgumentMatchers.eq;
 | 
			
		||||
import static org.mockito.Mockito.mock;
 | 
			
		||||
import static org.mockito.Mockito.reset;
 | 
			
		||||
import static org.mockito.Mockito.when;
 | 
			
		||||
 | 
			
		||||
@RunWith(MockitoJUnitRunner.class)
 | 
			
		||||
@ExtendWith(MockitoExtension.class)
 | 
			
		||||
public class RateLimitServiceTest {
 | 
			
		||||
 | 
			
		||||
    private RateLimitService rateLimitService;
 | 
			
		||||
    private DefaultTbTenantProfileCache tenantProfileCache;
 | 
			
		||||
    private TenantId tenantId;
 | 
			
		||||
 | 
			
		||||
    @Before
 | 
			
		||||
    @BeforeEach
 | 
			
		||||
    public void beforeEach() {
 | 
			
		||||
        tenantProfileCache = Mockito.mock(DefaultTbTenantProfileCache.class);
 | 
			
		||||
        rateLimitService = new DefaultRateLimitService(tenantProfileCache, mock(NotificationRuleProcessor.class), 60, 100);
 | 
			
		||||
@ -102,10 +101,10 @@ public class RateLimitServiceTest {
 | 
			
		||||
    private void testRateLimits(LimitedApi limitedApi, int max, Object level) {
 | 
			
		||||
        for (int i = 1; i <= max; i++) {
 | 
			
		||||
            boolean success = rateLimitService.checkRateLimit(limitedApi, tenantId, level);
 | 
			
		||||
            assertTrue(success);
 | 
			
		||||
            Assertions.assertTrue(success);
 | 
			
		||||
        }
 | 
			
		||||
        boolean success = rateLimitService.checkRateLimit(limitedApi, tenantId, level);
 | 
			
		||||
        assertFalse(success);
 | 
			
		||||
        Assertions.assertFalse(success);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    private void updateTenantProfileConfiguration(DefaultTenantProfileConfiguration profileConfiguration) {
 | 
			
		||||
 | 
			
		||||
@ -18,13 +18,15 @@ package org.thingsboard.server.service.queue.ruleengine;
 | 
			
		||||
import lombok.Getter;
 | 
			
		||||
import lombok.SneakyThrows;
 | 
			
		||||
import lombok.extern.slf4j.Slf4j;
 | 
			
		||||
import org.junit.After;
 | 
			
		||||
import org.junit.Before;
 | 
			
		||||
import org.junit.Test;
 | 
			
		||||
import org.junit.runner.RunWith;
 | 
			
		||||
import org.junit.jupiter.api.AfterEach;
 | 
			
		||||
import org.junit.jupiter.api.BeforeEach;
 | 
			
		||||
import org.junit.jupiter.api.Test;
 | 
			
		||||
import org.junit.jupiter.api.extension.ExtendWith;
 | 
			
		||||
import org.mockito.Mock;
 | 
			
		||||
import org.mockito.Mockito;
 | 
			
		||||
import org.mockito.junit.MockitoJUnitRunner;
 | 
			
		||||
import org.mockito.junit.jupiter.MockitoExtension;
 | 
			
		||||
import org.mockito.junit.jupiter.MockitoSettings;
 | 
			
		||||
import org.mockito.quality.Strictness;
 | 
			
		||||
import org.testcontainers.shaded.org.apache.commons.lang3.RandomUtils;
 | 
			
		||||
import org.thingsboard.common.util.JacksonUtil;
 | 
			
		||||
import org.thingsboard.common.util.ThingsBoardExecutors;
 | 
			
		||||
@ -95,7 +97,7 @@ import static org.mockito.Mockito.verifyNoMoreInteractions;
 | 
			
		||||
import static org.mockito.Mockito.when;
 | 
			
		||||
 | 
			
		||||
@Slf4j
 | 
			
		||||
@RunWith(MockitoJUnitRunner.class)
 | 
			
		||||
@MockitoSettings(strictness = Strictness.LENIENT)
 | 
			
		||||
public class TbRuleEngineQueueConsumerManagerTest {
 | 
			
		||||
 | 
			
		||||
    @Mock
 | 
			
		||||
@ -128,7 +130,7 @@ public class TbRuleEngineQueueConsumerManagerTest {
 | 
			
		||||
    private AtomicInteger totalConsumedMsgs;
 | 
			
		||||
    private AtomicInteger totalProcessedMsgs;
 | 
			
		||||
 | 
			
		||||
    @Before
 | 
			
		||||
    @BeforeEach
 | 
			
		||||
    public void beforeEach() {
 | 
			
		||||
        ruleEngineConsumerContext = new TbRuleEngineConsumerContext(
 | 
			
		||||
                actorContext, statsFactory, spy(new TbRuleEngineSubmitStrategyFactory()),
 | 
			
		||||
@ -190,7 +192,7 @@ public class TbRuleEngineQueueConsumerManagerTest {
 | 
			
		||||
                .build();
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @After
 | 
			
		||||
    @AfterEach
 | 
			
		||||
    public void afterEach() {
 | 
			
		||||
        consumerManager.stop();
 | 
			
		||||
        consumerManager.awaitStop();
 | 
			
		||||
 | 
			
		||||
@ -15,10 +15,10 @@
 | 
			
		||||
 */
 | 
			
		||||
package org.thingsboard.server.service.sms.smpp;
 | 
			
		||||
 | 
			
		||||
import org.junit.Before;
 | 
			
		||||
import org.junit.Test;
 | 
			
		||||
import org.junit.runner.RunWith;
 | 
			
		||||
import org.mockito.junit.MockitoJUnitRunner;
 | 
			
		||||
import org.junit.jupiter.api.BeforeEach;
 | 
			
		||||
import org.junit.jupiter.api.Test;
 | 
			
		||||
import org.junit.jupiter.api.extension.ExtendWith;
 | 
			
		||||
import org.mockito.junit.jupiter.MockitoExtension;
 | 
			
		||||
import org.smpp.Session;
 | 
			
		||||
import org.smpp.pdu.SubmitSMResp;
 | 
			
		||||
import org.thingsboard.server.common.data.StringUtils;
 | 
			
		||||
@ -35,14 +35,14 @@ import static org.mockito.Mockito.spy;
 | 
			
		||||
import static org.mockito.Mockito.verify;
 | 
			
		||||
import static org.mockito.Mockito.when;
 | 
			
		||||
 | 
			
		||||
@RunWith(MockitoJUnitRunner.class)
 | 
			
		||||
@ExtendWith(MockitoExtension.class)
 | 
			
		||||
public class SmppSmsSenderTest {
 | 
			
		||||
 | 
			
		||||
    SmppSmsSender smppSmsSender;
 | 
			
		||||
    SmppSmsProviderConfiguration smppConfig;
 | 
			
		||||
    Session smppSession;
 | 
			
		||||
 | 
			
		||||
    @Before
 | 
			
		||||
    @BeforeEach
 | 
			
		||||
    public void beforeEach() throws Exception {
 | 
			
		||||
        Constructor<SmppSmsSender> constructor = SmppSmsSender.class.getDeclaredConstructor();
 | 
			
		||||
        constructor.setAccessible(true);
 | 
			
		||||
 | 
			
		||||
@ -460,7 +460,7 @@ public class LwM2mClient {
 | 
			
		||||
            if (lwM2mPath.isObject()) {
 | 
			
		||||
                LwM2m.Version ver;
 | 
			
		||||
                if (link.getAttributes().get("ver")!= null) {
 | 
			
		||||
                    ver = (Version) link.getAttributes().get("ver").getValue();
 | 
			
		||||
                    ver = new LwM2m.Version(link.getAttributes().get("ver").getValue().toString());
 | 
			
		||||
                } else {
 | 
			
		||||
                    ver = getDefaultObjectIDVer();
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
@ -145,7 +145,7 @@ public abstract class TransportHealthChecker<C extends TransportMonitoringConfig
 | 
			
		||||
            profileData.setTransportConfiguration(new DefaultDeviceProfileTransportConfiguration());
 | 
			
		||||
            deviceProfile.setProfileData(profileData);
 | 
			
		||||
        } else {
 | 
			
		||||
            tbClient.getResources(new PageLink(1, 0, "lwm2m monitoring")).getData()
 | 
			
		||||
            tbClient.getResources(new PageLink(1, 0, "LwM2M Monitoring id=3 v1.0")).getData()
 | 
			
		||||
                    .stream().findFirst()
 | 
			
		||||
                    .orElseGet(() -> {
 | 
			
		||||
                        TbResource newResource = ResourceUtils.getResource("lwm2m/resource.json", TbResource.class);
 | 
			
		||||
 | 
			
		||||
@ -244,27 +244,27 @@ class AlarmState {
 | 
			
		||||
                return null;
 | 
			
		||||
            }
 | 
			
		||||
        } else {
 | 
			
		||||
            currentAlarm = new Alarm();
 | 
			
		||||
            currentAlarm.setType(alarmDefinition.getAlarmType());
 | 
			
		||||
            currentAlarm.setAcknowledged(false);
 | 
			
		||||
            currentAlarm.setCleared(false);
 | 
			
		||||
            currentAlarm.setSeverity(severity);
 | 
			
		||||
            var newAlarm = new Alarm();
 | 
			
		||||
            newAlarm.setType(alarmDefinition.getAlarmType());
 | 
			
		||||
            newAlarm.setAcknowledged(false);
 | 
			
		||||
            newAlarm.setCleared(false);
 | 
			
		||||
            newAlarm.setSeverity(severity);
 | 
			
		||||
            long startTs = dataSnapshot.getTs();
 | 
			
		||||
            if (startTs == 0L) {
 | 
			
		||||
                startTs = System.currentTimeMillis();
 | 
			
		||||
            }
 | 
			
		||||
            currentAlarm.setStartTs(startTs);
 | 
			
		||||
            currentAlarm.setEndTs(currentAlarm.getStartTs());
 | 
			
		||||
            currentAlarm.setDetails(createDetails(ruleState));
 | 
			
		||||
            currentAlarm.setOriginator(originator);
 | 
			
		||||
            currentAlarm.setTenantId(ctx.getTenantId());
 | 
			
		||||
            currentAlarm.setPropagate(alarmDefinition.isPropagate());
 | 
			
		||||
            currentAlarm.setPropagateToOwner(alarmDefinition.isPropagateToOwner());
 | 
			
		||||
            currentAlarm.setPropagateToTenant(alarmDefinition.isPropagateToTenant());
 | 
			
		||||
            newAlarm.setStartTs(startTs);
 | 
			
		||||
            newAlarm.setEndTs(startTs);
 | 
			
		||||
            newAlarm.setDetails(createDetails(ruleState));
 | 
			
		||||
            newAlarm.setOriginator(originator);
 | 
			
		||||
            newAlarm.setTenantId(ctx.getTenantId());
 | 
			
		||||
            newAlarm.setPropagate(alarmDefinition.isPropagate());
 | 
			
		||||
            newAlarm.setPropagateToOwner(alarmDefinition.isPropagateToOwner());
 | 
			
		||||
            newAlarm.setPropagateToTenant(alarmDefinition.isPropagateToTenant());
 | 
			
		||||
            if (alarmDefinition.getPropagateRelationTypes() != null) {
 | 
			
		||||
                currentAlarm.setPropagateRelationTypes(alarmDefinition.getPropagateRelationTypes());
 | 
			
		||||
                newAlarm.setPropagateRelationTypes(alarmDefinition.getPropagateRelationTypes());
 | 
			
		||||
            }
 | 
			
		||||
            AlarmApiCallResult result = ctx.getAlarmService().createAlarm(AlarmCreateOrUpdateActiveRequest.fromAlarm(currentAlarm));
 | 
			
		||||
            AlarmApiCallResult result = ctx.getAlarmService().createAlarm(AlarmCreateOrUpdateActiveRequest.fromAlarm(newAlarm));
 | 
			
		||||
            currentAlarm = result.getAlarm();
 | 
			
		||||
            return TbAlarmResult.fromAlarmResult(result);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user