json util class

src/main/java/io/zipcoder/tc_spring_poll_application/controller/ComputeResultController.java

@@ -38,8 +38,7 @@ public class ComputeResultController {
         long voteCount = StreamSupport.stream(allVotes.spliterator(), false).count();
         HashMap<Long, OptionCount> optionCountsMap = new HashMap<>();
 
-        for (Vote v: allVotes
-             ) {
+        for (Vote v: allVotes) {
             Long optionID = (v.getOption().getId());
             if (!optionCountsMap.keySet().contains(optionID)) {
                 optionCountsMap.put(optionID, new OptionCount());

src/test/java/io/zipcoder/tc_spring_poll_application/JsonTestUtilities.java

@@ -0,0 +1,14 @@
+package io.zipcoder.tc_spring_poll_application;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+public class JsonTestUtilities {
+    public static String asJsonString(final Object obj) {
+        try {
+            final ObjectMapper mapper = new ObjectMapper();
+            return mapper.writeValueAsString(obj);
+        } catch (Exception e) {
+            throw new RuntimeException(e);
+        }
+    }
+}

src/test/java/io/zipcoder/tc_spring_poll_application/controller/PollControllerTest.java

@@ -14,6 +14,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
 import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.boot.test.json.JacksonTester;
+import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
 import org.springframework.mock.web.MockHttpServletResponse;
@@ -21,10 +22,14 @@ import org.springframework.test.context.junit4.SpringRunner;
 import org.springframework.test.web.servlet.MockMvc;
 import org.springframework.test.web.servlet.setup.MockMvcBuilders;
 
+import java.net.URI;
 import java.util.Arrays;
 
+import static io.zipcoder.tc_spring_poll_application.JsonTestUtilities.asJsonString;
 import static org.mockito.BDDMockito.given;
+import static org.mockito.Matchers.endsWith;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
 
 @SpringBootTest
 @RunWith(SpringRunner.class)
@@ -103,4 +108,27 @@ public class PollControllerTest {
         Assert.assertEquals(response.getStatus(), HttpStatus.NOT_FOUND.value());
         Assert.assertTrue(response.getContentAsString().isEmpty());
     }
+
+    @Test /* This one is failing */
+    public void testCreatePollSuccess() throws Exception {
+        Poll testPoll2 = new Poll();
+        String expected = "http://localhost:8080/polls/2";
+
+        HttpHeaders responseHeader =  new HttpHeaders();
+        responseHeader.setLocation(URI.create(expected));
+
+        given(pollController.createPoll(testPoll2)).willReturn(null);
+
+        MockHttpServletResponse response = mvc.perform(post("/polls")
+                .content(asJsonString(testPoll2))
+                .contentType(MediaType.APPLICATION_JSON)
+                .accept(MediaType.APPLICATION_JSON))
+                .andReturn().getResponse();
+
+        String header = response.getHeader("Location");
+
+        Assert.assertEquals(HttpStatus.CREATED.value(),(response.getStatus()));
+        Assert.assertEquals(expected, header);
+    }
+
 }