Package org.springframework.context.annotation.scope

Source Code of org.springframework.context.annotation.scope.ClassPathBeanDefinitionScannerScopeIntegrationTests

/*
* Copyright 2002-2012 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
*
*      http://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.context.annotation.scope;

import org.junit.After;
import static org.junit.Assert.*;
import org.junit.Before;
import org.junit.Test;

import org.springframework.aop.support.AopUtils;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.beans.factory.support.BeanNameGenerator;
import org.springframework.context.ApplicationContext;
import org.springframework.context.annotation.ClassPathBeanDefinitionScanner;
import org.springframework.context.annotation.Scope;
import org.springframework.context.annotation.ScopedProxyMode;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpSession;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.springframework.web.context.support.GenericWebApplicationContext;

/**
* @author Mark Fisher
* @author Juergen Hoeller
* @author Chris Beams
*/
public class ClassPathBeanDefinitionScannerScopeIntegrationTests {

  private static final String DEFAULT_NAME = "default";

  private static final String MODIFIED_NAME = "modified";

  private ServletRequestAttributes oldRequestAttributes;

  private ServletRequestAttributes newRequestAttributes;

  private ServletRequestAttributes oldRequestAttributesWithSession;

  private ServletRequestAttributes newRequestAttributesWithSession;


  @Before
  public void setUp() {
    this.oldRequestAttributes = new ServletRequestAttributes(new MockHttpServletRequest());
    this.newRequestAttributes = new ServletRequestAttributes(new MockHttpServletRequest());

    MockHttpServletRequest oldRequestWithSession = new MockHttpServletRequest();
    oldRequestWithSession.setSession(new MockHttpSession());
    this.oldRequestAttributesWithSession = new ServletRequestAttributes(oldRequestWithSession);

    MockHttpServletRequest newRequestWithSession = new MockHttpServletRequest();
    newRequestWithSession.setSession(new MockHttpSession());
    this.newRequestAttributesWithSession = new ServletRequestAttributes(newRequestWithSession);
  }

  @After
  public void tearDown() throws Exception {
    RequestContextHolder.setRequestAttributes(null);
  }


  @Test
  public void testSingletonScopeWithNoProxy() {
    RequestContextHolder.setRequestAttributes(oldRequestAttributes);
    ApplicationContext context = createContext(ScopedProxyMode.NO);
    ScopedTestBean bean = (ScopedTestBean) context.getBean("singleton");

    // should not be a proxy
    assertFalse(AopUtils.isAopProxy(bean));

    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    RequestContextHolder.setRequestAttributes(newRequestAttributes);
    // not a proxy so this should not have changed
    assertEquals(MODIFIED_NAME, bean.getName());

    // singleton bean, so name should be modified even after lookup
    ScopedTestBean bean2 = (ScopedTestBean) context.getBean("singleton");
    assertEquals(MODIFIED_NAME, bean2.getName());
  }

  @Test
  public void testSingletonScopeIgnoresProxyInterfaces() {
    RequestContextHolder.setRequestAttributes(oldRequestAttributes);
    ApplicationContext context = createContext(ScopedProxyMode.INTERFACES);
    ScopedTestBean bean = (ScopedTestBean) context.getBean("singleton");

    // should not be a proxy
    assertFalse(AopUtils.isAopProxy(bean));

    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    RequestContextHolder.setRequestAttributes(newRequestAttributes);
    // not a proxy so this should not have changed
    assertEquals(MODIFIED_NAME, bean.getName());

    // singleton bean, so name should be modified even after lookup
    ScopedTestBean bean2 = (ScopedTestBean) context.getBean("singleton");
    assertEquals(MODIFIED_NAME, bean2.getName());
  }

  @Test
  public void testSingletonScopeIgnoresProxyTargetClass() {
    RequestContextHolder.setRequestAttributes(oldRequestAttributes);
    ApplicationContext context = createContext(ScopedProxyMode.TARGET_CLASS);
    ScopedTestBean bean = (ScopedTestBean) context.getBean("singleton");

    // should not be a proxy
    assertFalse(AopUtils.isAopProxy(bean));

    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    RequestContextHolder.setRequestAttributes(newRequestAttributes);
    // not a proxy so this should not have changed
    assertEquals(MODIFIED_NAME, bean.getName());

    // singleton bean, so name should be modified even after lookup
    ScopedTestBean bean2 = (ScopedTestBean) context.getBean("singleton");
    assertEquals(MODIFIED_NAME, bean2.getName());
  }

  @Test
  public void testRequestScopeWithNoProxy() {
    RequestContextHolder.setRequestAttributes(oldRequestAttributes);
    ApplicationContext context = createContext(ScopedProxyMode.NO);
    ScopedTestBean bean = (ScopedTestBean) context.getBean("request");

    // should not be a proxy
    assertFalse(AopUtils.isAopProxy(bean));

    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    RequestContextHolder.setRequestAttributes(newRequestAttributes);
    // not a proxy so this should not have changed
    assertEquals(MODIFIED_NAME, bean.getName());

    // but a newly retrieved bean should have the default name
    ScopedTestBean bean2 = (ScopedTestBean) context.getBean("request");
    assertEquals(DEFAULT_NAME, bean2.getName());
  }

  @Test
  public void testRequestScopeWithProxiedInterfaces() {
    RequestContextHolder.setRequestAttributes(oldRequestAttributes);
    ApplicationContext context = createContext(ScopedProxyMode.INTERFACES);
    IScopedTestBean bean = (IScopedTestBean) context.getBean("request");

    // should be dynamic proxy, implementing both interfaces
    assertTrue(AopUtils.isJdkDynamicProxy(bean));
    assertTrue(bean instanceof AnotherScopeTestInterface);

    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    RequestContextHolder.setRequestAttributes(newRequestAttributes);
    // this is a proxy so it should be reset to default
    assertEquals(DEFAULT_NAME, bean.getName());

    RequestContextHolder.setRequestAttributes(oldRequestAttributes);
    assertEquals(MODIFIED_NAME, bean.getName());
  }

