增加测试验证 issues #50 和 #52

Author: abel533

service/src/test/java/io/mybatis/service/ServiceTest.java

@@ -24,8 +24,10 @@
 import org.junit.Test;
 import org.springframework.context.support.ClassPathXmlApplicationContext;
 
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
+import java.util.stream.Collectors;
 
 
 public class ServiceTest {
@@ -178,6 +180,21 @@ public void testUserService() {
     Assert.assertEquals(3, userService.findList(user).size());
     user.setName("admin");
     Assert.assertTrue(userService.findOne(user) != null);
+
+    List<Integer> ids = new ArrayList<>();
+    user = new User();
+    userService.save(user);
+    ids.add(user.getId());
+    user = new User();
+    userService.save(user);
+    ids.add(user.getId());
+
+    // issues #52
+    List<User> users = ids.stream().map(userService::findById).collect(Collectors.toList());
+    users.forEach(u -> Assert.assertNotNull(u.getId()));
+
+    // issues #50
+    Assert.assertEquals(2, userService.deleteByIdList(ids));
   }
 
   @Test

service/src/test/java/io/mybatis/service/UserService.java

@@ -18,9 +18,13 @@
 
 import io.mybatis.service.model.User;
 
+import java.util.List;
+
 /**
  * 暴露所有通用方法
  */
 public interface UserService extends MyBaseService<User> {
 
+  int deleteByIdList(List<Integer> ids);
+
 }

service/src/test/java/io/mybatis/service/impl/BaseIdService.java

@@ -0,0 +1,18 @@
+package io.mybatis.service.impl;
+
+import io.mybatis.mapper.BaseMapper;
+import io.mybatis.service.AbstractService;
+import io.mybatis.service.model.BaseId;
+
+import java.util.List;
+
+public abstract class BaseIdService<T extends BaseId<T>, M extends BaseMapper<T, Integer>> extends AbstractService<T, Integer, M> {
+
+  /**
+   * 根据ID列表进行删除，issues #50
+   */
+  public int deleteByIdList(List<Integer> ids) {
+    return deleteByFieldList(T::getId, ids);
+  }
+
+}

service/src/test/java/io/mybatis/service/impl/UserServiceImpl.java

@@ -16,13 +16,12 @@
 
 package io.mybatis.service.impl;
 
-import io.mybatis.service.AbstractService;
 import io.mybatis.service.UserService;
 import io.mybatis.service.mapper.UserMapper;
 import io.mybatis.service.model.User;
 import org.springframework.stereotype.Service;
 
 @Service
-public class UserServiceImpl extends AbstractService<User, Integer, UserMapper> implements UserService {
+public class UserServiceImpl extends BaseIdService<User, UserMapper> implements UserService {
 
 }