Total Rooms added to census command

src/main/java/scot/carricksoftware/grants/commands/census/CensusCommand.java

@@ -44,4 +44,8 @@ public interface CensusCommand {
     String getFilledInBy();
 
     void setFilledInBy(String filledInBy);
+
+    String getTotalRooms();
+
+    void setTotalRooms(String totalRooms);
 }

src/main/java/scot/carricksoftware/grants/commands/census/CensusCommandImpl.java

@@ -31,6 +31,8 @@ public class CensusCommandImpl implements CensusCommand {
 
     private String filledInBy;
 
+    private String totalRooms;
+
     @Override
     public Long getId() {
         return id;
@@ -110,4 +112,14 @@ public class CensusCommandImpl implements CensusCommand {
     public void setFilledInBy(String filledInBy) {
         this.filledInBy = filledInBy;
     }
+
+    @Override
+    public String getTotalRooms() {
+        return totalRooms;
+    }
+
+    @Override
+    public void setTotalRooms(String totalRooms) {
+        this.totalRooms = totalRooms;
+    }
 }

src/main/java/scot/carricksoftware/grants/domains/census/Census.java

@@ -5,7 +5,15 @@
 
 package scot.carricksoftware.grants.domains.census;
 
-import jakarta.persistence.*;
+
+import jakarta.persistence.CascadeType;
+import jakarta.persistence.Column;
+import jakarta.persistence.Entity;
+import jakarta.persistence.EnumType;
+import jakarta.persistence.Enumerated;
+import jakarta.persistence.JoinColumn;
+import jakarta.persistence.ManyToOne;
+import jakarta.persistence.OneToMany;
 import scot.carricksoftware.grants.BaseEntity;
 import scot.carricksoftware.grants.domains.places.Place;
 import scot.carricksoftware.grants.enums.census.CensusBoundaryType;

src/test/java/scot/carricksoftware/grants/commands/census/CensusCommandTest.java

@@ -89,5 +89,18 @@ class CensusCommandTest {
         assertEquals(filledInBy, command.getFilledInBy());
     }
 
+    @Test
+    public void getTotalRoomsTest() {
+        assertNull(command.getTotalRooms());
+    }
+
+    @Test
+    public void setTotalRoomsTest() {
+        String totalRooms = GetRandomString();
+        command.setTotalRooms(totalRooms);
+        assertEquals(totalRooms, command.getTotalRooms());
+    }
+
+
 
 }

src/test/java/scot/carricksoftware/grants/domains/census/CensusTest.java

@@ -105,8 +105,8 @@ public class CensusTest {
     @Test
     public void setTotalRoomsTest() {
         String totalRooms = GetRandomString();
-        census.setFilledInBy(totalRooms);
-        assertEquals(totalRooms, census.getFilledInBy());
+        census.setTotalRooms(totalRooms);
+        assertEquals(totalRooms, census.getTotalRooms());
     }