add live test

persistence-modules/spring-data-yugabytedb/src/main/java/com/baeldung/Main.java

@@ -8,25 +8,25 @@ import org.springframework.boot.autoconfigure.SpringBootApplication;
 @SpringBootApplication
 public class Main implements CommandLineRunner {
 
-	@Autowired
-	private UserRepository userRepository;
+    @Autowired
+    private UserRepository userRepository;
 
-	public static void main(String[] args) {
-		SpringApplication.run(Main.class, args);
-	}
+    public static void main(String[] args) {
+        SpringApplication.run(Main.class, args);
+    }
 
-	@Override
-	public void run(String... args) throws InterruptedException {
+    @Override
+    public void run(String... args) throws InterruptedException {
 
-		int iterationCount = 1_000;
-		int elementsPerIteration = 100;
+        int iterationCount = 1_000;
+        int elementsPerIteration = 100;
 
-		for (int i = 0; i < iterationCount; i++) {
-			for (long j = 0; j < elementsPerIteration; j++) {
-				User user = new User();
-				userRepository.save(user);
-			}
-			Thread.sleep(1000);
-		}
-	}
+        for (int i = 0; i < iterationCount; i++) {
+            for (long j = 0; j < elementsPerIteration; j++) {
+                User user = new User();
+                userRepository.save(user);
+            }
+            Thread.sleep(1000);
+        }
+    }
 }

persistence-modules/spring-data-yugabytedb/src/main/java/com/baeldung/User.java

@@ -11,34 +11,31 @@ import javax.persistence.Table;
 @Table(name = "users")
 public class User {
 
-	@Id
-	@GeneratedValue(strategy = GenerationType.IDENTITY)
-	private Long id;
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
 
-	@Column
-	private String name;
+    @Column
+    private String name;
 
-	Long getId() {
-		return id;
-	}
+    public Long getId() {
+        return id;
+    }
 
-	void setId(Long id) {
-		this.id = id;
-	}
+    public void setId(Long id) {
+        this.id = id;
+    }
 
-	String getName() {
-		return name;
-	}
+    public String getName() {
+        return name;
+    }
 
-	void setName(String name) {
-		this.name = name;
-	}
+    public void setName(String name) {
+        this.name = name;
+    }
 
-	@Override
-	public String toString() {
-		return "User{" +
-				"id=" + id +
-				", name='" + name + '\'' +
-				'}';
-	}
+    @Override
+    public String toString() {
+        return "User{" + "id=" + id + ", name='" + name + '\'' + '}';
+    }
 }

persistence-modules/spring-data-yugabytedb/src/main/resources/docker-compose.yml

@@ -0,0 +1,13 @@
+version: '3'
+
+services:
+  yugabytedb:
+    image: yugabytedb/yugabyte:latest
+    container_name: yugabyte
+    user: root
+    ports:
+      - '5433:5433'
+      - '7000:7000'
+      - '9000:9000'
+    command: ["bin/yugabyted", "start", "--daemon=false"]
+

persistence-modules/spring-data-yugabytedb/src/test/java/com/baeldung/YugabyteDBLiveTest.java

@@ -0,0 +1,37 @@
+package com.baeldung;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import java.util.List;
+
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.TestPropertySource;
+import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;
+
+/*
+ To run this test we need to run the databases first.
+ A dedicated docker-compose.yml file is located under the resources directory.
+ We can run it by simple executing `docker-compose up`.
+ */
+@SpringJUnitConfig
+@SpringBootTest
+@TestPropertySource("classpath:application.properties")
+public class YugabyteDBLiveTest {
+
+	@Autowired
+	private UserRepository userRepository;
+
+	@Test
+	void givenTwoUsers_whenPersistUsingJPARepository_thenUserAreSaved() {
+	    User user1 = new User();
+	    user1.setName("Alex");
+	    User user2 = new User();
+	    user2.setName("John");
+	    userRepository.save(user1);
+	    userRepository.save(user2);
+	    List<User> allUsers = userRepository.findAll();
+	    assertEquals(2, allUsers.size());
+	}
+}