|
|
@@ -0,0 +1,91 @@
|
|
|
+package top.haijunit.train.mapper;
|
|
|
+
|
|
|
+import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
|
|
+import com.baomidou.mybatisplus.core.metadata.IPage;
|
|
|
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
|
|
|
+import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
|
|
+import jakarta.annotation.Resource;
|
|
|
+import org.assertj.core.api.Assertions;
|
|
|
+import org.junit.Assert;
|
|
|
+import org.junit.Test;
|
|
|
+import org.junit.runner.RunWith;
|
|
|
+import org.springframework.boot.test.context.SpringBootTest;
|
|
|
+import org.springframework.test.context.junit4.SpringRunner;
|
|
|
+import top.haijunit.train.model.UserDO;
|
|
|
+
|
|
|
+import java.util.List;
|
|
|
+
|
|
|
+import static org.assertj.core.api.Assertions.assertThat;
|
|
|
+
|
|
|
+@RunWith(SpringRunner.class)
|
|
|
+@SpringBootTest
|
|
|
+public class UserMapperTest {
|
|
|
+ @Resource
|
|
|
+ private UserMapper userMapper;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 测试单条查询
|
|
|
+ */
|
|
|
+ @Test
|
|
|
+ public void testSelectOne() {
|
|
|
+ UserDO user = userMapper.selectById(1L);
|
|
|
+ System.out.println(user);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 测试新增
|
|
|
+ */
|
|
|
+ @Test
|
|
|
+ public void testInsert() {
|
|
|
+ UserDO user = new UserDO();
|
|
|
+ user.setUsername("kk");
|
|
|
+ user.setEmail("kk@qq.com");
|
|
|
+ assertThat(userMapper.insert(user)).isGreaterThan(0);
|
|
|
+ assertThat(user.getUserId()).isNotNull();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 测试删除
|
|
|
+ */
|
|
|
+ @Test
|
|
|
+ public void testDelete() {
|
|
|
+ assertThat(userMapper.deleteById(3L)).isGreaterThan(0);
|
|
|
+ assertThat(userMapper.delete(new QueryWrapper<UserDO>()
|
|
|
+ .lambda().eq(UserDO::getUsername, "kk"))).isGreaterThan(0);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 测试更新
|
|
|
+ */
|
|
|
+ @Test
|
|
|
+ public void testUpdate() {
|
|
|
+ UserDO user = userMapper.selectById(2);
|
|
|
+ assertThat(user.getUsername()).isEqualTo("Jack");
|
|
|
+ userMapper.update(null, Wrappers.<UserDO>lambdaUpdate().set(UserDO::getEmail, "zz@qq.com").eq(UserDO::getUserId, 1));
|
|
|
+ Assertions.assertThat(userMapper.selectById(2).getEmail()).isEqualTo("11@qq.com");
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 测试查询列表
|
|
|
+ */
|
|
|
+ @Test
|
|
|
+ public void testSelect() {
|
|
|
+ List<UserDO> userList = userMapper.selectList(null);
|
|
|
+ Assert.assertEquals(5, userList.size());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 测试分页查询
|
|
|
+ */
|
|
|
+ @Test
|
|
|
+ public void testPage() {
|
|
|
+ Page<UserDO> page = new Page<>(1, 2);
|
|
|
+ IPage<UserDO> userIPage = userMapper.selectPage(page, new QueryWrapper<>());
|
|
|
+ assertThat(page).isSameAs(userIPage);
|
|
|
+ System.out.println("总条数: " + userIPage.getTotal());
|
|
|
+ System.out.println("当前页数: " + userIPage.getCurrent());
|
|
|
+ System.out.println("当前每页显示数: " + userIPage.getSize());
|
|
|
+ System.out.println("记录列表: " + userIPage.getRecords());
|
|
|
+ }
|
|
|
+
|
|
|
+}
|