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


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

package org.openecomp.sdc.asdctool.migration.tasks.handlers;

import org.apache.poi.ss.usermodel.Workbook;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.Spy;
import org.mockito.junit.MockitoJUnitRunner;

import java.io.FileOutputStream;
import java.io.IOException;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.*;

@RunWith(MockitoJUnitRunner.class)
public class XlsOutputHandlerTest {

    @Spy
    private XlsOutputHandler handler = new XlsOutputHandler(null, "mock");

    @Mock
    private Workbook workbook;
    @Mock
    private FileOutputStream xlsFile;

    @Test
    public void verifyThatFileIsNotCreatedIfNoRecordsAdded() throws IOException {
        assertFalse(handler.writeOutputAndCloseFile());
        verify(workbook, times(0)).write(any());
    }

    @Test
    public void verifyThatFileIsCreatedIfSomeRecordsAdded() throws IOException {
        handler.addRecord("mock");
        doReturn(xlsFile).when(handler).getXlsFile();
        assertTrue(handler.writeOutputAndCloseFile());
    }
    
    
    private XlsOutputHandler createTestSubject() {
	return new XlsOutputHandler("mock", "mockPath", new Object());
    }

    @Test
    public void testInitiate() throws Exception {
	XlsOutputHandler testSubject;
	Object[] title = new Object[] { null };
	// default test
	testSubject = createTestSubject();
	testSubject.initiate("mock", title);
    }

    @Test
    public void testAddRecord() throws Exception {
	XlsOutputHandler testSubject;
	Object[] record = new Object[] { null };

	// default test
	testSubject = createTestSubject();
	testSubject.addRecord(record);
    }
}