CensusTest

src/test/java/scot/carricksoftware/grantswriter/domains/census/CensusPartOneTest.java → src/test/java/scot/carricksoftware/grantswriter/domains/census/CensusTest.java

@@ -21,7 +21,7 @@ import static scot.carricksoftware.grantswriter.GenerateRandomNumberValues.GetRa
 
 
 @ExtendWith(MockitoExtension.class)
-public class CensusPartOneTest {
+public class CensusTest {
 
     private Census census;
 
@@ -46,6 +46,18 @@ public class CensusPartOneTest {
         assertEquals(id, census.getId());
     }
 
+    @Test
+    public void getCensusDateTest() {
+        assertNull(census.getCensusDate());
+    }
+
+    @Test
+    public void setCensusDateTest() {
+        CensusDate censusDate = CensusDate.CENSUS_1861;
+        census.setCensusDate(censusDate);
+        assertEquals(censusDate, census.getCensusDate());
+    }
+
     @Test
     public void toStringTest() {
         CensusDate censusDate = GetRandomCensusDate();