From 0c38db9f11f31019b476ab4da95b3ea5ee71f1f0 Mon Sep 17 00:00:00 2001
From: Piotr Gawron <piotr.gawron@uni.lu>
Date: Thu, 27 Jul 2017 15:16:03 +0200
Subject: [PATCH] additional unit test

---
 .../api/projects/ProjectRestImplTest.java        | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/rest-api/src/test/java/lcsb/mapviewer/api/projects/ProjectRestImplTest.java b/rest-api/src/test/java/lcsb/mapviewer/api/projects/ProjectRestImplTest.java
index 12a789d393..647f931a7c 100644
--- a/rest-api/src/test/java/lcsb/mapviewer/api/projects/ProjectRestImplTest.java
+++ b/rest-api/src/test/java/lcsb/mapviewer/api/projects/ProjectRestImplTest.java
@@ -24,6 +24,7 @@ import lcsb.mapviewer.api.RestTestFunctions;
 import lcsb.mapviewer.common.exception.InvalidArgumentException;
 import lcsb.mapviewer.converter.graphics.PdfImageGenerator;
 import lcsb.mapviewer.converter.model.celldesigner.CellDesignerXmlParser;
+import lcsb.mapviewer.model.cache.FileEntry;
 import lcsb.mapviewer.model.map.MiriamType;
 import lcsb.mapviewer.model.map.model.Model;
 import lcsb.mapviewer.services.interfaces.IModelService;
@@ -50,12 +51,23 @@ public class ProjectRestImplTest extends RestTestFunctions {
 	}
 
 	@Test
-	public void testGetModelAsImage() throws Exception {
+	public void testGetModelAsImageForInvalidConverter() throws Exception {
 		try {
 			ProjectRestImpl projectRest = createMockProjectRest("testFiles/model/sample.xml");
 			projectRest.getModelAsImage(token.getId(), "sample", "0", "", "", "", "", "");
 			fail("Exception expected");
-		} catch (InvalidArgumentException e) {
+		} catch (Exception e) {
+			e.printStackTrace();
+			throw e;
+		}
+	}
+
+	@Test
+	public void testGetModelAsImage() throws Exception {
+		try {
+			ProjectRestImpl projectRest = createMockProjectRest("testFiles/model/sample.xml");
+			FileEntry result = projectRest.getModelAsImage(token.getId(), "sample", "0", PdfImageGenerator.class.getCanonicalName(), "", "", "", "");
+			assertNotNull(result);
 		} catch (Exception e) {
 			e.printStackTrace();
 			throw e;
-- 
GitLab