Igor Kulikov
3 years ago
39 changed files with 425 additions and 118 deletions
@ -0,0 +1,158 @@ |
|||
/** |
|||
* 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.dao.sql.rule; |
|||
|
|||
import com.datastax.oss.driver.api.core.uuid.Uuids; |
|||
import com.google.common.util.concurrent.ListeningExecutorService; |
|||
import org.junit.After; |
|||
import org.junit.Before; |
|||
import org.junit.Test; |
|||
import org.springframework.beans.factory.annotation.Autowired; |
|||
import org.thingsboard.common.util.JacksonUtil; |
|||
import org.thingsboard.server.common.data.id.RuleChainId; |
|||
import org.thingsboard.server.common.data.id.RuleNodeId; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.common.data.page.PageData; |
|||
import org.thingsboard.server.common.data.page.PageLink; |
|||
import org.thingsboard.server.common.data.rule.RuleChain; |
|||
import org.thingsboard.server.common.data.rule.RuleNode; |
|||
import org.thingsboard.server.dao.AbstractJpaDaoTest; |
|||
import org.thingsboard.server.dao.rule.RuleChainDao; |
|||
import org.thingsboard.server.dao.rule.RuleNodeDao; |
|||
|
|||
import java.util.ArrayList; |
|||
import java.util.List; |
|||
import java.util.UUID; |
|||
|
|||
import static org.assertj.core.api.Assertions.assertThatThrownBy; |
|||
import static org.junit.Assert.assertEquals; |
|||
import static org.junit.Assert.assertNotNull; |
|||
|
|||
public class JpaRuleNodeDaoTest extends AbstractJpaDaoTest { |
|||
|
|||
public static final int COUNT = 40; |
|||
public static final String PREFIX_FOR_RULE_NODE_NAME = "SEARCH_TEXT_"; |
|||
List<UUID> ruleNodeIds; |
|||
TenantId tenantId1; |
|||
TenantId tenantId2; |
|||
RuleChainId ruleChainId1; |
|||
RuleChainId ruleChainId2; |
|||
|
|||
@Autowired |
|||
private RuleChainDao ruleChainDao; |
|||
|
|||
@Autowired |
|||
private RuleNodeDao ruleNodeDao; |
|||
|
|||
ListeningExecutorService executor; |
|||
|
|||
@Before |
|||
public void setUp() { |
|||
tenantId1 = TenantId.fromUUID(Uuids.timeBased()); |
|||
ruleChainId1 = new RuleChainId(UUID.randomUUID()); |
|||
tenantId2 = TenantId.fromUUID(Uuids.timeBased()); |
|||
ruleChainId2 = new RuleChainId(UUID.randomUUID()); |
|||
|
|||
ruleNodeIds = createRuleNodes(tenantId1, tenantId2, ruleChainId1, ruleChainId2, COUNT); |
|||
} |
|||
|
|||
@After |
|||
public void tearDown() throws Exception { |
|||
ruleNodeDao.removeAllByIds(ruleNodeIds); |
|||
if (executor != null) { |
|||
executor.shutdownNow(); |
|||
} |
|||
} |
|||
|
|||
@Test |
|||
public void testSaveRuleName0x00_thenSomeDatabaseException() { |
|||
RuleNode ruleNode = getRuleNode(ruleChainId1, "T", "\u0000"); |
|||
assertThatThrownBy(() -> ruleNodeIds.add(ruleNodeDao.save(tenantId1, ruleNode).getUuidId())); |
|||
} |
|||
|
|||
@Test |
|||
public void testFindRuleNodesByTenantIdAndType() { |
|||
List<RuleNode> ruleNodes1 = ruleNodeDao.findRuleNodesByTenantIdAndType(tenantId1, "A", PREFIX_FOR_RULE_NODE_NAME); |
|||
assertEquals(20, ruleNodes1.size()); |
|||
|
|||
List<RuleNode> ruleNodes2 = ruleNodeDao.findRuleNodesByTenantIdAndType(tenantId2, "B", PREFIX_FOR_RULE_NODE_NAME); |
|||
assertEquals(20, ruleNodes2.size()); |
|||
|
|||
ruleNodes1 = ruleNodeDao.findRuleNodesByTenantIdAndType(tenantId1, "A", null); |
|||
assertEquals(20, ruleNodes1.size()); |
|||
|
|||
ruleNodes2 = ruleNodeDao.findRuleNodesByTenantIdAndType(tenantId2, "B", null); |
|||
assertEquals(20, ruleNodes2.size()); |
|||
} |
|||
|
|||
@Test |
|||
public void testFindRuleNodesByType() { |
|||
PageData<RuleNode> ruleNodes = ruleNodeDao.findAllRuleNodesByType( "A", new PageLink(10, 0, PREFIX_FOR_RULE_NODE_NAME)); |
|||
assertEquals(20, ruleNodes.getTotalElements()); |
|||
assertEquals(2, ruleNodes.getTotalPages()); |
|||
assertEquals(10, ruleNodes.getData().size()); |
|||
|
|||
ruleNodes = ruleNodeDao.findAllRuleNodesByType( "A", new PageLink(10, 0)); |
|||
assertEquals(20, ruleNodes.getTotalElements()); |
|||
assertEquals(2, ruleNodes.getTotalPages()); |
|||
assertEquals(10, ruleNodes.getData().size()); |
|||
} |
|||
|
|||
@Test |
|||
public void testFindRuleNodesByTypeAndVersionLessThan() { |
|||
PageData<RuleNode> ruleNodes = ruleNodeDao.findAllRuleNodesByTypeAndVersionLessThan( "A", 1, new PageLink(10, 0, PREFIX_FOR_RULE_NODE_NAME)); |
|||
assertEquals(20, ruleNodes.getTotalElements()); |
|||
assertEquals(2, ruleNodes.getTotalPages()); |
|||
assertEquals(10, ruleNodes.getData().size()); |
|||
|
|||
ruleNodes = ruleNodeDao.findAllRuleNodesByTypeAndVersionLessThan( "A", 1, new PageLink(10, 0)); |
|||
assertEquals(20, ruleNodes.getTotalElements()); |
|||
assertEquals(2, ruleNodes.getTotalPages()); |
|||
assertEquals(10, ruleNodes.getData().size()); |
|||
} |
|||
|
|||
private List<UUID> createRuleNodes(TenantId tenantId1, TenantId tenantId2, RuleChainId ruleChainId1, RuleChainId ruleChainId2, int count) { |
|||
var chain1 = new RuleChain(ruleChainId1); |
|||
chain1.setTenantId(tenantId1); |
|||
chain1.setName(ruleChainId1.toString()); |
|||
ruleChainDao.save(tenantId1, chain1); |
|||
var chain2 = new RuleChain(ruleChainId2); |
|||
chain2.setTenantId(tenantId2); |
|||
chain2.setName(ruleChainId2.toString()); |
|||
ruleChainDao.save(tenantId2, chain2); |
|||
List<UUID> savedRuleNodeIds = new ArrayList<>(); |
|||
for (int i = 0; i < count / 2; i++) { |
|||
savedRuleNodeIds.add(ruleNodeDao.save(tenantId1, getRuleNode(ruleChainId1, "A", Integer.toString(i))).getUuidId()); |
|||
savedRuleNodeIds.add(ruleNodeDao.save(tenantId2, getRuleNode(ruleChainId2, "B", Integer.toString(i + count / 2))).getUuidId()); |
|||
} |
|||
return savedRuleNodeIds; |
|||
} |
|||
|
|||
private RuleNode getRuleNode(RuleChainId ruleChainId, String type, String nameSuffix) { |
|||
return getRuleNode(ruleChainId, Uuids.timeBased(), type, nameSuffix); |
|||
} |
|||
|
|||
private RuleNode getRuleNode(RuleChainId ruleChainId, UUID ruleNodeId, String type, String nameSuffix) { |
|||
RuleNode ruleNode = new RuleNode(); |
|||
ruleNode.setId(new RuleNodeId(ruleNodeId)); |
|||
ruleNode.setRuleChainId(ruleChainId); |
|||
ruleNode.setName(nameSuffix); |
|||
ruleNode.setType(type); |
|||
ruleNode.setConfiguration(JacksonUtil.newObjectNode().put("searchHint", PREFIX_FOR_RULE_NODE_NAME + nameSuffix)); |
|||
ruleNode.setConfigurationVersion(0); |
|||
return ruleNode; |
|||
} |
|||
} |
|||
Loading…
Reference in new issue