@ -32,6 +32,7 @@ import org.thingsboard.server.common.data.security.Authority;
import java.util.ArrayList ;
import java.util.ArrayList ;
import java.util.Collections ;
import java.util.Collections ;
import java.util.List ;
import java.util.List ;
import java.util.stream.Collectors ;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status ;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status ;
@ -79,7 +80,7 @@ public abstract class BaseTbResourceControllerTest extends AbstractControllerTes
resource . setFileName ( DEFAULT_FILE_NAME ) ;
resource . setFileName ( DEFAULT_FILE_NAME ) ;
resource . setData ( "Test Data" ) ;
resource . setData ( "Test Data" ) ;
TbResource savedResource = doPost ( "/api/resource" , resource , TbResource . class ) ;
TbResource savedResource = save ( resource ) ;
Assert . assertNotNull ( savedResource ) ;
Assert . assertNotNull ( savedResource ) ;
Assert . assertNotNull ( savedResource . getId ( ) ) ;
Assert . assertNotNull ( savedResource . getId ( ) ) ;
@ -92,7 +93,7 @@ public abstract class BaseTbResourceControllerTest extends AbstractControllerTes
savedResource . setTitle ( "My new resource" ) ;
savedResource . setTitle ( "My new resource" ) ;
doPost ( "/api/resource" , savedResource , TbResource . class ) ;
save ( savedResource ) ;
TbResource foundResource = doGet ( "/api/resource/" + savedResource . getId ( ) . getId ( ) . toString ( ) , TbResource . class ) ;
TbResource foundResource = doGet ( "/api/resource/" + savedResource . getId ( ) . getId ( ) . toString ( ) , TbResource . class ) ;
Assert . assertEquals ( foundResource . getTitle ( ) , savedResource . getTitle ( ) ) ;
Assert . assertEquals ( foundResource . getTitle ( ) , savedResource . getTitle ( ) ) ;
@ -106,10 +107,10 @@ public abstract class BaseTbResourceControllerTest extends AbstractControllerTes
resource . setFileName ( DEFAULT_FILE_NAME ) ;
resource . setFileName ( DEFAULT_FILE_NAME ) ;
resource . setData ( "Test Data" ) ;
resource . setData ( "Test Data" ) ;
TbResource savedResource = doPost ( "/api/resource" , resource , TbResource . class ) ;
TbResource savedResource = save ( resource ) ;
loginDifferentTenant ( ) ;
loginDifferentTenant ( ) ;
doPost ( "/api/resource" , savedResource , TbResource . class , status ( ) . isForbidden ( ) ) ;
doPostWithTypedResponse ( "/api/resource" , Collections . singletonList ( savedResource ) , new TypeReference < > ( ) { } , status ( ) . isBadRequest ( ) ) ;
deleteDifferentTenant ( ) ;
deleteDifferentTenant ( ) ;
}
}
@ -121,7 +122,7 @@ public abstract class BaseTbResourceControllerTest extends AbstractControllerTes
resource . setFileName ( DEFAULT_FILE_NAME ) ;
resource . setFileName ( DEFAULT_FILE_NAME ) ;
resource . setData ( "Test Data" ) ;
resource . setData ( "Test Data" ) ;
TbResource savedResource = doPost ( "/api/resource" , resource , TbResource . class ) ;
TbResource savedResource = save ( resource ) ;
TbResource foundResource = doGet ( "/api/resource/" + savedResource . getId ( ) . getId ( ) . toString ( ) , TbResource . class ) ;
TbResource foundResource = doGet ( "/api/resource/" + savedResource . getId ( ) . getId ( ) . toString ( ) , TbResource . class ) ;
Assert . assertNotNull ( foundResource ) ;
Assert . assertNotNull ( foundResource ) ;
@ -136,7 +137,7 @@ public abstract class BaseTbResourceControllerTest extends AbstractControllerTes
resource . setFileName ( DEFAULT_FILE_NAME ) ;
resource . setFileName ( DEFAULT_FILE_NAME ) ;
resource . setData ( "Test Data" ) ;
resource . setData ( "Test Data" ) ;
TbResource savedResource = doPost ( "/api/resource" , resource , TbResource . class ) ;
TbResource savedResource = save ( resource ) ;
doDelete ( "/api/resource/" + savedResource . getId ( ) . getId ( ) . toString ( ) )
doDelete ( "/api/resource/" + savedResource . getId ( ) . getId ( ) . toString ( ) )
. andExpect ( status ( ) . isOk ( ) ) ;
. andExpect ( status ( ) . isOk ( ) ) ;
@ -147,15 +148,18 @@ public abstract class BaseTbResourceControllerTest extends AbstractControllerTes
@Test
@Test
public void testFindTenantTbResources ( ) throws Exception {
public void testFindTenantTbResources ( ) throws Exception {
List < TbResourceInfo > resources = new ArrayList < > ( ) ;
List < TbResource > resourcesToSave = new ArrayList < > ( ) ;
for ( int i = 0 ; i < 173 ; i + + ) {
for ( int i = 0 ; i < 173 ; i + + ) {
TbResource resource = new TbResource ( ) ;
TbResource resource = new TbResource ( ) ;
resource . setTitle ( "Resource" + i ) ;
resource . setTitle ( "Resource" + i ) ;
resource . setResourceType ( ResourceType . JKS ) ;
resource . setResourceType ( ResourceType . JKS ) ;
resource . setFileName ( i + DEFAULT_FILE_NAME ) ;
resource . setFileName ( i + DEFAULT_FILE_NAME ) ;
resource . setData ( "Test Data" ) ;
resource . setData ( "Test Data" ) ;
resources . add ( new TbResourceInfo ( doPost ( "/api/ resource" , resource , TbResource . class ) ) ) ;
resourcesToSave . add ( resource ) ;
}
}
List < TbResourceInfo > resources = save ( resourcesToSave ) . stream ( ) . map ( TbResourceInfo : : new ) . collect ( Collectors . toList ( ) ) ;
List < TbResourceInfo > loadedResources = new ArrayList < > ( ) ;
List < TbResourceInfo > loadedResources = new ArrayList < > ( ) ;
PageLink pageLink = new PageLink ( 24 ) ;
PageLink pageLink = new PageLink ( 24 ) ;
PageData < TbResourceInfo > pageData ;
PageData < TbResourceInfo > pageData ;
@ -186,7 +190,7 @@ public abstract class BaseTbResourceControllerTest extends AbstractControllerTes
resource . setResourceType ( ResourceType . JKS ) ;
resource . setResourceType ( ResourceType . JKS ) ;
resource . setFileName ( i + DEFAULT_FILE_NAME ) ;
resource . setFileName ( i + DEFAULT_FILE_NAME ) ;
resource . setData ( "Test Data" ) ;
resource . setData ( "Test Data" ) ;
resources . add ( new TbResourceInfo ( doPost ( "/api/resource" , resource , TbResource . class ) ) ) ;
resources . add ( new TbResourceInfo ( save ( resource ) ) ) ;
}
}
List < TbResourceInfo > loadedResources = new ArrayList < > ( ) ;
List < TbResourceInfo > loadedResources = new ArrayList < > ( ) ;
PageLink pageLink = new PageLink ( 24 ) ;
PageLink pageLink = new PageLink ( 24 ) ;
@ -236,7 +240,7 @@ public abstract class BaseTbResourceControllerTest extends AbstractControllerTes
resource . setResourceType ( ResourceType . JKS ) ;
resource . setResourceType ( ResourceType . JKS ) ;
resource . setFileName ( i + DEFAULT_FILE_NAME ) ;
resource . setFileName ( i + DEFAULT_FILE_NAME ) ;
resource . setData ( "Test Data" ) ;
resource . setData ( "Test Data" ) ;
expectedResources . add ( new TbResourceInfo ( doPost ( "/api/resource" , resource , TbResource . class ) ) ) ;
expectedResources . add ( new TbResourceInfo ( save ( resource ) ) ) ;
}
}
loginSysAdmin ( ) ;
loginSysAdmin ( ) ;
@ -247,7 +251,7 @@ public abstract class BaseTbResourceControllerTest extends AbstractControllerTes
resource . setResourceType ( ResourceType . JKS ) ;
resource . setResourceType ( ResourceType . JKS ) ;
resource . setFileName ( i + DEFAULT_FILE_NAME ) ;
resource . setFileName ( i + DEFAULT_FILE_NAME ) ;
resource . setData ( "Test Data" ) ;
resource . setData ( "Test Data" ) ;
TbResourceInfo savedResource = new TbResourceInfo ( doPost ( "/api/resource" , resource , TbResource . class ) ) ;
TbResourceInfo savedResource = new TbResourceInfo ( save ( resource ) ) ;
systemResources . add ( savedResource ) ;
systemResources . add ( savedResource ) ;
if ( i > = 73 ) {
if ( i > = 73 ) {
expectedResources . add ( savedResource ) ;
expectedResources . add ( savedResource ) ;
@ -281,4 +285,12 @@ public abstract class BaseTbResourceControllerTest extends AbstractControllerTes
. andExpect ( status ( ) . isOk ( ) ) ;
. andExpect ( status ( ) . isOk ( ) ) ;
}
}
}
}
private TbResource save ( TbResource tbResource ) throws Exception {
return save ( Collections . singletonList ( tbResource ) ) . get ( 0 ) ;
}
private List < TbResource > save ( List < TbResource > tbResources ) throws Exception {
return doPostWithTypedResponse ( "/api/resource" , tbResources , new TypeReference < > ( ) { } ) ;
}
}
}