对于最新的稳定版本,请使用 Spring Security 6.5.3! |
在 Spring MVC 测试中以用户身份运行测试
通常希望以特定用户身份运行测试。 有两种简单的方法可以填充用户:
在使用 RequestPostProcessor 的 Spring MVC 测试中以用户身份运行
您可以使用许多选项将用户与当前HttpServletRequest
.
以下示例以用户名user
,其密码为password
,其角色是ROLE_USER
:
-
Java
-
Kotlin
mvc
.perform(get("/").with(user("user")))
mvc.get("/") {
with(user("user"))
}
该支持的工作原理是将用户关联到
|
您可以轻松进行自定义。 例如,以下内容将以用户名“admin”、密码“pass”以及角色“ROLE_USER”和“ROLE_ADMIN”的用户(不需要存在)身份运行。
-
Java
-
Kotlin
mvc
.perform(get("/admin").with(user("admin").password("pass").roles("USER","ADMIN")))
mvc.get("/admin") {
with(user("admin").password("pass").roles("USER","ADMIN"))
}
如果您有自定义UserDetails
你想使用的,你也可以轻松指定。
例如,以下将使用指定的UserDetails
(不需要存在)以使用UsernamePasswordAuthenticationToken
具有指定UserDetails
:
-
Java
-
Kotlin
mvc
.perform(get("/").with(user(userDetails)))
mvc.get("/") {
with(user(userDetails))
}
您可以使用以下命令以匿名用户身份运行:
-
Java
-
Kotlin
mvc
.perform(get("/").with(anonymous()))
mvc.get("/") {
with(anonymous())
}
如果您使用默认用户运行并希望以匿名用户身份处理一些请求,这将特别有用。
如果你想要一个自定义Authentication
(不需要存在),您可以使用以下命令来执行此作:
-
Java
-
Kotlin
mvc
.perform(get("/").with(authentication(authentication)))
mvc.get("/") {
with(authentication(authentication))
}
您甚至可以自定义SecurityContext
使用以下内容:
-
Java
-
Kotlin
mvc
.perform(get("/").with(securityContext(securityContext)))
mvc.get("/") {
with(securityContext(securityContext))
}
我们还可以通过使用MockMvcBuilders
的默认请求。例如,以下内容将以用户名“admin”、密码“password”和角色“ROLE_ADMIN”的用户(不需要存在)身份运行:
-
Java
-
Kotlin
mvc = MockMvcBuilders
.webAppContextSetup(context)
.defaultRequest(get("/").with(user("user").roles("ADMIN")))
.apply(springSecurity())
.build();
mvc = MockMvcBuilders
.webAppContextSetup(context)
.defaultRequest<DefaultMockMvcBuilder>(get("/").with(user("user").roles("ADMIN")))
.apply<DefaultMockMvcBuilder>(springSecurity())
.build()
如果您发现在许多测试中使用了同一用户,建议将用户移动到一个方法。例如,您可以在自己的类中指定以下名为CustomSecurityMockMvcRequestPostProcessors
:
-
Java
-
Kotlin
public static RequestPostProcessor rob() {
return user("rob").roles("ADMIN");
}
fun rob(): RequestPostProcessor {
return user("rob").roles("ADMIN")
}
现在,您可以在CustomSecurityMockMvcRequestPostProcessors
并在测试中使用它:
-
Java
-
Kotlin
import static sample.CustomSecurityMockMvcRequestPostProcessors.*;
...
mvc
.perform(get("/").with(rob()))
import sample.CustomSecurityMockMvcRequestPostProcessors.*
//...
mvc.get("/") {
with(rob())
}
在 Spring MVC 测试中使用注释以用户身份运行
作为使用RequestPostProcessor
要创建用户,可以使用测试方法安全性中所述的注释。
例如,以下内容将对用户名为“user”、密码为“password”和角色为“ROLE_USER”的用户运行测试:
-
Java
-
Kotlin
@Test
@WithMockUser
public void requestProtectedUrlWithUser() throws Exception {
mvc
.perform(get("/"))
...
}
@Test
@WithMockUser
fun requestProtectedUrlWithUser() {
mvc
.get("/")
// ...
}
或者,以下作将对用户名为“user”、密码为“password”和角色为“ROLE_ADMIN”的用户运行测试:
-
Java
-
Kotlin
@Test
@WithMockUser(roles="ADMIN")
public void requestProtectedUrlWithUser() throws Exception {
mvc
.perform(get("/"))
...
}
@Test
@WithMockUser(roles = ["ADMIN"])
fun requestProtectedUrlWithUser() {
mvc
.get("/")
// ...
}