  @Test
  public void testRequestScopeWithProxiedTargetClass() {
    RequestContextHolder.setRequestAttributes(oldRequestAttributes);
    ApplicationContext context = createContext(ScopedProxyMode.TARGET_CLASS);
    IScopedTestBean bean = (IScopedTestBean) context.getBean("request");

    // should be a class-based proxy
    assertTrue(AopUtils.isCglibProxy(bean));
    assertTrue(bean instanceof RequestScopedTestBean);

    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    RequestContextHolder.setRequestAttributes(newRequestAttributes);
    // this is a proxy so it should be reset to default
    assertEquals(DEFAULT_NAME, bean.getName());

    RequestContextHolder.setRequestAttributes(oldRequestAttributes);
    assertEquals(MODIFIED_NAME, bean.getName());
  }

  @Test
  public void testSessionScopeWithNoProxy() {
    RequestContextHolder.setRequestAttributes(oldRequestAttributesWithSession);
    ApplicationContext context = createContext(ScopedProxyMode.NO);
    ScopedTestBean bean = (ScopedTestBean) context.getBean("session");

    // should not be a proxy
    assertFalse(AopUtils.isAopProxy(bean));

    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    RequestContextHolder.setRequestAttributes(newRequestAttributesWithSession);
    // not a proxy so this should not have changed
    assertEquals(MODIFIED_NAME, bean.getName());

    // but a newly retrieved bean should have the default name
    ScopedTestBean bean2 = (ScopedTestBean) context.getBean("session");
    assertEquals(DEFAULT_NAME, bean2.getName());
  }

  @Test
  public void testSessionScopeWithProxiedInterfaces() {
    RequestContextHolder.setRequestAttributes(oldRequestAttributesWithSession);
    ApplicationContext context = createContext(ScopedProxyMode.INTERFACES);
    IScopedTestBean bean = (IScopedTestBean) context.getBean("session");

    // should be dynamic proxy, implementing both interfaces
    assertTrue(AopUtils.isJdkDynamicProxy(bean));
    assertTrue(bean instanceof AnotherScopeTestInterface);

    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    RequestContextHolder.setRequestAttributes(newRequestAttributesWithSession);
    // this is a proxy so it should be reset to default
    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    IScopedTestBean bean2 = (IScopedTestBean) context.getBean("session");
    assertEquals(MODIFIED_NAME, bean2.getName());
    bean2.setName(DEFAULT_NAME);
    assertEquals(DEFAULT_NAME, bean.getName());

    RequestContextHolder.setRequestAttributes(oldRequestAttributesWithSession);
    assertEquals(MODIFIED_NAME, bean.getName());
  }

  @Test
  public void testSessionScopeWithProxiedTargetClass() {
    RequestContextHolder.setRequestAttributes(oldRequestAttributesWithSession);
    ApplicationContext context = createContext(ScopedProxyMode.TARGET_CLASS);
    IScopedTestBean bean = (IScopedTestBean) context.getBean("session");

    // should be a class-based proxy
    assertTrue(AopUtils.isCglibProxy(bean));
    assertTrue(bean instanceof ScopedTestBean);
    assertTrue(bean instanceof SessionScopedTestBean);

    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    RequestContextHolder.setRequestAttributes(newRequestAttributesWithSession);
    // this is a proxy so it should be reset to default
    assertEquals(DEFAULT_NAME, bean.getName());
    bean.setName(MODIFIED_NAME);

    IScopedTestBean bean2 = (IScopedTestBean) context.getBean("session");
    assertEquals(MODIFIED_NAME, bean2.getName());
    bean2.setName(DEFAULT_NAME);
    assertEquals(DEFAULT_NAME, bean.getName());

    RequestContextHolder.setRequestAttributes(oldRequestAttributesWithSession);
    assertEquals(MODIFIED_NAME, bean.getName());
  }


  private ApplicationContext createContext(ScopedProxyMode scopedProxyMode) {
    GenericWebApplicationContext context = new GenericWebApplicationContext();
    ClassPathBeanDefinitionScanner scanner = new ClassPathBeanDefinitionScanner(context);
    scanner.setIncludeAnnotationConfig(false);
    scanner.setBeanNameGenerator(new BeanNameGenerator() {
      @Override
      public String generateBeanName(BeanDefinition definition, BeanDefinitionRegistry registry) {
        return definition.getScope();
      }
    });
    scanner.setScopedProxyMode(scopedProxyMode);

    // Scan twice in order to find errors in the bean definition compatibility check.
    scanner.scan(getClass().getPackage().getName());
    scanner.scan(getClass().getPackage().getName());

    context.refresh();
    return context;
  }


   public static interface IScopedTestBean {

     String getName();

     void setName(String name);
   }


  public static abstract class ScopedTestBean implements IScopedTestBean {

    private String name = DEFAULT_NAME;

    @Override
    public String getName() { return this.name; }

    @Override
    public void setName(String name) { this.name = name; }
  }


  @Component
  public static class SingletonScopedTestBean extends ScopedTestBean {
  }


  public static interface AnotherScopeTestInterface {
  }


  @Component
  @Scope("request")
  public static class RequestScopedTestBean extends ScopedTestBean implements AnotherScopeTestInterface {
  }


  @Component
  @Scope("session")
  public static class SessionScopedTestBean extends ScopedTestBean implements AnotherScopeTestInterface {
  }

}
TOP

Related Classes of org.springframework.context.annotation.scope.ClassPathBeanDefinitionScannerScopeIntegrationTests

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.