329 lines
17 KiB
Java
329 lines
17 KiB
Java
package cn.celess.blog.controller;
|
|
|
|
import cn.celess.blog.BaseTest;
|
|
import cn.celess.blog.entity.Article;
|
|
import cn.celess.blog.entity.Comment;
|
|
import cn.celess.blog.entity.model.CommentModel;
|
|
import cn.celess.blog.entity.request.CommentReq;
|
|
import cn.celess.blog.mapper.ArticleMapper;
|
|
import cn.celess.blog.mapper.CommentMapper;
|
|
import cn.celess.blog.mapper.UserMapper;
|
|
import com.github.pagehelper.PageInfo;
|
|
import net.sf.json.JSONObject;
|
|
import org.junit.Test;
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
import org.springframework.http.MediaType;
|
|
import org.springframework.test.web.servlet.result.MockMvcResultHandlers;
|
|
|
|
import java.util.ArrayList;
|
|
import java.util.Date;
|
|
import java.util.List;
|
|
import java.util.UUID;
|
|
import java.util.concurrent.atomic.AtomicReference;
|
|
|
|
import static org.junit.Assert.*;
|
|
import static cn.celess.blog.enmu.ResponseEnum.*;
|
|
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
|
|
|
|
public class CommentControllerTest extends BaseTest {
|
|
@Autowired
|
|
ArticleMapper articleMapper;
|
|
@Autowired
|
|
CommentMapper commentMapper;
|
|
|
|
@Test
|
|
public void addOne() throws Exception {
|
|
CommentReq commentReq = new CommentReq();
|
|
// 测试留言
|
|
commentReq.setArticleID(null);
|
|
commentReq.setComment(false);
|
|
commentReq.setContent(UUID.randomUUID().toString());
|
|
commentReq.setPid(-1L);
|
|
commentReq.setResponseId(null);
|
|
String token = userLogin();
|
|
CommentModel PC = null;
|
|
mockMvc.perform(post("/user/comment/create")
|
|
.contentType(MediaType.APPLICATION_JSON_UTF8)
|
|
.content(JSONObject.fromObject(commentReq).toString())
|
|
.header("Authorization", token)
|
|
).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
CommentModel model = (CommentModel) JSONObject.toBean(object.getJSONObject(Result), CommentModel.class);
|
|
assertNotEquals(0, model.getId());
|
|
assertEquals(commentReq.getPid().longValue(), model.getPid());
|
|
assertEquals(-1, model.getPid());
|
|
assertEquals(commentReq.getComment(), model.isComment());
|
|
assertEquals(commentReq.getContent(), model.getContent());
|
|
assertNotNull(model.getDate());
|
|
assertNotNull(model.getAuthorName());
|
|
assertNotNull(model.getAuthorAvatarImgUrl());
|
|
});
|
|
|
|
Article article = articleMapper.getLastestArticle();
|
|
// 测试评论
|
|
commentReq.setArticleID(article.getId());
|
|
commentReq.setComment(true);
|
|
commentReq.setContent(UUID.randomUUID().toString());
|
|
commentReq.setPid(-1L);
|
|
commentReq.setResponseId(null);
|
|
mockMvc.perform(post("/user/comment/create")
|
|
.contentType(MediaType.APPLICATION_JSON_UTF8)
|
|
.content(JSONObject.fromObject(commentReq).toString())
|
|
.header("Authorization", token)
|
|
).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
CommentModel model = (CommentModel) JSONObject.toBean(object.getJSONObject(Result), CommentModel.class);
|
|
// 响应数据的完整性
|
|
assertNotEquals(0, model.getId());
|
|
assertEquals(commentReq.getPid().longValue(), model.getPid());
|
|
assertEquals(-1, model.getPid());
|
|
assertEquals(commentReq.getComment(), model.isComment());
|
|
assertEquals(commentReq.getContent(), model.getContent());
|
|
assertEquals(commentReq.getArticleID().longValue(), model.getArticleID());
|
|
assertNotNull(model.getDate());
|
|
assertNotNull(model.getAuthorName());
|
|
assertNotNull(model.getAuthorAvatarImgUrl());
|
|
});
|
|
|
|
// 测试二级回复
|
|
Comment lastestComment = commentMapper.getLastestComment();
|
|
commentReq.setArticleID(lastestComment.getArticleID());
|
|
commentReq.setComment(lastestComment.getType());
|
|
commentReq.setContent(UUID.randomUUID().toString());
|
|
commentReq.setPid(lastestComment.getId());
|
|
commentReq.setResponseId(null);
|
|
mockMvc.perform(post("/user/comment/create")
|
|
.contentType(MediaType.APPLICATION_JSON_UTF8)
|
|
.content(JSONObject.fromObject(commentReq).toString())
|
|
.header("Authorization", token)
|
|
).andDo(MockMvcResultHandlers.print()).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
CommentModel model = (CommentModel) JSONObject.toBean(object.getJSONObject(Result), CommentModel.class);
|
|
// 重新获取父评论信息
|
|
Comment pCommon = commentMapper.findCommentById(lastestComment.getId());
|
|
assertEquals(pCommon.getId().longValue(), model.getPid());
|
|
// 判断父评论中是否有写入当前新增的评论的id
|
|
String[] ids = pCommon.getResponseId().split(",");
|
|
boolean contain = false;
|
|
for (String id : ids) {
|
|
if (!id.isEmpty() && Long.parseLong(id) == model.getId()) {
|
|
contain = true;
|
|
break;
|
|
}
|
|
}
|
|
assertTrue(contain);
|
|
});
|
|
}
|
|
|
|
@Test
|
|
public void deleteTest() throws Exception {
|
|
// 准备数据
|
|
Comment c = new Comment();
|
|
c.setArticleID(-1L);
|
|
c.setType(true);
|
|
c.setAuthorID(2L);
|
|
c.setDate(new Date());
|
|
c.setPid(-1L);
|
|
commentMapper.insert(c);
|
|
Comment comment = commentMapper.getLastestComment();
|
|
// 接口测试
|
|
long id = comment.getId();
|
|
assertNotEquals(0, id);
|
|
String token = userLogin();
|
|
mockMvc.perform(delete("/user/comment/del?id=" + id).header("Authorization", token)).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
assertTrue(object.getBoolean(Result));
|
|
});
|
|
mockMvc.perform(delete("/user/comment/del?id=" + id).header("Authorization", token)).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(COMMENT_NOT_EXIST.getCode(), object.getInt(Code));
|
|
});
|
|
}
|
|
|
|
@Test
|
|
public void update() throws Exception {
|
|
Comment comment = commentMapper.getLastestComment();
|
|
CommentReq commentReq = new CommentReq();
|
|
commentReq.setId(comment.getId());
|
|
commentReq.setPid(comment.getPid());
|
|
commentReq.setContent(UUID.randomUUID().toString());
|
|
commentReq.setArticleID(comment.getArticleID());
|
|
// 不合法数据 setResponseId
|
|
commentReq.setResponseId("xxxx");
|
|
commentReq.setComment(comment.getType());
|
|
mockMvc.perform(put("/user/comment/update")
|
|
.content(JSONObject.fromObject(commentReq).toString())
|
|
.contentType(MediaType.APPLICATION_JSON_UTF8)
|
|
.header("Authorization", userLogin())
|
|
).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
CommentModel c = (CommentModel) JSONObject.toBean(object.getJSONObject(Result), CommentModel.class);
|
|
assertEquals(commentReq.getContent(), c.getContent());
|
|
assertEquals(commentReq.getResponseId(), c.getResponseId());
|
|
assertNotNull(c.getAuthorAvatarImgUrl());
|
|
assertNotNull(c.getAuthorName());
|
|
assertNotNull(c.getDate());
|
|
assertNotEquals(0, c.getPid());
|
|
assertNotEquals(0, c.getArticleID());
|
|
});
|
|
}
|
|
|
|
@Test
|
|
public void commentsOfArticle() throws Exception {
|
|
mockMvc.perform(get("/comments?articleId=3&page=1&count=10")).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
PageInfo pageInfo = (PageInfo) JSONObject.toBean(object.getJSONObject(Result), PageInfo.class);
|
|
assertNotEquals(0, pageInfo.getStartRow());
|
|
assertNotEquals(0, pageInfo.getEndRow());
|
|
assertEquals(1, pageInfo.getPageNum());
|
|
assertEquals(10, pageInfo.getPageSize());
|
|
pageInfo.getList().forEach(o -> {
|
|
CommentModel model = (CommentModel) JSONObject.toBean(JSONObject.fromObject(o), CommentModel.class);
|
|
assertEquals(3, model.getArticleID());
|
|
assertNotNull(model.getDate());
|
|
assertNotNull(model.getAuthorName());
|
|
assertNotNull(model.getAuthorAvatarImgUrl());
|
|
assertNotNull(model.getArticleTitle());
|
|
assertNotNull(model.getContent());
|
|
assertNotNull(model.getResponseId());
|
|
});
|
|
});
|
|
}
|
|
|
|
@Test
|
|
public void retrievePage() throws Exception {
|
|
long pid = 17;
|
|
mockMvc.perform(get("/comment/pid/" + pid + "?articleId=3&page=1&count=10")).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
PageInfo pageInfo = (PageInfo) JSONObject.toBean(object.getJSONObject(Result), PageInfo.class);
|
|
assertNotEquals(0, pageInfo.getStartRow());
|
|
assertNotEquals(0, pageInfo.getEndRow());
|
|
assertEquals(1, pageInfo.getPageNum());
|
|
assertEquals(10, pageInfo.getPageSize());
|
|
pageInfo.getList().forEach(o -> {
|
|
CommentModel model = (CommentModel) JSONObject.toBean(JSONObject.fromObject(o), CommentModel.class);
|
|
assertEquals(3, model.getArticleID());
|
|
assertNotNull(model.getDate());
|
|
assertNotNull(model.getAuthorName());
|
|
assertNotNull(model.getAuthorAvatarImgUrl());
|
|
assertNotNull(model.getArticleTitle());
|
|
assertNotNull(model.getContent());
|
|
assertNotNull(model.getResponseId());
|
|
});
|
|
});
|
|
}
|
|
|
|
@Test
|
|
public void retrievePageOfLeaveMsg() throws Exception {
|
|
mockMvc.perform(get("/leaveMsg?page=1&count=10")).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
PageInfo pageInfo = (PageInfo) JSONObject.toBean(object.getJSONObject(Result), PageInfo.class);
|
|
assertNotEquals(0, pageInfo.getStartRow());
|
|
assertNotEquals(0, pageInfo.getEndRow());
|
|
assertEquals(1, pageInfo.getPageNum());
|
|
assertEquals(10, pageInfo.getPageSize());
|
|
pageInfo.getList().forEach(o -> {
|
|
CommentModel model = (CommentModel) JSONObject.toBean(JSONObject.fromObject(o), CommentModel.class);
|
|
assertEquals(-1, model.getArticleID());
|
|
assertNotNull(model.getDate());
|
|
assertNotNull(model.getAuthorName());
|
|
assertNotNull(model.getAuthorAvatarImgUrl());
|
|
assertNotNull(model.getContent());
|
|
assertNotNull(model.getResponseId());
|
|
assertFalse(model.isComment());
|
|
});
|
|
});
|
|
}
|
|
|
|
@Test
|
|
public void retrievePageAdmin() throws Exception {
|
|
mockMvc.perform(get("/admin/comment/type/1?page=1&count=10").header("Authorization", adminLogin())).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
PageInfo pageInfo = (PageInfo) JSONObject.toBean(object.getJSONObject(Result), PageInfo.class);
|
|
assertNotEquals(0, pageInfo.getStartRow());
|
|
assertNotEquals(0, pageInfo.getEndRow());
|
|
assertEquals(1, pageInfo.getPageNum());
|
|
assertEquals(10, pageInfo.getPageSize());
|
|
pageInfo.getList().forEach(o -> {
|
|
CommentModel model = (CommentModel) JSONObject.toBean(JSONObject.fromObject(o), CommentModel.class);
|
|
assertNotEquals(-1, model.getArticleID());
|
|
assertNotNull(model.getDate());
|
|
assertNotNull(model.getAuthorName());
|
|
assertNotNull(model.getAuthorAvatarImgUrl());
|
|
assertNotNull(model.getContent());
|
|
assertNotNull(model.getResponseId());
|
|
assertTrue(model.isComment());
|
|
});
|
|
});
|
|
mockMvc.perform(get("/admin/comment/type/0?page=1&count=10").header("Authorization", adminLogin())).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
PageInfo pageInfo = (PageInfo) JSONObject.toBean(object.getJSONObject(Result), PageInfo.class);
|
|
assertNotEquals(0, pageInfo.getStartRow());
|
|
assertNotEquals(0, pageInfo.getEndRow());
|
|
assertEquals(1, pageInfo.getPageNum());
|
|
assertEquals(10, pageInfo.getPageSize());
|
|
pageInfo.getList().forEach(o -> {
|
|
CommentModel model = (CommentModel) JSONObject.toBean(JSONObject.fromObject(o), CommentModel.class);
|
|
assertEquals(-1, model.getArticleID());
|
|
assertNotNull(model.getDate());
|
|
assertNotNull(model.getAuthorName());
|
|
assertNotNull(model.getAuthorAvatarImgUrl());
|
|
assertNotNull(model.getContent());
|
|
assertNotNull(model.getResponseId());
|
|
assertFalse(model.isComment());
|
|
});
|
|
});
|
|
}
|
|
|
|
@Test
|
|
public void retrievePageByAuthor() throws Exception {
|
|
mockMvc.perform(get("/user/comment/type/1?page=1&count=10").header("Authorization", userLogin())).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
PageInfo pageInfo = (PageInfo) JSONObject.toBean(object.getJSONObject(Result), PageInfo.class);
|
|
assertNotEquals(0, pageInfo.getStartRow());
|
|
assertNotEquals(0, pageInfo.getEndRow());
|
|
assertEquals(1, pageInfo.getPageNum());
|
|
assertEquals(10, pageInfo.getPageSize());
|
|
pageInfo.getList().forEach(o -> {
|
|
CommentModel model = (CommentModel) JSONObject.toBean(JSONObject.fromObject(o), CommentModel.class);
|
|
assertNotEquals(-1, model.getArticleID());
|
|
assertNotNull(model.getDate());
|
|
assertNotNull(model.getAuthorName());
|
|
assertNotNull(model.getAuthorAvatarImgUrl());
|
|
assertNotNull(model.getContent());
|
|
assertNotNull(model.getResponseId());
|
|
assertTrue(model.isComment());
|
|
});
|
|
});
|
|
mockMvc.perform(get("/user/comment/type/0?page=1&count=10").header("Authorization", userLogin())).andDo(result -> {
|
|
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
|
|
assertEquals(SUCCESS.getCode(), object.getInt(Code));
|
|
PageInfo pageInfo = (PageInfo) JSONObject.toBean(object.getJSONObject(Result), PageInfo.class);
|
|
assertNotEquals(0, pageInfo.getStartRow());
|
|
assertNotEquals(0, pageInfo.getEndRow());
|
|
assertEquals(1, pageInfo.getPageNum());
|
|
assertEquals(10, pageInfo.getPageSize());
|
|
pageInfo.getList().forEach(o -> {
|
|
CommentModel model = (CommentModel) JSONObject.toBean(JSONObject.fromObject(o), CommentModel.class);
|
|
assertEquals(-1, model.getArticleID());
|
|
assertNotNull(model.getDate());
|
|
assertNotNull(model.getAuthorName());
|
|
assertNotNull(model.getAuthorAvatarImgUrl());
|
|
assertNotNull(model.getContent());
|
|
assertNotNull(model.getResponseId());
|
|
assertFalse(model.isComment());
|
|
});
|
|
});
|
|
}
|
|
} |