Java Code Examples for org.springframework.web.context.WebApplicationContext#getServletContext()

The following examples show how to use org.springframework.web.context.WebApplicationContext#getServletContext() . You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.

Example 1

Source File: EurekaStore.java From qconfig with MIT License

6 votes

@Override
public void onApplicationEvent(ContextRefreshedEvent event) {
    ApplicationContext applicationContext = event.getApplicationContext();
    if (applicationContext.getParent() != null) {
        return;
    }

    logger.info("init eureka store");
    try {
        context = (WebApplicationContext) applicationContext;

        initEurekaEnvironment();
        initEurekaServerContext();

        ServletContext sc = context.getServletContext();
        sc.setAttribute(EurekaServerContext.class.getName(), serverContext);
    } catch (Throwable e) {
        logger.error("Cannot bootstrap eureka server :", e);
        throw new RuntimeException("Cannot bootstrap eureka server :", e);
    }
}

Example 2

Source File: AbstractMockMvcBuilder.java From java-technology-stack with MIT License

6 votes

/**
 * Build a {@link org.springframework.test.web.servlet.MockMvc} instance.
 */
@Override
@SuppressWarnings("rawtypes")
public final MockMvc build() {
	WebApplicationContext wac = initWebAppContext();
	ServletContext servletContext = wac.getServletContext();
	MockServletConfig mockServletConfig = new MockServletConfig(servletContext);

	for (MockMvcConfigurer configurer : this.configurers) {
		RequestPostProcessor processor = configurer.beforeMockMvcCreated(this, wac);
		if (processor != null) {
			if (this.defaultRequestBuilder == null) {
				this.defaultRequestBuilder = MockMvcRequestBuilders.get("/");
			}
			if (this.defaultRequestBuilder instanceof ConfigurableSmartRequestBuilder) {
				((ConfigurableSmartRequestBuilder) this.defaultRequestBuilder).with(processor);
			}
		}
	}

	Filter[] filterArray = this.filters.toArray(new Filter[0]);

	return super.createMockMvc(filterArray, mockServletConfig, wac, this.defaultRequestBuilder,
			this.globalResultMatchers, this.globalResultHandlers, this.dispatcherServletCustomizers);
}

Example 3

Source File: MockMvcBuilderSupport.java From spring4-understanding with Apache License 2.0

6 votes

protected final MockMvc createMockMvc(Filter[] filters, MockServletConfig servletConfig,
		WebApplicationContext webAppContext, RequestBuilder defaultRequestBuilder,
		List<ResultMatcher> globalResultMatchers, List<ResultHandler> globalResultHandlers,
		Boolean dispatchOptions) {

	ServletContext servletContext = webAppContext.getServletContext();

	TestDispatcherServlet dispatcherServlet = new TestDispatcherServlet(webAppContext);
	dispatcherServlet.setDispatchOptionsRequest(dispatchOptions);
	try {
		dispatcherServlet.init(servletConfig);
	}
	catch (ServletException ex) {
		// should never happen..
		throw new MockMvcBuildException("Failed to initialize TestDispatcherServlet", ex);
	}

	MockMvc mockMvc = new MockMvc(dispatcherServlet, filters, servletContext);
	mockMvc.setDefaultRequest(defaultRequestBuilder);
	mockMvc.setGlobalResultMatchers(globalResultMatchers);
	mockMvc.setGlobalResultHandlers(globalResultHandlers);

	return mockMvc;
}

Example 4

Source File: ControllerIntegrationTests.java From spring4-understanding with Apache License 2.0

6 votes

@Test
public void verifyRootWacSupport() {
	assertEquals("foo", foo);
	assertEquals("bar", bar);

	ApplicationContext parent = wac.getParent();
	assertNotNull(parent);
	assertTrue(parent instanceof WebApplicationContext);
	WebApplicationContext root = (WebApplicationContext) parent;
	assertFalse(root.getBeansOfType(String.class).containsKey("bar"));

	ServletContext childServletContext = wac.getServletContext();
	assertNotNull(childServletContext);
	ServletContext rootServletContext = root.getServletContext();
	assertNotNull(rootServletContext);
	assertSame(childServletContext, rootServletContext);

	assertSame(root, rootServletContext.getAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE));
	assertSame(root, childServletContext.getAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE));
}

