CapitaliseCensusEntryTest

src/test/java/scot/carricksoftware/grants/capitalisation/census/census/CapitaliseCensusTest.java

@@ -38,6 +38,5 @@ class CapitaliseCensusTest {
         censusCommand.setFilledInBy("lower");
         capitaliseCensus.capitalise(censusCommand);
         assertEquals("Lower", censusCommand.getFilledInBy());
-
     }
 }

src/test/java/scot/carricksoftware/grants/capitalisation/census/censusentry/CapitaliseCensusEntryTest.java

@@ -6,11 +6,48 @@
 package scot.carricksoftware.grants.capitalisation.census.censusentry;
 
 
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.Mock;
 import org.mockito.junit.jupiter.MockitoExtension;
+import scot.carricksoftware.grants.capitalisation.CapitaliseStringImpl;
+import scot.carricksoftware.grants.commands.census.CensusEntryCommand;
+import scot.carricksoftware.grants.commands.census.CensusEntryCommandImpl;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.when;
 
 
 @ExtendWith(MockitoExtension.class)
 class CapitaliseCensusEntryTest {
 
+    private CapitaliseCensusEntry capitaliseCensusEntry;
+
+    @Mock
+    private CapitaliseStringImpl capitaliseStringMock;
+
+    @BeforeEach
+    void setUp() {
+        capitaliseCensusEntry = new CapitaliseCensusEntryImpl(capitaliseStringMock);
+    }
+
+    @Test
+    void nameTest() {
+        CensusEntryCommand censusEntryCommand = new CensusEntryCommandImpl();
+        when(capitaliseStringMock.capitalise(any())).thenReturn("Lower");
+        censusEntryCommand.setName("lower");
+        capitaliseCensusEntry.capitalise(censusEntryCommand);
+        assertEquals("Lower", censusEntryCommand.getName());
+    }
+
+    @Test
+    void industryOrServiceTest() {
+        CensusEntryCommand censusEntryCommand = new CensusEntryCommandImpl();
+        when(capitaliseStringMock.capitalise(any())).thenReturn("Lower");
+        censusEntryCommand.setIndustryOrService("lower");
+        capitaliseCensusEntry.capitalise(censusEntryCommand);
+        assertEquals("Lower", censusEntryCommand.getIndustryOrService());
+    }
 }