Volodymyr Babak
9 years ago
24 changed files with 97 additions and 374 deletions
@ -1,284 +0,0 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2017 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.service; |
|
||||
|
|
||||
import com.datastax.driver.core.utils.UUIDs; |
|
||||
import com.google.common.util.concurrent.ListenableFuture; |
|
||||
import org.junit.After; |
|
||||
import org.junit.Assert; |
|
||||
import org.junit.Before; |
|
||||
import org.junit.Test; |
|
||||
import org.thingsboard.server.common.data.EntityType; |
|
||||
import org.thingsboard.server.common.data.asset.Asset; |
|
||||
import org.thingsboard.server.common.data.id.AssetId; |
|
||||
import org.thingsboard.server.common.data.id.DeviceId; |
|
||||
import org.thingsboard.server.common.data.relation.EntityRelation; |
|
||||
import org.thingsboard.server.common.data.relation.RelationTypeGroup; |
|
||||
import org.thingsboard.server.dao.exception.DataValidationException; |
|
||||
import org.thingsboard.server.dao.relation.EntityRelationsQuery; |
|
||||
import org.thingsboard.server.dao.relation.EntitySearchDirection; |
|
||||
import org.thingsboard.server.dao.relation.EntityTypeFilter; |
|
||||
import org.thingsboard.server.dao.relation.RelationsSearchParameters; |
|
||||
|
|
||||
import java.util.Collections; |
|
||||
import java.util.List; |
|
||||
import java.util.concurrent.ExecutionException; |
|
||||
|
|
||||
public class RelationServiceTest extends AbstractServiceTest { |
|
||||
|
|
||||
@Before |
|
||||
public void before() { |
|
||||
} |
|
||||
|
|
||||
@After |
|
||||
public void after() { |
|
||||
} |
|
||||
|
|
||||
@Test |
|
||||
public void testSaveRelation() throws ExecutionException, InterruptedException { |
|
||||
AssetId parentId = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId childId = new AssetId(UUIDs.timeBased()); |
|
||||
|
|
||||
EntityRelation relation = new EntityRelation(parentId, childId, EntityRelation.CONTAINS_TYPE); |
|
||||
|
|
||||
Assert.assertTrue(saveRelation(relation)); |
|
||||
|
|
||||
Assert.assertTrue(relationService.checkRelation(parentId, childId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
|
||||
|
|
||||
Assert.assertFalse(relationService.checkRelation(parentId, childId, "NOT_EXISTING_TYPE", RelationTypeGroup.COMMON).get()); |
|
||||
|
|
||||
Assert.assertFalse(relationService.checkRelation(childId, parentId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
|
||||
|
|
||||
Assert.assertFalse(relationService.checkRelation(childId, parentId, "NOT_EXISTING_TYPE", RelationTypeGroup.COMMON).get()); |
|
||||
} |
|
||||
|
|
||||
@Test |
|
||||
public void testDeleteRelation() throws ExecutionException, InterruptedException { |
|
||||
AssetId parentId = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId childId = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId subChildId = new AssetId(UUIDs.timeBased()); |
|
||||
|
|
||||
EntityRelation relationA = new EntityRelation(parentId, childId, EntityRelation.CONTAINS_TYPE); |
|
||||
EntityRelation relationB = new EntityRelation(childId, subChildId, EntityRelation.CONTAINS_TYPE); |
|
||||
|
|
||||
saveRelation(relationA); |
|
||||
saveRelation(relationB); |
|
||||
|
|
||||
Assert.assertTrue(relationService.deleteRelation(relationA).get()); |
|
||||
|
|
||||
Assert.assertFalse(relationService.checkRelation(parentId, childId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
|
||||
|
|
||||
Assert.assertTrue(relationService.checkRelation(childId, subChildId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
|
||||
|
|
||||
Assert.assertTrue(relationService.deleteRelation(childId, subChildId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
|
||||
} |
|
||||
|
|
||||
@Test |
|
||||
public void testDeleteEntityRelations() throws ExecutionException, InterruptedException { |
|
||||
AssetId parentId = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId childId = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId subChildId = new AssetId(UUIDs.timeBased()); |
|
||||
|
|
||||
EntityRelation relationA = new EntityRelation(parentId, childId, EntityRelation.CONTAINS_TYPE); |
|
||||
EntityRelation relationB = new EntityRelation(childId, subChildId, EntityRelation.CONTAINS_TYPE); |
|
||||
|
|
||||
saveRelation(relationA); |
|
||||
saveRelation(relationB); |
|
||||
|
|
||||
Assert.assertTrue(relationService.deleteEntityRelations(childId).get()); |
|
||||
|
|
||||
Assert.assertFalse(relationService.checkRelation(parentId, childId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
|
||||
|
|
||||
Assert.assertFalse(relationService.checkRelation(childId, subChildId, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get()); |
|
||||
} |
|
||||
|
|
||||
@Test |
|
||||
public void testFindFrom() throws ExecutionException, InterruptedException { |
|
||||
AssetId parentA = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId parentB = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId childA = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId childB = new AssetId(UUIDs.timeBased()); |
|
||||
|
|
||||
EntityRelation relationA1 = new EntityRelation(parentA, childA, EntityRelation.CONTAINS_TYPE); |
|
||||
EntityRelation relationA2 = new EntityRelation(parentA, childB, EntityRelation.CONTAINS_TYPE); |
|
||||
|
|
||||
EntityRelation relationB1 = new EntityRelation(parentB, childA, EntityRelation.MANAGES_TYPE); |
|
||||
EntityRelation relationB2 = new EntityRelation(parentB, childB, EntityRelation.MANAGES_TYPE); |
|
||||
|
|
||||
saveRelation(relationA1); |
|
||||
saveRelation(relationA2); |
|
||||
|
|
||||
saveRelation(relationB1); |
|
||||
saveRelation(relationB2); |
|
||||
|
|
||||
List<EntityRelation> relations = relationService.findByFrom(parentA, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(2, relations.size()); |
|
||||
for (EntityRelation relation : relations) { |
|
||||
Assert.assertEquals(EntityRelation.CONTAINS_TYPE, relation.getType()); |
|
||||
Assert.assertEquals(parentA, relation.getFrom()); |
|
||||
Assert.assertTrue(childA.equals(relation.getTo()) || childB.equals(relation.getTo())); |
|
||||
} |
|
||||
|
|
||||
relations = relationService.findByFromAndType(parentA, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(2, relations.size()); |
|
||||
|
|
||||
relations = relationService.findByFromAndType(parentA, EntityRelation.MANAGES_TYPE, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(0, relations.size()); |
|
||||
|
|
||||
relations = relationService.findByFrom(parentB, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(2, relations.size()); |
|
||||
for (EntityRelation relation : relations) { |
|
||||
Assert.assertEquals(EntityRelation.MANAGES_TYPE, relation.getType()); |
|
||||
Assert.assertEquals(parentB, relation.getFrom()); |
|
||||
Assert.assertTrue(childA.equals(relation.getTo()) || childB.equals(relation.getTo())); |
|
||||
} |
|
||||
|
|
||||
relations = relationService.findByFromAndType(parentB, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(0, relations.size()); |
|
||||
|
|
||||
relations = relationService.findByFromAndType(parentB, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(0, relations.size()); |
|
||||
} |
|
||||
|
|
||||
private Boolean saveRelation(EntityRelation relationA1) throws ExecutionException, InterruptedException { |
|
||||
return relationService.saveRelation(relationA1).get(); |
|
||||
} |
|
||||
|
|
||||
@Test |
|
||||
public void testFindTo() throws ExecutionException, InterruptedException { |
|
||||
AssetId parentA = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId parentB = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId childA = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId childB = new AssetId(UUIDs.timeBased()); |
|
||||
|
|
||||
EntityRelation relationA1 = new EntityRelation(parentA, childA, EntityRelation.CONTAINS_TYPE); |
|
||||
EntityRelation relationA2 = new EntityRelation(parentA, childB, EntityRelation.CONTAINS_TYPE); |
|
||||
|
|
||||
EntityRelation relationB1 = new EntityRelation(parentB, childA, EntityRelation.MANAGES_TYPE); |
|
||||
EntityRelation relationB2 = new EntityRelation(parentB, childB, EntityRelation.MANAGES_TYPE); |
|
||||
|
|
||||
saveRelation(relationA1); |
|
||||
saveRelation(relationA2); |
|
||||
|
|
||||
saveRelation(relationB1); |
|
||||
saveRelation(relationB2); |
|
||||
|
|
||||
// Data propagation to views is async
|
|
||||
Thread.sleep(3000); |
|
||||
|
|
||||
List<EntityRelation> relations = relationService.findByTo(childA, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(2, relations.size()); |
|
||||
for (EntityRelation relation : relations) { |
|
||||
Assert.assertEquals(childA, relation.getTo()); |
|
||||
Assert.assertTrue(parentA.equals(relation.getFrom()) || parentB.equals(relation.getFrom())); |
|
||||
} |
|
||||
|
|
||||
relations = relationService.findByToAndType(childA, EntityRelation.CONTAINS_TYPE, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(1, relations.size()); |
|
||||
|
|
||||
relations = relationService.findByToAndType(childB, EntityRelation.MANAGES_TYPE, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(1, relations.size()); |
|
||||
|
|
||||
relations = relationService.findByToAndType(parentA, EntityRelation.MANAGES_TYPE, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(0, relations.size()); |
|
||||
|
|
||||
relations = relationService.findByToAndType(parentB, EntityRelation.MANAGES_TYPE, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(0, relations.size()); |
|
||||
|
|
||||
relations = relationService.findByTo(childB, RelationTypeGroup.COMMON).get(); |
|
||||
Assert.assertEquals(2, relations.size()); |
|
||||
for (EntityRelation relation : relations) { |
|
||||
Assert.assertEquals(childB, relation.getTo()); |
|
||||
Assert.assertTrue(parentA.equals(relation.getFrom()) || parentB.equals(relation.getFrom())); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
@Test |
|
||||
public void testCyclicRecursiveRelation() throws ExecutionException, InterruptedException { |
|
||||
// A -> B -> C -> A
|
|
||||
AssetId assetA = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId assetB = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId assetC = new AssetId(UUIDs.timeBased()); |
|
||||
|
|
||||
EntityRelation relationA = new EntityRelation(assetA, assetB, EntityRelation.CONTAINS_TYPE); |
|
||||
EntityRelation relationB = new EntityRelation(assetB, assetC, EntityRelation.CONTAINS_TYPE); |
|
||||
EntityRelation relationC = new EntityRelation(assetC, assetA, EntityRelation.CONTAINS_TYPE); |
|
||||
|
|
||||
saveRelation(relationA); |
|
||||
saveRelation(relationB); |
|
||||
saveRelation(relationC); |
|
||||
|
|
||||
EntityRelationsQuery query = new EntityRelationsQuery(); |
|
||||
query.setParameters(new RelationsSearchParameters(assetA, EntitySearchDirection.FROM, -1)); |
|
||||
query.setFilters(Collections.singletonList(new EntityTypeFilter(EntityRelation.CONTAINS_TYPE, Collections.singletonList(EntityType.ASSET)))); |
|
||||
List<EntityRelation> relations = relationService.findByQuery(query).get(); |
|
||||
Assert.assertEquals(3, relations.size()); |
|
||||
Assert.assertTrue(relations.contains(relationA)); |
|
||||
Assert.assertTrue(relations.contains(relationB)); |
|
||||
Assert.assertTrue(relations.contains(relationC)); |
|
||||
} |
|
||||
|
|
||||
@Test |
|
||||
public void testRecursiveRelation() throws ExecutionException, InterruptedException { |
|
||||
// A -> B -> [C,D]
|
|
||||
AssetId assetA = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId assetB = new AssetId(UUIDs.timeBased()); |
|
||||
AssetId assetC = new AssetId(UUIDs.timeBased()); |
|
||||
DeviceId deviceD = new DeviceId(UUIDs.timeBased()); |
|
||||
|
|
||||
EntityRelation relationAB = new EntityRelation(assetA, assetB, EntityRelation.CONTAINS_TYPE); |
|
||||
EntityRelation relationBC = new EntityRelation(assetB, assetC, EntityRelation.CONTAINS_TYPE); |
|
||||
EntityRelation relationBD = new EntityRelation(assetB, deviceD, EntityRelation.CONTAINS_TYPE); |
|
||||
|
|
||||
|
|
||||
saveRelation(relationAB); |
|
||||
saveRelation(relationBC); |
|
||||
saveRelation(relationBD); |
|
||||
|
|
||||
EntityRelationsQuery query = new EntityRelationsQuery(); |
|
||||
query.setParameters(new RelationsSearchParameters(assetA, EntitySearchDirection.FROM, -1)); |
|
||||
query.setFilters(Collections.singletonList(new EntityTypeFilter(EntityRelation.CONTAINS_TYPE, Collections.singletonList(EntityType.ASSET)))); |
|
||||
List<EntityRelation> relations = relationService.findByQuery(query).get(); |
|
||||
Assert.assertEquals(2, relations.size()); |
|
||||
Assert.assertTrue(relations.contains(relationAB)); |
|
||||
Assert.assertTrue(relations.contains(relationBC)); |
|
||||
} |
|
||||
|
|
||||
|
|
||||
@Test(expected = DataValidationException.class) |
|
||||
public void testSaveRelationWithEmptyFrom() throws ExecutionException, InterruptedException { |
|
||||
EntityRelation relation = new EntityRelation(); |
|
||||
relation.setTo(new AssetId(UUIDs.timeBased())); |
|
||||
relation.setType(EntityRelation.CONTAINS_TYPE); |
|
||||
Assert.assertTrue(saveRelation(relation)); |
|
||||
} |
|
||||
|
|
||||
@Test(expected = DataValidationException.class) |
|
||||
public void testSaveRelationWithEmptyTo() throws ExecutionException, InterruptedException { |
|
||||
EntityRelation relation = new EntityRelation(); |
|
||||
relation.setFrom(new AssetId(UUIDs.timeBased())); |
|
||||
relation.setType(EntityRelation.CONTAINS_TYPE); |
|
||||
Assert.assertTrue(saveRelation(relation)); |
|
||||
} |
|
||||
|
|
||||
@Test(expected = DataValidationException.class) |
|
||||
public void testSaveRelationWithEmptyType() throws ExecutionException, InterruptedException { |
|
||||
EntityRelation relation = new EntityRelation(); |
|
||||
relation.setFrom(new AssetId(UUIDs.timeBased())); |
|
||||
relation.setTo(new AssetId(UUIDs.timeBased())); |
|
||||
Assert.assertTrue(saveRelation(relation)); |
|
||||
} |
|
||||
} |
|
||||
@ -0,0 +1,2 @@ |
|||||
|
TRUNCATE TABLE plugin; |
||||
|
TRUNCATE TABLE rule; |
||||
Loading…
Reference in new issue