- File file = new File("src/test/resources/BlackList.xls");
- try(FileInputStream targetStream = new FileInputStream(file)){
- ExportAndImportDecisionBlackListEntriesTest testController = Mockito.mock(ExportAndImportDecisionBlackListEntriesTest.class);
- ServletInputStream inputStream = testController.getInputStream(getBytes(targetStream));
- Mockito.when(request.getInputStream()).thenReturn(inputStream);
- String boundary = "===" + System.currentTimeMillis() + "===";
- request.addHeader("Content-Type", "multipart/form-data; boundary=" + boundary);
- request.addHeader("name", "BlackList.xls");
- controller.importBlackListFile(request, response);
- assertTrue(response.getContentAsString().contains("data"));
- }catch(Exception e){
- fail("Not expecting Exception while importing BlackListEntries.");
- }
- }
-
- 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;
+ MockHttpServletResponse resp = new MockHttpServletResponse();
+ controller.importBlackListFile(req, resp);
+ assertTrue(resp.getContentAsString().contains("data"));
+