/* * Copyright 2002-2019 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.web.reactive.result.view.script; import java.util.HashMap; import java.util.Locale; import java.util.Map; import org.junit.Test; import org.springframework.context.annotation.AnnotationConfigApplicationContext; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.context.support.ResourceBundleMessageSource; import org.springframework.http.MediaType; import org.springframework.mock.http.server.reactive.test.MockServerHttpRequest; import org.springframework.mock.http.server.reactive.test.MockServerHttpResponse; import org.springframework.mock.web.test.server.MockServerWebExchange; import static org.assertj.core.api.Assertions.assertThat; /** * Unit tests for Kotlin script templates running on Kotlin JSR-223 support. * * @author Sebastien Deleuze */ public class KotlinScriptTemplateTests { @Test public void renderTemplateWithFrenchLocale() throws Exception { Map model = new HashMap<>(); model.put("foo", "Foo"); String url = "org/springframework/web/reactive/result/view/script/kotlin/template.kts"; MockServerHttpResponse response = render(url, model, Locale.FRENCH, ScriptTemplatingConfiguration.class); assertThat(response.getBodyAsString().block()).isEqualTo("\n

Bonjour Foo

\n"); } @Test public void renderTemplateWithEnglishLocale() throws Exception { Map model = new HashMap<>(); model.put("foo", "Foo"); String url = "org/springframework/web/reactive/result/view/script/kotlin/template.kts"; MockServerHttpResponse response = render(url, model, Locale.ENGLISH, ScriptTemplatingConfiguration.class); assertThat(response.getBodyAsString().block()).isEqualTo("\n

Hello Foo

\n"); } @Test public void renderTemplateWithoutRenderFunction() throws Exception { Map model = new HashMap<>(); model.put("header", ""); model.put("hello", "Hello"); model.put("foo", "Foo"); model.put("footer", ""); String url = "org/springframework/web/reactive/result/view/script/kotlin/eval.kts"; Class configClass = ScriptTemplatingConfigurationWithoutRenderFunction.class; MockServerHttpResponse response = render(url, model, Locale.ENGLISH, configClass); assertThat(response.getBodyAsString().block()).isEqualTo("\n

Hello Foo

\n"); } private MockServerHttpResponse render(String viewUrl, Map model, Locale locale, Class configuration) throws Exception { ScriptTemplateView view = createViewWithUrl(viewUrl, configuration); MockServerHttpRequest request = MockServerHttpRequest.get("/").acceptLanguageAsLocales(locale).build(); MockServerWebExchange exchange = MockServerWebExchange.from(request); view.renderInternal(model, MediaType.TEXT_HTML, exchange).block(); return exchange.getResponse(); } private ScriptTemplateView createViewWithUrl(String viewUrl, Class configuration) throws Exception { AnnotationConfigApplicationContext ctx = new AnnotationConfigApplicationContext(); ctx.register(configuration); ctx.refresh(); ScriptTemplateView view = new ScriptTemplateView(); view.setApplicationContext(ctx); view.setUrl(viewUrl); view.afterPropertiesSet(); return view; } @Configuration static class ScriptTemplatingConfiguration { @Bean public ScriptTemplateConfigurer kotlinScriptConfigurer() { ScriptTemplateConfigurer configurer = new ScriptTemplateConfigurer(); configurer.setEngineName("kotlin"); configurer.setScripts("org/springframework/web/reactive/result/view/script/kotlin/render.kts"); configurer.setRenderFunction("render"); return configurer; } @Bean public ResourceBundleMessageSource messageSource() { ResourceBundleMessageSource messageSource = new ResourceBundleMessageSource(); messageSource.setBasename("org/springframework/web/reactive/result/view/script/messages"); return messageSource; } } @Configuration static class ScriptTemplatingConfigurationWithoutRenderFunction { @Bean public ScriptTemplateConfigurer kotlinScriptConfigurer() { return new ScriptTemplateConfigurer("kotlin"); } } }