Browse Source
- Add support for request variables - Added additional tests Issue gh-11105pull/11230/head
3 changed files with 138 additions and 0 deletions
@ -0,0 +1,94 @@ |
|||||||
|
/* |
||||||
|
* Copyright 2002-2016 the original author or authors. |
||||||
|
* |
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
||||||
|
* you may not use this file except in compliance with the License. |
||||||
|
* You may obtain a copy of the License at |
||||||
|
* |
||||||
|
* https://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
* |
||||||
|
* Unless required by applicable law or agreed to in writing, software |
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS, |
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
||||||
|
* See the License for the specific language governing permissions and |
||||||
|
* limitations under the License. |
||||||
|
*/ |
||||||
|
|
||||||
|
package org.springframework.security.web.access.expression; |
||||||
|
|
||||||
|
import org.junit.jupiter.api.AfterEach; |
||||||
|
import org.junit.jupiter.api.BeforeEach; |
||||||
|
import org.junit.jupiter.api.Test; |
||||||
|
import org.junit.jupiter.api.extension.ExtendWith; |
||||||
|
import org.mockito.Mock; |
||||||
|
import org.mockito.junit.jupiter.MockitoExtension; |
||||||
|
|
||||||
|
import org.springframework.beans.factory.support.RootBeanDefinition; |
||||||
|
import org.springframework.context.support.StaticApplicationContext; |
||||||
|
import org.springframework.expression.EvaluationContext; |
||||||
|
import org.springframework.expression.Expression; |
||||||
|
import org.springframework.expression.ExpressionParser; |
||||||
|
import org.springframework.security.access.SecurityConfig; |
||||||
|
import org.springframework.security.authentication.AuthenticationTrustResolver; |
||||||
|
import org.springframework.security.core.Authentication; |
||||||
|
import org.springframework.security.core.context.SecurityContextHolder; |
||||||
|
import org.springframework.security.web.access.intercept.RequestAuthorizationContext; |
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat; |
||||||
|
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException; |
||||||
|
import static org.mockito.Mockito.mock; |
||||||
|
import static org.mockito.Mockito.verify; |
||||||
|
|
||||||
|
@ExtendWith(MockitoExtension.class) |
||||||
|
public class DefaultHttpSecurityExpressionHandlerTests { |
||||||
|
|
||||||
|
@Mock |
||||||
|
private AuthenticationTrustResolver trustResolver; |
||||||
|
|
||||||
|
@Mock |
||||||
|
private Authentication authentication; |
||||||
|
|
||||||
|
@Mock |
||||||
|
private RequestAuthorizationContext context; |
||||||
|
|
||||||
|
private DefaultHttpSecurityExpressionHandler handler; |
||||||
|
|
||||||
|
@BeforeEach |
||||||
|
public void setup() { |
||||||
|
this.handler = new DefaultHttpSecurityExpressionHandler(); |
||||||
|
} |
||||||
|
|
||||||
|
@AfterEach |
||||||
|
public void cleanup() { |
||||||
|
SecurityContextHolder.clearContext(); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void expressionPropertiesAreResolvedAgainstAppContextBeans() { |
||||||
|
StaticApplicationContext appContext = new StaticApplicationContext(); |
||||||
|
RootBeanDefinition bean = new RootBeanDefinition(SecurityConfig.class); |
||||||
|
bean.getConstructorArgumentValues().addGenericArgumentValue("ROLE_A"); |
||||||
|
appContext.registerBeanDefinition("role", bean); |
||||||
|
this.handler.setApplicationContext(appContext); |
||||||
|
EvaluationContext ctx = this.handler.createEvaluationContext(mock(Authentication.class), |
||||||
|
mock(RequestAuthorizationContext.class)); |
||||||
|
ExpressionParser parser = this.handler.getExpressionParser(); |
||||||
|
assertThat(parser.parseExpression("@role.getAttribute() == 'ROLE_A'").getValue(ctx, Boolean.class)).isTrue(); |
||||||
|
assertThat(parser.parseExpression("@role.attribute == 'ROLE_A'").getValue(ctx, Boolean.class)).isTrue(); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void setTrustResolverNull() { |
||||||
|
assertThatIllegalArgumentException().isThrownBy(() -> this.handler.setTrustResolver(null)); |
||||||
|
} |
||||||
|
|
||||||
|
@Test |
||||||
|
public void createEvaluationContextCustomTrustResolver() { |
||||||
|
this.handler.setTrustResolver(this.trustResolver); |
||||||
|
Expression expression = this.handler.getExpressionParser().parseExpression("anonymous"); |
||||||
|
EvaluationContext context = this.handler.createEvaluationContext(this.authentication, this.context); |
||||||
|
assertThat(expression.getValue(context, Boolean.class)).isFalse(); |
||||||
|
verify(this.trustResolver).isAnonymous(this.authentication); |
||||||
|
} |
||||||
|
|
||||||
|
} |
||||||
Loading…
Reference in new issue