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

refactor(Test): 重构 AuthorizationFilter

Browse Source
This commit is contained in:
禾几海
2020-08-15 12:10:48 +08:00
parent 938a051487
commit 3b6439c5cf
1 changed files with 18 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

59
src/test/java/cn/celess/blog/filter/AuthorizationFilter.java
View File

@@ -1,12 +1,12 @@
package cn.celess.blog.filter; package cn.celess.blog.filter;
import cn.celess.blog.BaseTest; import cn.celess.blog.BaseTest;
import net.sf.json.JSONObject; import cn.celess.blog.entity.Response;
import org.junit.Test; import org.junit.Test;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
import static org.junit.Assert.*;
import static cn.celess.blog.enmu.ResponseEnum.*; import static cn.celess.blog.enmu.ResponseEnum.*;
import static org.junit.Assert.*;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
/** /**
* @Author: 小海 * @Author: 小海
@@ -17,60 +17,37 @@ public class AuthorizationFilter extends BaseTest {
@Test @Test
public void UserAccess() throws Exception { public void UserAccess() throws Exception {
String token = "";
// 未登录 // 未登录
mockMvc.perform(get("/user/userInfo").header("Authorization", token)).andDo(result -> { getMockData(get("/user/userInfo")).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), getResponse(result).getCode()));
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString()); // user权限登录
assertEquals(HAVE_NOT_LOG_IN.getCode(), object.getInt(Code)); getMockData(get("/user/userInfo"), userLogin()).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result).getCode()));
});
token = userLogin();
mockMvc.perform(get("/user/userInfo").header("Authorization", token)).andDo(result -> {
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
assertEquals(SUCCESS.getCode(), object.getInt(Code));
});
} }
@Test @Test
public void AdminAccess() throws Exception { public void AdminAccess() throws Exception {
String token = "";
// 未登录 // 未登录
mockMvc.perform(get("/admin/articles?page=1&count=1").header("Authorization", token)).andDo(result -> { getMockData(get("/admin/articles?page=1&count=1")).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), getResponse(result).getCode()));
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString()); // user权限
assertEquals(HAVE_NOT_LOG_IN.getCode(), object.getInt(Code)); getMockData(get("/admin/articles?page=1&count=1"), userLogin()).andDo(result -> assertEquals(PERMISSION_ERROR.getCode(), getResponse(result).getCode()));
}); // admin 权限
token = userLogin(); getMockData(get("/admin/articles?page=1&count=1"), adminLogin()).andDo(result -> assertEquals(SUCCESS.getCode(), getResponse(result).getCode()));
mockMvc.perform(get("/admin/articles?page=1&count=1").header("Authorization", token)).andDo(result -> {
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
assertEquals(PERMISSION_ERROR.getCode(), object.getInt(Code));
});
token = adminLogin();
mockMvc.perform(get("/admin/articles?page=1&count=1").header("Authorization", token)).andDo(result -> {
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
assertEquals(SUCCESS.getCode(), object.getInt(Code));
});
} }
@Test @Test
public void VisitorAccess() throws Exception { public void VisitorAccess() throws Exception {
mockMvc.perform(get("/user/userInfo")).andDo(result -> { getMockData(get("/user/userInfo")).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), getResponse(result).getCode()));
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString()); getMockData(get("/admin/articles?page=1&count=1")).andDo(result -> assertEquals(HAVE_NOT_LOG_IN.getCode(), getResponse(result).getCode()));
assertEquals(HAVE_NOT_LOG_IN.getCode(), object.getInt(Code));
});
mockMvc.perform(get("/admin/articles?page=1&count=1")).andDo(result -> {
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString());
assertEquals(HAVE_NOT_LOG_IN.getCode(), object.getInt(Code));
});
} }
@Test @Test
public void authorizationTest() throws Exception { public void authorizationTest() throws Exception {
// 测试response中有无Authorization字段 // 测试response中有无Authorization字段
String s = userLogin(); String token = userLogin();
mockMvc.perform(get("/user/userInfo").header("Authorization", s)).andDo(result -> { getMockData(get("/user/userInfo"), token).andDo(result -> {
JSONObject object = JSONObject.fromObject(result.getResponse().getContentAsString()); Response<Object> response = getResponse(result);
assertEquals(SUCCESS.getCode(), object.getInt(Code)); assertEquals(SUCCESS.getCode(), response.getCode());
assertNotNull(result.getResponse().getHeader("Authorization")); assertNotNull(result.getResponse().getHeader("Authorization"));
assertNotEquals(s, result.getResponse().getHeader("Authorization")); assertNotEquals(token, result.getResponse().getHeader("Authorization"));
}); });
} }
} }