TimeLine Class tests

src/main/java/data/TimelineData.java → src/main/java/scot/carricksoftware/grantswriter/data/TimelineData.java

@@ -3,7 +3,7 @@
  *
  */
 
-package data;
+package scot.carricksoftware.grantswriter.data;
 
 import scot.carricksoftware.grantswriter.domains.CensusEntry;
 

src/main/java/data/TimelineDataImpl.java → src/main/java/scot/carricksoftware/grantswriter/data/TimelineDataImpl.java

@@ -3,7 +3,7 @@
  *
  */
 
-package data;
+package scot.carricksoftware.grantswriter.data;
 
 import org.springframework.stereotype.Component;
 import scot.carricksoftware.grantswriter.domains.CensusEntry;
@@ -11,6 +11,8 @@ import scot.carricksoftware.grantswriter.domains.CensusEntry;
 import java.util.List;
 import java.util.SortedMap;
 import java.util.SortedSet;
+import java.util.TreeMap;
+import java.util.TreeSet;
 
 @Component
 public class TimelineDataImpl implements TimelineData {
@@ -19,6 +21,11 @@ public class TimelineDataImpl implements TimelineData {
 
     private SortedSet<String> refs;
 
+    public TimelineDataImpl() {
+        this.timeline = new TreeMap<>();
+        this.refs = new TreeSet<>();
+    }
+
     @Override
     public void clear() {
         timeline.clear();

src/test/java/data/TimelineDataTest.java

@@ -1,28 +0,0 @@
-/*
- * Copyright (c) 2025.  Andrew Grant Carrick Software. All rights reserved
- *
- */
-
-package data;
-
-import org.junit.jupiter.api.BeforeEach;
-import org.junit.jupiter.api.Test;
-
-import static org.junit.jupiter.api.Assertions.*;
-
-class TimelineDataTest {
-
-    private TimelineData timelineData;
-
-    @BeforeEach
-    void setUp() {
-        timelineData = new TimelineDataImpl();
-    }
-
-    @Test
-    void constructorTest() {
-        assertNotNull(timelineData);
-    }
-
-
-}

src/test/java/scot/carricksoftware/grantswriter/data/TimelineDataTest.java

@@ -0,0 +1,56 @@
+/*
+ * Copyright (c) 2025.  Andrew Grant Carrick Software. All rights reserved
+ *
+ */
+
+package scot.carricksoftware.grantswriter.data;
+
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+
+import java.util.SortedMap;
+import java.util.SortedSet;
+import java.util.TreeMap;
+import java.util.TreeSet;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static scot.carricksoftware.grantswriter.GenerateRandomNumberValues.GetRandomString;
+
+class TimelineDataTest {
+
+    private TimelineData timelineData;
+
+    @BeforeEach
+    void setUp() {
+        timelineData = new TimelineDataImpl();
+    }
+
+    @Test
+    void getRefsTest() {
+        assertNotNull(timelineData.getRefs());
+    }
+
+    @Test
+    void setRefsTest() {
+        SortedSet<String> refs = new TreeSet<>();
+        refs.add(GetRandomString());
+        timelineData.setRefs(refs);
+        assertEquals(refs, timelineData.getRefs());
+    }
+
+    @Test
+    void getTimelineTest() {
+        assertEquals(0, timelineData.getTimeline().size());
+    }
+
+    @Test
+    void setTimelineTest() {
+        SortedMap<String, String> timeline   = new TreeMap<>();
+        timeline.put(GetRandomString(), GetRandomString());
+        timelineData.setTimeline(timeline);
+        assertEquals(timeline, timelineData.getTimeline());
+    }
+
+
+}