AppendixPartWriterTest

src/main/java/scot/carricksoftware/grantswriter/writer/latex/parts/appendix/AppendixPartWriterImpl.java

@@ -8,23 +8,22 @@ package scot.carricksoftware.grantswriter.writer.latex.parts.appendix;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.springframework.stereotype.Component;
-import scot.carricksoftware.grantswriter.writer.latex.LatexPartHeader;
+import scot.carricksoftware.grantswriter.writer.latex.parts.appendix.headers.AppendixPartHeader;
 
 @Component
 public class AppendixPartWriterImpl implements AppendixPartWriter {
 
     private static final Logger logger = LogManager.getLogger(AppendixPartWriterImpl.class);
 
-    private final LatexPartHeader latexPartHeader;
+    private final AppendixPartHeader appendixPartHeader;
 
-    public AppendixPartWriterImpl(LatexPartHeader latexPartHeader) {
-        this.latexPartHeader = latexPartHeader;
+    public AppendixPartWriterImpl(AppendixPartHeader appendixPartHeader) {
+        this.appendixPartHeader = appendixPartHeader;
     }
 
-
     @Override
     public void write() {
         logger.info("AppendixPartsImpl::write()");
-        latexPartHeader.write("Appendix");
+        appendixPartHeader.write();
     }
 }

src/test/java/scot/carricksoftware/grantswriter/writer/latex/parts/appendix/AppendixPartWriterTest.java

@@ -10,9 +10,9 @@ import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.mockito.Mock;
 import org.mockito.junit.jupiter.MockitoExtension;
-import scot.carricksoftware.grantswriter.writer.latex.LatexPartHeader;
+import scot.carricksoftware.grantswriter.writer.latex.parts.appendix.headers.AppendixPartHeader;
 
-import static org.junit.jupiter.api.Assertions.*;
+import static org.mockito.Mockito.verify;
 
 @ExtendWith(MockitoExtension.class)
 class AppendixPartWriterTest {
@@ -20,15 +20,16 @@ class AppendixPartWriterTest {
     private AppendixPartWriter writer;
 
     @Mock
-    private LatexPartHeader latexPartHeaderMock;
+    private AppendixPartHeader appendixPartHeaderMock;
 
     @BeforeEach
     void setUp() {
-        writer = new AppendixPartWriterImpl(latexPartHeaderMock);
+        writer = new AppendixPartWriterImpl(appendixPartHeaderMock);
     }
 
     @Test
-    void constructorTest(){
-        assertNotNull(writer);
+    void partHeaderIsCalled(){
+        writer.write();
+        verify(appendixPartHeaderMock).write();
     }
 }