Add createdBy endpoint to pollsController

Author: Ferenc Hammerl

src/main/java/com/fhammerl/polls/controllers/PollsController.java

@@ -14,9 +14,18 @@ public class PollsController {
     @Autowired
     private PollRepository pollRepository;
 
+    // Test: http://localhost:8080/api/v1/polls/list
     @GetMapping
     @RequestMapping("list")
     public List<Poll> list() {
         return pollRepository.findAll();
     }
+
+
+    // Test: http://localhost:8080/api/v1/polls/createdBy?userId=4
+    @GetMapping
+    @RequestMapping("createdBy")
+    public List<Poll> createdBy(int userId) {
+        return pollRepository.findByCreatorId(userId);
+    }
 }

src/main/java/com/fhammerl/polls/models/Poll.java

@@ -22,6 +22,8 @@ public class Poll {
     private String preferencesType;
     private String state;
     private String title;
+    @Column(name = "created_by_user_id")
+    private int createdByUserId;
 
     @OneToMany(mappedBy = "poll", cascade = CascadeType.ALL, orphanRemoval = true, fetch = FetchType.LAZY)
     private List<Participant> participants;
@@ -89,5 +91,12 @@ public List<Participant> getParticipants() {
     public void setParticipants(List<Participant> participant) {
         this.participants = participant;
     }
+    public int getCreatedByUserId() {
+        return this.createdByUserId;
+    }
+
+    public void setCreatedByUserId(int createdByUserId) {
+        this.createdByUserId = createdByUserId;
+    }
 
 }

src/main/java/com/fhammerl/polls/repositories/PollRepository.java

@@ -1,7 +1,14 @@
 package com.fhammerl.polls.repositories;
 
 import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.jpa.repository.Query;
+import org.springframework.data.repository.query.Param;
+
+import java.util.List;
+
 import com.fhammerl.polls.models.Poll;
 
-public interface PollRepository extends JpaRepository<Poll, Long> {
+public interface PollRepository extends JpaRepository<Poll, Integer> {
+    @Query("SELECT p FROM poll p where p.createdByUserId = :id") 
+    List<Poll> findByCreatorId(@Param("id") Integer userId);
 }

src/test/java/com/fhammerl/polls/PollsControllerTests.java

@@ -27,6 +27,12 @@ class PollsControllerTests {
 	@Test
 	void listAllWorks() throws Exception {
 		List<Poll> polls = pollsController.list();
-		assertEquals(polls.toArray().length, 2, "Unexpected number of polls found!");
+		assertEquals(polls.toArray().length, 3, "Unexpected number of polls found!");
+	}
+
+	@Test
+	void listCreatedByUserWorks() throws Exception {
+		List<Poll> polls = pollsController.createdBy(4);
+		assertEquals(polls.toArray().length, 2, "Unexpected number of polls found for user 4!");
 	}
 }