POLICY-SDK-APP/src/test/java/org/onap/policy/controller/ExportAndImportDecisionBlackListEntriesTest.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * ONAP Policy Engine
   4  * ================================================================================
   5  * Copyright (C) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20 package org.onap.policy.controller;
  21
  22 import static org.junit.Assert.assertTrue;
  23 import static org.junit.Assert.fail;
  24 import static org.mockito.Mockito.mock;
  25
  26 import java.io.BufferedReader;
  27 import java.io.ByteArrayInputStream;
  28 import java.io.ByteArrayOutputStream;
  29 import java.io.File;
  30 import java.io.FileInputStream;
  31 import java.io.IOException;
  32 import java.io.InputStream;
  33 import java.io.StringReader;
  34
  35 import javax.servlet.ReadListener;
  36 import javax.servlet.ServletInputStream;
  37 import javax.servlet.http.HttpServletRequest;
  38
  39 import org.apache.commons.io.IOUtils;
  40 import org.junit.Before;
  41 import org.junit.Test;
  42 import org.mockito.Mockito;
  43 import org.springframework.mock.web.MockHttpServletRequest;
  44 import org.springframework.mock.web.MockHttpServletResponse;
  45
  46 public class ExportAndImportDecisionBlackListEntriesTest {
  47
  48         private HttpServletRequest request;
  49         private MockHttpServletResponse response;
  50         String jsonString;
  51
  52         @Before
  53         public void setUp() throws Exception {
  54                 request = mock(HttpServletRequest.class);
  55                 response =  new MockHttpServletResponse();
  56         }
  57
  58         @Test
  59         public void testExportBlackList() throws IOException{
  60                 ClassLoader classLoader = getClass().getClassLoader();
  61                 jsonString = IOUtils.toString(classLoader.getResourceAsStream("DecisionPolicyData.txt"));
  62                 try(BufferedReader reader = new BufferedReader(new StringReader(jsonString))){
  63                         Mockito.when(request.getReader()).thenReturn(reader);
  64                         ExportAndImportDecisionBlackListEntries controller = new ExportAndImportDecisionBlackListEntries();
  65                         controller.exportBlackList(request, response);
  66                         assertTrue("".equals(response.getContentAsString()));
  67                 }catch(Exception e){
  68                         fail("Not expecting Exception while Exporting BlackListEntries.");
  69                 }
  70         }
  71
  72         @Test
  73         public void testImportBlackList() throws Exception{
  74                 MockHttpServletRequest request =  new MockHttpServletRequest();
  75                 ExportAndImportDecisionBlackListEntries controller = new ExportAndImportDecisionBlackListEntries();
  76                 File file = new File("src/test/resources/BlackList.xls");
  77                 try(FileInputStream targetStream = new FileInputStream(file)){
  78                         ExportAndImportDecisionBlackListEntriesTest testController = Mockito.mock(ExportAndImportDecisionBlackListEntriesTest.class);
  79                         ServletInputStream inputStream = testController.getInputStream(getBytes(targetStream));
  80                         Mockito.when(request.getInputStream()).thenReturn(inputStream);
  81                         String boundary = "===" + System.currentTimeMillis() + "===";
  82                         request.addHeader("Content-Type", "multipart/form-data; boundary=" + boundary);
  83                         request.addHeader("name", "BlackList.xls");
  84                         controller.importBlackListFile(request, response);
  85                         assertTrue(response.getContentAsString().contains("data"));
  86                 }catch(Exception e){
  87                         fail("Not expecting Exception while importing BlackListEntries.");
  88                 }
  89         }
  90
  91         public static byte[] getBytes(InputStream is) throws IOException {
  92                 int len;
  93                 int size = 1024;
  94                 byte[] buf;
  95                 ByteArrayOutputStream bos = new ByteArrayOutputStream();
  96                 buf = new byte[size];
  97                 while ((len = is.read(buf, 0, size)) != -1)
  98                         bos.write(buf, 0, len);
  99                 buf = bos.toByteArray();
 100                 return buf;
 101         }
 102
 103         public ServletInputStream getInputStream(byte[] body) throws IOException {
 104         final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body);
 105         ServletInputStream servletInputStream = new ServletInputStream() {
 106             public int read() throws IOException {
 107                 return byteArrayInputStream.read();
 108             }
 109
 110                         @Override
 111                         public boolean isFinished() {
 112                                 return false;
 113                         }
 114
 115                         @Override
 116                         public boolean isReady() {
 117                                 return false;
 118                         }
 119
 120                         @Override
 121                         public void setReadListener(ReadListener readListener) {
 122                         }
 123         };
 124         return servletInputStream;
 125     }
 126 }