|
@@ -1,5 +1,22 @@
|
|
|
package com.baomidou.mybatisplus.test.h2;
|
|
|
|
|
|
+import static java.util.stream.Collectors.toList;
|
|
|
+
|
|
|
+import java.io.IOException;
|
|
|
+import java.sql.SQLException;
|
|
|
+import java.util.HashMap;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+
|
|
|
+import javax.annotation.Resource;
|
|
|
+
|
|
|
+import org.junit.Assert;
|
|
|
+import org.junit.BeforeClass;
|
|
|
+import org.junit.Test;
|
|
|
+import org.junit.runner.RunWith;
|
|
|
+import org.springframework.test.context.ContextConfiguration;
|
|
|
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
|
|
|
+
|
|
|
import com.baomidou.mybatisplus.core.conditions.Wrapper;
|
|
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
|
|
@@ -7,23 +24,9 @@ import com.baomidou.mybatisplus.core.metadata.IPage;
|
|
|
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
|
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
import com.baomidou.mybatisplus.test.h2.config.H2Db;
|
|
|
+import com.baomidou.mybatisplus.test.h2.entity.enums.AgeEnum;
|
|
|
import com.baomidou.mybatisplus.test.h2.entity.mapper.H2UserMapper;
|
|
|
import com.baomidou.mybatisplus.test.h2.entity.persistent.H2User;
|
|
|
-import org.junit.Assert;
|
|
|
-import org.junit.BeforeClass;
|
|
|
-import org.junit.Test;
|
|
|
-import org.junit.runner.RunWith;
|
|
|
-import org.springframework.test.context.ContextConfiguration;
|
|
|
-import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
|
|
|
-
|
|
|
-import javax.annotation.Resource;
|
|
|
-import java.io.IOException;
|
|
|
-import java.sql.SQLException;
|
|
|
-import java.util.HashMap;
|
|
|
-import java.util.List;
|
|
|
-import java.util.Map;
|
|
|
-
|
|
|
-import static java.util.stream.Collectors.toList;
|
|
|
|
|
|
/**
|
|
|
* <p>
|
|
@@ -49,7 +52,7 @@ public class H2UserMapperTest extends BaseTest {
|
|
|
public void crudTest() {
|
|
|
H2User h2User = new H2User();
|
|
|
h2User.setName(NQQ);
|
|
|
- h2User.setAge(1);
|
|
|
+ h2User.setAge(AgeEnum.ONE);
|
|
|
h2User.setDeleted(0);
|
|
|
h2User.setDesc("这是一个不错的小伙子");
|
|
|
Assert.assertTrue(1 == userMapper.insert(h2User));
|
|
@@ -61,17 +64,17 @@ public class H2UserMapperTest extends BaseTest {
|
|
|
h2User.setName("测试");
|
|
|
userMapper.insert(h2User);
|
|
|
for (int i = 0; i < 10; i++) {
|
|
|
- userMapper.insert(new H2User("mp" + i, i));
|
|
|
+ userMapper.insert(new H2User("mp" + i, AgeEnum.ONE));
|
|
|
}
|
|
|
Assert.assertTrue(1 == userMapper.deleteById(1L));
|
|
|
|
|
|
Map<String, Object> map = new HashMap<>();
|
|
|
map.put("name", "mp0");
|
|
|
- map.put("age", 0);
|
|
|
+ map.put("age", AgeEnum.ONE);
|
|
|
|
|
|
// 根据 map 查询
|
|
|
h2User = userMapper.selectByMap(map).get(0);
|
|
|
- Assert.assertTrue(0 == h2User.getAge());
|
|
|
+ Assert.assertTrue(AgeEnum.ONE == h2User.getAge());
|
|
|
|
|
|
// 根据 map 删除
|
|
|
Assert.assertTrue(1 == userMapper.deleteByMap(map));
|
|
@@ -93,24 +96,24 @@ public class H2UserMapperTest extends BaseTest {
|
|
|
|
|
|
// 更新
|
|
|
h2User = new H2User();
|
|
|
- h2User.setAge(2);
|
|
|
+ h2User.setAge(AgeEnum.TWO);
|
|
|
h2User.setDesc("测试置空");
|
|
|
Assert.assertTrue(1 == userMapper.update(h2User,
|
|
|
new QueryWrapper<H2User>().eq("name", NQQ)));
|
|
|
|
|
|
- log(userMapper.selectOne(new QueryWrapper<>(new H2User().setName(NQQ).setAge(2))));
|
|
|
+ log(userMapper.selectOne(new QueryWrapper<>(new H2User().setName(NQQ).setAge(AgeEnum.TWO))));
|
|
|
|
|
|
- h2User.setAge(3);
|
|
|
+ h2User.setAge(AgeEnum.THREE);
|
|
|
h2User.setDesc(null);
|
|
|
Assert.assertTrue(userMapper.update(h2User,
|
|
|
new UpdateWrapper<H2User>().lambda()
|
|
|
.set(H2User::getDesc, "")
|
|
|
.eq(H2User::getName, "Jerry")) > 0);
|
|
|
|
|
|
- log(userMapper.selectOne(new QueryWrapper<>(new H2User().setName(NQQ).setAge(3))));
|
|
|
+ log(userMapper.selectOne(new QueryWrapper<>(new H2User().setName(NQQ).setAge(AgeEnum.THREE))));
|
|
|
|
|
|
// 根据主键更新 age = 18
|
|
|
- h2User.setAge(18);
|
|
|
+ h2User.setAge(AgeEnum.TWO);
|
|
|
Assert.assertNotNull(1 == userMapper.updateById(h2User));
|
|
|
|
|
|
// 查询一条记录
|
|
@@ -136,12 +139,12 @@ public class H2UserMapperTest extends BaseTest {
|
|
|
System.out.println(m);
|
|
|
}
|
|
|
}
|
|
|
- Assert.assertTrue(CollectionUtils.isNotEmpty(userMapper.selectMaps(new QueryWrapper<>(new H2User().setAge(18)))));
|
|
|
+ Assert.assertTrue(CollectionUtils.isNotEmpty(userMapper.selectMaps(new QueryWrapper<>(new H2User().setAge(AgeEnum.TWO)))));
|
|
|
}
|
|
|
|
|
|
@Test
|
|
|
public void delete() {
|
|
|
- userMapper.delete(new QueryWrapper<>(new H2User().setAge(2))
|
|
|
+ userMapper.delete(new QueryWrapper<>(new H2User().setAge(AgeEnum.TWO))
|
|
|
.eq("name", "Tony"));
|
|
|
}
|
|
|
}
|