Java WithSecurityContextFactory类代码示例

import org.springframework.security.test.context.support.WithSecurityContextFactory; //导入依赖的package包/类
/**
 * Run the test corresponding to the child, which can be assumed to be an element
 * of the list returned by getChildren(). Ensures that relevant test events are
 * reported through the notifier.
 *
 * The implementation wraps the inherited runChild() method to insert the mock/test
 * user into the SecurityContext prior to execution and removes it after execution.
 *
 * @param frameworkMethod The method representing the child test.
 * @param notifier The notifier for the test execution.
 */
@SuppressWarnings("unchecked")
@Override
protected void runChild(FrameworkMethod frameworkMethod, RunNotifier notifier) {
    if (frameworkMethod instanceof AnnotationFrameworkMethod) {
        AnnotationFrameworkMethod annotationFrameworkMethod = (AnnotationFrameworkMethod) frameworkMethod;

        Annotation userAnnotation = annotationFrameworkMethod.getAnnotation();
        WithSecurityContext withSecurityContext = userAnnotation.annotationType().getAnnotation(WithSecurityContext.class);

        Class<? extends WithSecurityContextFactory<? extends Annotation>> clazz = withSecurityContext.factory();

        WithSecurityContextFactory withSecurityContextFactory = buildWithSecurityContextFactory(clazz);

        SecurityContext securityContext = null;
        if (withSecurityContextFactory != null) {
            securityContext = withSecurityContextFactory.createSecurityContext(userAnnotation);
        }

        if (securityContext == null) {
            securityContext = SecurityContextHolder.createEmptyContext();
        }
        SecurityContextHolder.setContext(securityContext);
    }
    super.runChild(frameworkMethod, notifier);

    SecurityContextHolder.clearContext();
}
 

import org.springframework.security.test.context.support.WithSecurityContextFactory; //导入依赖的package包/类
/**
 * Construct a new TestMethodContext with the provided details.
 * @param methodName The name of the method being tested.
 * @param withTestUsers A list of annotations representing mock/test users.
 * @param context The ApplicationContext for the test.
 */
public TestMethodContext(String methodName, List<Annotation> withTestUsers, ApplicationContext context) {
    this.methodName = methodName;

    WithSecurityContextFactory factory;
    for (Annotation withTestUser : withTestUsers) {
        factory = buildWithSecurityContextFactory(withTestUser, context);
        executionStack.add((User)factory.createSecurityContext(withTestUser).getAuthentication().getPrincipal());
    }
}
 

import org.springframework.security.test.context.support.WithSecurityContextFactory; //导入依赖的package包/类
/**
 * Construct a new WithSecurityContextFactory for the provided class name.
 * @param withUser The annotation representing the mock/test user.
 * @param context The ApplicationContext under which the test is executing.
 * @return The instance if it could be constructed, otherwise null.
 */
private WithSecurityContextFactory buildWithSecurityContextFactory(Annotation withUser, ApplicationContext context) {
    WithSecurityContext withSecurityContext = withUser.annotationType().getAnnotation(WithSecurityContext.class);

    Class<? extends WithSecurityContextFactory<? extends Annotation>> clazz = withSecurityContext.factory();

    return buildWithSecurityContextFactory(clazz, context);
}