Browse Source

Refactor tenant deletion in tests

pull/10201/head
ViacheslavKlimov 2 years ago
parent
commit
572b680b29
  1. 21
      application/src/test/java/org/thingsboard/server/controller/AbstractWebTest.java
  2. 6
      application/src/test/java/org/thingsboard/server/controller/AssetControllerTest.java
  3. 3
      application/src/test/java/org/thingsboard/server/controller/AssetProfileControllerTest.java
  4. 3
      application/src/test/java/org/thingsboard/server/controller/AuditLogControllerTest.java
  5. 3
      application/src/test/java/org/thingsboard/server/controller/ComponentDescriptorControllerTest.java
  6. 3
      application/src/test/java/org/thingsboard/server/controller/CustomerControllerTest.java
  7. 6
      application/src/test/java/org/thingsboard/server/controller/DashboardControllerTest.java
  8. 3
      application/src/test/java/org/thingsboard/server/controller/DeviceConnectivityControllerTest.java
  9. 10
      application/src/test/java/org/thingsboard/server/controller/DeviceControllerTest.java
  10. 3
      application/src/test/java/org/thingsboard/server/controller/DeviceProfileControllerTest.java
  11. 3
      application/src/test/java/org/thingsboard/server/controller/EdgeControllerTest.java
  12. 4
      application/src/test/java/org/thingsboard/server/controller/EntityQueryControllerTest.java
  13. 3
      application/src/test/java/org/thingsboard/server/controller/EntityRelationControllerTest.java
  14. 3
      application/src/test/java/org/thingsboard/server/controller/EntityViewControllerTest.java
  15. 2
      application/src/test/java/org/thingsboard/server/controller/HomePageApiTest.java
  16. 3
      application/src/test/java/org/thingsboard/server/controller/OtaPackageControllerTest.java
  17. 3
      application/src/test/java/org/thingsboard/server/controller/RpcControllerTest.java
  18. 3
      application/src/test/java/org/thingsboard/server/controller/RuleChainControllerTest.java
  19. 3
      application/src/test/java/org/thingsboard/server/controller/TbResourceControllerTest.java
  20. 14
      application/src/test/java/org/thingsboard/server/controller/TenantControllerTest.java
  21. 3
      application/src/test/java/org/thingsboard/server/controller/TenantProfileControllerTest.java
  22. 3
      application/src/test/java/org/thingsboard/server/controller/UserControllerTest.java
  23. 3
      application/src/test/java/org/thingsboard/server/controller/WidgetTypeControllerTest.java
  24. 3
      application/src/test/java/org/thingsboard/server/controller/WidgetsBundleControllerTest.java
  25. 2
      application/src/test/java/org/thingsboard/server/rules/flow/AbstractRuleEngineFlowIntegrationTest.java
  26. 9
      application/src/test/java/org/thingsboard/server/service/resource/sql/BaseTbResourceServiceTest.java
  27. 2
      application/src/test/java/org/thingsboard/server/service/sql/SequentialTimeseriesPersistenceTest.java

21
application/src/test/java/org/thingsboard/server/controller/AbstractWebTest.java