Example 5

Source File: JavaConfigTests.java From java-technology-stack with MIT License

6 votes

/**
 * Verify that the breaking change introduced in <a
 * href="https://jira.spring.io/browse/SPR-12553">SPR-12553</a> has been reverted.
 *
 * <p>This code has been copied from
 * {@link org.springframework.test.context.hierarchies.web.ControllerIntegrationTests}.
 *
 * @see org.springframework.test.context.hierarchies.web.ControllerIntegrationTests#verifyRootWacSupport()
 */
private void verifyRootWacSupport() {
	assertNotNull(personDao);
	assertNotNull(personController);

	ApplicationContext parent = wac.getParent();
	assertNotNull(parent);
	assertTrue(parent instanceof WebApplicationContext);
	WebApplicationContext root = (WebApplicationContext) parent;

	ServletContext childServletContext = wac.getServletContext();
	assertNotNull(childServletContext);
	ServletContext rootServletContext = root.getServletContext();
	assertNotNull(rootServletContext);
	assertSame(childServletContext, rootServletContext);

	assertSame(root, rootServletContext.getAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE));
	assertSame(root, childServletContext.getAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE));
}

Example 6

Source File: ControllerIntegrationTests.java From java-technology-stack with MIT License

6 votes

@Test
public void verifyRootWacSupport() {
	assertEquals("foo", foo);
	assertEquals("bar", bar);

	ApplicationContext parent = wac.getParent();
	assertNotNull(parent);
	assertTrue(parent instanceof WebApplicationContext);
	WebApplicationContext root = (WebApplicationContext) parent;
	assertFalse(root.getBeansOfType(String.class).containsKey("bar"));

	ServletContext childServletContext = wac.getServletContext();
	assertNotNull(childServletContext);
	ServletContext rootServletContext = root.getServletContext();
	assertNotNull(rootServletContext);
	assertSame(childServletContext, rootServletContext);

	assertSame(root, rootServletContext.getAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE));
	assertSame(root, childServletContext.getAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE));
}

Example 7

Source File: JavaConfigTests.java From spring4-understanding with Apache License 2.0

6 votes

/**
 * Verify that the breaking change introduced in <a
 * href="https://jira.spring.io/browse/SPR-12553">SPR-12553</a> has been reverted.
 *
 * <p>This code has been copied from
 * {@link org.springframework.test.context.hierarchies.web.ControllerIntegrationTests}.
 *
 * @see org.springframework.test.context.hierarchies.web.ControllerIntegrationTests#verifyRootWacSupport()
 */
private void verifyRootWacSupport() {
	assertNotNull(personDao);
	assertNotNull(personController);

	ApplicationContext parent = wac.getParent();
	assertNotNull(parent);
	assertTrue(parent instanceof WebApplicationContext);
	WebApplicationContext root = (WebApplicationContext) parent;

	ServletContext childServletContext = wac.getServletContext();
	assertNotNull(childServletContext);
	ServletContext rootServletContext = root.getServletContext();
	assertNotNull(rootServletContext);
	assertSame(childServletContext, rootServletContext);

	assertSame(root, rootServletContext.getAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE));
	assertSame(root, childServletContext.getAttribute(WebApplicationContext.ROOT_WEB_APPLICATION_CONTEXT_ATTRIBUTE));
}

Example 8

Source File: MVCConfig.java From convergent-ui with Apache License 2.0

5 votes

@Bean
public ThymeleafViewResolver thymeleafViewResolver(WebApplicationContext wac) {
    ThymeleafViewResolver resolver = new ThymeleafViewResolver();
    SpringTemplateEngine templateEngine = new SpringTemplateEngine();
    ServletContextTemplateResolver templateResolver = new ServletContextTemplateResolver(wac.getServletContext());
    templateResolver.setPrefix("/WEB-INF/templates/");
    templateResolver.setSuffix(".html");
    templateResolver.setTemplateMode("HTML");
    templateResolver.setCharacterEncoding("UTF-8");
    templateEngine.setTemplateResolver(templateResolver);
    resolver.setTemplateEngine(templateEngine);
    resolver.setOrder(2);
    resolver.setApplicationContext(wac);
    resolver.setCharacterEncoding("UTF-8");

    //Enable the Spring Security Thymeleaf integration
    templateEngine.addDialect(new LayoutDialect());

    // caching
    String[] activeProfiles = env.getActiveProfiles();
    if (Arrays.asList(activeProfiles).contains("dev")) {
        log.info("DEV Profile is active. Disabling template caching.");
        templateResolver.setCacheable(false);
        templateEngine.setCacheManager(null);
        resolver.setCache(false);
    }
    return resolver;
}

