fixed getProjectUserByTest

Author: knysh

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

@@ -46,7 +46,7 @@ public TestDto create(TestDto template) throws AqualityException {
     }
 
     public List<TestDto> get(TestDto template, boolean withChildren) throws AqualityException {
-        if(baseUser.isFromGlobalManagement() || baseUser.getProjectUserByTestId(template.getId()).isViewer()){
+        if(baseUser.isFromGlobalManagement() || baseUser.getProjectUserByTest(template).isViewer()){
             return fillTests(testDao.searchAll(template), withChildren);
         }else{
             throw new AqualityPermissionsException("Account is not allowed to view Tests", baseUser);

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

@@ -109,12 +109,17 @@ 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 getProjectUserByTest(TestDto testDto) throws AqualityException {
+        if(testDto.getProject_id() != null){
+            return getProjectUser(testDto.getProject_id());
+        }
+
+        try {
+            TestDao testDao = new TestDao();
+            return getProjectUser(testDao.searchAll(testDto).get(0).getProject_id());
+        }catch (IndexOutOfBoundsException e){
+            throw new AqualityException("Cannot define project id");
+        }
     }
 
     public ProjectUserDto getProjectUserByMilestoneId(Integer milestone_id) throws AqualityException {