Examples with AbstractMessage com.google.protobuf.AbstractMessage used on opensource projects

Example 26 with AbstractMessage

use of com.google.protobuf.AbstractMessage in project thingsboard by thingsboard.

the class BaseEdgeTest method testDevices.

@Test
public void testDevices() throws Exception {
    // 1
    Device savedDevice = saveDeviceOnCloudAndVerifyDeliveryToEdge();
    // 2
    edgeImitator.expectMessageAmount(1);
    doDelete("/api/edge/" + edge.getUuidId() + "/device/" + savedDevice.getUuidId(), Device.class);
    Assert.assertTrue(edgeImitator.waitForMessages());
    AbstractMessage latestMessage = edgeImitator.getLatestMessage();
    Assert.assertTrue(latestMessage instanceof DeviceUpdateMsg);
    DeviceUpdateMsg deviceUpdateMsg = (DeviceUpdateMsg) latestMessage;
    Assert.assertEquals(UpdateMsgType.ENTITY_DELETED_RPC_MESSAGE, deviceUpdateMsg.getMsgType());
    Assert.assertEquals(deviceUpdateMsg.getIdMSB(), savedDevice.getUuidId().getMostSignificantBits());
    Assert.assertEquals(deviceUpdateMsg.getIdLSB(), savedDevice.getUuidId().getLeastSignificantBits());
    // 3
    edgeImitator.expectMessageAmount(1);
    doDelete("/api/device/" + savedDevice.getUuidId()).andExpect(status().isOk());
    // we should not get any message because device is not assigned to edge any more
    Assert.assertFalse(edgeImitator.waitForMessages(1));
    // 4
    edgeImitator.expectMessageAmount(1);
    savedDevice = saveDevice("Edge Device 3", "Default");
    doPost("/api/edge/" + edge.getUuidId() + "/device/" + savedDevice.getUuidId(), Device.class);
    Assert.assertTrue(edgeImitator.waitForMessages());
    latestMessage = edgeImitator.getLatestMessage();
    Assert.assertTrue(latestMessage instanceof DeviceUpdateMsg);
    deviceUpdateMsg = (DeviceUpdateMsg) latestMessage;
    Assert.assertEquals(UpdateMsgType.ENTITY_CREATED_RPC_MESSAGE, deviceUpdateMsg.getMsgType());
    Assert.assertEquals(deviceUpdateMsg.getIdMSB(), savedDevice.getUuidId().getMostSignificantBits());
    Assert.assertEquals(deviceUpdateMsg.getIdLSB(), savedDevice.getUuidId().getLeastSignificantBits());
    Assert.assertEquals(deviceUpdateMsg.getName(), savedDevice.getName());
    Assert.assertEquals(deviceUpdateMsg.getType(), savedDevice.getType());
    // 5
    edgeImitator.expectMessageAmount(1);
    doDelete("/api/device/" + savedDevice.getUuidId()).andExpect(status().isOk());
    // in this case we should get messages because device was assigned to edge
    Assert.assertTrue(edgeImitator.waitForMessages());
    latestMessage = edgeImitator.getLatestMessage();
    Assert.assertTrue(latestMessage instanceof DeviceUpdateMsg);
    deviceUpdateMsg = (DeviceUpdateMsg) latestMessage;
    Assert.assertEquals(UpdateMsgType.ENTITY_DELETED_RPC_MESSAGE, deviceUpdateMsg.getMsgType());
    Assert.assertEquals(deviceUpdateMsg.getIdMSB(), savedDevice.getUuidId().getMostSignificantBits());
    Assert.assertEquals(deviceUpdateMsg.getIdLSB(), savedDevice.getUuidId().getLeastSignificantBits());
}

Example 27 with AbstractMessage

use of com.google.protobuf.AbstractMessage in project thingsboard by thingsboard.

the class BaseEdgeTest method testSendRelationRequestToCloud.

@Test
public void testSendRelationRequestToCloud() throws Exception {
    Device device = findDeviceByName("Edge Device 1");
    Asset asset = findAssetByName("Edge Asset 1");
    EntityRelation relation = new EntityRelation();
    relation.setType("test");
    relation.setFrom(device.getId());
    relation.setTo(asset.getId());
    relation.setTypeGroup(RelationTypeGroup.COMMON);
    edgeImitator.expectMessageAmount(1);
    doPost("/api/relation", relation);
    Assert.assertTrue(edgeImitator.waitForMessages());
    UplinkMsg.Builder uplinkMsgBuilder = UplinkMsg.newBuilder();
    RelationRequestMsg.Builder relationRequestMsgBuilder = RelationRequestMsg.newBuilder();
    relationRequestMsgBuilder.setEntityIdMSB(device.getId().getId().getMostSignificantBits());
    relationRequestMsgBuilder.setEntityIdLSB(device.getId().getId().getLeastSignificantBits());
    relationRequestMsgBuilder.setEntityType(device.getId().getEntityType().name());
    testAutoGeneratedCodeByProtobuf(relationRequestMsgBuilder);
    uplinkMsgBuilder.addRelationRequestMsg(relationRequestMsgBuilder.build());
    testAutoGeneratedCodeByProtobuf(uplinkMsgBuilder);
    edgeImitator.expectResponsesAmount(1);
    edgeImitator.expectMessageAmount(1);
    edgeImitator.sendUplinkMsg(uplinkMsgBuilder.build());
    Assert.assertTrue(edgeImitator.waitForResponses());
    Assert.assertTrue(edgeImitator.waitForMessages());
    AbstractMessage latestMessage = edgeImitator.getLatestMessage();
    Assert.assertTrue(latestMessage instanceof RelationUpdateMsg);
    RelationUpdateMsg relationUpdateMsg = (RelationUpdateMsg) latestMessage;
    Assert.assertEquals(UpdateMsgType.ENTITY_CREATED_RPC_MESSAGE, relationUpdateMsg.getMsgType());
    Assert.assertEquals(relation.getType(), relationUpdateMsg.getType());
    UUID fromUUID = new UUID(relationUpdateMsg.getFromIdMSB(), relationUpdateMsg.getFromIdLSB());
    EntityId fromEntityId = EntityIdFactory.getByTypeAndUuid(relationUpdateMsg.getFromEntityType(), fromUUID);
    Assert.assertEquals(relation.getFrom(), fromEntityId);
    UUID toUUID = new UUID(relationUpdateMsg.getToIdMSB(), relationUpdateMsg.getToIdLSB());
    EntityId toEntityId = EntityIdFactory.getByTypeAndUuid(relationUpdateMsg.getToEntityType(), toUUID);
    Assert.assertEquals(relation.getTo(), toEntityId);
    Assert.assertEquals(relation.getTypeGroup().name(), relationUpdateMsg.getTypeGroup());
}