1 files changed, 40 insertions, 8 deletions

diff --git a/ecomp-sdk/epsdk-analytics/src/test/java/org/onap/portalsdk/analytics/gmap/line/LineTest.java b/ecomp-sdk/epsdk-analytics/src/test/java/org/onap/portalsdk/analytics/gmap/line/LineTest.java
index c3ea5ead..dc4171ce 100644
--- a/ecomp-sdk/epsdk-analytics/src/test/java/org/onap/portalsdk/analytics/gmap/line/LineTest.java
+++ b/ecomp-sdk/epsdk-analytics/src/test/java/org/onap/portalsdk/analytics/gmap/line/LineTest.java
@@ -37,23 +37,55 @@
  */
 package org.onap.portalsdk.analytics.gmap.line;
 
+import static org.junit.Assert.assertNotNull;
+
+import java.awt.geom.AffineTransform;
+import java.awt.geom.Point2D;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.junit.Before;
 import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.Mockito;
 import org.onap.portalsdk.analytics.gmap.map.NovaMap;
-
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
+import org.springframework.test.util.ReflectionTestUtils;
+@RunWith(PowerMockRunner.class)
+@PrepareForTest({Line.class})
 public class LineTest {
+	Line line;
+	Point2D screenPoint;
+	private String LINE_ID = "lineID";	
+	private String LINE_TYPE = "test";
+	private String NODE1 = "nodeID1";
+	private String NODE2 = "nodeID2";
 	
-    public Line mockLineTest()
-    {
+	@Before
+    public void init() throws Exception{
     	NovaMap novaMap= new NovaMap();
     	novaMap.setDataLoaded("dataLoaded");
-    	Line line = new Line(novaMap);
-    	return line;
+    	ReflectionTestUtils.setField(novaMap, "transform", new AffineTransform());
+    	LineCollection lineCollection = Mockito.mock(LineCollection.class);
+    	ArrayList<LineInfo> list = new ArrayList<LineInfo>();
+    	LineInfo lineInfo = new LineInfo(NODE1, NODE2);
+		lineInfo.setLineID(LINE_ID);
+		lineInfo.setLineType(LINE_TYPE);
+		lineCollection.addLine(lineInfo);
+		lineCollection.addSelectedLine(LINE_ID);
+		list.add(lineInfo);
+    	PowerMockito.whenNew(LineCollection.class).withNoArguments().thenReturn(lineCollection);
+    	Mockito.when(lineCollection.getLineCollection()).thenReturn(list);
+    	screenPoint = Mockito.mock(Point2D.class);
+    	line = new Line(novaMap);
     }
 	
     @Test
-    public void lineExistTest()
-    {
-    	Line line = mockLineTest();
+    public void lineExistTest(){
+    	assertNotNull(line);
+    	line.lineExist(screenPoint);
     }
 	
 }