testing

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

@@ -5,27 +5,20 @@ import io.zipcoder.tc_spring_poll_application.domain.Option;
 import io.zipcoder.tc_spring_poll_application.domain.Poll;
 import io.zipcoder.tc_spring_poll_application.exception.ResourceNotFoundException;
 import io.zipcoder.tc_spring_poll_application.repositories.PollRepository;
-import io.zipcoder.tc_spring_poll_application.repositories.VoteRepository;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.InjectMocks;
-import org.mockito.Mock;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
 import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest;
-import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.boot.test.json.JacksonTester;
 import org.springframework.boot.test.mock.mockito.MockBean;
-import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
-import org.springframework.http.ResponseEntity;
 import org.springframework.mock.web.MockHttpServletResponse;
 import org.springframework.test.context.junit4.SpringRunner;
 import org.springframework.test.web.servlet.MockMvc;
-import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;
 import org.springframework.test.web.servlet.setup.MockMvcBuilders;
 
 import java.util.Arrays;

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

@@ -1,6 +1,7 @@
 package io.zipcoder.tc_spring_poll_application.controller;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
+import io.zipcoder.tc_spring_poll_application.domain.Option;
 import io.zipcoder.tc_spring_poll_application.domain.Poll;
 import io.zipcoder.tc_spring_poll_application.domain.Vote;
 import io.zipcoder.tc_spring_poll_application.exception.ResourceNotFoundException;
@@ -15,21 +16,21 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest;
 import org.springframework.boot.test.json.JacksonTester;
 import org.springframework.boot.test.mock.mockito.MockBean;
-import org.springframework.http.HttpHeaders;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.MediaType;
 import org.springframework.mock.web.MockHttpServletResponse;
 import org.springframework.test.context.junit4.SpringRunner;
 import org.springframework.test.web.servlet.MockMvc;
 import org.springframework.test.web.servlet.setup.MockMvcBuilders;
-import org.springframework.web.servlet.support.ServletUriComponentsBuilder;
 
 import java.util.Arrays;
+import java.util.HashSet;
 
 import static io.zipcoder.tc_spring_poll_application.JsonTestUtilities.asJsonString;
 import static org.mockito.BDDMockito.given;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
 import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
 
 @RunWith(SpringRunner.class)
 @WebMvcTest(VoteController.class)
@@ -41,7 +42,8 @@ public class VoteControllerTest {
     private Vote testVote;
     private Iterable<Vote> votes;
     private final Long TEST_ID = 1L;
-    private final Long BAD_ID = 400L;
+    private Option option1, option2;
+    private HashSet<Option> options1;
 
     @MockBean
     private VoteRepository voteRepository;
@@ -73,6 +75,20 @@ public class VoteControllerTest {
 
         testPoll = new Poll();
         testPoll.setId(TEST_ID);
+        options1 = new HashSet<>();
+
+        option1 = new Option();
+        option1.setValue("Yes");
+        option1.setId(1L);
+
+        option2 = new Option();
+        option2.setValue("No");
+        option2.setId(2L);
+
+        options1.add(option1);
+        options1.add(option2);
+
+        testPoll.setOptions(options1);
     }
 
     @Test
@@ -121,16 +137,15 @@ public class VoteControllerTest {
     @Test
     public void testCreateVoteSuccess() throws Exception {
         Vote testVote2 = new Vote();
+        testVote2.setId(TEST_ID);
 
-        given(voteRepository.save(testVote2))
-                .willReturn(testVote2);
+        given(pollRepository.findOne(TEST_ID)).willReturn(testPoll);
+        given(voteRepository.save(testVote2)).willReturn(testVote2);
 
-        MockHttpServletResponse response = mvc.perform(post("/polls/2/votes")
+        mvc.perform(post("/polls/1/votes/1")
+                .content(asJsonString(testVote2))
                 .contentType(MediaType.APPLICATION_JSON)
                 .accept(MediaType.APPLICATION_JSON))
-                .andReturn().getResponse();
-
-
-        Assert.assertEquals(response.getStatus(), HttpStatus.CREATED.value());
+                .andExpect(status().isCreated());
     }
 }