@ -25,7 +25,6 @@ import io.jsonwebtoken.Header;
import io.jsonwebtoken.Jwt;
import io.jsonwebtoken.Jwts;
import lombok.extern.slf4j.Slf4j;
import org.apache.http.HttpStatus;
import org.awaitility.Awaitility;
import org.hamcrest.Matcher;
import org.hibernate.exception.ConstraintViolationException;
@ -367,21 +366,11 @@ public abstract class AbstractWebTest extends AbstractInMemoryStorageTest {
assertThat(loadedTenants).as("All tenants expected to be deleted, but some tenants left in the database").isEmpty();
}
private void deleteTenant(TenantId tenantId) {
int status = 0;
int retries = 0;
while (status != HttpStatus.SC_OK && retries < CLEANUP_TENANT_RETRIES_COUNT) {
retries++;
try {
status = doDelete("/api/tenant/" + tenantId.getId().toString())
.andReturn().getResponse().getStatus();
if (status != HttpStatus.SC_OK) {
log.warn("Tenant deletion failed, tenantId: {}", tenantId.getId().toString());
Thread.sleep(1000L);
}
} catch (Exception e) {
throw new RuntimeException(e);
}
protected void deleteTenant(TenantId tenantId) {
try {
doDelete("/api/tenant/" + tenantId.getId()).andExpect(status().isOk());
} catch (Exception e) {
throw new RuntimeException(e);
}
Awaitility.await("tenant cleanup finish").atMost(30, TimeUnit.SECONDS)
.until(() -> attributesService.find(TenantId.SYS_TENANT_ID, tenantId, AttributeScope.SERVER_SCOPE, "test").get().isEmpty());

6
application/src/test/java/org/thingsboard/server/controller/AssetControllerTest.java

@ -107,8 +107,7 @@ public class AssetControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test
@ -588,8 +587,7 @@ public class AssetControllerTest extends AbstractControllerTest {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant2.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant2.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/AssetProfileControllerTest.java

@ -99,8 +99,7 @@ public class AssetProfileControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/AuditLogControllerTest.java

@ -92,8 +92,7 @@ public class AuditLogControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/ComponentDescriptorControllerTest.java

@ -64,8 +64,7 @@ public class ComponentDescriptorControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/CustomerControllerTest.java

@ -104,8 +104,7 @@ public class CustomerControllerTest extends AbstractControllerTest {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

6
application/src/test/java/org/thingsboard/server/controller/DashboardControllerTest.java

@ -94,8 +94,7 @@ public class DashboardControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test
@ -321,8 +320,7 @@ public class DashboardControllerTest extends AbstractControllerTest {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant2.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant2.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/DeviceConnectivityControllerTest.java

@ -187,8 +187,7 @@ public class DeviceConnectivityControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

10
application/src/test/java/org/thingsboard/server/controller/DeviceControllerTest.java

@ -151,8 +151,7 @@ public class DeviceControllerTest extends AbstractControllerTest {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test
@ -721,9 +720,7 @@ public class DeviceControllerTest extends AbstractControllerTest {
testNotificationUpdateGatewayNever();
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant2.getId().getId())
.andExpect(status().isOk());
deleteTenant(savedTenant2.getId());
}
@Test
@ -1380,8 +1377,7 @@ public class DeviceControllerTest extends AbstractControllerTest {
.andExpect(statusReason(containsString(msgErrorNoFound("Device", savedAnotherDevice.getId().getId().toString()))));
loginSysAdmin();
doDelete("/api/tenant/" + savedDifferentTenant.getId().getId())
.andExpect(status().isOk());
deleteTenant(savedDifferentTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/DeviceProfileControllerTest.java

@ -110,8 +110,7 @@ public class DeviceProfileControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/EdgeControllerTest.java

@ -394,8 +394,7 @@ public class EdgeControllerTest extends AbstractControllerTest {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant2.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant2.getId());
}
@Test

4
application/src/test/java/org/thingsboard/server/controller/EntityQueryControllerTest.java

@ -99,9 +99,7 @@ public class EntityQueryControllerTest extends AbstractControllerTest {
@After
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/EntityRelationControllerTest.java

@ -93,8 +93,7 @@ public class EntityRelationControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/EntityViewControllerTest.java

@ -398,8 +398,7 @@ public class EntityViewControllerTest extends AbstractControllerTest {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant2.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant2.getId());
}
@Test

2
application/src/test/java/org/thingsboard/server/controller/HomePageApiTest.java

@ -115,7 +115,7 @@ public class HomePageApiTest extends AbstractControllerTest {
Assert.assertEquals(initialCount + 100, update.getCount());
for (Tenant tenant : tenants) {
doDelete("/api/tenant/" + tenant.getId().toString());
deleteTenant(tenant.getId());
}
}

3
application/src/test/java/org/thingsboard/server/controller/OtaPackageControllerTest.java

@ -94,8 +94,7 @@ public class OtaPackageControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/RpcControllerTest.java

@ -66,8 +66,7 @@ public class RpcControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
private Device createDefaultDevice() {

3
application/src/test/java/org/thingsboard/server/controller/RuleChainControllerTest.java

@ -99,8 +99,7 @@ public class RuleChainControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/TbResourceControllerTest.java

@ -90,8 +90,7 @@ public class TbResourceControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

14
application/src/test/java/org/thingsboard/server/controller/TenantControllerTest.java

@ -155,8 +155,7 @@ public class TenantControllerTest extends AbstractControllerTest {
testBroadcastEntityStateChangeEventTimeManyTimeTenant(savedTenant, ComponentLifecycleEvent.UPDATED, 1);
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
testBroadcastEntityStateChangeEventTimeManyTimeTenant(savedTenant, ComponentLifecycleEvent.DELETED, 1);
}
@ -185,8 +184,7 @@ public class TenantControllerTest extends AbstractControllerTest {
Tenant foundTenant = doGet("/api/tenant/" + savedTenant.getId().getId().toString(), Tenant.class);
Assert.assertNotNull(foundTenant);
Assert.assertEquals(savedTenant, foundTenant);
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test
@ -198,8 +196,7 @@ public class TenantControllerTest extends AbstractControllerTest {
TenantInfo foundTenant = doGet("/api/tenant/info/" + savedTenant.getId().getId().toString(), TenantInfo.class);
Assert.assertNotNull(foundTenant);
Assert.assertEquals(new TenantInfo(savedTenant, "Default"), foundTenant);
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test
@ -240,8 +237,7 @@ public class TenantControllerTest extends AbstractControllerTest {
Tenant savedTenant = doPost("/api/tenant", tenant, Tenant.class);
String tenantIdStr = savedTenant.getId().getId().toString();
doDelete("/api/tenant/" + tenantIdStr)
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
doGet("/api/tenant/" + tenantIdStr)
.andExpect(status().isNotFound())
.andExpect(statusReason(containsString(msgErrorNoFound("Tenant", tenantIdStr))));
@ -559,7 +555,7 @@ public class TenantControllerTest extends AbstractControllerTest {
}
loginSysAdmin();
doDelete("/api/tenant/" + tenant.getId().getId().toString()).andExpect(status().isOk());
deleteTenant(tenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/TenantProfileControllerTest.java

@ -186,8 +186,7 @@ public class TenantProfileControllerTest extends AbstractControllerTest {
testBroadcastEntityStateChangeEventNeverTenantProfile();
doDelete("/api/tenant/"+savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/UserControllerTest.java

@ -422,8 +422,7 @@ public class UserControllerTest extends AbstractControllerTest {
assertThat(tenantAdmins).as("admins list size").hasSameSizeAs(loadedTenantAdmins);
assertThat(tenantAdmins).as("admins list content").isEqualTo(loadedTenantAdmins);
doDelete("/api/tenant/" + tenantId.getId().toString())
.andExpect(status().isOk());
deleteTenant(tenantId);
pageLink = new PageLink(33);
pageData = doGetTypedWithPageLink("/api/tenant/" + tenantId.getId().toString() + "/users?",

3
application/src/test/java/org/thingsboard/server/controller/WidgetTypeControllerTest.java

@ -70,8 +70,7 @@ public class WidgetTypeControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/"+savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

3
application/src/test/java/org/thingsboard/server/controller/WidgetsBundleControllerTest.java

@ -70,8 +70,7 @@ public class WidgetsBundleControllerTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

2
application/src/test/java/org/thingsboard/server/rules/flow/AbstractRuleEngineFlowIntegrationTest.java

@ -121,7 +121,7 @@ public abstract class AbstractRuleEngineFlowIntegrationTest extends AbstractRule
public void afterTest() throws Exception {
loginSysAdmin();
if (savedTenant != null) {
doDelete("/api/tenant/" + savedTenant.getId().getId().toString()).andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
}

9
application/src/test/java/org/thingsboard/server/service/resource/sql/BaseTbResourceServiceTest.java

@ -147,8 +147,7 @@ public class BaseTbResourceServiceTest extends AbstractControllerTest {
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString())
.andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test
@ -441,8 +440,7 @@ public class BaseTbResourceServiceTest extends AbstractControllerTest {
Assert.assertFalse(pageData.hasNext());
Assert.assertTrue(pageData.getData().isEmpty());
doDelete("/api/tenant/" + tenantId.getId().toString())
.andExpect(status().isOk());
deleteTenant(tenantId);
}
@Test
@ -518,8 +516,7 @@ public class BaseTbResourceServiceTest extends AbstractControllerTest {
Assert.assertFalse(pageData.hasNext());
Assert.assertTrue(pageData.getData().isEmpty());
doDelete("/api/tenant/" + tenantId.getId().toString())
.andExpect(status().isOk());
deleteTenant(tenantId);
}
}

2
application/src/test/java/org/thingsboard/server/service/sql/SequentialTimeseriesPersistenceTest.java

@ -93,7 +93,7 @@ public class SequentialTimeseriesPersistenceTest extends AbstractControllerTest
@After
public void afterTest() throws Exception {
loginSysAdmin();
doDelete("/api/tenant/" + savedTenant.getId().getId().toString()).andExpect(status().isOk());
deleteTenant(savedTenant.getId());
}
@Test

Loading…
Cancel
Save