Validate Poll objects

src/main/java/io/zipcoder/tc_spring_poll_application/controllers/PollController.java

@@ -10,6 +10,7 @@ import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.servlet.support.ServletUriComponentsBuilder;
 
+import javax.validation.Valid;
 import java.net.URI;
 import java.util.Optional;
 
@@ -37,7 +38,7 @@ public class PollController {
     }
 
     @RequestMapping(value = "/polls", method = RequestMethod.POST)
-    public ResponseEntity<Poll> createPoll(@RequestBody Poll poll) {
+    public ResponseEntity<Poll> createPoll(@RequestBody @Valid Poll poll) {
         Poll createdPoll = pollRepository.save(poll);
 
         URI newPollUri;
@@ -69,7 +70,7 @@ public class PollController {
     }
 
     @RequestMapping(value = "/polls/{pollId}", method = RequestMethod.PUT)
-    public ResponseEntity<Poll> updatePoll(@RequestBody Poll poll, @PathVariable Long pollId) {
+    public ResponseEntity<Poll> updatePoll(@RequestBody @Valid Poll poll, @PathVariable Long pollId) {
         verifyPoll(pollId);
         return new ResponseEntity<>(pollRepository.save(poll), HttpStatus.OK);
     }

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

@@ -1,6 +1,8 @@
 package io.zipcoder.tc_spring_poll_application.domain;
 
 import javax.persistence.*;
+import javax.validation.constraints.NotEmpty;
+import javax.validation.constraints.Size;
 import java.util.Set;
 
 /**
@@ -12,17 +14,20 @@ import java.util.Set;
 @Entity
 public class Poll {
 
+
     @Id
     @GeneratedValue
     @Column(name = "POLL_ID")
     private Long Id;
 
     @Column(name = "QUESTION")
+    @NotEmpty
     private String question;
 
     @OneToMany(cascade = CascadeType.ALL)
     @JoinColumn(name = "POLL_ID")
     @OrderBy
+    @Size(min=2, max = 6)
     private Set<Option> options;
 
     public Poll() {

src/main/java/io/zipcoder/tc_spring_poll_application/error/RestExceptionHandler.java

@@ -0,0 +1,32 @@
+package io.zipcoder.tc_spring_poll_application.error;
+
+import io.zipcoder.tc_spring_poll_application.exception.ResourceNotFoundException;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.ControllerAdvice;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+
+import javax.servlet.http.HttpServletRequest;
+import java.util.Arrays;
+import java.util.Date;
+
+/**
+ * project: spring-demo
+ * package: io.zipcoder.tc_spring_poll_application.error
+ * author: https://github.com/vvmk
+ * date: 4/7/18
+ */
+@ControllerAdvice
+public class RestExceptionHandler {
+    @ExceptionHandler(ResourceNotFoundException.class)
+    public ResponseEntity<?> handleResourceNotFoundException(ResourceNotFoundException rnfe,
+                                                             HttpServletRequest request) {
+        ErrorDetail err = new ErrorDetail();
+        err.setTimestamp(new Date().getTime());
+        err.setStatus(HttpStatus.NOT_FOUND.value());
+        err.setDetail(rnfe.getMessage());
+        err.setDeveloperMessage(Arrays.toString(rnfe.getStackTrace()));
+        err.setTitle(rnfe.getClass().toString());
+        return new ResponseEntity<>(err, HttpStatus.NOT_FOUND);
+    }
+}

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

@@ -143,8 +143,6 @@ public class PollControllerTest {
 
     @Test
     public void testVerifyPoll() {
-
-
         try {
             pollCtrl.verifyPoll(exists);
         } catch (ResourceNotFoundException rnfe) {