LatexDocumentEndTest

src/test/java/scot/carricksoftware/grantswriter/writer/latex/LatexDocumentEndTest.java

@@ -13,6 +13,7 @@ import org.springframework.test.context.junit.jupiter.SpringExtension;
 import scot.carricksoftware.grantswriter.writer.FileWriter;
 
 import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.mockito.Mockito.verify;
 
 @ExtendWith(SpringExtension.class)
 class LatexDocumentEndTest {
@@ -31,4 +32,10 @@ class LatexDocumentEndTest {
     public void constructorTest() {
         assertNotNull(documentEnd);
     }
+
+    @Test
+    public void writeTest() {
+        documentEnd.write();
+        verify(fileWriterMock).writeLine("\\end{document}");
+    }
 }

src/test/java/scot/carricksoftware/grantswriter/writer/latex/LatexDocumentStartTest.java

@@ -13,6 +13,7 @@ import org.springframework.test.context.junit.jupiter.SpringExtension;
 import scot.carricksoftware.grantswriter.writer.FileWriter;
 
 import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.mockito.Mockito.verify;
 
 @ExtendWith(SpringExtension.class)
 class LatexDocumentStartTest {
@@ -31,4 +32,10 @@ class LatexDocumentStartTest {
     public void constructorTest() {
         assertNotNull(documentStart);
     }
+
+    @Test
+    public void writeTest() {
+        documentStart.write();
+        verify(fileWriterMock).writeLine("\\begin{document}");
+    }
 }