use of org.thingsboard.server.common.data.edge.Edge in project thingsboard by thingsboard.
the class BaseEntityServiceTest method createEdge.
private Edge createEdge(int i, String type) {
Edge edge = new Edge();
edge.setTenantId(tenantId);
edge.setName("Edge" + i);
edge.setType(type);
edge.setLabel("EdgeLabel" + i);
edge.setSecret(RandomStringUtils.randomAlphanumeric(20));
edge.setRoutingKey(RandomStringUtils.randomAlphanumeric(20));
return edge;
}
use of org.thingsboard.server.common.data.edge.Edge in project thingsboard by thingsboard.
the class BaseEdgeServiceTest method testFindEdgesByTenantIdAndType.
@Test
public void testFindEdgesByTenantIdAndType() {
String title1 = "Edge title 1";
String type1 = "typeA";
List<Edge> edgesType1 = new ArrayList<>();
for (int i = 0; i < 143; i++) {
String suffix = RandomStringUtils.randomAlphanumeric(15);
String name = title1 + suffix;
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase();
Edge edge = constructEdge(name, type1);
edgesType1.add(edgeService.saveEdge(edge));
}
String title2 = "Edge title 2";
String type2 = "typeB";
List<Edge> edgesType2 = new ArrayList<>();
for (int i = 0; i < 175; i++) {
String suffix = RandomStringUtils.randomAlphanumeric(15);
String name = title2 + suffix;
name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase();
Edge edge = constructEdge(name, type2);
edgesType2.add(edgeService.saveEdge(edge));
}
List<Edge> loadedEdgesType1 = new ArrayList<>();
PageLink pageLink = new PageLink(15, 0, title1);
PageData<Edge> pageData = null;
do {
pageData = edgeService.findEdgesByTenantIdAndType(tenantId, type1, pageLink);
loadedEdgesType1.addAll(pageData.getData());
if (pageData.hasNext()) {
pageLink = pageLink.nextPageLink();
}
} while (pageData.hasNext());
Collections.sort(edgesType1, idComparator);
Collections.sort(loadedEdgesType1, idComparator);
Assert.assertEquals(edgesType1, loadedEdgesType1);
List<Edge> loadedEdgesType2 = new ArrayList<>();
pageLink = new PageLink(4, 0, title2);
do {
pageData = edgeService.findEdgesByTenantIdAndType(tenantId, type2, pageLink);
loadedEdgesType2.addAll(pageData.getData());
if (pageData.hasNext()) {
pageLink = pageLink.nextPageLink();
}
} while (pageData.hasNext());
Collections.sort(edgesType2, idComparator);
Collections.sort(loadedEdgesType2, idComparator);
Assert.assertEquals(edgesType2, loadedEdgesType2);
for (Edge edge : loadedEdgesType1) {
edgeService.deleteEdge(tenantId, edge.getId());
}
pageLink = new PageLink(4, 0, title1);
pageData = edgeService.findEdgesByTenantIdAndType(tenantId, type1, pageLink);
Assert.assertFalse(pageData.hasNext());
Assert.assertEquals(0, pageData.getData().size());
for (Edge edge : loadedEdgesType2) {
edgeService.deleteEdge(tenantId, edge.getId());
}
pageLink = new PageLink(4, 0, title2);
pageData = edgeService.findEdgesByTenantIdAndType(tenantId, type2, pageLink);
Assert.assertFalse(pageData.hasNext());
Assert.assertEquals(0, pageData.getData().size());
}
use of org.thingsboard.server.common.data.edge.Edge in project thingsboard by thingsboard.
the class BaseEdgeServiceTest method testFindEdgeTypesByTenantId.
@Test
public void testFindEdgeTypesByTenantId() throws Exception {
List<Edge> edges = new ArrayList<>();
try {
for (int i = 0; i < 3; i++) {
Edge edge = constructEdge("My edge B" + i, "typeB");
edges.add(edgeService.saveEdge(edge));
}
for (int i = 0; i < 7; i++) {
Edge edge = constructEdge("My edge C" + i, "typeC");
edges.add(edgeService.saveEdge(edge));
}
for (int i = 0; i < 9; i++) {
Edge edge = constructEdge("My edge A" + i, "typeA");
edges.add(edgeService.saveEdge(edge));
}
List<EntitySubtype> edgeTypes = edgeService.findEdgeTypesByTenantId(tenantId).get();
Assert.assertNotNull(edgeTypes);
Assert.assertEquals(3, edgeTypes.size());
Assert.assertEquals("typeA", edgeTypes.get(0).getType());
Assert.assertEquals("typeB", edgeTypes.get(1).getType());
Assert.assertEquals("typeC", edgeTypes.get(2).getType());
} finally {
edges.forEach((edge) -> {
edgeService.deleteEdge(tenantId, edge.getId());
});
}
}
use of org.thingsboard.server.common.data.edge.Edge in project thingsboard by thingsboard.
the class BaseEdgeServiceTest method testSaveEdgeWithEmptyName.
@Test(expected = DataValidationException.class)
public void testSaveEdgeWithEmptyName() {
Edge edge = new Edge();
edge.setType("default");
edge.setTenantId(tenantId);
edgeService.saveEdge(edge);
}
use of org.thingsboard.server.common.data.edge.Edge in project thingsboard by thingsboard.
the class BaseEdgeServiceTest method testAssignEdgeToCustomerFromDifferentTenant.
@Test(expected = DataValidationException.class)
public void testAssignEdgeToCustomerFromDifferentTenant() {
Edge edge = constructEdge("My edge", "default");
edge = edgeService.saveEdge(edge);
Tenant tenant = new Tenant();
tenant.setTitle("Test different tenant");
tenant = tenantService.saveTenant(tenant);
Customer customer = new Customer();
customer.setTenantId(tenant.getId());
customer.setTitle("Test different customer");
customer = customerService.saveCustomer(customer);
try {
edgeService.assignEdgeToCustomer(tenantId, edge.getId(), customer.getId());
} finally {
edgeService.deleteEdge(tenantId, edge.getId());
tenantService.deleteTenant(tenant.getId());
}
}
Aggregations