Delete a post

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

@@ -29,13 +29,13 @@ public class PollController {
         this.pollRepository = pollRepository;
     }
 
-    @RequestMapping(name = "/polls", method = RequestMethod.GET)
+    @RequestMapping(value = "/polls", method = RequestMethod.GET)
     public ResponseEntity<Iterable<Poll>> getAllPolls() {
         Iterable<Poll> allPolls = pollRepository.findAll();
         return new ResponseEntity<>(allPolls, HttpStatus.OK);
     }
 
-    @RequestMapping(name = "/polls", method = RequestMethod.POST)
+    @RequestMapping(value = "/polls", method = RequestMethod.POST)
     public ResponseEntity<Poll> createPoll(@RequestBody Poll poll) {
         Poll createdPoll = pollRepository.save(poll);
 
@@ -61,9 +61,24 @@ public class PollController {
         return new ResponseEntity<>(createdPoll, headers, HttpStatus.CREATED);
     }
 
-    @RequestMapping(name = "/polls/{pollId}", method = RequestMethod.GET)
+    @RequestMapping(value = "/polls/{pollId}", method = RequestMethod.GET)
     public ResponseEntity<Poll> getPoll(@PathVariable Long pollId) {
         Optional<Poll> poll = pollRepository.findById(pollId);
         return new ResponseEntity<>(poll.orElse(new Poll()), HttpStatus.OK);
     }
+
+    @RequestMapping(value = "/polls/{pollId}", method = RequestMethod.PUT)
+    public ResponseEntity<Poll> updatePoll(@RequestBody Poll poll, @PathVariable Long pollId) {
+        //TODO: check ids
+        //if (pollId != poll.getId())
+        //return BAD_REQUEST
+
+        return new ResponseEntity<>(pollRepository.save(poll), HttpStatus.OK);
+    }
+
+    @RequestMapping(value = "/polls/{pollId}", method = RequestMethod.DELETE)
+    public ResponseEntity<?> deletePoll(@PathVariable Long pollId) {
+        pollRepository.deleteById(pollId);
+        return new ResponseEntity<>(null, HttpStatus.OK);
+    }
 }

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

@@ -47,7 +47,8 @@ public class PollControllerTest {
         // GET one poll
         when(pollRepo.findById(anyLong())).thenReturn(Optional.of(new Poll()));
 
-        //POST create poll
+        // POST create poll
+        // PUT update poll
         when(pollRepo.save(any(Poll.class))).thenAnswer(inv -> inv.getArgument(0));
 
 //        when(pollRepo::save)
@@ -106,4 +107,22 @@ public class PollControllerTest {
         verify(pollRepo).findById(anyLong());
         assertEquals(expected, actual);
     }
+
+    @Test
+    public void updatesPoll() {
+        HttpStatus expected = HttpStatus.OK;
+        HttpStatus actual = pollCtrl.updatePoll(mock(Poll.class), anyLong()).getStatusCode();
+
+        verify(pollRepo).save(any(Poll.class));
+        assertEquals(expected, actual);
+    }
+
+    @Test
+    public void deletePoll() {
+        HttpStatus expected = HttpStatus.OK;
+        HttpStatus actual = pollCtrl.deletePoll(anyLong()).getStatusCode();
+
+        verify(pollRepo).deleteById(anyLong());
+        assertEquals(expected, actual);
+    }
 }