Merge pull request #3 from RIT-SWEN-261-02/update-need

Update need

1 files changed, 15 insertions, 2 deletions

diff --git a/ufund-api/src/main/java/com/ufund/api/ufundapi/controller/CupboardController.java b/ufund-api/src/main/java/com/ufund/api/ufundapi/controller/CupboardController.java
index 106b2e0..8d394c1 100644
--- a/ufund-api/src/main/java/com/ufund/api/ufundapi/controller/CupboardController.java
+++ b/ufund-api/src/main/java/com/ufund/api/ufundapi/controller/CupboardController.java
@@ -40,9 +40,22 @@ public class CupboardController {
         return cupboard.getNeed(id);
     }
 
+
+    /**
+     * Updates a Need with the provided one
+     *
+     * @param need The need to update
+     * @return OK response and the need if it was successful, or INTERNAL_SERVER_ERROR if there was an issue
+     */
+
     @PutMapping("")
-    public void updateNeed(@RequestBody Need need) {
-        cupboard.updateNeed(need);
+    public ResponseEntity<Need> updateNeed(@RequestBody Need need) {
+        try {
+            need = cupboard.updateNeed(need);
+            return new ResponseEntity<>(need, HttpStatus.OK);
+        } catch (IOException e) {
+            return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR);
+        }
     }
 
     @DeleteMapping("/{id}")