Examples with AttributesResponse org.thingsboard.server.msa.mapper.AttributesResponse used on opensource projects

Example 1 with AttributesResponse

use of org.thingsboard.server.msa.mapper.AttributesResponse in project thingsboard by thingsboard.

the class MqttClientTest method requestAttributeValuesFromServer.

@Test
public void requestAttributeValuesFromServer() throws Exception {
    restClient.login("tenant@thingsboard.org", "tenant");
    Device device = createDevice("mqtt_");
    DeviceCredentials deviceCredentials = restClient.getDeviceCredentialsByDeviceId(device.getId()).get();
    WsClient wsClient = subscribeToWebSocket(device.getId(), "CLIENT_SCOPE", CmdsType.ATTR_SUB_CMDS);
    MqttMessageListener listener = new MqttMessageListener();
    MqttClient mqttClient = getMqttClient(deviceCredentials, listener);
    // Add a new client attribute
    JsonObject clientAttributes = new JsonObject();
    String clientAttributeValue = RandomStringUtils.randomAlphanumeric(8);
    clientAttributes.addProperty("clientAttr", clientAttributeValue);
    mqttClient.publish("v1/devices/me/attributes", Unpooled.wrappedBuffer(clientAttributes.toString().getBytes())).get();
    WsTelemetryResponse actualLatestTelemetry = wsClient.getLastMessage();
    log.info("Received ws telemetry: {}", actualLatestTelemetry);
    wsClient.closeBlocking();
    Assert.assertEquals(1, actualLatestTelemetry.getData().size());
    Assert.assertEquals(Sets.newHashSet("clientAttr"), actualLatestTelemetry.getLatestValues().keySet());
    Assert.assertTrue(verify(actualLatestTelemetry, "clientAttr", clientAttributeValue));
    // Add a new shared attribute
    JsonObject sharedAttributes = new JsonObject();
    String sharedAttributeValue = RandomStringUtils.randomAlphanumeric(8);
    sharedAttributes.addProperty("sharedAttr", sharedAttributeValue);
    ResponseEntity sharedAttributesResponse = restClient.getRestTemplate().postForEntity(HTTPS_URL + "/api/plugins/telemetry/DEVICE/{deviceId}/SHARED_SCOPE", mapper.readTree(sharedAttributes.toString()), ResponseEntity.class, device.getId());
    Assert.assertTrue(sharedAttributesResponse.getStatusCode().is2xxSuccessful());
    // Subscribe to attributes response
    mqttClient.on("v1/devices/me/attributes/response/+", listener, MqttQoS.AT_LEAST_ONCE).get();
    // Wait until subscription is processed
    TimeUnit.SECONDS.sleep(3);
    // Request attributes
    JsonObject request = new JsonObject();
    request.addProperty("clientKeys", "clientAttr");
    request.addProperty("sharedKeys", "sharedAttr");
    mqttClient.publish("v1/devices/me/attributes/request/" + new Random().nextInt(100), Unpooled.wrappedBuffer(request.toString().getBytes())).get();
    MqttEvent event = listener.getEvents().poll(10, TimeUnit.SECONDS);
    AttributesResponse attributes = mapper.readValue(Objects.requireNonNull(event).getMessage(), AttributesResponse.class);
    log.info("Received telemetry: {}", attributes);
    Assert.assertEquals(1, attributes.getClient().size());
    Assert.assertEquals(clientAttributeValue, attributes.getClient().get("clientAttr"));
    Assert.assertEquals(1, attributes.getShared().size());
    Assert.assertEquals(sharedAttributeValue, attributes.getShared().get("sharedAttr"));
    restClient.getRestTemplate().delete(HTTPS_URL + "/api/device/" + device.getId());
}