Example 9

Source File: VSCrawlerManager.java From vscrawler with Apache License 2.0

5 votes

@Override
public void onApplicationEvent(ContextRefreshedEvent contextRefreshedEvent) {
    ApplicationContext applicationContext = contextRefreshedEvent.getApplicationContext();
    webApplicationContext = (WebApplicationContext) applicationContext;
    ServletContext servletContext = webApplicationContext.getServletContext();
    webAppPath = servletContext.getRealPath("/");
    init();
}

Example 10

Source File: AbstractMockMvcBuilder.java From spring4-understanding with Apache License 2.0

5 votes

/**
 * Build a {@link org.springframework.test.web.servlet.MockMvc} instance.
 */
@Override
@SuppressWarnings("rawtypes")
public final MockMvc build() {

	WebApplicationContext wac = initWebAppContext();

	ServletContext servletContext = wac.getServletContext();
	MockServletConfig mockServletConfig = new MockServletConfig(servletContext);

	for (MockMvcConfigurer configurer : this.configurers) {
		RequestPostProcessor processor = configurer.beforeMockMvcCreated(this, wac);
		if (processor != null) {
			if (this.defaultRequestBuilder == null) {
				this.defaultRequestBuilder = MockMvcRequestBuilders.get("/");
			}
			if (this.defaultRequestBuilder instanceof ConfigurableSmartRequestBuilder) {
				((ConfigurableSmartRequestBuilder) this.defaultRequestBuilder).with(processor);
			}
		}
	}

	Filter[] filterArray = this.filters.toArray(new Filter[this.filters.size()]);

	return super.createMockMvc(filterArray, mockServletConfig, wac, this.defaultRequestBuilder,
			this.globalResultMatchers, this.globalResultHandlers, this.dispatchOptions);
}

Example 11

Source File: CacheUtil.java From singleton with Eclipse Public License 2.0

5 votes

public static void cacheSessionAndToken(WebApplicationContext webApplicationContext,String authenticationResult) {
    JSONObject jsonObj=JSONUtils.string2JSON(authenticationResult);
    String sessionID=(String) jsonObj.get("sessionID");
    String token=(String) jsonObj.get("token");
    MockServletContext application = (MockServletContext) webApplicationContext.getServletContext();
    application.setAttribute("sessionID", sessionID);
    application.setAttribute("token", token);
}

Example 12

Source File: MockServerContainerContextCustomizer.java From spring-analysis-note with MIT License

5 votes

@Override
public void customizeContext(ConfigurableApplicationContext context, MergedContextConfiguration mergedConfig) {
	if (context instanceof WebApplicationContext) {
		WebApplicationContext wac = (WebApplicationContext) context;
		ServletContext sc = wac.getServletContext();
		if (sc != null) {
			sc.setAttribute("javax.websocket.server.ServerContainer", new MockServerContainer());
		}
	}
}

Example 13

Source File: MockServerContainerContextCustomizer.java From java-technology-stack with MIT License

5 votes

@Override
public void customizeContext(ConfigurableApplicationContext context, MergedContextConfiguration mergedConfig) {
	if (context instanceof WebApplicationContext) {
		WebApplicationContext wac = (WebApplicationContext) context;
		ServletContext sc = wac.getServletContext();
		if (sc != null) {
			sc.setAttribute("javax.websocket.server.ServerContainer", new MockServerContainer());
		}
	}
}

Example 14

Source File: RequestUtil.java From singleton with Eclipse Public License 2.0

5 votes

public static HttpHeaders getHeaders(WebApplicationContext webApplicationContext) {
    MockServletContext application = (MockServletContext) webApplicationContext.getServletContext();
    HttpHeaders httpHeaders = new HttpHeaders();
    httpHeaders.add("sessionID", (String) application.getAttribute("sessionID"));
    httpHeaders.add("token", (String) application.getAttribute("token"));
    return httpHeaders;
}

