|
|
|
@ -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); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
} |
|
|
|
|