Examples with WsClient org.thingsboard.server.msa.WsClient used on opensource projects

Example 1 with WsClient

use of org.thingsboard.server.msa.WsClient in project thingsboard by thingsboard.

the class MqttGatewayClientTest method telemetryUpload.

@Test
public void telemetryUpload() throws Exception {
    WsClient wsClient = subscribeToWebSocket(createdDevice.getId(), "LATEST_TELEMETRY", CmdsType.TS_SUB_CMDS);
    mqttClient.publish("v1/gateway/telemetry", Unpooled.wrappedBuffer(createGatewayPayload(createdDevice.getName(), -1).toString().getBytes())).get();
    WsTelemetryResponse actualLatestTelemetry = wsClient.getLastMessage();
    log.info("Received telemetry: {}", actualLatestTelemetry);
    wsClient.closeBlocking();
    Assert.assertEquals(4, actualLatestTelemetry.getData().size());
    Assert.assertEquals(Sets.newHashSet("booleanKey", "stringKey", "doubleKey", "longKey"), actualLatestTelemetry.getLatestValues().keySet());
    Assert.assertTrue(verify(actualLatestTelemetry, "booleanKey", Boolean.TRUE.toString()));
    Assert.assertTrue(verify(actualLatestTelemetry, "stringKey", "value1"));
    Assert.assertTrue(verify(actualLatestTelemetry, "doubleKey", Double.toString(42.0)));
    Assert.assertTrue(verify(actualLatestTelemetry, "longKey", Long.toString(73)));
}

Example 2 with WsClient

use of org.thingsboard.server.msa.WsClient in project thingsboard by thingsboard.

the class MqttClientTest method telemetryUpload.

@Test
public void telemetryUpload() throws Exception {
    restClient.login("tenant@thingsboard.org", "tenant");
    Device device = createDevice("mqtt_");
    DeviceCredentials deviceCredentials = restClient.getDeviceCredentialsByDeviceId(device.getId()).get();
    WsClient wsClient = subscribeToWebSocket(device.getId(), "LATEST_TELEMETRY", CmdsType.TS_SUB_CMDS);
    MqttClient mqttClient = getMqttClient(deviceCredentials, null);
    mqttClient.publish("v1/devices/me/telemetry", Unpooled.wrappedBuffer(createPayload().toString().getBytes())).get();
    WsTelemetryResponse actualLatestTelemetry = wsClient.getLastMessage();
    log.info("Received telemetry: {}", actualLatestTelemetry);
    wsClient.closeBlocking();
    Assert.assertEquals(4, actualLatestTelemetry.getData().size());
    Assert.assertEquals(Sets.newHashSet("booleanKey", "stringKey", "doubleKey", "longKey"), actualLatestTelemetry.getLatestValues().keySet());
    Assert.assertTrue(verify(actualLatestTelemetry, "booleanKey", Boolean.TRUE.toString()));
    Assert.assertTrue(verify(actualLatestTelemetry, "stringKey", "value1"));
    Assert.assertTrue(verify(actualLatestTelemetry, "doubleKey", Double.toString(42.0)));
    Assert.assertTrue(verify(actualLatestTelemetry, "longKey", Long.toString(73)));
    restClient.getRestTemplate().delete(HTTPS_URL + "/api/device/" + device.getId());
}

Example 3 with WsClient

use of org.thingsboard.server.msa.WsClient 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());
}

Example 4 with WsClient

use of org.thingsboard.server.msa.WsClient in project thingsboard by thingsboard.

the class HttpClientTest method telemetryUpload.

@Test
public void telemetryUpload() throws Exception {
    restClient.login("tenant@thingsboard.org", "tenant");
    Device device = createDevice("http_");
    DeviceCredentials deviceCredentials = restClient.getDeviceCredentialsByDeviceId(device.getId()).get();
    WsClient wsClient = subscribeToWebSocket(device.getId(), "LATEST_TELEMETRY", CmdsType.TS_SUB_CMDS);
    ResponseEntity deviceTelemetryResponse = restClient.getRestTemplate().postForEntity(HTTPS_URL + "/api/v1/{credentialsId}/telemetry", mapper.readTree(createPayload().toString()), ResponseEntity.class, deviceCredentials.getCredentialsId());
    Assert.assertTrue(deviceTelemetryResponse.getStatusCode().is2xxSuccessful());
    WsTelemetryResponse actualLatestTelemetry = wsClient.getLastMessage();
    wsClient.closeBlocking();
    Assert.assertEquals(Sets.newHashSet("booleanKey", "stringKey", "doubleKey", "longKey"), actualLatestTelemetry.getLatestValues().keySet());
    Assert.assertTrue(verify(actualLatestTelemetry, "booleanKey", Boolean.TRUE.toString()));
    Assert.assertTrue(verify(actualLatestTelemetry, "stringKey", "value1"));
    Assert.assertTrue(verify(actualLatestTelemetry, "doubleKey", Double.toString(42.0)));
    Assert.assertTrue(verify(actualLatestTelemetry, "longKey", Long.toString(73)));
    restClient.deleteDevice(device.getId());
}

Example 5 with WsClient

use of org.thingsboard.server.msa.WsClient in project thingsboard by thingsboard.

the class MqttGatewayClientTest method publishAttributeUpdateToServer.

@Test
public void publishAttributeUpdateToServer() throws Exception {
    Optional<DeviceCredentials> createdDeviceCredentials = restClient.getDeviceCredentialsByDeviceId(createdDevice.getId());
    Assert.assertTrue(createdDeviceCredentials.isPresent());
    WsClient wsClient = subscribeToWebSocket(createdDevice.getId(), "CLIENT_SCOPE", CmdsType.ATTR_SUB_CMDS);
    JsonObject clientAttributes = new JsonObject();
    clientAttributes.addProperty("attr1", "value1");
    clientAttributes.addProperty("attr2", true);
    clientAttributes.addProperty("attr3", 42.0);
    clientAttributes.addProperty("attr4", 73);
    JsonObject gatewayClientAttributes = new JsonObject();
    gatewayClientAttributes.add(createdDevice.getName(), clientAttributes);
    mqttClient.publish("v1/gateway/attributes", Unpooled.wrappedBuffer(gatewayClientAttributes.toString().getBytes())).get();
    WsTelemetryResponse actualLatestTelemetry = wsClient.getLastMessage();
    log.info("Received attributes: {}", actualLatestTelemetry);
    wsClient.closeBlocking();
    Assert.assertEquals(4, actualLatestTelemetry.getData().size());
    Assert.assertEquals(Sets.newHashSet("attr1", "attr2", "attr3", "attr4"), actualLatestTelemetry.getLatestValues().keySet());
    Assert.assertTrue(verify(actualLatestTelemetry, "attr1", "value1"));
    Assert.assertTrue(verify(actualLatestTelemetry, "attr2", Boolean.TRUE.toString()));
    Assert.assertTrue(verify(actualLatestTelemetry, "attr3", Double.toString(42.0)));
    Assert.assertTrue(verify(actualLatestTelemetry, "attr4", Long.toString(73)));
}