natas_hw
/
dolphinscheduler
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Improvement-12850][UT] Improve the ut of DataSourceMapperTest (#12851)

This commit is contained in:
rickchengx 2022-11-12 23:57:40 +08:00 committed by GitHub
parent 2ba5bd47dd
commit 7c711e622f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
dolphinscheduler-dao/src/test/java/org/apache/dolphinscheduler/dao/mapper/DataSourceMapperTest.java
View File

@ -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