/** * Copyright © 2016-2026 The Thingsboard Authors * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package org.thingsboard.server.controller; import com.datastax.oss.driver.api.core.uuid.Uuids; import com.fasterxml.jackson.core.type.TypeReference; import com.fasterxml.jackson.databind.JsonNode; import com.google.common.util.concurrent.Futures; import com.google.common.util.concurrent.ListenableFuture; import com.google.common.util.concurrent.ListeningExecutorService; import com.google.common.util.concurrent.MoreExecutors; import org.junit.After; import org.junit.Assert; import org.junit.Before; import org.junit.Ignore; import org.junit.Test; import org.mockito.AdditionalAnswers; import org.mockito.ArgumentCaptor; import org.mockito.Mockito; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.mock.mockito.SpyBean; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Primary; import org.springframework.test.context.ContextConfiguration; import org.testcontainers.shaded.org.awaitility.Awaitility; import org.thingsboard.common.util.JacksonUtil; import org.thingsboard.common.util.ThingsBoardExecutors; import org.thingsboard.server.common.data.AttributeScope; import org.thingsboard.server.common.data.Customer; import org.thingsboard.server.common.data.Device; import org.thingsboard.server.common.data.DeviceInfo; import org.thingsboard.server.common.data.DeviceProfile; import org.thingsboard.server.common.data.EntitySubtype; import org.thingsboard.server.common.data.EntityType; import org.thingsboard.server.common.data.OtaPackageInfo; import org.thingsboard.server.common.data.SaveDeviceWithCredentialsRequest; import org.thingsboard.server.common.data.SaveOtaPackageInfoRequest; import org.thingsboard.server.common.data.StringUtils; import org.thingsboard.server.common.data.Tenant; import org.thingsboard.server.common.data.User; import org.thingsboard.server.common.data.alarm.Alarm; import org.thingsboard.server.common.data.alarm.AlarmInfo; import org.thingsboard.server.common.data.alarm.AlarmSeverity; import org.thingsboard.server.common.data.audit.ActionType; import org.thingsboard.server.common.data.edge.Edge; import org.thingsboard.server.common.data.id.CustomerId; import org.thingsboard.server.common.data.id.DeviceCredentialsId; import org.thingsboard.server.common.data.id.DeviceId; import org.thingsboard.server.common.data.id.EntityId; import org.thingsboard.server.common.data.kv.AttributeKvEntry; import org.thingsboard.server.common.data.page.PageData; import org.thingsboard.server.common.data.page.PageLink; import org.thingsboard.server.common.data.relation.EntityRelation; import org.thingsboard.server.common.data.relation.RelationTypeGroup; import org.thingsboard.server.common.data.security.Authority; import org.thingsboard.server.common.data.security.DeviceCredentials; import org.thingsboard.server.common.data.security.DeviceCredentialsType; import org.thingsboard.server.common.data.sync.ie.importing.csv.BulkImportColumnType; import org.thingsboard.server.common.data.sync.ie.importing.csv.BulkImportRequest; import org.thingsboard.server.common.data.sync.ie.importing.csv.BulkImportResult; import org.thingsboard.server.dao.device.DeviceDao; import org.thingsboard.server.exception.DataValidationException; import org.thingsboard.server.dao.exception.DeviceCredentialsValidationException; import org.thingsboard.server.dao.model.ModelConstants; import org.thingsboard.server.dao.service.DaoSqlTest; import org.thingsboard.server.gen.transport.TransportProtos; import org.thingsboard.server.service.gateway_device.GatewayNotificationsService; import org.thingsboard.server.service.state.DeviceStateService; import org.thingsboard.server.utils.CsvUtils; import java.nio.charset.StandardCharsets; import java.util.ArrayList; import java.util.Arrays; import java.util.LinkedList; import java.util.List; import java.util.Map; import java.util.Optional; import java.util.concurrent.TimeUnit; import static org.assertj.core.api.Assertions.assertThat; import static org.awaitility.Awaitility.await; import static org.hamcrest.Matchers.containsString; import static org.mockito.ArgumentMatchers.any; import static org.mockito.Mockito.never; import static org.mockito.Mockito.times; import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status; import static org.thingsboard.server.common.data.ota.OtaPackageType.FIRMWARE; import static org.thingsboard.server.common.data.ota.OtaPackageType.SOFTWARE; import static org.thingsboard.server.dao.model.ModelConstants.NULL_UUID; @ContextConfiguration(classes = {DeviceControllerTest.Config.class}) @DaoSqlTest public class DeviceControllerTest extends AbstractControllerTest { static final TypeReference> PAGE_DATA_DEVICE_TYPE_REF = new TypeReference<>() { }; ListeningExecutorService executor; List> futures; PageData pageData; private Tenant savedTenant; private User tenantAdmin; @SpyBean private GatewayNotificationsService gatewayNotificationsService; @SpyBean private DeviceStateService deviceStateService; @Autowired private DeviceDao deviceDao; static class Config { @Bean @Primary public DeviceDao deviceDao(DeviceDao deviceDao) { return Mockito.mock(DeviceDao.class, AdditionalAnswers.delegatesTo(deviceDao)); } } @Before public void beforeTest() throws Exception { executor = MoreExecutors.listeningDecorator(ThingsBoardExecutors.newWorkStealingPool(8, getClass())); loginSysAdmin(); Tenant tenant = new Tenant(); tenant.setTitle("My tenant"); savedTenant = saveTenant(tenant); Assert.assertNotNull(savedTenant); tenantAdmin = new User(); tenantAdmin.setAuthority(Authority.TENANT_ADMIN); tenantAdmin.setTenantId(savedTenant.getId()); tenantAdmin.setEmail("tenant2@thingsboard.org"); tenantAdmin.setFirstName("Joe"); tenantAdmin.setLastName("Downs"); tenantAdmin = createUserAndLogin(tenantAdmin, "testPassword1"); } @After public void afterTest() throws Exception { executor.shutdownNow(); loginSysAdmin(); deleteTenant(savedTenant.getId()); } @Test public void testSaveDevice() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); Device savedDevice = doPost("/api/device", device, Device.class); Device oldDevice = new Device(savedDevice); testNotifyEntityAllOneTime(savedDevice, savedDevice.getId(), savedDevice.getId(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ADDED); testNotificationUpdateGatewayNever(); Assert.assertNotNull(savedDevice); Assert.assertNotNull(savedDevice.getId()); Assert.assertTrue(savedDevice.getCreatedTime() > 0); Assert.assertEquals(savedTenant.getId(), savedDevice.getTenantId()); Assert.assertNotNull(savedDevice.getCustomerId()); Assert.assertEquals(NULL_UUID, savedDevice.getCustomerId().getId()); Assert.assertEquals(device.getName(), savedDevice.getName()); DeviceCredentials deviceCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); Assert.assertNotNull(deviceCredentials); Assert.assertNotNull(deviceCredentials.getId()); Assert.assertEquals(savedDevice.getId(), deviceCredentials.getDeviceId()); Assert.assertEquals(DeviceCredentialsType.ACCESS_TOKEN, deviceCredentials.getCredentialsType()); Assert.assertNotNull(deviceCredentials.getCredentialsId()); Assert.assertEquals(20, deviceCredentials.getCredentialsId().length()); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); savedDevice.setName("My new device"); savedDevice = doPost("/api/device", savedDevice, Device.class); testNotifyEntityAllOneTime(savedDevice, savedDevice.getId(), savedDevice.getId(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.UPDATED); testNotificationUpdateGatewayOneTime(savedDevice, oldDevice); Device foundDevice = doGet("/api/device/" + savedDevice.getId().getId(), Device.class); Assert.assertEquals(foundDevice.getName(), savedDevice.getName()); } @Test public void testSaveDeviceWithCredentials() throws Exception { String testToken = "TEST_TOKEN"; Device device = new Device(); device.setName("My device"); device.setType("default"); DeviceCredentials deviceCredentials = new DeviceCredentials(); deviceCredentials.setCredentialsType(DeviceCredentialsType.ACCESS_TOKEN); deviceCredentials.setCredentialsId(testToken); SaveDeviceWithCredentialsRequest saveRequest = new SaveDeviceWithCredentialsRequest(device, deviceCredentials); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); Device savedDevice = readResponse(doPost("/api/device-with-credentials", saveRequest).andExpect(status().isOk()), Device.class); Device oldDevice = new Device(savedDevice); testNotifyEntityAllOneTime(savedDevice, savedDevice.getId(), savedDevice.getId(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ADDED); testNotificationUpdateGatewayNever(); Assert.assertNotNull(savedDevice); Assert.assertNotNull(savedDevice.getId()); Assert.assertTrue(savedDevice.getCreatedTime() > 0); Assert.assertEquals(savedTenant.getId(), savedDevice.getTenantId()); Assert.assertNotNull(savedDevice.getCustomerId()); Assert.assertEquals(NULL_UUID, savedDevice.getCustomerId().getId()); Assert.assertEquals(device.getName(), savedDevice.getName()); DeviceCredentials foundDeviceCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); Assert.assertNotNull(foundDeviceCredentials); Assert.assertNotNull(foundDeviceCredentials.getId()); Assert.assertEquals(savedDevice.getId(), foundDeviceCredentials.getDeviceId()); Assert.assertEquals(DeviceCredentialsType.ACCESS_TOKEN, foundDeviceCredentials.getCredentialsType()); Assert.assertEquals(testToken, foundDeviceCredentials.getCredentialsId()); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); savedDevice.setName("My new device"); savedDevice = doPost("/api/device", savedDevice, Device.class); testNotifyEntityAllOneTime(savedDevice, savedDevice.getId(), savedDevice.getId(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.UPDATED); testNotificationUpdateGatewayOneTime(savedDevice, oldDevice); } @Test public void testSaveDeviceWithCredentials_CredentialsIsNull() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); SaveDeviceWithCredentialsRequest saveRequest = new SaveDeviceWithCredentialsRequest(device, null); doPost("/api/device-with-credentials", saveRequest).andExpect(status().isBadRequest()) .andExpect(statusReason(containsString("Validation error: credentials must not be null"))); } @Test public void testSaveDeviceWithCredentials_DeviceIsNull() throws Exception { String testToken = "TEST_TOKEN"; DeviceCredentials deviceCredentials = new DeviceCredentials(); deviceCredentials.setCredentialsType(DeviceCredentialsType.ACCESS_TOKEN); deviceCredentials.setCredentialsId(testToken); SaveDeviceWithCredentialsRequest saveRequest = new SaveDeviceWithCredentialsRequest(null, deviceCredentials); doPost("/api/device-with-credentials", saveRequest).andExpect(status().isBadRequest()) .andExpect(statusReason(containsString("Validation error: device must not be null"))); } @Test public void testSaveDeviceWithCredentials_WithExistingName() throws Exception { String testToken = "TEST_TOKEN"; Device device = new Device(); device.setName("My device"); device.setType("default"); DeviceCredentials deviceCredentials = new DeviceCredentials(); deviceCredentials.setCredentialsType(DeviceCredentialsType.ACCESS_TOKEN); deviceCredentials.setCredentialsId(testToken); SaveDeviceWithCredentialsRequest saveRequest = new SaveDeviceWithCredentialsRequest(device, deviceCredentials); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); Device savedDevice = readResponse(doPost("/api/device-with-credentials", saveRequest).andExpect(status().isOk()), Device.class); Assert.assertNotNull(savedDevice); doPost("/api/device-with-credentials", saveRequest).andExpect(status().isBadRequest()) .andExpect(statusReason(containsString("Device with such name already exists!"))); } @Test public void saveDeviceWithViolationOfValidation() throws Exception { Device device = new Device(); device.setName(StringUtils.randomAlphabetic(300)); device.setType("default"); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); String msgError = msgErrorFieldLength("name"); doPost("/api/device", device) .andExpect(status().isBadRequest()) .andExpect(statusReason(containsString(msgError))); testNotifyEntityEqualsOneTimeServiceNeverError(device, savedTenant.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ADDED, new DataValidationException(msgError)); testNotificationUpdateGatewayNever(); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); device.setTenantId(savedTenant.getId()); msgError = msgErrorFieldLength("type"); device.setType(StringUtils.randomAlphabetic(300)); doPost("/api/device", device) .andExpect(status().isBadRequest()) .andExpect(statusReason(containsString(msgError))); testNotifyEntityEqualsOneTimeServiceNeverError(device, savedTenant.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ADDED, new DataValidationException(msgError)); testNotificationUpdateGatewayNever(); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); msgError = msgErrorFieldLength("label"); device.setType("Normal type"); device.setLabel(StringUtils.randomAlphabetic(300)); doPost("/api/device", device) .andExpect(status().isBadRequest()) .andExpect(statusReason(containsString(msgError))); testNotifyEntityEqualsOneTimeServiceNeverError(device, savedTenant.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ADDED, new DataValidationException(msgError)); testNotificationUpdateGatewayNever(); } @Test public void testUpdateDeviceFromDifferentTenant() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); loginDifferentTenant(); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); String savedDeviceIdStr = savedDevice.getId().getId().toString(); doPost("/api/device", savedDevice) .andExpect(status().isNotFound()) .andExpect(statusReason(containsString(msgErrorNoFound("Device", savedDeviceIdStr)))); testNotifyEntityNever(savedDevice.getId(), savedDevice); testNotificationUpdateGatewayNever(); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); doDelete("/api/device/" + savedDeviceIdStr) .andExpect(status().isNotFound()) .andExpect(statusReason(containsString(msgErrorNoFound("Device", savedDeviceIdStr)))); testNotifyEntityNever(savedDevice.getId(), savedDevice); testNotificationUpdateGatewayNever(); deleteDifferentTenant(); } @Test public void testSaveDeviceWithProfileFromDifferentTenant() throws Exception { loginDifferentTenant(); DeviceProfile differentProfile = createDeviceProfile("Different profile"); differentProfile = doPost("/api/deviceProfile", differentProfile, DeviceProfile.class); loginTenantAdmin(); Device device = new Device(); device.setName("My device"); device.setDeviceProfileId(differentProfile.getId()); doPost("/api/device", device).andExpect(status().isBadRequest()) .andExpect(statusReason(containsString("Device can`t be referencing to device profile from different tenant!"))); } @Test public void testSaveDeviceWithFirmware() throws Exception { loginTenantAdmin(); DeviceProfile profile = createDeviceProfile("Profile to test ota updates"); profile = doPost("/api/deviceProfile", profile, DeviceProfile.class); SaveOtaPackageInfoRequest firmwareInfo = new SaveOtaPackageInfoRequest(); firmwareInfo.setDeviceProfileId(profile.getId()); firmwareInfo.setType(FIRMWARE); String title = "title"; firmwareInfo.setTitle(title); String fwVersion = "1.0"; firmwareInfo.setVersion(fwVersion); String url = "test.url"; firmwareInfo.setUrl(url); firmwareInfo.setUsesUrl(true); OtaPackageInfo savedFw = doPost("/api/otaPackage", firmwareInfo, OtaPackageInfo.class); Device device = new Device(); device.setName("My ota device"); device.setDeviceProfileId(profile.getId()); device.setFirmwareId(savedFw.getId()); device = doPost("/api/device", device, Device.class); //check shared attributes Device finalDevice = device; await().atMost(TIMEOUT, TimeUnit.SECONDS).until(() -> { List> attributes = doGetAsyncTyped("/api/plugins/telemetry/DEVICE/" + finalDevice.getId() + "/values/attributes/SHARED_SCOPE", new TypeReference>>() { }); return findAttrValue("fw_version", attributes).equals(fwVersion) && findAttrValue("fw_title", attributes).equals(title) && findAttrValue("fw_url", attributes).equals(url); }); } private static Object findAttrValue(String key, List> attributes) { Optional> attr = attributes.stream() .filter(att -> att.get("key").equals(key)).findFirst(); return attr.isPresent() ? attr.get().get("value") : ""; } @Test public void testSaveDeviceWithFirmwareFromDifferentTenant() throws Exception { loginDifferentTenant(); DeviceProfile differentProfile = createDeviceProfile("Different profile"); differentProfile = doPost("/api/deviceProfile", differentProfile, DeviceProfile.class); SaveOtaPackageInfoRequest firmwareInfo = new SaveOtaPackageInfoRequest(); firmwareInfo.setDeviceProfileId(differentProfile.getId()); firmwareInfo.setType(FIRMWARE); firmwareInfo.setTitle("title"); firmwareInfo.setVersion("1.0"); firmwareInfo.setUrl("test.url"); firmwareInfo.setUsesUrl(true); OtaPackageInfo savedFw = doPost("/api/otaPackage", firmwareInfo, OtaPackageInfo.class); loginTenantAdmin(); Device device = new Device(); device.setName("My device"); device.setType("default"); device.setFirmwareId(savedFw.getId()); doPost("/api/device", device).andExpect(status().isBadRequest()) .andExpect(statusReason(containsString("Can't assign firmware from different tenant!"))); } @Test public void testSaveDeviceWithSoftwareFromDifferentTenant() throws Exception { loginDifferentTenant(); DeviceProfile differentProfile = createDeviceProfile("Different profile"); differentProfile = doPost("/api/deviceProfile", differentProfile, DeviceProfile.class); SaveOtaPackageInfoRequest softwareInfo = new SaveOtaPackageInfoRequest(); softwareInfo.setDeviceProfileId(differentProfile.getId()); softwareInfo.setType(SOFTWARE); softwareInfo.setTitle("title"); softwareInfo.setVersion("1.0"); softwareInfo.setUrl("test.url"); softwareInfo.setUsesUrl(true); OtaPackageInfo savedSw = doPost("/api/otaPackage", softwareInfo, OtaPackageInfo.class); loginTenantAdmin(); Device device = new Device(); device.setName("My device"); device.setType("default"); device.setSoftwareId(savedSw.getId()); doPost("/api/device", device).andExpect(status().isBadRequest()) .andExpect(statusReason(containsString("Can't assign software from different tenant!"))); } @Test public void testFindDeviceById() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); Device foundDevice = doGet("/api/device/" + savedDevice.getId().getId(), Device.class); Assert.assertNotNull(foundDevice); Assert.assertEquals(savedDevice, foundDevice); } @Test public void testFindDeviceTypesByTenantId() throws Exception { DeviceProfile deviceProfile = createDeviceProfile("typeB"); deviceProfile = doPost("/api/deviceProfile", deviceProfile, DeviceProfile.class); List devices = new ArrayList<>(); int cntEntity = 3; Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); for (int i = 0; i < cntEntity; i++) { Device device = new Device(); device.setName("My device B" + i); device.setType("typeB"); device.setDeviceProfileId(deviceProfile.getId()); devices.add(doPost("/api/device", device, Device.class)); } testNotifyManyEntityManyTimeMsgToEdgeServiceEntityEqAny(new Device(), new Device(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ADDED, cntEntity, cntEntity, cntEntity); testNotificationUpdateGatewayNever(); for (int i = 0; i < 7; i++) { Device device = new Device(); device.setName("My device C" + i); device.setType("typeC"); devices.add(doPost("/api/device", device, Device.class)); } for (int i = 0; i < 9; i++) { Device device = new Device(); device.setName("My device A" + i); device.setType("typeA"); devices.add(doPost("/api/device", device, Device.class)); } List deviceTypes = doGetTyped("/api/device/types", new TypeReference<>() { }); Assert.assertNotNull(deviceTypes); Assert.assertEquals(3, deviceTypes.size()); Assert.assertEquals("typeA", deviceTypes.get(0).getType()); Assert.assertEquals("typeB", deviceTypes.get(1).getType()); Assert.assertEquals("typeC", deviceTypes.get(2).getType()); deleteEntitiesAsync("/api/device/", devices, executor).get(TIMEOUT, TimeUnit.SECONDS); } @Test public void testDeleteDevice() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); doDelete("/api/device/" + savedDevice.getId().getId()) .andExpect(status().isOk()); testNotifyEntityAllOneTime(savedDevice, savedDevice.getId(), savedDevice.getId(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.DELETED, savedDevice.getId().getId().toString()); testNotificationDeleteGatewayOneTime(savedDevice); EntityId savedDeviceId = savedDevice.getId(); doGet("/api/device/" + savedDeviceId) .andExpect(status().isNotFound()) .andExpect(statusReason(containsString(msgErrorNoFound("Device", savedDeviceId.getId().toString())))); } @Test public void testDeleteDeviceWithAlarmsAndAlarmTypes() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); Alarm alarm = Alarm.builder() .tenantId(tenantId) .originator(savedDevice.getId()) .severity(AlarmSeverity.CRITICAL) .type("test_type") .build(); alarm = doPost("/api/alarm", alarm, Alarm.class); Assert.assertNotNull(alarm); AlarmInfo foundAlarm = doGet("/api/alarm/info/" + alarm.getId(), AlarmInfo.class); Assert.assertNotNull(foundAlarm); doDelete("/api/device/" + savedDevice.getId().getId().toString()) .andExpect(status().isOk()); String DeviceIdStr = savedDevice.getId().getId().toString(); doGet("/api/device/" + DeviceIdStr) .andExpect(status().isNotFound()) .andExpect(statusReason(containsString(msgErrorNoFound("Device", DeviceIdStr)))); doGet("/api/device/info/" + alarm.getId()) .andExpect(status().isNotFound()) .andExpect(statusReason(containsString(msgErrorNoFound("Device", alarm.getId().getId().toString())))); } @Test public void testSaveDeviceWithEmptyType() throws Exception { Device device = new Device(); device.setName("My device"); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); Device savedDevice = doPost("/api/device", device, Device.class); Assert.assertEquals("default", savedDevice.getType()); testNotifyEntityAllOneTime(savedDevice, savedDevice.getId(), savedDevice.getId(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ADDED); testNotificationUpdateGatewayNever(); } @Test public void testSaveDeviceWithEmptyName() throws Exception { Device device = new Device(); device.setType("default"); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); String msgError = "Device name " + msgErrorShouldBeSpecified; doPost("/api/device", device) .andExpect(status().isBadRequest()) .andExpect(statusReason(containsString(msgError))); testNotifyEntityEqualsOneTimeServiceNeverError(device, savedTenant.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ADDED, new DataValidationException(msgError)); testNotificationUpdateGatewayNever(); } @Test public void testAssignUnassignDeviceToCustomer() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); Customer customer = new Customer(); customer.setTitle("My customer"); Customer savedCustomer = doPost("/api/customer", customer, Customer.class); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); Device assignedDevice = doPost("/api/customer/" + savedCustomer.getId().getId() + "/device/" + savedDevice.getId().getId(), Device.class); Assert.assertEquals(savedCustomer.getId(), assignedDevice.getCustomerId()); testNotifyAssignUnassignEntityAllOneTime(assignedDevice, assignedDevice.getId(), assignedDevice.getId(), savedTenant.getId(), savedCustomer.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ASSIGNED_TO_CUSTOMER, ActionType.UPDATED, assignedDevice.getId().getId().toString(), savedCustomer.getId().getId().toString(), savedCustomer.getTitle()); testNotificationUpdateGatewayNever(); Device foundDevice = doGet("/api/device/" + savedDevice.getId().getId(), Device.class); Assert.assertEquals(savedCustomer.getId(), foundDevice.getCustomerId()); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); Device unassignedDevice = doDelete("/api/customer/device/" + savedDevice.getId().getId(), Device.class); Assert.assertEquals(ModelConstants.NULL_UUID, unassignedDevice.getCustomerId().getId()); testNotifyAssignUnassignEntityAllOneTime(unassignedDevice, unassignedDevice.getId(), unassignedDevice.getId(), savedTenant.getId(), savedCustomer.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.UNASSIGNED_FROM_CUSTOMER, ActionType.UPDATED, unassignedDevice.getId().getId().toString(), savedCustomer.getId().getId().toString(), savedCustomer.getTitle()); testNotificationDeleteGatewayNever(); foundDevice = doGet("/api/device/" + savedDevice.getId().getId(), Device.class); Assert.assertEquals(ModelConstants.NULL_UUID, foundDevice.getCustomerId().getId()); } @Test public void testAssignUnassignDeviceToPublicCustomer() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); Device assignedDevice = doPost("/api/customer/public/device/" + savedDevice.getId().getId(), Device.class); Customer publicCustomer = doGet("/api/customer/" + assignedDevice.getCustomerId(), Customer.class); Assert.assertTrue(publicCustomer.isPublic()); testNotifyAssignUnassignEntityAllOneTime(assignedDevice, assignedDevice.getId(), assignedDevice.getId(), savedTenant.getId(), publicCustomer.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ASSIGNED_TO_CUSTOMER, ActionType.UPDATED, assignedDevice.getId().getId().toString(), publicCustomer.getId().getId().toString(), publicCustomer.getTitle()); testNotificationUpdateGatewayNever(); Device foundDevice = doGet("/api/device/" + savedDevice.getId().getId(), Device.class); Assert.assertEquals(publicCustomer.getId(), foundDevice.getCustomerId()); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); Device unassignedDevice = doDelete("/api/customer/device/" + savedDevice.getId().getId(), Device.class); Assert.assertEquals(ModelConstants.NULL_UUID, unassignedDevice.getCustomerId().getId()); testNotifyAssignUnassignEntityAllOneTime(unassignedDevice, unassignedDevice.getId(), unassignedDevice.getId(), savedTenant.getId(), publicCustomer.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.UNASSIGNED_FROM_CUSTOMER, ActionType.UPDATED, unassignedDevice.getId().getId().toString(), publicCustomer.getId().getId().toString(), publicCustomer.getTitle()); testNotificationDeleteGatewayNever(); foundDevice = doGet("/api/device/" + savedDevice.getId().getId(), Device.class); Assert.assertEquals(ModelConstants.NULL_UUID, foundDevice.getCustomerId().getId()); } @Test public void testAssignDeviceToNonExistentCustomer() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); String customerIdStr = savedDevice.getId().toString(); doPost("/api/customer/" + customerIdStr + "/device/" + savedDevice.getId().getId()) .andExpect(status().isNotFound()) .andExpect(statusReason(containsString(msgErrorNoFound("Customer", customerIdStr)))); testNotifyEntityNever(savedDevice.getId(), savedDevice); testNotificationUpdateGatewayNever(); } @Test public void testAssignDeviceToCustomerFromDifferentTenant() throws Exception { loginSysAdmin(); Tenant tenant2 = new Tenant(); tenant2.setTitle("Different tenant"); Tenant savedTenant2 = saveTenant(tenant2); Assert.assertNotNull(savedTenant2); User tenantAdmin2 = new User(); tenantAdmin2.setAuthority(Authority.TENANT_ADMIN); tenantAdmin2.setTenantId(savedTenant2.getId()); tenantAdmin2.setEmail("tenant3@thingsboard.org"); tenantAdmin2.setFirstName("Joe"); tenantAdmin2.setLastName("Downs"); createUserAndLogin(tenantAdmin2, "testPassword1"); Customer customer = new Customer(); customer.setTitle("Different customer"); Customer savedCustomer = doPost("/api/customer", customer, Customer.class); login(tenantAdmin.getEmail(), "testPassword1"); Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); doPost("/api/customer/" + savedCustomer.getId().getId() + "/device/" + savedDevice.getId().getId()) .andExpect(status().isForbidden()) .andExpect(statusReason(containsString(msgErrorPermission))); testNotifyEntityNever(savedDevice.getId(), savedDevice); testNotificationUpdateGatewayNever(); loginSysAdmin(); deleteTenant(savedTenant2.getId()); } @Test public void testFindDeviceCredentialsByDeviceId() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); DeviceCredentials deviceCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); Assert.assertEquals(savedDevice.getId(), deviceCredentials.getDeviceId()); } @Test public void testSaveDeviceCredentials() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); DeviceCredentials deviceCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); Assert.assertEquals(savedDevice.getId(), deviceCredentials.getDeviceId()); deviceCredentials.setCredentialsType(DeviceCredentialsType.ACCESS_TOKEN); deviceCredentials.setCredentialsId("access_token"); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); deviceCredentials = doPost("/api/device/credentials", deviceCredentials, DeviceCredentials.class); testNotifyEntityMsgToEdgePushMsgToCoreOneTime(savedDevice, savedDevice.getId(), savedDevice.getId(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.CREDENTIALS_UPDATED, deviceCredentials); testNotificationUpdateGatewayNever(); DeviceCredentials foundDeviceCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); Assert.assertEquals(deviceCredentials, foundDeviceCredentials); } @Test public void testSaveDeviceCredentialsWithEmptyDevice() throws Exception { DeviceCredentials deviceCredentials = new DeviceCredentials(); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); doPost("/api/device/credentials", deviceCredentials) .andExpect(status().isBadRequest()) .andExpect(statusReason(containsString("Invalid entity id"))); testNotifyEntityNever(deviceCredentials.getDeviceId(), new Device()); testNotificationUpdateGatewayNever(); } @Test public void testSaveDeviceCredentialsWithEmptyCredentialsType() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); DeviceCredentials deviceCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); deviceCredentials.setCredentialsType(null); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); String msgError = "Device credentials type " + msgErrorShouldBeSpecified; doPost("/api/device/credentials", deviceCredentials) .andExpect(status().isBadRequest()) .andExpect(statusReason(containsString(msgError))); testNotifyEntityIsNullOneTimeEdgeServiceNeverError(device, savedTenant.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.CREDENTIALS_UPDATED, new DataValidationException(msgError), deviceCredentials); testNotificationUpdateGatewayNever(); } @Test public void testSaveDeviceCredentialsWithEmptyCredentialsId() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); DeviceCredentials deviceCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); deviceCredentials.setCredentialsId(null); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); String msgError = "Device credentials id " + msgErrorShouldBeSpecified; doPost("/api/device/credentials", deviceCredentials) .andExpect(status().isBadRequest()) .andExpect(statusReason(containsString(msgError))); testNotifyEntityIsNullOneTimeEdgeServiceNeverError(device, savedTenant.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.CREDENTIALS_UPDATED, new DeviceCredentialsValidationException(msgError), deviceCredentials); testNotificationUpdateGatewayNever(); } @Test public void testSaveNonExistentDeviceCredentials() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); DeviceCredentials deviceCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); DeviceCredentials newDeviceCredentials = new DeviceCredentials(new DeviceCredentialsId(Uuids.timeBased())); newDeviceCredentials.setCreatedTime(deviceCredentials.getCreatedTime()); newDeviceCredentials.setDeviceId(deviceCredentials.getDeviceId()); newDeviceCredentials.setCredentialsType(deviceCredentials.getCredentialsType()); newDeviceCredentials.setCredentialsId(deviceCredentials.getCredentialsId()); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); String msgError = "Unable to update non-existent device credentials"; doPost("/api/device/credentials", newDeviceCredentials) .andExpect(status().isBadRequest()) .andExpect(statusReason(containsString(msgError))); testNotifyEntityIsNullOneTimeEdgeServiceNeverError(device, savedTenant.getId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.CREDENTIALS_UPDATED, new DeviceCredentialsValidationException(msgError), newDeviceCredentials); testNotificationUpdateGatewayNever(); } @Test public void testSaveDeviceCredentialsWithNonExistentDevice() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); DeviceId deviceTimeBasedId = new DeviceId(Uuids.timeBased()); DeviceCredentials deviceCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); deviceCredentials.setDeviceId(deviceTimeBasedId); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); doPost("/api/device/credentials", deviceCredentials) .andExpect(status().isNotFound()) .andExpect(statusReason(containsString(msgErrorNoFound("Device", deviceTimeBasedId.toString())))); testNotifyEntityNever(savedDevice.getId(), savedDevice); testNotificationUpdateGatewayNever(); } @Test public void testFindTenantDevices() throws Exception { int cntEntity = 178; Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); futures = new ArrayList<>(cntEntity); for (int i = 0; i < cntEntity; i++) { Device device = new Device(); device.setName("Device" + i); device.setType("default"); futures.add(executor.submit(() -> doPost("/api/device", device, Device.class))); } List devices = Futures.allAsList(futures).get(TIMEOUT, TimeUnit.SECONDS); testNotifyManyEntityManyTimeMsgToEdgeServiceEntityEqAny(new Device(), new Device(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ADDED, cntEntity, cntEntity, cntEntity); testNotificationUpdateGatewayNever(); List loadedDevices = new ArrayList<>(cntEntity); PageLink pageLink = new PageLink(23); do { pageData = doGetTypedWithPageLink("/api/tenant/devices?", PAGE_DATA_DEVICE_TYPE_REF, pageLink); loadedDevices.addAll(pageData.getData()); if (pageData.hasNext()) { pageLink = pageLink.nextPageLink(); } } while (pageData.hasNext()); assertThat(devices).containsExactlyInAnyOrderElementsOf(loadedDevices); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); deleteEntitiesAsync("/api/device/", loadedDevices, executor).get(TIMEOUT, TimeUnit.SECONDS); testNotifyManyEntityManyTimeMsgToEdgeServiceEntityEqAnyAdditionalInfoAny(new Device(), new Device(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.DELETED, ActionType.DELETED, cntEntity, cntEntity, 1); testNotificationUpdateGatewayNever(); } @Test public void testFindTenantDevicesByName() throws Exception { String title1 = "Device title 1"; futures = new ArrayList<>(143); for (int i = 0; i < 143; i++) { Device device = new Device(); String suffix = StringUtils.randomAlphanumeric(15); String name = title1 + suffix; name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); device.setName(name); device.setType("default"); futures.add(executor.submit(() -> doPost("/api/device", device, Device.class))); } List devicesTitle1 = Futures.allAsList(futures).get(TIMEOUT, TimeUnit.SECONDS); String title2 = "Device title 2"; futures = new ArrayList<>(75); for (int i = 0; i < 75; i++) { Device device = new Device(); String suffix = StringUtils.randomAlphanumeric(15); String name = title2 + suffix; name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); device.setName(name); device.setType("default"); futures.add(executor.submit(() -> doPost("/api/device", device, Device.class))); } List devicesTitle2 = Futures.allAsList(futures).get(TIMEOUT, TimeUnit.SECONDS); List loadedDevicesTitle1 = new ArrayList<>(143); PageLink pageLink = new PageLink(15, 0, title1); do { pageData = doGetTypedWithPageLink("/api/tenant/devices?", PAGE_DATA_DEVICE_TYPE_REF, pageLink); loadedDevicesTitle1.addAll(pageData.getData()); if (pageData.hasNext()) { pageLink = pageLink.nextPageLink(); } } while (pageData.hasNext()); assertThat(devicesTitle1).as(title1).containsExactlyInAnyOrderElementsOf(loadedDevicesTitle1); List loadedDevicesTitle2 = new ArrayList<>(75); pageLink = new PageLink(4, 0, title2); do { pageData = doGetTypedWithPageLink("/api/tenant/devices?", PAGE_DATA_DEVICE_TYPE_REF, pageLink); loadedDevicesTitle2.addAll(pageData.getData()); if (pageData.hasNext()) { pageLink = pageLink.nextPageLink(); } } while (pageData.hasNext()); assertThat(devicesTitle2).as(title2).containsExactlyInAnyOrderElementsOf(loadedDevicesTitle2); deleteEntitiesAsync("/api/device/", loadedDevicesTitle1, executor).get(TIMEOUT, TimeUnit.SECONDS); pageLink = new PageLink(4, 0, title1); pageData = doGetTypedWithPageLink("/api/tenant/devices?", PAGE_DATA_DEVICE_TYPE_REF, pageLink); Assert.assertFalse(pageData.hasNext()); Assert.assertEquals(0, pageData.getData().size()); deleteEntitiesAsync("/api/device/", loadedDevicesTitle2, executor).get(TIMEOUT, TimeUnit.SECONDS); pageLink = new PageLink(4, 0, title2); pageData = doGetTypedWithPageLink("/api/tenant/devices?", PAGE_DATA_DEVICE_TYPE_REF, pageLink); Assert.assertFalse(pageData.hasNext()); Assert.assertEquals(0, pageData.getData().size()); } @Test public void testFindTenantDevicesByType() throws Exception { String title1 = "Device title 1"; String type1 = "typeA"; futures = new ArrayList<>(143); for (int i = 0; i < 143; i++) { Device device = new Device(); String suffix = StringUtils.randomAlphanumeric(15); String name = title1 + suffix; name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); device.setName(name); device.setType(type1); futures.add(executor.submit(() -> doPost("/api/device", device, Device.class))); if (i == 0) { futures.get(0).get(TIMEOUT, TimeUnit.SECONDS); // wait for the device profile created first time } } List devicesType1 = Futures.allAsList(futures).get(TIMEOUT, TimeUnit.SECONDS); String title2 = "Device title 2"; String type2 = "typeB"; futures = new ArrayList<>(75); for (int i = 0; i < 75; i++) { Device device = new Device(); String suffix = StringUtils.randomAlphanumeric(15); String name = title2 + suffix; name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); device.setName(name); device.setType(type2); futures.add(executor.submit(() -> doPost("/api/device", device, Device.class))); if (i == 0) { futures.get(0).get(TIMEOUT, TimeUnit.SECONDS); // wait for the device profile created first time } } List devicesType2 = Futures.allAsList(futures).get(TIMEOUT, TimeUnit.SECONDS); List loadedDevicesType1 = new ArrayList<>(143); PageLink pageLink = new PageLink(15); do { pageData = doGetTypedWithPageLink("/api/tenant/devices?type={type}&", PAGE_DATA_DEVICE_TYPE_REF, pageLink, type1); loadedDevicesType1.addAll(pageData.getData()); if (pageData.hasNext()) { pageLink = pageLink.nextPageLink(); } } while (pageData.hasNext()); assertThat(devicesType1).as(title1).containsExactlyInAnyOrderElementsOf(loadedDevicesType1); List loadedDevicesType2 = new ArrayList<>(75); pageLink = new PageLink(4); do { pageData = doGetTypedWithPageLink("/api/tenant/devices?type={type}&", PAGE_DATA_DEVICE_TYPE_REF, pageLink, type2); loadedDevicesType2.addAll(pageData.getData()); if (pageData.hasNext()) { pageLink = pageLink.nextPageLink(); } } while (pageData.hasNext()); assertThat(devicesType2).as(title2).containsExactlyInAnyOrderElementsOf(loadedDevicesType2); deleteEntitiesAsync("/api/device/", loadedDevicesType1, executor).get(TIMEOUT, TimeUnit.SECONDS); pageLink = new PageLink(4); pageData = doGetTypedWithPageLink("/api/tenant/devices?type={type}&", PAGE_DATA_DEVICE_TYPE_REF, pageLink, type1); Assert.assertFalse(pageData.hasNext()); Assert.assertEquals(0, pageData.getData().size()); deleteEntitiesAsync("/api/device/", loadedDevicesType2, executor).get(TIMEOUT, TimeUnit.SECONDS); pageLink = new PageLink(4); pageData = doGetTypedWithPageLink("/api/tenant/devices?type={type}&", PAGE_DATA_DEVICE_TYPE_REF, pageLink, type2); Assert.assertFalse(pageData.hasNext()); Assert.assertEquals(0, pageData.getData().size()); } @Test public void testFindCustomerDevices() throws Exception { Customer customer = new Customer(); customer.setTitle("Test customer"); customer = doPost("/api/customer", customer, Customer.class); CustomerId customerId = customer.getId(); int cntEntity = 128; Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); futures = new ArrayList<>(cntEntity); for (int i = 0; i < cntEntity; i++) { Device device = new Device(); device.setName("Device" + i); device.setType("default"); ListenableFuture future = executor.submit(() -> doPost("/api/device", device, Device.class)); futures.add(Futures.transform(future, (dev) -> doPost("/api/customer/" + customerId.getId() + "/device/" + dev.getId().getId(), Device.class), MoreExecutors.directExecutor())); } List devices = Futures.allAsList(futures).get(TIMEOUT, TimeUnit.SECONDS); testNotifyManyEntityManyTimeMsgToEdgeServiceEntityEqAny(new Device(), new Device(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ADDED, cntEntity, cntEntity, cntEntity * 2); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); testNotificationUpdateGatewayNever(); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); List loadedDevices = new ArrayList<>(cntEntity); PageLink pageLink = new PageLink(23); do { pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId() + "/devices?", PAGE_DATA_DEVICE_TYPE_REF, pageLink); loadedDevices.addAll(pageData.getData()); if (pageData.hasNext()) { pageLink = pageLink.nextPageLink(); } } while (pageData.hasNext()); assertThat(devices).containsExactlyInAnyOrderElementsOf(loadedDevices); deleteEntitiesAsync("/api/customer/device/", loadedDevices, executor).get(TIMEOUT, TimeUnit.SECONDS); testNotifyManyEntityManyTimeMsgToEdgeServiceEntityEqAnyAdditionalInfoAny(new Device(), new Device(), savedTenant.getId(), customerId, tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.UNASSIGNED_FROM_CUSTOMER, ActionType.UPDATED, cntEntity, cntEntity, 3); testNotificationUpdateGatewayNever(); } @Test public void testFindCustomerDevicesByName() throws Exception { Customer customer = new Customer(); customer.setTitle("Test customer"); customer = doPost("/api/customer", customer, Customer.class); CustomerId customerId = customer.getId(); String title1 = "Device title 1"; futures = new ArrayList<>(125); for (int i = 0; i < 125; i++) { Device device = new Device(); String suffix = StringUtils.randomAlphanumeric(15); String name = title1 + suffix; name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); device.setName(name); device.setType("default"); ListenableFuture future = executor.submit(() -> doPost("/api/device", device, Device.class)); futures.add(Futures.transform(future, (dev) -> doPost("/api/customer/" + customerId.getId() + "/device/" + dev.getId().getId(), Device.class), MoreExecutors.directExecutor())); } List devicesTitle1 = Futures.allAsList(futures).get(TIMEOUT, TimeUnit.SECONDS); String title2 = "Device title 2"; futures = new ArrayList<>(143); for (int i = 0; i < 143; i++) { Device device = new Device(); String suffix = StringUtils.randomAlphanumeric(15); String name = title2 + suffix; name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); device.setName(name); device.setType("default"); ListenableFuture future = executor.submit(() -> doPost("/api/device", device, Device.class)); futures.add(Futures.transform(future, (dev) -> doPost("/api/customer/" + customerId.getId() + "/device/" + dev.getId().getId(), Device.class), MoreExecutors.directExecutor())); } List devicesTitle2 = Futures.allAsList(futures).get(TIMEOUT, TimeUnit.SECONDS); List loadedDevicesTitle1 = new ArrayList<>(125); PageLink pageLink = new PageLink(15, 0, title1); do { pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId() + "/devices?", PAGE_DATA_DEVICE_TYPE_REF, pageLink); loadedDevicesTitle1.addAll(pageData.getData()); if (pageData.hasNext()) { pageLink = pageLink.nextPageLink(); } } while (pageData.hasNext()); assertThat(devicesTitle1).as(title1).containsExactlyInAnyOrderElementsOf(loadedDevicesTitle1); List loadedDevicesTitle2 = new ArrayList<>(143); pageLink = new PageLink(4, 0, title2); do { pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId() + "/devices?", PAGE_DATA_DEVICE_TYPE_REF, pageLink); loadedDevicesTitle2.addAll(pageData.getData()); if (pageData.hasNext()) { pageLink = pageLink.nextPageLink(); } } while (pageData.hasNext()); assertThat(devicesTitle2).as(title2).containsExactlyInAnyOrderElementsOf(loadedDevicesTitle2); deleteEntitiesAsync("/api/customer/device/", loadedDevicesTitle1, executor).get(TIMEOUT, TimeUnit.SECONDS); pageLink = new PageLink(4, 0, title1); pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId() + "/devices?", PAGE_DATA_DEVICE_TYPE_REF, pageLink); Assert.assertFalse(pageData.hasNext()); Assert.assertEquals(0, pageData.getData().size()); deleteEntitiesAsync("/api/customer/device/", loadedDevicesTitle2, executor).get(TIMEOUT, TimeUnit.SECONDS); pageLink = new PageLink(4, 0, title2); pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId() + "/devices?", PAGE_DATA_DEVICE_TYPE_REF, pageLink); Assert.assertFalse(pageData.hasNext()); Assert.assertEquals(0, pageData.getData().size()); } @Test public void testFindCustomerDevicesByType() throws Exception { Customer customer = new Customer(); customer.setTitle("Test customer"); customer = doPost("/api/customer", customer, Customer.class); CustomerId customerId = customer.getId(); String title1 = "Device title 1"; String type1 = "typeC"; futures = new ArrayList<>(125); for (int i = 0; i < 125; i++) { Device device = new Device(); String suffix = StringUtils.randomAlphanumeric(15); String name = title1 + suffix; name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); device.setName(name); device.setType(type1); ListenableFuture future = executor.submit(() -> doPost("/api/device", device, Device.class)); futures.add(Futures.transform(future, (dev) -> doPost("/api/customer/" + customerId.getId() + "/device/" + dev.getId().getId(), Device.class), MoreExecutors.directExecutor())); if (i == 0) { futures.get(0).get(TIMEOUT, TimeUnit.SECONDS); // wait for the device profile created first time } } List devicesType1 = Futures.allAsList(futures).get(TIMEOUT, TimeUnit.SECONDS); String title2 = "Device title 2"; String type2 = "typeD"; futures = new ArrayList<>(143); for (int i = 0; i < 143; i++) { Device device = new Device(); String suffix = StringUtils.randomAlphanumeric(15); String name = title2 + suffix; name = i % 2 == 0 ? name.toLowerCase() : name.toUpperCase(); device.setName(name); device.setType(type2); ListenableFuture future = executor.submit(() -> doPost("/api/device", device, Device.class)); futures.add(Futures.transform(future, (dev) -> doPost("/api/customer/" + customerId.getId() + "/device/" + dev.getId().getId(), Device.class), MoreExecutors.directExecutor())); if (i == 0) { futures.get(0).get(TIMEOUT, TimeUnit.SECONDS); // wait for the device profile created first time } } List devicesType2 = Futures.allAsList(futures).get(TIMEOUT, TimeUnit.SECONDS); List loadedDevicesType1 = new ArrayList<>(125); PageLink pageLink = new PageLink(15); do { pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId() + "/devices?type={type}&", PAGE_DATA_DEVICE_TYPE_REF, pageLink, type1); loadedDevicesType1.addAll(pageData.getData()); if (pageData.hasNext()) { pageLink = pageLink.nextPageLink(); } } while (pageData.hasNext()); assertThat(devicesType1).as(title1).containsExactlyInAnyOrderElementsOf(loadedDevicesType1); List loadedDevicesType2 = new ArrayList<>(143); pageLink = new PageLink(4); do { pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId() + "/devices?type={type}&", PAGE_DATA_DEVICE_TYPE_REF, pageLink, type2); loadedDevicesType2.addAll(pageData.getData()); if (pageData.hasNext()) { pageLink = pageLink.nextPageLink(); } } while (pageData.hasNext()); assertThat(devicesType2).as(title2).containsExactlyInAnyOrderElementsOf(loadedDevicesType2); deleteEntitiesAsync("/api/customer/device/", loadedDevicesType1, executor).get(TIMEOUT, TimeUnit.SECONDS); pageLink = new PageLink(4); pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId() + "/devices?type={type}&", PAGE_DATA_DEVICE_TYPE_REF, pageLink, type1); Assert.assertFalse(pageData.hasNext()); Assert.assertEquals(0, pageData.getData().size()); deleteEntitiesAsync("/api/customer/device/", loadedDevicesType2, executor).get(TIMEOUT, TimeUnit.SECONDS); pageLink = new PageLink(4); pageData = doGetTypedWithPageLink("/api/customer/" + customerId.getId() + "/devices?type={type}&", PAGE_DATA_DEVICE_TYPE_REF, pageLink, type2); Assert.assertFalse(pageData.hasNext()); Assert.assertEquals(0, pageData.getData().size()); } @Test public void testAssignDeviceToTenant() throws Exception { Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); Device anotherDevice = new Device(); anotherDevice.setName("My device1"); anotherDevice.setType("default"); Device savedAnotherDevice = doPost("/api/device", anotherDevice, Device.class); EntityRelation relation = new EntityRelation(); relation.setFrom(savedDevice.getId()); relation.setTo(savedAnotherDevice.getId()); relation.setTypeGroup(RelationTypeGroup.COMMON); relation.setType("Contains"); doPost("/api/relation", relation).andExpect(status().isOk()); loginSysAdmin(); Tenant tenant = new Tenant(); tenant.setTitle("Different tenant"); Tenant savedDifferentTenant = saveTenant(tenant); Assert.assertNotNull(savedDifferentTenant); User user = new User(); user.setAuthority(Authority.TENANT_ADMIN); user.setTenantId(savedDifferentTenant.getId()); user.setEmail("tenant9@thingsboard.org"); user.setFirstName("Sam"); user.setLastName("Downs"); createUserAndLogin(user, "testPassword1"); login("tenant2@thingsboard.org", "testPassword1"); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); Device assignedDevice = doPost("/api/tenant/" + savedDifferentTenant.getId().getId() + "/device/" + savedDevice.getId().getId(), Device.class); doGet("/api/device/" + assignedDevice.getId().getId()) .andExpect(status().isNotFound()) .andExpect(statusReason(containsString(msgErrorNoFound("Device", assignedDevice.getId().getId().toString())))); testNotifyEntityOneTimeMsgToEdgeServiceNever(assignedDevice, assignedDevice.getId(), assignedDevice.getId(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ASSIGNED_TO_TENANT, savedDifferentTenant.getId().getId().toString(), savedDifferentTenant.getTitle()); testNotificationUpdateGatewayNever(); ArgumentCaptor protoCaptor = ArgumentCaptor.forClass(TransportProtos.DeviceStateServiceMsgProto.class); Awaitility.await().atMost(5, TimeUnit.SECONDS).until(() -> { Mockito.verify(deviceStateService, Mockito.atLeastOnce()).onQueueMsg(protoCaptor.capture(), any()); return protoCaptor.getAllValues().stream().anyMatch(proto -> proto.getTenantIdMSB() == savedTenant.getUuidId().getMostSignificantBits() && proto.getTenantIdLSB() == savedTenant.getUuidId().getLeastSignificantBits() && proto.getDeviceIdMSB() == savedDevice.getUuidId().getMostSignificantBits() && proto.getDeviceIdLSB() == savedDevice.getUuidId().getLeastSignificantBits() && proto.getDeleted()); }); Awaitility.await().atMost(5, TimeUnit.SECONDS).until(() -> { Mockito.verify(deviceStateService, Mockito.atLeastOnce()).onQueueMsg(protoCaptor.capture(), any()); return protoCaptor.getAllValues().stream().anyMatch(proto -> proto.getTenantIdMSB() == savedDifferentTenant.getUuidId().getMostSignificantBits() && proto.getTenantIdLSB() == savedDifferentTenant.getUuidId().getLeastSignificantBits() && proto.getDeviceIdMSB() == savedDevice.getUuidId().getMostSignificantBits() && proto.getDeviceIdLSB() == savedDevice.getUuidId().getLeastSignificantBits() && proto.getAdded()); }); login("tenant9@thingsboard.org", "testPassword1"); Device foundDevice1 = doGet("/api/device/" + assignedDevice.getId().getId(), Device.class); Assert.assertNotNull(foundDevice1); doGet("/api/relation?fromId=" + savedDevice.getId().getId() + "&fromType=DEVICE&relationType=Contains&toId=" + savedAnotherDevice.getId().getId() + "&toType=DEVICE") .andExpect(status().isNotFound()) .andExpect(statusReason(containsString(msgErrorNoFound("Device", savedAnotherDevice.getId().getId().toString())))); loginSysAdmin(); deleteTenant(savedDifferentTenant.getId()); } @Test public void testAssignDeviceToEdge() throws Exception { Edge edge = constructEdge("My edge", "default"); Edge savedEdge = doPost("/api/edge", edge, Edge.class); Device device = new Device(); device.setName("My device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); doPost("/api/edge/" + savedEdge.getId().getId() + "/device/" + savedDevice.getId().getId(), Device.class); testNotifyEntityAllOneTime(savedDevice, savedDevice.getId(), savedDevice.getId(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.ASSIGNED_TO_EDGE, savedDevice.getId().getId().toString(), savedEdge.getId().getId().toString(), savedEdge.getName()); testNotificationUpdateGatewayNever(); PageData pageData = doGetTypedWithPageLink("/api/edge/" + savedEdge.getId().getId() + "/devices?", new TypeReference<>() {}, new PageLink(100)); Assert.assertEquals(1, pageData.getData().size()); Mockito.reset(tbClusterService, auditLogService, gatewayNotificationsService); doDelete("/api/edge/" + savedEdge.getId().getId() + "/device/" + savedDevice.getId().getId(), Device.class); testNotifyEntityAllOneTime(savedDevice, savedDevice.getId(), savedDevice.getId(), savedTenant.getId(), tenantAdmin.getCustomerId(), tenantAdmin.getId(), tenantAdmin.getEmail(), ActionType.UNASSIGNED_FROM_EDGE, savedDevice.getId().getId().toString(), savedEdge.getId().getId().toString(), savedEdge.getName()); testNotificationUpdateGatewayNever(); pageData = doGetTypedWithPageLink("/api/edge/" + savedEdge.getId().getId() + "/devices?", new TypeReference<>() {}, new PageLink(100)); Assert.assertEquals(0, pageData.getData().size()); } protected void testNotificationUpdateGatewayOneTime(Device device, Device oldDevice) { Mockito.verify(gatewayNotificationsService, times(1)).onDeviceUpdated(Mockito.eq(device), Mockito.eq(oldDevice)); } protected void testNotificationUpdateGatewayNever() { Mockito.verify(gatewayNotificationsService, never()).onDeviceUpdated(Mockito.any(Device.class), Mockito.any(Device.class)); } protected void testNotificationDeleteGatewayOneTime(Device device) { Mockito.verify(gatewayNotificationsService, times(1)).onDeviceDeleted(device); } protected void testNotificationDeleteGatewayNever() { Mockito.verify(gatewayNotificationsService, never()).onDeviceDeleted(Mockito.any(Device.class)); } @Test public void testDeleteDashboardWithDeleteRelationsOk() throws Exception { DeviceId deviceId = createDevice("Device for Test WithRelationsOk").getId(); testEntityDaoWithRelationsOk(savedTenant.getId(), deviceId, "/api/device/" + deviceId); } @Ignore @Test public void testDeleteDeviceExceptionWithRelationsTransactional() throws Exception { DeviceId deviceId = createDevice("Device for Test WithRelations Transactional Exception").getId(); testEntityDaoWithRelationsTransactionalException(deviceDao, savedTenant.getId(), deviceId, "/api/device/" + deviceId); } @Test public void testBulkImportDeviceWithoutCredentials() throws Exception { String deviceName = "some_device"; String deviceType = "some_type"; BulkImportRequest request = new BulkImportRequest(); request.setFile(String.format("NAME,TYPE\n%s,%s", deviceName, deviceType)); BulkImportRequest.Mapping mapping = new BulkImportRequest.Mapping(); BulkImportRequest.ColumnMapping name = new BulkImportRequest.ColumnMapping(); name.setType(BulkImportColumnType.NAME); BulkImportRequest.ColumnMapping type = new BulkImportRequest.ColumnMapping(); type.setType(BulkImportColumnType.TYPE); List columns = new ArrayList<>(); columns.add(name); columns.add(type); mapping.setColumns(columns); mapping.setDelimiter(','); mapping.setUpdate(true); mapping.setHeader(true); request.setMapping(mapping); BulkImportResult deviceBulkImportResult = doPostWithTypedResponse("/api/device/bulk_import", request, new TypeReference<>() {}); Assert.assertEquals(1, deviceBulkImportResult.getCreated().get()); Assert.assertEquals(0, deviceBulkImportResult.getErrors().get()); Assert.assertEquals(0, deviceBulkImportResult.getUpdated().get()); Assert.assertTrue(deviceBulkImportResult.getErrorsList().isEmpty()); Device savedDevice = doGet("/api/tenant/devices?deviceName=" + deviceName, Device.class); Assert.assertNotNull(savedDevice); Assert.assertEquals(deviceName, savedDevice.getName()); Assert.assertEquals(deviceType, savedDevice.getType()); DeviceCredentials savedCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); Assert.assertNotNull(savedCredentials); Assert.assertNotNull(savedCredentials.getId()); Assert.assertEquals(savedDevice.getId(), savedCredentials.getDeviceId()); Assert.assertEquals(DeviceCredentialsType.ACCESS_TOKEN, savedCredentials.getCredentialsType()); Assert.assertNotNull(savedCredentials.getCredentialsId()); Assert.assertEquals(20, savedCredentials.getCredentialsId().length()); deviceBulkImportResult = doPostWithTypedResponse("/api/device/bulk_import", request, new TypeReference<>() {}); Assert.assertEquals(0, deviceBulkImportResult.getCreated().get()); Assert.assertEquals(0, deviceBulkImportResult.getErrors().get()); Assert.assertEquals(1, deviceBulkImportResult.getUpdated().get()); Assert.assertTrue(deviceBulkImportResult.getErrorsList().isEmpty()); Device updatedDevice = doGet("/api/device/" + savedDevice.getId().getId(), Device.class); savedDevice.setVersion(updatedDevice.getVersion()); Assert.assertEquals(savedDevice, updatedDevice); DeviceCredentials updatedCredentials = doGet("/api/device/" + savedDevice.getId().getId() + "/credentials", DeviceCredentials.class); savedCredentials.setVersion(updatedCredentials.getVersion()); Assert.assertEquals(savedCredentials, updatedCredentials); } @Test public void testBulkImportDeviceWithUpdateFalse() throws Exception { String deviceName = "firstDevice"; String attributeValue = "testValue"; BulkImportRequest request = new BulkImportRequest(); request.setFile(String.format("NAME,TYPE,DATA\n%s,%s,%s", deviceName, "thermostat", attributeValue)); BulkImportRequest.Mapping mapping = new BulkImportRequest.Mapping(); BulkImportRequest.ColumnMapping name = new BulkImportRequest.ColumnMapping(); name.setType(BulkImportColumnType.NAME); BulkImportRequest.ColumnMapping type = new BulkImportRequest.ColumnMapping(); type.setType(BulkImportColumnType.TYPE); BulkImportRequest.ColumnMapping attribute = new BulkImportRequest.ColumnMapping(); attribute.setType(BulkImportColumnType.SERVER_ATTRIBUTE); attribute.setKey("DATA"); List columns = new ArrayList<>(); columns.add(name); columns.add(type); columns.add(attribute); mapping.setColumns(columns); mapping.setDelimiter(','); mapping.setUpdate(true); mapping.setHeader(true); request.setMapping(mapping); //import device doPostWithTypedResponse("/api/device/bulk_import", request, new TypeReference<>() {}); Device savedDevice = doGet("/api/tenant/devices?deviceName=" + deviceName, Device.class); //check server attribute value await().atMost(TIMEOUT, TimeUnit.SECONDS).untilAsserted(() -> { Map actualAttribute = doGetAsyncTyped("/api/plugins/telemetry/DEVICE/" + savedDevice.getId() + "/values/attributes/SERVER_SCOPE", new TypeReference>>() {}).stream() .filter(att -> att.get("key").equals("DATA")).findFirst().get(); Assert.assertEquals(attributeValue, actualAttribute.get("value")); }); //update server attribute value String newAttributeValue = "testValue2"; JsonNode content = JacksonUtil.toJsonNode("{\"DATA\": \"" + newAttributeValue + "\"}"); doPost("/api/plugins/telemetry/" + EntityType.DEVICE.name() + "/" + savedDevice.getUuidId() + "/SERVER_SCOPE", content) .andExpect(status().isOk()); await().atMost(TIMEOUT, TimeUnit.SECONDS).untilAsserted(() -> { Map actualAttribute = doGetAsyncTyped("/api/plugins/telemetry/DEVICE/" + savedDevice.getId() + "/values/attributes/SERVER_SCOPE", new TypeReference>>() {}).stream() .filter(att -> att.get("key").equals("DATA")).findFirst().get(); Assert.assertEquals(newAttributeValue, actualAttribute.get("value")); }); //reimport devices String deviceName2 = "secondDevice"; String attributeValue2 = "testValue3"; request.setFile(String.format("NAME,TYPE,DATA\n%s,%s,%s\n%s,%s,%s", deviceName, "thermostat", attributeValue, deviceName2, "thermostat", attributeValue2)); mapping.setUpdate(false); doPostWithTypedResponse("/api/device/bulk_import", request, new TypeReference<>() {}); Device savedDevice2 = doGet("/api/tenant/devices?deviceName=" + deviceName2, Device.class); //check attribute for second device await().atMost(TIMEOUT, TimeUnit.SECONDS).untilAsserted(() -> { Map actualAttribute = doGetAsyncTyped("/api/plugins/telemetry/DEVICE/" + savedDevice2.getId() + "/values/attributes/SERVER_SCOPE", new TypeReference>>() {}).stream() .filter(att -> att.get("key").equals("DATA")).findFirst().get(); Assert.assertEquals(attributeValue2, actualAttribute.get("value")); }); //check attribute value was not changed after reimport Map actualAttribute = doGetAsyncTyped("/api/plugins/telemetry/DEVICE/" + savedDevice.getId() + "/values/attributes/SERVER_SCOPE", new TypeReference>>() {}).stream() .filter(att -> att.get("key").equals("DATA")).findFirst().get(); Assert.assertEquals(newAttributeValue, actualAttribute.get("value")); } @Test public void testBulkImportDeviceWithJsonAttr() throws Exception { String deviceName = "some_device"; String deviceType = "some_type"; String deviceAttr = "{\"threshold\":45}"; List> content = new LinkedList<>(); content.add(Arrays.asList("NAME", "TYPE", "ATTR")); content.add(Arrays.asList(deviceName, deviceType, deviceAttr)); byte[] bytes = CsvUtils.generateCsv(content); BulkImportRequest request = new BulkImportRequest(); request.setFile(new String(bytes, StandardCharsets.UTF_8)); BulkImportRequest.Mapping mapping = new BulkImportRequest.Mapping(); BulkImportRequest.ColumnMapping name = new BulkImportRequest.ColumnMapping(); name.setType(BulkImportColumnType.NAME); BulkImportRequest.ColumnMapping type = new BulkImportRequest.ColumnMapping(); type.setType(BulkImportColumnType.TYPE); BulkImportRequest.ColumnMapping attr = new BulkImportRequest.ColumnMapping(); attr.setType(BulkImportColumnType.SERVER_ATTRIBUTE); attr.setKey("attr"); List columns = new ArrayList<>(); columns.add(name); columns.add(type); columns.add(attr); mapping.setColumns(columns); mapping.setDelimiter(','); mapping.setUpdate(true); mapping.setHeader(true); request.setMapping(mapping); BulkImportResult deviceBulkImportResult = doPostWithTypedResponse("/api/device/bulk_import", request, new TypeReference<>() {}); Assert.assertEquals(1, deviceBulkImportResult.getCreated().get()); Assert.assertEquals(0, deviceBulkImportResult.getErrors().get()); Assert.assertEquals(0, deviceBulkImportResult.getUpdated().get()); Assert.assertTrue(deviceBulkImportResult.getErrorsList().isEmpty()); Device savedDevice = doGet("/api/tenant/devices?deviceName=" + deviceName, Device.class); Assert.assertNotNull(savedDevice); Assert.assertEquals(deviceName, savedDevice.getName()); Assert.assertEquals(deviceType, savedDevice.getType()); Optional retrieved = await().atMost(5, TimeUnit.SECONDS) .until(() -> attributesService.find(tenantId, savedDevice.getId(), AttributeScope.SERVER_SCOPE, "attr").get(), Optional::isPresent); assertThat(retrieved.get().getJsonValue().get()).isEqualTo(deviceAttr); assertThat(retrieved.get().getStrValue()).isNotPresent(); } @Test public void testSaveDeviceWithOutdatedVersion() throws Exception { Device device = createDevice("Device v1.0"); assertThat(device.getVersion()).isOne(); device.setName("Device v2.0"); device = doPost("/api/device", device, Device.class); assertThat(device.getVersion()).isEqualTo(2); device.setName("Device v1.1"); device.setVersion(1L); String response = doPost("/api/device", device).andExpect(status().isConflict()) .andReturn().getResponse().getContentAsString(); assertThat(JacksonUtil.toJsonNode(response).get("message").asText()) .containsIgnoringCase("already changed by someone else"); device.setVersion(null); // overriding entity device = doPost("/api/device", device, Device.class); assertThat(device.getName()).isEqualTo("Device v1.1"); assertThat(device.getVersion()).isEqualTo(3); } @Test public void testSaveDeviceWithUniquifyStrategy() throws Exception { Device device = new Device(); device.setName("My unique device"); device.setType("default"); Device savedDevice = doPost("/api/device", device, Device.class); doPost("/api/device", device).andExpect(status().isBadRequest()); doPost("/api/device?nameConflictPolicy=FAIL", device).andExpect(status().isBadRequest()); Device secondDevice = doPost("/api/device?nameConflictPolicy=UNIQUIFY", device, Device.class); assertThat(secondDevice.getName()).startsWith("My unique device_"); Device thirdDevice = doPost("/api/device?nameConflictPolicy=UNIQUIFY&uniquifySeparator=-", device, Device.class); assertThat(thirdDevice.getName()).startsWith("My unique device-"); Device fourthDevice = doPost("/api/device?nameConflictPolicy=UNIQUIFY&uniquifyStrategy=INCREMENTAL", device, Device.class); assertThat(fourthDevice.getName()).isEqualTo("My unique device_1"); Device fifthDevice = doPost("/api/device?nameConflictPolicy=UNIQUIFY&uniquifyStrategy=INCREMENTAL", device, Device.class); assertThat(fifthDevice.getName()).isEqualTo("My unique device_2"); } }