Create ResourceNotFoundException and verifyPolls

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

@@ -1,6 +1,7 @@
 package io.zipcoder.tc_spring_poll_application.controllers;
 
 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 org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.HttpHeaders;
@@ -81,4 +82,9 @@ public class PollController {
         pollRepository.deleteById(pollId);
         return new ResponseEntity<>(null, HttpStatus.OK);
     }
+
+    protected void verifyPoll(Long pollId) throws ResourceNotFoundException {
+        if (!pollRepository.existsById(pollId))
+            throw new ResourceNotFoundException();
+    }
 }

src/main/java/io/zipcoder/tc_spring_poll_application/exception/ResourceNotFoundException.java

@@ -0,0 +1,25 @@
+package io.zipcoder.tc_spring_poll_application.exception;
+
+import org.springframework.http.HttpStatus;
+import org.springframework.web.bind.annotation.ResponseStatus;
+
+/**
+ * project: spring-demo
+ * package: io.zipcoder.tc_spring_poll_application.exception
+ * author: https://github.com/vvmk
+ * date: 4/7/18
+ */
+@ResponseStatus(HttpStatus.NOT_FOUND)
+public class ResourceNotFoundException extends RuntimeException {
+    public ResourceNotFoundException() {
+        super();
+    }
+
+    public ResourceNotFoundException(String message) {
+        super(message);
+    }
+
+    public ResourceNotFoundException(String message, Throwable throwable) {
+        super(message, throwable);
+    }
+}

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

@@ -1,6 +1,7 @@
 package io.zipcoder.tc_spring_poll_application.controllers;
 
 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 org.junit.Before;
 import org.junit.Test;
@@ -15,6 +16,7 @@ import java.util.List;
 import java.util.Optional;
 import java.util.regex.Pattern;
 
+import static junit.framework.TestCase.fail;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;
 import static org.mockito.ArgumentMatchers.any;
@@ -51,6 +53,9 @@ public class PollControllerTest {
         // PUT update poll
         when(pollRepo.save(any(Poll.class))).thenAnswer(inv -> inv.getArgument(0));
 
+        when(pollRepo.existsById(10L)).thenReturn(true);
+        when(pollRepo.existsById(999L)).thenReturn(false);
+
 //        when(pollRepo::save)
 //                .onSuccess(HttpStatus.CREATED)
 //                .onFail(HttpStatus.BAD_REQUEST);
@@ -125,4 +130,21 @@ public class PollControllerTest {
         verify(pollRepo).deleteById(anyLong());
         assertEquals(expected, actual);
     }
+
+    @Test
+    public void testVerifyPoll() {
+
+        Long exists = 10L;
+
+        try {
+            pollCtrl.verifyPoll(exists);
+        } catch (ResourceNotFoundException rnfe) {
+            fail();
+        }
+    }
+
+    @Test(expected = ResourceNotFoundException.class)
+    public void testVerifyPollNotFound() {
+        pollCtrl.verifyPoll(999L);
+    }
 }