View Javadoc
1   /*
2    * Copyright 2020-2022 the original author or authors.
3    *
4    * Licensed under the Apache License, Version 2.0 (the "License");
5    * you may not use this file except in compliance with the License.
6    * You may obtain a copy of the License at
7    *
8    *      http://www.apache.org/licenses/LICENSE-2.0
9    *
10   * Unless required by applicable law or agreed to in writing, software
11   * distributed under the License is distributed on an "AS IS" BASIS,
12   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13   * See the License for the specific language governing permissions and
14   * limitations under the License.
15   */
16  
17  package org.bremersee.spring.boot.autoconfigure.thymeleaf;
18  
19  import static org.assertj.core.api.Assertions.assertThat;
20  import static org.mockito.Mockito.mock;
21  import static org.mockito.Mockito.when;
22  
23  import java.util.Collections;
24  import org.bremersee.spring.boot.autoconfigure.thymeleaf.AdditionalThymeleafProperties.ResolverProperties;
25  import org.junit.jupiter.api.Test;
26  import org.springframework.beans.factory.ObjectProvider;
27  import org.springframework.core.io.DefaultResourceLoader;
28  import org.thymeleaf.TemplateEngine;
29  import org.thymeleaf.templatemode.TemplateMode;
30  
31  /**
32   * The additional thymeleaf autoconfiguration test.
33   *
34   * @author Christian Bremer
35   */
36  class AdditionalThymeleafAutoConfigurationTest {
37  
38    /**
39     * Register template resolver.
40     */
41    @Test
42    void registerTemplateResolver() {
43      // Mockito cannot mock TemplateEngine because getTemplateResolvers() is final.
44      TemplateEngine templateEngine = new TemplateEngine();
45      //noinspection unchecked
46      ObjectProvider<TemplateEngine> provider = mock(ObjectProvider.class);
47      when(provider.getIfAvailable()).thenReturn(templateEngine);
48  
49      AdditionalThymeleafProperties properties = new AdditionalThymeleafProperties();
50      AdditionalThymeleafAutoConfiguration configuration
51          = new AdditionalThymeleafAutoConfiguration(properties, provider);
52      configuration.setResourceLoader(new DefaultResourceLoader());
53      configuration.registerTemplateResolver();
54  
55      ResolverProperties resolver0 = new ResolverProperties();
56      resolver0.setName("resolver0");
57      resolver0.setCacheablePatterns(Collections.singleton("1234"));
58      resolver0.setCacheTtlms(123456789L);
59      resolver0.setCssTemplateModePatterns(Collections.singleton("2345"));
60      resolver0.setHtmlTemplateModePatterns(Collections.singleton("3456"));
61      resolver0.setJavaScriptTemplateModePatterns(Collections.singleton("4567"));
62      resolver0.setNonCacheablePatterns(Collections.singleton("5678"));
63      resolver0.setPrefix("6789");
64      resolver0.setRawTemplateModePatterns(Collections.singleton("7890"));
65      resolver0.setSuffix("12345");
66      resolver0.setTemplateAliases(Collections.singletonMap("key", "23456"));
67      resolver0.setTemplateMode(TemplateMode.HTML);
68      resolver0.setTextTemplateModePatterns(Collections.singleton("34567"));
69      resolver0.setXmlTemplateModePatterns(Collections.singleton("45678"));
70  
71      ResolverProperties resolver1 = new ResolverProperties();
72      resolver1.setName("");
73      resolver1.setPrefix("");
74      resolver1.setSuffix("");
75      resolver0.setTemplateMode(null);
76  
77      properties.getResolvers().add(resolver0);
78      properties.getResolvers().add(resolver1);
79      configuration.registerTemplateResolver();
80  
81      assertThat(templateEngine.getTemplateResolvers())
82          .hasSize(2);
83    }
84  
85  }