diff --git a/rest-api/src/test/java/lcsb/mapviewer/api/RestTestFunctions.java b/rest-api/src/test/java/lcsb/mapviewer/api/RestTestFunctions.java
index df9125be04e1ad57d96e1d790ca5652324cf7b67..8cd10e6d9f24cdbcab486403fc0779ed15889a00 100644
--- a/rest-api/src/test/java/lcsb/mapviewer/api/RestTestFunctions.java
+++ b/rest-api/src/test/java/lcsb/mapviewer/api/RestTestFunctions.java
@@ -1,5 +1,8 @@
 package lcsb.mapviewer.api;
 
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyString;
+
 import java.io.BufferedReader;
 import java.io.File;
 import java.io.FileInputStream;
@@ -32,6 +35,7 @@ import org.apache.log4j.Logger;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.runner.RunWith;
+import org.mockito.Mockito;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.security.crypto.password.PasswordEncoder;
 import org.springframework.test.annotation.Rollback;
@@ -46,6 +50,7 @@ import org.xml.sax.InputSource;
 import org.xml.sax.SAXException;
 
 import lcsb.mapviewer.common.Configuration;
+import lcsb.mapviewer.common.exception.InvalidStateException;
 import lcsb.mapviewer.common.exception.InvalidXmlSchemaException;
 import lcsb.mapviewer.converter.ComplexZipConverter;
 import lcsb.mapviewer.converter.ComplexZipConverterParams;
@@ -59,6 +64,8 @@ import lcsb.mapviewer.model.map.model.Model;
 import lcsb.mapviewer.model.map.model.ModelSubmodelConnection;
 import lcsb.mapviewer.model.map.model.SubmodelType;
 import lcsb.mapviewer.persist.DbUtils;
+import lcsb.mapviewer.services.interfaces.IModelService;
+import lcsb.mapviewer.services.interfaces.IProjectService;
 import lcsb.mapviewer.services.interfaces.IUserService;
 
 @Transactional
@@ -312,4 +319,21 @@ public abstract class RestTestFunctions {
     return tmp.toString();
   }
 
+  protected IProjectService createProjectMockServiceForModel(Model model) {
+    try {
+      IProjectService mockModelService = Mockito.mock(IProjectService.class);
+      Mockito.when(mockModelService.getProjectByProjectId(anyString(), anyString())).thenReturn(model.getProject());
+      return mockModelService;
+    } catch (lcsb.mapviewer.services.SecurityException e) {
+      throw new InvalidStateException(e);
+    }
+  }
+
+  protected IModelService createModelMockServiceForModel(Model model) throws lcsb.mapviewer.services.SecurityException {
+    IModelService mockModelService = Mockito.mock(IModelService.class);
+    Mockito.when(mockModelService.getLastModelByProjectId(anyString(), any())).thenReturn(model);
+    return mockModelService;
+  }
+
+
 }
diff --git a/rest-api/src/test/java/lcsb/mapviewer/api/projects/drugs/DrugRestImplTest.java b/rest-api/src/test/java/lcsb/mapviewer/api/projects/drugs/DrugRestImplTest.java
index e5e83e30b94de235311ac7b3fbe3fa615e6df847..dd5eb15d048d1075e838db11ad95f6b4796b6780 100644
--- a/rest-api/src/test/java/lcsb/mapviewer/api/projects/drugs/DrugRestImplTest.java
+++ b/rest-api/src/test/java/lcsb/mapviewer/api/projects/drugs/DrugRestImplTest.java
@@ -1,8 +1,8 @@
 package lcsb.mapviewer.api.projects.drugs;
 
 import static org.junit.Assert.assertNotNull;
-import static org.mockito.Matchers.any;
-import static org.mockito.Matchers.anyString;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.ArgumentMatchers.anyString;
 
 import java.util.List;
 import java.util.Map;
@@ -16,8 +16,11 @@ import org.mockito.Mockito;
 import org.springframework.beans.factory.annotation.Autowired;
 
 import lcsb.mapviewer.api.RestTestFunctions;
+import lcsb.mapviewer.common.exception.InvalidStateException;
 import lcsb.mapviewer.model.map.model.Model;
+import lcsb.mapviewer.services.SecurityException;
 import lcsb.mapviewer.services.interfaces.IModelService;
+import lcsb.mapviewer.services.interfaces.IProjectService;
 
 public class DrugRestImplTest extends RestTestFunctions {
   Logger logger = Logger.getLogger(DrugRestImplTest.class);
@@ -53,7 +56,7 @@ public class DrugRestImplTest extends RestTestFunctions {
       throw e;
     }
   }
-  
+
   @Test
   public void testTargetWithEmptyMechanism() throws Exception {
     try {
@@ -69,9 +72,8 @@ public class DrugRestImplTest extends RestTestFunctions {
 
   private DrugRestImpl createMockProjectRest(String string) throws Exception {
     Model model = super.getModelForFile(string, true);
-    IModelService mockModelService = Mockito.mock(IModelService.class);
-    Mockito.when(mockModelService.getLastModelByProjectId(anyString(), any())).thenReturn(model);
-    _drugRestImpl.setModelService(mockModelService);
+    _drugRestImpl.setModelService(createModelMockServiceForModel(model));
+    _drugRestImpl.setProjectService(createProjectMockServiceForModel(model));
     return _drugRestImpl;
   }
 
diff --git a/rest-api/src/test/java/lcsb/mapviewer/api/projects/mirnas/MiRnaRestImplTest.java b/rest-api/src/test/java/lcsb/mapviewer/api/projects/mirnas/MiRnaRestImplTest.java
index 624dcd70c9b73e9edd75479012112025e8437c26..56da6872af889beb83c7963f003363112b0ef1af 100644
--- a/rest-api/src/test/java/lcsb/mapviewer/api/projects/mirnas/MiRnaRestImplTest.java
+++ b/rest-api/src/test/java/lcsb/mapviewer/api/projects/mirnas/MiRnaRestImplTest.java
@@ -58,9 +58,8 @@ public class MiRnaRestImplTest extends RestTestFunctions {
 
   private MiRnaRestImpl createMockProjectRest(String string) throws Exception {
     Model model = super.getModelForFile(string, true);
-    IModelService mockModelService = Mockito.mock(IModelService.class);
-    Mockito.when(mockModelService.getLastModelByProjectId(anyString(), any())).thenReturn(model);
-    _miRnaRestImpl.setModelService(mockModelService);
+    _miRnaRestImpl.setModelService(createModelMockServiceForModel(model));
+    _miRnaRestImpl.setProjectService(createProjectMockServiceForModel(model));
     return _miRnaRestImpl;
   }
 
diff --git a/rest-api/src/test/resources/log4j.properties b/rest-api/src/test/resources/log4j.properties
index c36f0e8f0f2d1bc6becf9ac41a23e8ecd76ef358..e3cf076f834ff213be76567ff8498531f0b1d474 100644
--- a/rest-api/src/test/resources/log4j.properties
+++ b/rest-api/src/test/resources/log4j.properties
@@ -19,4 +19,7 @@ log4j.logger.org.springframework=warn
 log4j.logger.org.hibernate=warn
 log4j.logger.lcsb=debug
 log4j.logger.lcsb.mapviewer.annotation.cache=info
-log4j.logger.lcsb.mapviewer.persist=info
\ No newline at end of file
+log4j.logger.lcsb.mapviewer.persist=info
+
+#There are plenty of useless warnings in jsbml library
+log4j.logger.org.sbml.jsbml=error
\ No newline at end of file