1 files changed, 74 insertions, 0 deletions
diff --git a/ecomp-portal-widget-ms/widget-ms/src/test/java/org/openecomp/portalapp/widget/test/controller/WidgetFileControllerTest.java b/ecomp-portal-widget-ms/widget-ms/src/test/java/org/openecomp/portalapp/widget/test/controller/WidgetFileControllerTest.java
new file mode 100644
index 00000000..2045435c
--- /dev/null
+++ b/ecomp-portal-widget-ms/widget-ms/src/test/java/org/openecomp/portalapp/widget/test/controller/WidgetFileControllerTest.java
@@ -0,0 +1,74 @@
+package org.openecomp.portalapp.widget.test.controller;
+
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Mockito.times;
+
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.ArgumentCaptor;
+import org.mockito.InjectMocks;
+import org.mockito.Mock;
+import org.mockito.Mockito;
+import org.mockito.MockitoAnnotations;
+import org.mockito.runners.MockitoJUnitRunner;
+import org.openecomp.portalapp.widget.controller.DatabaseFileUploadController;
+import org.openecomp.portalapp.widget.service.impl.StorageServiceImpl;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
+import org.springframework.test.web.servlet.setup.MockMvcBuilders;
+
+
+@RunWith(MockitoJUnitRunner.class)
+public class WidgetFileControllerTest {
+	private MockMvc mockMvc;
+	
+	@Mock
+	private StorageServiceImpl storageService;
+	
+	@InjectMocks
+	private DatabaseFileUploadController controller;
+	
+	@Before
+	public void setUp() {
+		MockitoAnnotations.initMocks(this);
+		mockMvc = MockMvcBuilders.standaloneSetup(controller).build();
+	}
+	
+	@Test
+	public void getWidgetMarkup_NoError() throws Exception{
+		ArgumentCaptor<Long> storageServiceArg = ArgumentCaptor.forClass(Long.class);
+		Long widgetId = new Long(1);
+		mockMvc.perform(MockMvcRequestBuilders.get("/microservices/markup/" + widgetId)).andReturn();;
+		Mockito.verify(storageService, times(1)).getWidgetMarkup(storageServiceArg.capture());
+		assertEquals(storageServiceArg.getValue(), widgetId);
+	}
+	
+	@Test
+	public void getWidgetController_NoError() throws Exception{
+		ArgumentCaptor<Long> storageServiceArg = ArgumentCaptor.forClass(Long.class);
+		Long widgetId = new Long(1);
+		mockMvc.perform(MockMvcRequestBuilders.get("/microservices/" + widgetId + "/controller.js")).andReturn();;
+		Mockito.verify(storageService, times(1)).getWidgetController(storageServiceArg.capture());
+		assertEquals(storageServiceArg.getValue(), widgetId);
+	}
+	
+	@Test
+	public void getWidgetFramework_NoError() throws Exception{
+		ArgumentCaptor<Long> storageServiceArg = ArgumentCaptor.forClass(Long.class);
+		Long widgetId = new Long(1);
+		mockMvc.perform(MockMvcRequestBuilders.get("/microservices/" + widgetId + "/framework.js")).andReturn();;
+		Mockito.verify(storageService, times(1)).getWidgetFramework(storageServiceArg.capture());
+		assertEquals(storageServiceArg.getValue(), widgetId);
+	}
+	
+	@Test
+	public void getWidgetCSS_NoError() throws Exception{
+		ArgumentCaptor<Long> storageServiceArg = ArgumentCaptor.forClass(Long.class);
+		Long widgetId = new Long(1);
+		mockMvc.perform(MockMvcRequestBuilders.get("/microservices/" + widgetId + "/styles.css")).andReturn();;
+		Mockito.verify(storageService, times(1)).getWidgetCSS(storageServiceArg.capture());
+		assertEquals(storageServiceArg.getValue(), widgetId);
+	}
+	
+}