From ca2f67cb2565c761984589bb089c5954612a77ae Mon Sep 17 00:00:00 2001
From: Erik Foris <erikfor@outlook.de>
Date: Tue, 25 Jun 2024 16:12:31 +0200
Subject: [PATCH] fix: closes #26

---
 .../group3/DataDash/controler/DatasetController.java     | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/main/java/de/uni_passau/fim/PADAS/group3/DataDash/controler/DatasetController.java b/src/main/java/de/uni_passau/fim/PADAS/group3/DataDash/controler/DatasetController.java
index 1fc3c2e..a9e5126 100644
--- a/src/main/java/de/uni_passau/fim/PADAS/group3/DataDash/controler/DatasetController.java
+++ b/src/main/java/de/uni_passau/fim/PADAS/group3/DataDash/controler/DatasetController.java
@@ -54,23 +54,22 @@ public class DatasetController {
     @PutMapping("/id/{id}/upvote")
     public Dataset upvote(@PathVariable("id") UUID id) {
         datasetService.upvoteDataset(id);
-        return null;
+        return datasetService.getDatasetById(id);
     }
 
     @PutMapping("/id/{id}/downvote")
     public Dataset downvote(@PathVariable("id") UUID id) {
         datasetService.downvoteDataset(id);
-        return null; // new ResponseEntity<>(null, HttpStatus.OK);
+        return getDatasetById(id); // new ResponseEntity<>(null, HttpStatus.OK);
     }
 
     @PutMapping("/id/{id}/vote")
-    public String postMethodName(@PathVariable("id") UUID id,
+    public Dataset postMethodName(@PathVariable("id") UUID id,
             @RequestParam("stars") int stars) {
         if (stars > 0 && stars < 6) {
             datasetService.voteDataset(id, stars);
-            return null;
         }
-        return "Invalid vote";
+        return datasetService.getDatasetById(id);
     }
 
     @GetMapping