nhu fixed the vote test :)

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

@@ -23,6 +23,8 @@ public class VoteController {
     public ResponseEntity<?> createVote(@PathVariable Long pollId, @RequestBody Vote
             vote) {
         vote = voteRepository.save(vote);
+
+        System.out.println("votes is " + vote);
         // Set the headers for the newly created resource
         HttpHeaders responseHeaders = new HttpHeaders();
         responseHeaders.setLocation(ServletUriComponentsBuilder.

src/main/java/io/zipcoder/tc_spring_poll_application/domain/Option.java

@@ -1,6 +1,7 @@
 package io.zipcoder.tc_spring_poll_application.domain;
 
 import javax.persistence.*;
+import java.util.Objects;
 
 @Entity
 public class Option {
@@ -32,4 +33,18 @@ public class Option {
         return "{\"id\":" + id
                 + ",\"value\":\"" + value + "\"}";
     }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof Option)) return false;
+        Option option = (Option) o;
+        return Objects.equals(getId(), option.getId()) &&
+                Objects.equals(getValue(), option.getValue());
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(getId(), getValue());
+    }
 }

src/main/java/io/zipcoder/tc_spring_poll_application/domain/Poll.java

@@ -4,6 +4,7 @@ import org.hibernate.validator.constraints.NotEmpty;
 
 import javax.persistence.*;
 import javax.validation.constraints.Size;
+import java.util.Objects;
 import java.util.Set;
 
 @Entity
@@ -55,4 +56,19 @@ public class Poll {
                 + ",\"question\":\"" + question + "\""
                 + ",\"options\":" + options + "}";
     }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof Poll)) return false;
+        Poll poll = (Poll) o;
+        return Objects.equals(getId(), poll.getId()) &&
+                Objects.equals(getQuestion(), poll.getQuestion()) &&
+                Objects.equals(getOptions(), poll.getOptions());
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(getId(), getQuestion(), getOptions());
+    }
 }

src/main/java/io/zipcoder/tc_spring_poll_application/domain/Vote.java

@@ -1,6 +1,7 @@
 package io.zipcoder.tc_spring_poll_application.domain;
 
 import javax.persistence.*;
+import java.util.Objects;
 
 @Entity
 public class Vote {
@@ -34,4 +35,18 @@ public class Vote {
         return "{\"id\":" + id
                 + ",\"option\":" + option + "}";
     }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof Vote)) return false;
+        Vote vote = (Vote) o;
+        return Objects.equals(getId(), vote.getId()) &&
+                Objects.equals(getOption(), vote.getOption());
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(getId(), getOption());
+    }
 }

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

@@ -155,13 +155,12 @@ public class PollControllerTest {
     public void testCreatePollSuccess() throws Exception {
         given(pollRepository.save(testPoll2)).willReturn(testPoll2);
 
-        MockHttpServletResponse response = mvc.perform(post("/polls/")
+        mvc.perform(post("/polls/")
                 .content(asJsonString(testPoll2))
                 .contentType(MediaType.APPLICATION_JSON)
                 .accept(MediaType.APPLICATION_JSON))
-                .andReturn().getResponse();
+                .andExpect(status().isCreated());
 
-        Assert.assertEquals(HttpStatus.CREATED.value(),(response.getStatus()));
     }
 
     // =================== Update Poll =================== //

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

@@ -75,6 +75,8 @@ public class VoteControllerTest {
 
         testPoll = new Poll();
         testPoll.setId(TEST_ID);
+
+        testPoll.setQuestion("Do you like cheese?");
         options1 = new HashSet<>();
 
         option1 = new Option();
@@ -126,7 +128,7 @@ public class VoteControllerTest {
                 .willThrow(new ResourceNotFoundException());
 
         MockHttpServletResponse response = mvc.perform(
-                get("polls/votesBADBAD")
+                get("/polls/votesBADBAD")
                     .accept(MediaType.APPLICATION_JSON))
                     .andReturn().getResponse();
 
@@ -138,11 +140,17 @@ public class VoteControllerTest {
     public void testCreateVoteSuccess() throws Exception {
         Vote testVote2 = new Vote();
         testVote2.setId(TEST_ID);
+        testVote2.setOption(option1);
+
+        /* Fixed by adding an equals method and hashcode to each  object.
+         * during mocking the 'save' method is checking the absolute equality
+          * and since it's technically a different object, we have to override the equals method
+          * to properly compare the objects. */
 
         given(pollRepository.findOne(TEST_ID)).willReturn(testPoll);
         given(voteRepository.save(testVote2)).willReturn(testVote2);
 
-        mvc.perform(post("/polls/1/votes/1")
+        mvc.perform(post("/polls/1/votes")
                 .content(asJsonString(testVote2))
                 .contentType(MediaType.APPLICATION_JSON)
                 .accept(MediaType.APPLICATION_JSON))