zheng/blog-backEnd
1
0
Fork 0
Code Issues 5 Pull Requests 12 Actions Packages Projects Releases Wiki Activity

调整UserController的测试类

Browse Source
This commit is contained in:
禾几海
2020-07-25 11:23:03 +08:00
parent f65c96fa2d
commit 7922ea558f
3 changed files with 135 additions and 141 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/cn/celess/blog/entity/request/LoginReq.java
View File

@@ -1,12 +1,16 @@
package cn.celess.blog.entity.request; package cn.celess.blog.entity.request;
import lombok.AllArgsConstructor;
import lombok.Data; import lombok.Data;
import lombok.NoArgsConstructor;
/** /**
* @author : xiaohai * @author : xiaohai
* @date : 2019/06/01 22:47 * @date : 2019/06/01 22:47
*/ */
@Data @Data
@NoArgsConstructor
@AllArgsConstructor
public class LoginReq { public class LoginReq {
private String email; private String email;
private String password; private String password;

10
src/test/java/cn/celess/blog/BaseTest.java
View File

@@ -54,6 +54,8 @@ public class BaseTest {
protected MockMvc mockMvc; protected MockMvc mockMvc;
protected final static String Code = "code"; protected final static String Code = "code";
protected final static String Result = "result"; protected final static String Result = "result";
protected final static String USERE_MAIL = "zh56462271@qq.com";
protected final static String ADMIN_EMAIL = "a@celess.cn";
/** /**
* jackson 序列化/反序列化Json * jackson 序列化/反序列化Json
*/ */
@@ -87,7 +89,7 @@ public class BaseTest {
*/ */
protected String adminLogin() { protected String adminLogin() {
LoginReq req = new LoginReq(); LoginReq req = new LoginReq();
req.setEmail("a@celess.cn"); req.setEmail(ADMIN_EMAIL);
req.setPassword("123456789"); req.setPassword("123456789");
req.setIsRememberMe(true); req.setIsRememberMe(true);
String token = login(req); String token = login(req);
@@ -102,7 +104,7 @@ public class BaseTest {
*/ */
protected String userLogin() { protected String userLogin() {
LoginReq req = new LoginReq(); LoginReq req = new LoginReq();
req.setEmail("zh56462271@qq.com"); req.setEmail(USERE_MAIL);
req.setPassword("123456789"); req.setPassword("123456789");
req.setIsRememberMe(true); req.setIsRememberMe(true);
String token = login(req); String token = login(req);
@@ -116,7 +118,7 @@ public class BaseTest {
* @param req 用户信息 * @param req 用户信息
* @return token | null * @return token | null
*/ */
private String login(LoginReq req) { protected String login(LoginReq req) {
String str = null; String str = null;
try { try {
str = getMockData(post("/login"), null, req) str = getMockData(post("/login"), null, req)
@@ -208,6 +210,7 @@ public class BaseTest {
} }
if (content != null) { if (content != null) {
builder.content(mapper.writeValueAsString(content)).contentType(MediaType.APPLICATION_JSON); builder.content(mapper.writeValueAsString(content)).contentType(MediaType.APPLICATION_JSON);
logger.debug("param::json->{}", mapper.writeValueAsString(content));
} }
return mockMvc.perform(builder).andExpect(status().isOk()); return mockMvc.perform(builder).andExpect(status().isOk());
} }
@@ -230,7 +233,6 @@ public class BaseTest {
*/ */
protected <T> Response<T> getResponse(String json, TypeReference<?> responseType) { protected <T> Response<T> getResponse(String json, TypeReference<?> responseType) {
Response<T> response = null; Response<T> response = null;
System.out.println(responseType.getType());
try { try {
response = mapper.readValue(json, responseType); response = mapper.readValue(json, responseType);
} catch (IOException e) { } catch (IOException e) {

184
src/test/java/cn/celess/blog/controller/UserControllerTest.java
View File

@@ -1,6 +1,7 @@
package cn.celess.blog.controller; package cn.celess.blog.controller;
import cn.celess.blog.BaseTest; import cn.celess.blog.BaseTest;
import cn.celess.blog.entity.Response;
import cn.celess.blog.entity.User; import cn.celess.blog.entity.User;
import cn.celess.blog.entity.model.PageData; import cn.celess.blog.entity.model.PageData;
import cn.celess.blog.entity.model.UserModel; import cn.celess.blog.entity.model.UserModel;
@@ -8,12 +9,13 @@ import cn.celess.blog.entity.request.LoginReq;
import cn.celess.blog.entity.request.UserReq; import cn.celess.blog.entity.request.UserReq;
import cn.celess.blog.mapper.UserMapper; import cn.celess.blog.mapper.UserMapper;
import cn.celess.blog.util.MD5Util; import cn.celess.blog.util.MD5Util;
import net.sf.json.JSONArray; import cn.celess.blog.util.RedisUtil;
import net.sf.json.JSONObject; import com.fasterxml.jackson.core.type.TypeReference;
import org.junit.Test; import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType; import org.springframework.http.MediaType;
import org.springframework.mock.web.MockMultipartFile; import org.springframework.mock.web.MockMultipartFile;
import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
import org.springframework.util.LinkedMultiValueMap; import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap; import org.springframework.util.MultiValueMap;
@@ -21,16 +23,25 @@ import java.io.InputStream;
import java.net.HttpURLConnection; import java.net.HttpURLConnection;
import java.net.URL; import java.net.URL;
import java.util.*; import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;
import static cn.celess.blog.enmu.ResponseEnum.*;
import static org.junit.Assert.*; import static org.junit.Assert.*;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*; import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.*;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
import static cn.celess.blog.enmu.ResponseEnum.*;
public class UserControllerTest extends BaseTest { public class UserControllerTest extends BaseTest {
@Autowired @Autowired
UserMapper userMapper; UserMapper userMapper;
@Autowired
RedisUtil redisUtil;
private static final TypeReference<?> USER_MODEL_TYPE = new TypeReference<Response<UserModel>>() {
};
private static final TypeReference<?> USER_MODEL_PAGE_TYPE = new TypeReference<Response<PageData<UserModel>>>() {
};
private static final TypeReference<?> USER_MODEL_LIST_TYPE = new TypeReference<Response<List<Map<String, Object>>>>() {
};
@Test @Test
public void login() throws Exception { public void login() throws Exception {
@@ -41,11 +52,7 @@ public class UserControllerTest extends BaseTest {
req.setEmail("zh@celess.cn"); req.setEmail("zh@celess.cn");
req.setPassword("123456789"); req.setPassword("123456789");
req.setIsRememberMe(false); req.setIsRememberMe(false);
JSONObject loginReq = JSONObject.fromObject(req); getMockData(post("/login"), null, req).andDo(result -> assertEquals(USER_NOT_EXIST.getCode(), getResponse(result, STRING_TYPE).getCode()));
mockMvc.perform(post("/login").content(loginReq.toString()).contentType("application/json"))
.andDo(result ->
assertEquals(USER_NOT_EXIST.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))
);
} }
@Test @Test
@@ -55,19 +62,18 @@ public class UserControllerTest extends BaseTest {
@Test @Test
public void logout() throws Exception { public void logout() throws Exception {
mockMvc.perform(get("/logout")).andDo(result -> assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))); getMockData(get("/logout")).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result, STRING_TYPE).getCode()));
mockMvc.perform(get("/logout").header("Authorization", userLogin())).andDo(result -> assertEquals(SUCCESS.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))); getMockData(get("/logout"), adminLogin()).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result).getCode()));
} }
@Test @Test
public void updateInfo() throws Exception { public void updateInfo() throws Exception {
String desc = UUID.randomUUID().toString().substring(0, 4); String desc = UUID.randomUUID().toString().substring(0, 4);
String disPlayName = UUID.randomUUID().toString().substring(0, 4); String disPlayName = UUID.randomUUID().toString().substring(0, 4);
mockMvc.perform(put("/user/userInfo/update?desc=" + desc + "&displayName=" + disPlayName).header("Authorization", userLogin())) getMockData(put("/user/userInfo/update?desc=" + desc + "&displayName=" + disPlayName), userLogin()).andDo(result -> {
.andDo(result -> { Response<UserModel> response = getResponse(result, USER_MODEL_TYPE);
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString()); assertEquals(SUCCESS.getCode(), response.getCode());
assertEquals(SUCCESS.getCode(), object.getInt(Code)); UserModel u = response.getResult();
UserModel u = (UserModel) JSONObject.toBean(object.getJSONObject(Result), UserModel.class);
assertEquals(desc, u.getDesc()); assertEquals(desc, u.getDesc());
assertEquals(disPlayName, u.getDisplayName()); assertEquals(disPlayName, u.getDisplayName());
assertNotNull(u.getId()); assertNotNull(u.getId());
@@ -76,11 +82,10 @@ public class UserControllerTest extends BaseTest {
@Test @Test
public void getUserInfo() throws Exception { public void getUserInfo() throws Exception {
mockMvc.perform(get("/user/userInfo").header("Authorization", userLogin())) getMockData(get("/user/userInfo"), userLogin()).andDo(result -> {
.andDo(result -> { Response<UserModel> response = getResponse(result, USER_MODEL_TYPE);
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString()); assertEquals(SUCCESS.getCode(), response.getCode());
assertEquals(SUCCESS.getCode(), object.getInt(Code)); UserModel u = response.getResult();
UserModel u = (UserModel) JSONObject.toBean(object.getJSONObject(Result), UserModel.class);
assertNotNull(u.getId()); assertNotNull(u.getId());
assertNotNull(u.getEmail()); assertNotNull(u.getEmail());
assertNotNull(u.getDisplayName()); assertNotNull(u.getDisplayName());
@@ -100,11 +105,10 @@ public class UserControllerTest extends BaseTest {
InputStream inputStream = connection.getInputStream(); InputStream inputStream = connection.getInputStream();
assertNotNull(inputStream); assertNotNull(inputStream);
MockMultipartFile file = new MockMultipartFile("file", "logo.png", MediaType.IMAGE_PNG_VALUE, inputStream); MockMultipartFile file = new MockMultipartFile("file", "logo.png", MediaType.IMAGE_PNG_VALUE, inputStream);
mockMvc.perform(multipart("/user/imgUpload").file(file)).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))); getMockData(multipart("/user/imgUpload").file(file), userLogin()).andDo(result -> {
mockMvc.perform(multipart("/user/imgUpload").file(file).header("Authorization", userLogin())).andDo(result -> { Response<Object> response = getResponse(result, OBJECT_TYPE);
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString()); assertEquals(SUCCESS.getCode(), response.getCode());
assertEquals(SUCCESS.getCode(), object.getInt(Code)); assertNotNull(response.getResult());
assertNotNull(object.getString(Result));
}); });
} }
@@ -119,13 +123,34 @@ public class UserControllerTest extends BaseTest {
} }
@Test @Test
public void emailVerify() { public void emailVerify() throws Exception {
// ignore String email = randomStr(4) + "@celess.cn";
String pwd = MD5Util.getMD5("123456789");
userMapper.addUser(new User(email, pwd));
String verifyId = randomStr();
LoginReq req = new LoginReq(email, "123456789", true);
redisUtil.setEx(email + "-verify", verifyId, 2, TimeUnit.DAYS);
getMockData(post("/emailVerify").param("verifyId", verifyId).param("email", email), login(req)).andDo(result ->
assertEquals(SUCCESS.getCode(), getResponse(result, OBJECT_TYPE).getCode())
);
} }
@Test @Test
public void resetPwd() { public void resetPwd() throws Exception {
// ignore String email = randomStr(4) + "@celess.cn";
String pwd = MD5Util.getMD5("1234567890");
userMapper.addUser(new User(email, pwd));
LoginReq req = new LoginReq(email, "1234567890", true);
String verifyId = randomStr();
// 设置验证id
redisUtil.setEx(email + "-resetPwd", verifyId, 2, TimeUnit.DAYS);
MockHttpServletRequestBuilder resetPwd = post("/resetPwd").param("verifyId", verifyId).param("email", email).param("pwd", "123456789");
// 未验证
getMockData(resetPwd, login(req)).andDo(result -> assertEquals(USEREMAIL_NOT_VERIFY.getCode(), getResponse(result, OBJECT_TYPE).getCode()));
// 设置未验证
userMapper.updateEmailStatus(email, true);
// 正常
getMockData(resetPwd, login(req)).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result, OBJECT_TYPE).getCode()));
} }
@Test @Test
@@ -146,25 +171,16 @@ public class UserControllerTest extends BaseTest {
userMapper.setUserRole(userMapper.findByEmail(email).getId(), "admin"); userMapper.setUserRole(userMapper.findByEmail(email).getId(), "admin");
} }
} }
List<Long> idList = new ArrayList<>(); List<Integer> idList = userList.stream().map(user -> user.getId().intValue()).collect(Collectors.toList());
userList.forEach(user -> idList.add(user.getId())); getMockData(delete("/admin/user/delete"), adminLogin(), idList).andDo(result -> {
System.out.println("id :: == > " + idList.toString()); Response<List<Map<String, Object>>> response = getResponse(result, USER_MODEL_LIST_TYPE);
mockMvc.perform(delete("/admin/user/delete").content(idList.toString()).contentType("application/json")) assertEquals(SUCCESS.getCode(), response.getCode());
.andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))); response.getResult().forEach(o -> {
mockMvc.perform(delete("/admin/user/delete").content(idList.toString()).contentType("application/json").header("Authorization", userLogin()))
.andDo(result -> assertEquals(PERMISSION_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
mockMvc.perform(delete("/admin/user/delete").content(idList.toString()).contentType("application/json").header("Authorization", adminLogin()))
.andDo(result -> {
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
assertEquals(SUCCESS.getCode(), object.getInt(Code));
JSONArray jsonArray = object.getJSONArray(Result);
jsonArray.forEach(o -> {
JSONObject json = JSONObject.fromObject(o);
// 判断响应数据中是否包含输入的id // 判断响应数据中是否包含输入的id
assertTrue(idList.contains((long) json.getInt("id"))); assertTrue(idList.contains((int) o.get("id")));
// 判断处理状态 // 判断处理状态
boolean status = json.getBoolean("status"); boolean status = (boolean) o.get("status");
if (json.containsKey("msg")) if (o.containsKey("msg"))
assertFalse(status); assertFalse(status);
else else
assertTrue(status); assertTrue(status);
@@ -176,26 +192,21 @@ public class UserControllerTest extends BaseTest {
@Test @Test
public void updateInfoByAdmin() throws Exception { public void updateInfoByAdmin() throws Exception {
UserReq userReq = new UserReq(); UserReq userReq = new UserReq();
String email = UUID.randomUUID().toString().substring(0, 4) + "@celess.cn"; String email = randomStr(4) + "@celess.cn";
User user = new User(email, MD5Util.getMD5("123456789")); User user = new User(email, MD5Util.getMD5("123456789"));
userMapper.addUser(user); userMapper.addUser(user);
User userByDb = userMapper.findByEmail(email); User userByDb = userMapper.findByEmail(email);
userReq.setId(userByDb.getId()); userReq.setId(userByDb.getId());
userReq.setPwd(UUID.randomUUID().toString().replaceAll("-", "").substring(0, 10)); userReq.setPwd(randomStr().substring(0, 10));
userReq.setDesc(UUID.randomUUID().toString()); userReq.setDesc(randomStr());
userReq.setEmailStatus(new Random().nextBoolean()); userReq.setEmailStatus(new Random().nextBoolean());
userReq.setRole("admin"); userReq.setRole("admin");
userReq.setDisplayName(UUID.randomUUID().toString().substring(0, 4)); userReq.setDisplayName(randomStr(4));
userReq.setEmail(UUID.randomUUID().toString().substring(0, 5) + "@celess.cn"); userReq.setEmail(randomStr().substring(0, 5) + "@celess.cn");
mockMvc.perform(put("/admin/user").contentType("application/json").content(JSONObject.fromObject(userReq).toString())) getMockData(put("/admin/user"), adminLogin(), userReq).andDo(result -> {
.andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))); Response<UserModel> response = getResponse(result, USER_MODEL_TYPE);
mockMvc.perform(put("/admin/user").contentType("application/json").header("Authorization", userLogin()).content(JSONObject.fromObject(userReq).toString())) assertEquals(SUCCESS.getCode(), response.getCode());
.andDo(result -> assertEquals(PERMISSION_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))); UserModel userModel = response.getResult();
mockMvc.perform(put("/admin/user").contentType("application/json").header("Authorization", adminLogin()).content(JSONObject.fromObject(userReq).toString()))
.andDo(result -> {
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
assertEquals(SUCCESS.getCode(), object.getInt(Code));
UserModel userModel = (UserModel) JSONObject.toBean(object.getJSONObject(Result), UserModel.class);
assertEquals(userReq.getId(), userModel.getId()); assertEquals(userReq.getId(), userModel.getId());
assertEquals(userReq.getRole(), userModel.getRole()); assertEquals(userReq.getRole(), userModel.getRole());
assertEquals(userReq.getEmail(), userModel.getEmail()); assertEquals(userReq.getEmail(), userModel.getEmail());
@@ -206,25 +217,18 @@ public class UserControllerTest extends BaseTest {
@Test @Test
public void getAllUser() throws Exception { public void getAllUser() throws Exception {
mockMvc.perform(get("/admin/users?page=1&count=10")) getMockData(get("/admin/users?page=1&count=10"), adminLogin()).andDo(result -> {
.andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code))); Response<PageData<UserModel>> response = getResponse(result, USER_MODEL_PAGE_TYPE);
mockMvc.perform(get("/admin/users?page=1&count=10").header("authorization", userLogin())) assertEquals(SUCCESS.getCode(), response.getCode());
.andDo(result -> assertEquals(PERMISSION_ERROR.getCode(), JSONObject.fromObject(result.getResponse().getContentAsString()).getInt(Code)));
mockMvc.perform(get("/admin/users?page=1&count=10").header("Authorization", adminLogin()))
.andDo(result -> {
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
assertEquals(SUCCESS.getCode(), object.getInt(Code));
// 结果集非空 // 结果集非空
assertNotNull(object.getJSONObject(Result)); assertNotNull(response.getResult());
// 判断pageInfo是否包装完全 // 判断pageInfo是否包装完全
JSONObject resultJson = JSONObject.fromObject(object.getJSONObject(Result)); PageData<UserModel> pageData = response.getResult();
PageData<UserModel> pageData = (PageData<UserModel>) JSONObject.toBean(resultJson, PageData.class);
assertNotEquals(0, pageData.getTotal()); assertNotEquals(0, pageData.getTotal());
assertEquals(1, pageData.getPageNum()); assertEquals(1, pageData.getPageNum());
assertEquals(10, pageData.getPageSize()); assertEquals(10, pageData.getPageSize());
// 内容完整 // 内容完整
for (Object user : pageData.getList()) { for (UserModel u : pageData.getList()) {
UserModel u = (UserModel) JSONObject.toBean(JSONObject.fromObject(user), UserModel.class);
assertNotNull(u.getId()); assertNotNull(u.getId());
assertNotNull(u.getEmail()); assertNotNull(u.getEmail());
assertNotNull(u.getRole()); assertNotNull(u.getRole());
@@ -236,40 +240,24 @@ public class UserControllerTest extends BaseTest {
@Test @Test
public void getEmailStatus() throws Exception { public void getEmailStatus() throws Exception {
String email = UUID.randomUUID().toString().substring(0, 4) + "@celess.cn"; String email = randomStr(4) + "@celess.cn";
mockMvc.perform(get("/emailStatus/" + email)).andDo(result -> { getMockData(get("/emailStatus/" + email)).andDo(result -> assertFalse((Boolean) getResponse(result, BOOLEAN_TYPE).getResult()));
String content = result.getResponse().getContentAsString(); getMockData(get("/emailStatus/" + ADMIN_EMAIL)).andDo(result -> assertTrue((Boolean) getResponse(result, BOOLEAN_TYPE).getResult()));
assertEquals(SUCCESS.getCode(), JSONObject.fromObject(content).getInt(Code));
assertFalse(JSONObject.fromObject(content).getBoolean(Result));
});
email = "a@celess.cn";
mockMvc.perform(get("/emailStatus/" + email)).andDo(result -> {
String content = result.getResponse().getContentAsString();
assertEquals(SUCCESS.getCode(), JSONObject.fromObject(content).getInt(Code));
assertTrue(JSONObject.fromObject(content).getBoolean(Result));
});
} }
@Test @Test
public void setPwd() throws Exception { public void setPwd() throws Exception {
String email = UUID.randomUUID().toString().substring(0, 4) + "@celess.cn"; String email = randomStr(4) + "@celess.cn";
assertEquals(1, userMapper.addUser(new User(email, MD5Util.getMD5("1234567890")))); assertEquals(1, userMapper.addUser(new User(email, MD5Util.getMD5("1234567890"))));
LoginReq req = new LoginReq(); LoginReq req = new LoginReq(email, "1234567890", false);
req.setEmail(email); String token = login(req);
req.setPassword("1234567890");
req.setIsRememberMe(false);
JSONObject loginReq = JSONObject.fromObject(req);
String contentAsString = mockMvc.perform(post("/login").content(loginReq.toString()).contentType("application/json")).andReturn().getResponse().getContentAsString();
assertNotNull(contentAsString);
String token = JSONObject.fromObject(contentAsString).getJSONObject(Result).getString("token");
assertNotNull(token); assertNotNull(token);
MultiValueMap<String, String> param = new LinkedMultiValueMap<String, String>(); MultiValueMap<String, String> param = new LinkedMultiValueMap<String, String>();
param.add("pwd", "1234567890"); param.add("pwd", "1234567890");
param.add("newPwd", "aaabbbccc"); param.add("newPwd", "aaabbbccc");
param.add("confirmPwd", "aaabbbccc"); param.add("confirmPwd", "aaabbbccc");
mockMvc.perform(post("/user/setPwd").header("Authorization", token).params(param)).andDo(result -> { getMockData(post("/user/setPwd").params(param), token).andDo(result -> {
String content = result.getResponse().getContentAsString(); assertEquals(SUCCESS.getCode(), getResponse(result).getCode());
assertEquals(SUCCESS.getCode(), JSONObject.fromObject(content).getInt(Code));
assertEquals(MD5Util.getMD5("aaabbbccc"), userMapper.getPwd(email)); assertEquals(MD5Util.getMD5("aaabbbccc"), userMapper.getPwd(email));
}); });
} }