PdfFactoryTest

PdfFactoryTest.java

package com.trailblazers.freewheelers.model;

import com.itextpdf.text.Document;
import static org.mockito.Mockito.verify;
import com.itextpdf.text.pdf.PdfWriter;
import org.junit.Test;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

public class PdfFactoryTest {


    @Test
    public void shouldGetPdfInstance() throws Exception {

       //Given
       PdfFactory pdfFactory = mock(PdfFactory.class);
       Document mockPdf = mock(Document.class);
       ReserveOrder mockReserveOrder = mock(ReserveOrder.class);
       PdfWriter mockPdfWriter = mock(PdfWriter.class);

       when(pdfFactory.getPdfWriter()).thenReturn(mockPdfWriter);

        //When
        pdfFactory.create(mockPdf, mockReserveOrder);

        //Then
        verify(pdfFactory.getPdfWriter());
    }
}