Votes updated

src/main/java/com/groupfour/chatapp/chatapp/poll/Poll.java

@@ -10,8 +10,6 @@
 
 @Entity
 public class Poll {
-
-
     @Id
     @GeneratedValue
     @Column(name = "POLL_ID")
@@ -34,6 +32,7 @@ public class Poll {
     @ManyToOne
     private User pollCreator;
 
+
     public void setPollId(Long pollId) {
         this.pollId = pollId;
     }
@@ -49,6 +48,7 @@ public void setChat(Chat chat) {
     @ManyToOne
     private Chat chat;
 
+
     public Long getPollId() {
         return pollId;
     }

src/main/java/com/groupfour/chatapp/chatapp/poll/PollService.java

@@ -48,4 +48,5 @@ public Boolean delete(Long id) {
         pollRepository.deleteById(id);
         return true;
     }
+
 }

src/main/java/com/groupfour/chatapp/chatapp/poll/Vote.java

@@ -16,6 +16,9 @@ public class Vote {
     @JoinColumn(name = "OPTION_ID")
     private Option option;
 
+    @ManyToOne
+    private Poll poll;
+
     @OneToOne
     private User voter;
 

src/main/java/com/groupfour/chatapp/chatapp/poll/VoteController.java

@@ -11,32 +11,31 @@
 @RestController
 public class VoteController {
 
-    private VoteRepository voteRepository;
     private VoteService voteService;
 
     @Autowired
-    public VoteController(VoteRepository voteRepository) {
-        this.voteRepository = voteRepository;
+    public VoteController(VoteService voteService) {
+        this.voteService = voteService;
     }
 
-    @RequestMapping(value = "/polls/{pollId}/votes", method = RequestMethod.POST)
+    @PostMapping(value = "/polls/{pollId}/votes")
     public ResponseEntity<?> createVote(@PathVariable Long pollId, @RequestBody Vote vote) {
-        vote = voteRepository.save(vote);
+        vote = voteService.save(vote);
         // Set the headers for the newly created resource
         HttpHeaders responseHeaders = new HttpHeaders();
         responseHeaders.setLocation(ServletUriComponentsBuilder.
-                fromCurrentRequest().path("/{id}").buildAndExpand(vote.getVoteId()).toUri());
+        fromCurrentRequest().path("/{id}").buildAndExpand(vote.getVoteId()).toUri());
         return new ResponseEntity<>(null, responseHeaders, HttpStatus.CREATED);
     }
 
     @RequestMapping(value="/polls/votes", method=RequestMethod.GET)
     public Iterable<Vote> getAllVotes() {
-        return voteRepository.findAll();
+        return voteService.findAll();
     }
 
     @RequestMapping(value="/polls/{pollId}/votes", method=RequestMethod.GET)
     public Iterable<Vote> getVote(@PathVariable Long pollId) {
-        return voteRepository.findVotesByPoll(pollId);
+        return voteService.findVotesByPoll(pollId);
     }
 
 

src/main/java/com/groupfour/chatapp/chatapp/poll/VoteRepository.java

@@ -7,9 +7,13 @@
 @Repository
 public interface VoteRepository extends CrudRepository<Vote, Long> {
     @Query(value = "SELECT v.* " +
-            "FROM Option o, Vote v " +
+            "FROM Chat o, Vote v " +
             "WHERE o.POLL_ID = ?1 " +
             "AND v.OPTION_ID = o.OPTION_ID", nativeQuery = true)
     public Iterable<Vote> findVotesByPoll(Long pollId);
+
+    public Iterable<Vote> findVotesBy();
+
+    Vote findVotesByPoll_PollId(Long id);
 }
 

src/main/java/com/groupfour/chatapp/chatapp/poll/VoteService.java

@@ -12,4 +12,16 @@ public class VoteService {
     public VoteService(VoteRepository voteRepository) {
         this.voteRepository = voteRepository;
     }
+
+    public Vote save(Vote vote){
+      return voteRepository.save(vote);
+    }
+
+    public Iterable<Vote> findAll() {
+        return null;
+    }
+
+    public Iterable<Vote> findVotesByPoll(Long pollId) {
+        return null;
+    }
 }