An example endpoint to get users by first name with in memory (H2) repository test

Author: kasramp

src/main/java/com/madadipouya/springkafkatest/controller/UserController.java

@@ -3,15 +3,15 @@
 import com.github.javafaker.Faker;
 import com.madadipouya.springkafkatest.dto.User;
 import com.madadipouya.springkafkatest.kafka.producer.UserKafkaProducer;
+import com.madadipouya.springkafkatest.service.UserService;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.http.HttpStatus;
-import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.ResponseStatus;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
+import java.util.List;
 import java.util.UUID;
+import java.util.stream.Collectors;
 
 @RestController
 @RequestMapping("v1/users")
@@ -20,10 +20,13 @@ public class UserController {
 
     private final UserKafkaProducer kafkaProducer;
 
+    private final UserService userService;
+
     private final Faker faker;
 
-    public UserController(UserKafkaProducer kafkaProducer) {
+    public UserController(UserKafkaProducer kafkaProducer, UserService userService) {
         this.kafkaProducer = kafkaProducer;
+        this.userService = userService;
         faker = new Faker();
     }
 
@@ -33,4 +36,12 @@ public UserController(UserKafkaProducer kafkaProducer) {
     public void generateRandomUser() {
         kafkaProducer.writeToKafka(new User(UUID.randomUUID().toString(), faker.name().firstName(), faker.name().lastName()));
     }
+
+    @GetMapping("/{firstName}")
+    @ResponseStatus
+    @ApiOperation(value = "Returns a list of users that matchers the given name")
+    public List<User> getUsers(@PathVariable(name = "firstName") String name) {
+        List<com.madadipouya.springkafkatest.entity.User> users = userService.getUsers(name);
+        return users.stream().map(user -> new User(user.getId(), user.getFirstName(), user.getLastName())).collect(Collectors.toList());
+    }
 }

src/main/java/com/madadipouya/springkafkatest/repository/UserRepository.java

@@ -4,7 +4,11 @@
 import org.springframework.data.jpa.repository.JpaRepository;
 import org.springframework.stereotype.Repository;
 
+import java.util.List;
+
 @Repository
 public interface UserRepository extends JpaRepository<User, String> {
 
+    List<User> getByFirstName(String firstName);
+
 }

src/main/java/com/madadipouya/springkafkatest/service/UserService.java

@@ -2,7 +2,11 @@
 
 import com.madadipouya.springkafkatest.dto.User;
 
+import java.util.List;
+
 public interface UserService {
 
     void save(User user);
+
+    List<com.madadipouya.springkafkatest.entity.User> getUsers(String firstName);
 }

src/main/java/com/madadipouya/springkafkatest/service/impl/DefaultUserService.java

@@ -7,6 +7,8 @@
 import org.slf4j.LoggerFactory;
 import org.springframework.stereotype.Service;
 
+import java.util.List;
+
 @Service
 public class DefaultUserService implements UserService {
 
@@ -23,4 +25,9 @@ public void save(User user) {
         logger.info("Saving user with id = {}", user.getUuid());
         userRepository.save(new com.madadipouya.springkafkatest.entity.User(user.getUuid(), user.getFirstName(), user.getLastName()));
     }
+
+    @Override
+    public List<com.madadipouya.springkafkatest.entity.User> getUsers(String firstName) {
+        return userRepository.getByFirstName(firstName);
+    }
 }

src/main/resources/db/migration/V2__sample_users.sql

@@ -0,0 +1,6 @@
+INSERT INTO users (id, first_name, last_name) VALUES
+    ( 'b9e7100b-ff81-4a42-ace7-18802e00a9ab', 'Alejandro', 'Gillick'),
+    ( 'd8096b43-091d-46c4-acca-0869b0502e07', 'Frank', 'Castle'),
+    ( 'c656d2cd-8fe8-4640-b4ee-e9675790e926', 'John', 'Wick'),
+    ( '21494008-a54e-41d1-9c71-13344161378b', 'John', 'Rambo'),
+    ( 'cca1c70e-77d5-415e-8c2f-97ee55c3845d', 'Robert', 'McCall');

src/test/java/com/madadipouya/springkafkatest/kafka/repository/UserRepositoryTest.java

@@ -0,0 +1,46 @@
+package com.madadipouya.springkafkatest.kafka.repository;
+
+import com.madadipouya.springkafkatest.entity.User;
+import com.madadipouya.springkafkatest.repository.UserRepository;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
+import org.springframework.boot.test.autoconfigure.orm.jpa.TestEntityManager;
+import org.springframework.test.annotation.DirtiesContext;
+
+import java.util.List;
+import java.util.UUID;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertFalse;
+
+@DataJpaTest
+@DirtiesContext
+public class UserRepositoryTest {
+
+    @Autowired
+    private UserRepository userRepository;
+
+    @Autowired
+    private TestEntityManager testEntityManager;
+
+    @Test
+    void testGetByFirstName() {
+        testEntityManager.persist(stubUser("John", "Wick"));
+        testEntityManager.persist(stubUser("Robert", "McCall"));
+        testEntityManager.persist(stubUser("John", "Rambo"));
+
+        List<User> users = userRepository.getByFirstName("John");
+
+        assertFalse(users.isEmpty());
+        assertEquals(2, users.size());
+        assertEquals("John", users.get(0).getFirstName());
+        assertEquals("Wick", users.get(0).getLastName());
+        assertEquals("John", users.get(1).getFirstName());
+        assertEquals("Rambo", users.get(1).getLastName());
+    }
+
+    private User stubUser(String firstName, String lastName) {
+        return new User(UUID.randomUUID().toString(), firstName, lastName);
+    }
+}

src/test/resources/application.properties

@@ -0,0 +1,3 @@
+spring.jpa.show-sql=true
+spring.datasource.url=jdbc:h2:file:~/testdb
+spring.flyway.enabled=false