added possibility to get ProjectUser by test Id

Author: knysh

src/main/java/main/controllers/Project/TestController.java

@@ -60,7 +60,7 @@ public List<TestDto> get(TestDto template) throws AqualityException {
 
     @Override
     public boolean delete(TestDto template) throws AqualityException {
-        if(baseUser.isManager() || baseUser.getProjectUser(template.getProject_id()).isEditor()){
+        if(baseUser.isManager() || baseUser.getProjectUserByTestId(template.getProject_id()).isEditor()){
             return testDao.delete(template);
         }else{
             throw new AqualityPermissionsException("Account is not allowed to delete Test", baseUser);

src/main/java/main/model/dto/UserDto.java

@@ -7,6 +7,7 @@
 import main.annotations.DataBaseSearchable;
 import main.exceptions.AqualityException;
 import main.model.db.dao.project.MilestoneDao;
+import main.model.db.dao.project.TestDao;
 import main.model.db.dao.project.TestSuiteDao;
 import main.utils.BooleanUtil;
 
@@ -94,15 +95,10 @@ public boolean isFromGlobalManagement(){
     }
 
     public ProjectUserDto getProjectUser(Integer projectId){
-        ProjectUserDto emptyPU = new ProjectUserDto();
-        emptyPU.setViewer(0);
-        emptyPU.setAdmin(0);
-        emptyPU.setManager(0);
-        emptyPU.setEngineer(0);
         if(projectUsers != null){
-           return projectUsers.stream().filter(x -> x.getProject_id().equals(projectId)).findFirst().orElse(emptyPU);
+           return projectUsers.stream().filter(x -> x.getProject_id().equals(projectId)).findFirst().orElse(getEmptyProjectUser());
         }
-        return emptyPU;
+        return getEmptyProjectUser();
     }
 
     public ProjectUserDto getProjectUserBySuiteId(Integer suite_id) throws AqualityException {
@@ -113,6 +109,14 @@ public ProjectUserDto getProjectUserBySuiteId(Integer suite_id) throws AqualityE
         return getProjectUser(template.getProject_id());
     }
 
+    public ProjectUserDto getProjectUserByTestId(Integer test_id) throws AqualityException {
+        TestDao testDao = new TestDao();
+        TestDto template = new TestDto();
+        template.setId(test_id);
+        template = testDao.searchAll(template).get(0);
+        return getProjectUser(template.getProject_id());
+    }
+
     public ProjectUserDto getProjectUserByMilestoneId(Integer milestone_id) throws AqualityException {
         MilestoneDao milestoneDao = new MilestoneDao();
         MilestoneDto template = new MilestoneDto();
@@ -126,4 +130,13 @@ public UserDto toPublic(){
         this.setSession_code("");
         return this;
     }
+
+    private ProjectUserDto getEmptyProjectUser(){
+        ProjectUserDto emptyPU = new ProjectUserDto();
+        emptyPU.setViewer(0);
+        emptyPU.setAdmin(0);
+        emptyPU.setManager(0);
+        emptyPU.setEngineer(0);
+        return emptyPU;
+    }
 }