Search in sources :

Example 31 with Edge

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;
}
Also used : Edge(org.thingsboard.server.common.data.edge.Edge)

Example 32 with 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());
}
Also used : ArrayList(java.util.ArrayList) PageLink(org.thingsboard.server.common.data.page.PageLink) Edge(org.thingsboard.server.common.data.edge.Edge) Test(org.junit.Test)

Example 33 with Edge

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());
        });
    }
}
Also used : ArrayList(java.util.ArrayList) EntitySubtype(org.thingsboard.server.common.data.EntitySubtype) Edge(org.thingsboard.server.common.data.edge.Edge) Test(org.junit.Test)

Example 34 with Edge

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);
}
Also used : Edge(org.thingsboard.server.common.data.edge.Edge) Test(org.junit.Test)

Example 35 with 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());
    }
}
Also used : Tenant(org.thingsboard.server.common.data.Tenant) Customer(org.thingsboard.server.common.data.Customer) Edge(org.thingsboard.server.common.data.edge.Edge) Test(org.junit.Test)

Aggregations

Edge (org.thingsboard.server.common.data.edge.Edge)98 Test (org.junit.Test)46 EdgeId (org.thingsboard.server.common.data.id.EdgeId)32 PageLink (org.thingsboard.server.common.data.page.PageLink)31 ArrayList (java.util.ArrayList)25 ThingsboardException (org.thingsboard.server.common.data.exception.ThingsboardException)25 ApiOperation (io.swagger.annotations.ApiOperation)23 PreAuthorize (org.springframework.security.access.prepost.PreAuthorize)23 RequestMapping (org.springframework.web.bind.annotation.RequestMapping)23 ResponseBody (org.springframework.web.bind.annotation.ResponseBody)22 DataValidationException (org.thingsboard.server.dao.exception.DataValidationException)22 CustomerId (org.thingsboard.server.common.data.id.CustomerId)21 Customer (org.thingsboard.server.common.data.Customer)20 IncorrectParameterException (org.thingsboard.server.dao.exception.IncorrectParameterException)20 EntityRelation (org.thingsboard.server.common.data.relation.EntityRelation)16 TypeReference (com.fasterxml.jackson.core.type.TypeReference)15 TenantId (org.thingsboard.server.common.data.id.TenantId)14 Device (org.thingsboard.server.common.data.Device)13 Asset (org.thingsboard.server.common.data.asset.Asset)13 PageData (org.thingsboard.server.common.data.page.PageData)13