Browse Source

[Improvement-12763][UT] Improve testListAuthorizedProjects() in ProjectMapperTest (#12768)

3.2.0-release
rickchengx 2 years ago committed by GitHub
parent
commit
43719f7c15
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 40
      dolphinscheduler-dao/src/test/java/org/apache/dolphinscheduler/dao/mapper/ProjectMapperTest.java

40
dolphinscheduler-dao/src/test/java/org/apache/dolphinscheduler/dao/mapper/ProjectMapperTest.java

@ -22,7 +22,6 @@ import org.apache.dolphinscheduler.dao.entity.Project;
import org.apache.dolphinscheduler.dao.entity.ProjectUser;
import org.apache.dolphinscheduler.dao.entity.User;
import java.util.Collections;
import java.util.Date;
import java.util.List;
@ -217,14 +216,41 @@ public class ProjectMapperTest extends BaseDaoTest {
Assertions.assertNotEquals(allProject.size(), 0);
}
/**
* test query project permission
*/
private Project insertProject(long projectCode, String projectName, int userId) {
Project project = new Project();
project.setName(projectName);
project.setUserId(userId);
project.setCode(projectCode);
project.setCreateTime(new Date());
project.setUpdateTime(new Date());
projectMapper.insert(project);
return project;
}
@Test
public void testListAuthorizedProjects() {
Project project = insertOne();
List<Project> projects = projectMapper.listAuthorizedProjects(1, Collections.singletonList(project.getId()));
Assertions.assertEquals(projects.size(), 0);
User user = new User();
user.setUserName("ut user");
userMapper.insert(user);
// project1: created by user
insertProject(1, "project 1", user.getId());
// project2: not created by user
insertProject(2, "project 2", user.getId() + 1);
// project3: not created by user but is authorized to user
Project project3 = insertProject(3, "project 3", user.getId() + 1);
ProjectUser projectUser = new ProjectUser();
projectUser.setProjectId(project3.getId());
projectUser.setUserId(user.getId());
projectUser.setCreateTime(new Date());
projectUser.setUpdateTime(new Date());
projectUserMapper.insert(projectUser);
List<Project> projects = projectMapper.listAuthorizedProjects(user.getId(), null);
Assertions.assertEquals(projects.size(), 2);
}
}

Loading…
Cancel
Save