|
|
|
@ -272,6 +272,38 @@ public class DataSourceMapperTest extends BaseDaoTest {
|
|
|
|
|
.containsAll(Arrays.asList(dataSourceIds))); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
@Test |
|
|
|
|
public void testSelectPagingByIds() { |
|
|
|
|
User user1 = createGeneralUser("user1"); |
|
|
|
|
User user2 = createGeneralUser("user2"); |
|
|
|
|
DataSource dataSource1ForUser1 = createDataSource(user1.getId(), "dataSource1ForUser1"); |
|
|
|
|
DataSource dataSource2ForUser2 = createDataSource(user2.getId(), "dataSource2ForUser2"); |
|
|
|
|
DataSource dataSource3ForUser1 = createDataSource(user1.getId(), dataSource1ForUser1.getName() + "test"); |
|
|
|
|
|
|
|
|
|
// select without conditions
|
|
|
|
|
Page page = new Page(0, 4); |
|
|
|
|
List<DataSource> actualDataSources = dataSourceMapper.selectPagingByIds(page, null, null).getRecords(); |
|
|
|
|
Assertions.assertEquals(3, actualDataSources.size()); |
|
|
|
|
Assertions.assertTrue(actualDataSources.stream().map(t -> t.getId()).collect(toList()) |
|
|
|
|
.containsAll(Arrays.asList(dataSource1ForUser1.getId(), dataSource2ForUser2.getId(), |
|
|
|
|
dataSource3ForUser1.getId()))); |
|
|
|
|
|
|
|
|
|
// select with name
|
|
|
|
|
actualDataSources = dataSourceMapper.selectPagingByIds(page, null, dataSource1ForUser1.getName()).getRecords(); |
|
|
|
|
Assertions.assertEquals(2, actualDataSources.size()); |
|
|
|
|
Assertions.assertTrue(actualDataSources.stream().map(t -> t.getId()).collect(toList()) |
|
|
|
|
.containsAll(Arrays.asList(dataSource1ForUser1.getId(), dataSource3ForUser1.getId()))); |
|
|
|
|
|
|
|
|
|
// select with dataSourceIds and name
|
|
|
|
|
actualDataSources = dataSourceMapper |
|
|
|
|
.selectPagingByIds(page, Arrays.asList(dataSource1ForUser1.getId(), dataSource2ForUser2.getId()), |
|
|
|
|
dataSource1ForUser1.getName()) |
|
|
|
|
.getRecords(); |
|
|
|
|
Assertions.assertEquals(1, actualDataSources.size()); |
|
|
|
|
Assertions.assertTrue(actualDataSources.stream().map(t -> t.getId()).collect(toList()) |
|
|
|
|
.containsAll(Arrays.asList(dataSource1ForUser1.getId()))); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
/** |
|
|
|
|
* create datasource relation |
|
|
|
|
* @param userId |
|
|
|
|