我想编写某种依赖于 Spring Security 的单元测试。
例如,我有一些服务方法,它使用一些存储库并用 @PreAuthorize 注释标记。存储库我可以用 Mockito 模拟,没有问题。我也可以通过@WithSecurityContext注释来模拟安全上下文。但是当我运行测试时,@PreAuthorize 注释被忽略了。当然,我可以使用 @SpringBootTest 注释作为集成测试运行该测试,在这种情况下,安全上下文已启动,但这种方式既繁重又缓慢。
有没有办法在只引发 Spring Security Context 的情况下运行单元测试?
更新
做了一个这样的测试的例子。感谢@Sam Brannen 给出正确的方向。
@ActiveProfiles("method-security-test")
@RunWith(SpringRunner.class)
@SpringBootTest(classes = {ExampleService.class, ExampleServiceTest.MethodSecurityConfiguration.class})
public class ExampleServiceTest {
private ExampleService service;
@Autowired
public void setService(ExampleService service) {
this.service = service;
}
@Test
@WithMockUser(username = "john_doe")
public void testAuthenticated() {
String actualMessage = service.example();
Assert.assertEquals("Message of john_doe", actualMessage);
}
@Test(expected = AuthenticationException.class)
public void testNotAuthenticated() {
service.example();
Assert.fail();
}
@TestConfiguration
@EnableGlobalMethodSecurity(prePostEnabled = true)
static class MethodSecurityConfiguration extends GlobalMethodSecurityConfiguration {
}
}
@Service
class ExampleService {
@PreAuthorize("isAuthenticated()")
String example() {
Principal principal = SecurityContextHolder.getContext().getAuthentication();
return "Message of " + principal.getName();
}