asdctool/src/test/java/org/openecomp/sdc/asdctool/migration/tasks/handlers/XlsOutputHandlerTest.java

   1 package org.openecomp.sdc.asdctool.migration.tasks.handlers;
   2
   3 import org.apache.poi.ss.usermodel.Workbook;
   4 import org.junit.Test;
   5 import org.junit.runner.RunWith;
   6 import org.mockito.Mock;
   7 import org.mockito.Spy;
   8 import org.mockito.junit.MockitoJUnitRunner;
   9
  10 import java.io.FileOutputStream;
  11 import java.io.IOException;
  12
  13 import static org.junit.Assert.assertFalse;
  14 import static org.junit.Assert.assertTrue;
  15 import static org.mockito.ArgumentMatchers.any;
  16 import static org.mockito.Mockito.*;
  17
  18 @RunWith(MockitoJUnitRunner.class)
  19 public class XlsOutputHandlerTest {
  20
  21     @Spy
  22     private XlsOutputHandler handler = new XlsOutputHandler(null, "mock");
  23
  24     @Mock
  25     private Workbook workbook;
  26     @Mock
  27     private FileOutputStream xlsFile;
  28
  29     @Test
  30     public void verifyThatFileIsNotCreatedIfNoRecordsAdded() throws IOException {
  31         assertFalse(handler.writeOutputAndCloseFile());
  32         verify(workbook, times(0)).write(any());
  33     }
  34
  35     @Test
  36     public void verifyThatFileIsCreatedIfSomeRecordsAdded() throws IOException {
  37         handler.addRecord("mock");
  38         doReturn(xlsFile).when(handler).getXlsFile();
  39         assertTrue(handler.writeOutputAndCloseFile());
  40     }
  41
  42
  43     private XlsOutputHandler createTestSubject() {
  44         return new XlsOutputHandler("mock", "mockPath", new Object());
  45     }
  46
  47     @Test
  48     public void testInitiate() throws Exception {
  49         XlsOutputHandler testSubject;
  50         Object[] title = new Object[] { null };
  51         // default test
  52         testSubject = createTestSubject();
  53         testSubject.initiate("mock", title);
  54     }
  55
  56     @Test
  57     public void testAddRecord() throws Exception {
  58         XlsOutputHandler testSubject;
  59         Object[] record = new Object[] { null };
  60
  61         // default test
  62         testSubject = createTestSubject();
  63         testSubject.addRecord(record);
  64     }
  65 }