WriteReferencesTest

src/test/java/scot/carricksoftware/grantswriter/writer/latex/parts/people/subsections/helpers/WriteReferencesTest.java

@@ -8,6 +8,7 @@ package scot.carricksoftware.grantswriter.writer.latex.parts.people.subsections.
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.InOrder;
 import org.mockito.Mock;
 import org.mockito.junit.jupiter.MockitoExtension;
 import scot.carricksoftware.grantswriter.data.TimeLineData;
@@ -15,7 +16,13 @@ import scot.carricksoftware.grantswriter.writer.FileWriter;
 import scot.carricksoftware.grantswriter.writer.latex.LatexLongTabLeEnd;
 import scot.carricksoftware.grantswriter.writer.latex.LatexLongTableStart;
 
-import static org.junit.jupiter.api.Assertions.assertNotNull;
+import java.util.SortedSet;
+import java.util.TreeSet;
+
+import static org.mockito.Mockito.inOrder;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+import static scot.carricksoftware.grantswriter.GenerateCertificateRandomValues.GetRandomString;
 
 @ExtendWith(MockitoExtension.class)
 class WriteReferencesTest {
@@ -45,8 +52,25 @@ class WriteReferencesTest {
     }
 
     @Test
-    void constructorTest() {
-        assertNotNull(writeReferences);
+    void writeTest() {
+        writeReferences.write();
+        InOrder inOrder = inOrder(latexLongTableStartMock, latexLongTabLeEndMock);
+        inOrder.verify(latexLongTableStartMock).write("l");
+        inOrder.verify(latexLongTabLeEndMock).write();
+    }
+
+    @Test
+    void writeTheDataTest() {
+        String event = GetRandomString();
+
+        SortedSet<String> references = new TreeSet<>();
+        references.add(event);
+
+        when(timeLineDataMock.getRefs()).thenReturn(references);
+        writeReferences.write();
+
+        verify(fileWriterMock).writeLine(event + "\\\\");
     }
 
+
 }

src/test/java/scot/carricksoftware/grantswriter/writer/latex/parts/people/subsections/helpers/WriteTimeLineTest.java

@@ -75,7 +75,7 @@ class WriteTimeLineTest {
         when(timeLineDataMock.getTimeLine()).thenReturn(map);
         writeTimeLine.write();
 
-        verify(fileWriterMock).writeLine(dmy + "&" + event + "\\\\" );
+        verify(fileWriterMock).writeLine(dmy + "&" + event + "\\\\");
     }