X-Git-Url: https://gerrit.onap.org/r/gitweb?p=policy%2Fengine.git;a=blobdiff_plain;f=ONAP-PAP-REST%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Fpolicy%2Fpap%2Fxacml%2Frest%2Fcontroller%2FDictionaryImportControllerTest.java;h=ef723a3fdecf5521acb6cf0c9d5456b16c80abac;hp=e8d602ce5f16cde7dd6fbb6a8c9bba7589c1a1d1;hb=1e61676b77dd09659027b8984f050df7e8538526;hpb=e22a2b613bda0e683043ef870d6b16e5a7e04df7

diff --git a/ONAP-PAP-REST/src/test/java/org/onap/policy/pap/xacml/rest/controller/DictionaryImportControllerTest.java b/ONAP-PAP-REST/src/test/java/org/onap/policy/pap/xacml/rest/controller/DictionaryImportControllerTest.java
index e8d602ce5..ef723a3fd 100644
--- a/ONAP-PAP-REST/src/test/java/org/onap/policy/pap/xacml/rest/controller/DictionaryImportControllerTest.java
+++ b/ONAP-PAP-REST/src/test/java/org/onap/policy/pap/xacml/rest/controller/DictionaryImportControllerTest.java
@@ -2,14 +2,14 @@
  * ============LICENSE_START=======================================================
  * ONAP-PAP-REST
  * ================================================================================
- * Copyright (C) 2018 AT&T Intellectual Property. All rights reserved.
+ * Copyright (C) 2018-2019 AT&T Intellectual Property. All rights reserved.
  * ================================================================================
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
- * 
+ *
  *      http://www.apache.org/licenses/LICENSE-2.0
- * 
+ *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -17,36 +17,125 @@
  * limitations under the License.
  * ============LICENSE_END=========================================================
  */
+
 package org.onap.policy.pap.xacml.rest.controller;
 
 import static org.junit.Assert.assertTrue;
-import org.onap.policy.pap.xacml.rest.controller.DictionaryImportController;
-import org.junit.After;
+import static org.junit.Assert.fail;
+
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.ArrayList;
+import java.util.List;
+
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
 import org.junit.Before;
 import org.junit.Test;
+import org.mockito.Mockito;
 import org.onap.policy.common.logging.flexlogger.FlexLogger;
 import org.onap.policy.common.logging.flexlogger.Logger;
+import org.onap.policy.rest.dao.CommonClassDao;
+import org.springframework.mock.web.MockHttpServletResponse;
+
+public class DictionaryImportControllerTest extends Mockito {
+
+    private static Logger logger = FlexLogger.getLogger(DictionaryImportController.class);
+
+    private static CommonClassDao commonClassDao;
+    private HttpServletRequest request = null;
+    private HttpServletResponse response = null;
+    private DictionaryImportController controller = null;
+
+    @Before
+    public void setUp() throws Exception {
+        logger.info("setUp: Entering");
+        commonClassDao = Mockito.mock(CommonClassDao.class);
+        doNothing().when(commonClassDao).save(new Object());
+        controller = new DictionaryImportController();
+        new DictionaryImportController(commonClassDao);
+        request = Mockito.mock(HttpServletRequest.class);
+        response = new MockHttpServletResponse();
+    }
+
+    @Test
+    public void testIsValidDictionaryName() {
+        DictionaryImportController cotroller = new DictionaryImportController();
+        // test invalid name
+        assertTrue(!cotroller.isValidDictionaryName("wrong-name"));
+        // test valid name
+        assertTrue(cotroller.isValidDictionaryName("ActionList"));
+    }
+
+    @Test
+    public void testImportDictionaryData() throws ServletException, IOException {
+        List<String> fileNames = new ArrayList<>();
+        fileNames.add("Attribute.csv");
+        fileNames.add("ActionPolicyDictionary.csv");
+        fileNames.add("OnapName.csv");
+        fileNames.add("MSPolicyDictionary.csv");
+        fileNames.add("OptimizationPolicyDictionary.csv");
+        fileNames.add("ClosedLoopService.csv");
+        fileNames.add("ClosedLoopSite.csv");
+        fileNames.add("VarbindDictionary.csv");
+        fileNames.add("BRMSParamDictionary.csv");
+        fileNames.add("BRMSControllerDictionary.csv");
+        fileNames.add("BRMSDependencyDictionary.csv");
+        fileNames.add("PrefixList.csv");
+        fileNames.add("SecurityZone.csv");
+        fileNames.add("ServiceList.csv");
+        fileNames.add("ServiceGroup.csv");
+        fileNames.add("AddressGroup.csv");
+        fileNames.add("ProtocolList.csv");
+        fileNames.add("TermList.csv");
+        fileNames.add("SearchCriteria.csv");
+        fileNames.add("VNFType.csv");
+        fileNames.add("VSCLAction.csv");
+        fileNames.add("PEPOptions.csv");
+        fileNames.add("Settings.csv");
+        fileNames.add("Zone.csv");
+        fileNames.add("ActionList.csv");
+        for (int i = 0; i < fileNames.size(); i++) {
+            File file = new File("src/test/resources/dictionaryImport/" + fileNames.get(i));
+            try (FileInputStream targetStream = new FileInputStream(file)) {
+                PushPolicyControllerTest pushController = new PushPolicyControllerTest();
+                when(request.getInputStream()).thenReturn(pushController.getInputStream(getBytes(targetStream)));
+                when(request.getParameter("userId")).thenReturn("demo");
+                when(request.getParameter("dictionaryName")).thenReturn(fileNames.get(i));
+                controller.importDictionaryData(request, response);
+                assertTrue(HttpServletResponse.SC_OK == response.getStatus());
+            } catch (IOException e) {
+                fail();
+            }
+        }
+        when(request.getParameter("dictionaryName")).thenReturn("WrongName");
+        controller.importDictionaryData(request, response);
+        assertTrue(HttpServletResponse.SC_BAD_REQUEST == response.getStatus());
+
+        when(request.getParameter("dictionaryName")).thenReturn("");
+        controller.importDictionaryData(request, response);
+        assertTrue(HttpServletResponse.SC_BAD_REQUEST == response.getStatus());
+
+        when(request.getInputStream()).thenReturn(null);
+        when(request.getParameter("dictionaryName")).thenReturn("Attribute.csv");
+        controller.importDictionaryData(request, response);
+        assertTrue(HttpServletResponse.SC_INTERNAL_SERVER_ERROR == response.getStatus());
+    }
 
-public class DictionaryImportControllerTest {
-	
-	private static Logger logger = FlexLogger.getLogger(DictionaryImportController.class);
-
-	@Before
-	public void setUp() throws Exception {
-		logger.info("setUp: Entering");
-	}
-	
-	@Test
-	public void testIsValidDictionaryName(){
-		DictionaryImportController cotroller = new DictionaryImportController();
-		//test invalid name
-		assertTrue(!cotroller.isValidDictionaryName("wrong-name"));
-		//test valid name
-		assertTrue(cotroller.isValidDictionaryName("ActionList"));
-	}
-	
-	@After
-	 public void destroy(){
-
-	 }
+    public static byte[] getBytes(InputStream is) throws IOException {
+        int len;
+        int size = 1024;
+        byte[] buf;
+        ByteArrayOutputStream bos = new ByteArrayOutputStream();
+        buf = new byte[size];
+        while ((len = is.read(buf, 0, size)) != -1)
+            bos.write(buf, 0, len);
+        buf = bos.toByteArray();
+        return buf;
+    }
 }