add test

Author: xlorne

src/test/java/com/example/dbstream/repository/User1Repository.java

@@ -32,5 +32,10 @@ public interface User1Repository extends JpaRepository<User1, Long> {
     @Query("insert into User1(email,username,password,nickname) select u.email,u.username,u.password,u.nickname from User1 u")
     void insertIntoFromSelect();
 
+
+    @Modifying
+    @Query("delete from User1")
+    int deleteAllData();
+
 }
 

src/test/java/com/example/dbstream/tests/User1RepositoryTest.java

@@ -16,6 +16,7 @@
 
 import javax.persistence.EntityManager;
 import javax.transaction.Transactional;
+import java.util.ArrayList;
 import java.util.List;
 import java.util.concurrent.atomic.AtomicBoolean;
 
@@ -288,4 +289,37 @@ public void push(JdbcQuery jdbcQuery, List<DBEvent> events) {
         entityManager.clear();
     }
 
+    /**
+     * 批量数据插入测试
+     */
+    @Test
+    @Transactional
+    @Rollback(value = false)
+    @Order(9)
+    void test9() {
+        DBStreamContext.getInstance().cleanEventPushers();
+        DBStreamContext.getInstance().addEventPusher(new DBEventPusher() {
+            @Override
+            public void push(JdbcQuery jdbcQuery, List<DBEvent> events) {
+                assertTrue(events.size()>=10);
+                for (DBEvent event:events){
+                    assertTrue(event.hasPrimaryKeys());
+                }
+            }
+        });
+        List<User1> list = new ArrayList<>();
+        for (int i = 0; i < 5; i++) {
+            User1 user1 = new User1();
+            user1.setUsername("admin1");
+            user1.setPassword("admin1");
+            user1.setEmail("[email protected]");
+            user1.setNickname("admin1");
+            list.add(user1);
+        }
+        userRepository.saveAll(list);
+
+        userRepository.deleteAllData();
+
+    }
+
 }