Example 15

Source File: ServletTestExecutionListener.java From java-technology-stack with MIT License

5 votes

private void setUpRequestContextIfNecessary(TestContext testContext) {
	if (!isActivated(testContext) || alreadyPopulatedRequestContextHolder(testContext)) {
		return;
	}

	ApplicationContext context = testContext.getApplicationContext();

	if (context instanceof WebApplicationContext) {
		WebApplicationContext wac = (WebApplicationContext) context;
		ServletContext servletContext = wac.getServletContext();
		Assert.state(servletContext instanceof MockServletContext, () -> String.format(
					"The WebApplicationContext for test context %s must be configured with a MockServletContext.",
					testContext));

		if (logger.isDebugEnabled()) {
			logger.debug(String.format(
					"Setting up MockHttpServletRequest, MockHttpServletResponse, ServletWebRequest, and RequestContextHolder for test context %s.",
					testContext));
		}

		MockServletContext mockServletContext = (MockServletContext) servletContext;
		MockHttpServletRequest request = new MockHttpServletRequest(mockServletContext);
		request.setAttribute(CREATED_BY_THE_TESTCONTEXT_FRAMEWORK, Boolean.TRUE);
		MockHttpServletResponse response = new MockHttpServletResponse();
		ServletWebRequest servletWebRequest = new ServletWebRequest(request, response);

		RequestContextHolder.setRequestAttributes(servletWebRequest);
		testContext.setAttribute(POPULATED_REQUEST_CONTEXT_HOLDER_ATTRIBUTE, Boolean.TRUE);
		testContext.setAttribute(RESET_REQUEST_CONTEXT_HOLDER_ATTRIBUTE, Boolean.TRUE);

		if (wac instanceof ConfigurableApplicationContext) {
			@SuppressWarnings("resource")
			ConfigurableApplicationContext configurableApplicationContext = (ConfigurableApplicationContext) wac;
			ConfigurableListableBeanFactory bf = configurableApplicationContext.getBeanFactory();
			bf.registerResolvableDependency(MockHttpServletResponse.class, response);
			bf.registerResolvableDependency(ServletWebRequest.class, servletWebRequest);
		}
	}
}

Example 16

Source File: Tools.java From ApiManager with GNU Affero General Public License v3.0

4 votes

public static ServletContext getServletContext() {
    WebApplicationContext webApplicationContext = ContextLoader.getCurrentWebApplicationContext();
    return webApplicationContext.getServletContext();
}

Example 17

Source File: SpringServiceNameInstrumentation.java From apm-agent-java with Apache License 2.0

4 votes

@Advice.OnMethodExit(suppress = Throwable.class)
public static void afterInitPropertySources(@Advice.This WebApplicationContext applicationContext) {
    if (tracer != null && applicationContext.getServletContext() != null) {
        tracer.overrideServiceNameForClassLoader(applicationContext.getServletContext().getClassLoader(), applicationContext.getEnvironment().getProperty("spring.application.name"));
    }
}

Example 18

Source File: RequestUtil.java From singleton with Eclipse Public License 2.0

4 votes

public static MockHttpSession getSession(WebApplicationContext webApplicationContext) {
    MockServletContext application = (MockServletContext) webApplicationContext.getServletContext();
    MockHttpSession session=new MockHttpSession();
    session.setAttribute("token", application.getAttribute("token"));
    return session;
}

Example 19

Source File: RequestUtil.java From singleton with Eclipse Public License 2.0

4 votes

public static Cookie getCookies(WebApplicationContext webApplicationContext) {
    MockServletContext application = (MockServletContext) webApplicationContext.getServletContext();
    Cookie cookies = new Cookie("JSESSIONID",(String) application.getAttribute("sessionID"));
    return cookies;
}

Example 20

Source File: RequestUtil.java From singleton with Eclipse Public License 2.0

4 votes

public static MockHttpSession getSession(WebApplicationContext webApplicationContext) {
    MockServletContext application = (MockServletContext) webApplicationContext.getServletContext();
    MockHttpSession session=new MockHttpSession();
    session.setAttribute("token", application.getAttribute("token"));
    return session;
}