content: /*
   * Licensed to the Apache Software Foundation (ASF) under one or more
   * contributor license agreements.  See the NOTICE file distributed with
   * this work for additional information regarding copyright ownership.
   * The ASF licenses this file to You 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.apache.camel.component.ehcache;
  
  import org.apache.camel.BindToRegistry;
  import org.apache.camel.EndpointInject;
  import org.apache.camel.builder.RouteBuilder;
  import org.apache.camel.test.junit5.CamelTestSupport;
  import org.ehcache.Cache;
  import org.ehcache.CacheManager;
  import org.ehcache.UserManagedCache;
  import org.ehcache.config.CacheConfiguration;
  import org.ehcache.config.ResourcePools;
  import org.ehcache.config.ResourceType;
  import org.ehcache.config.SizedResourcePool;
  import org.ehcache.config.builders.CacheConfigurationBuilder;
  import org.ehcache.config.builders.CacheManagerBuilder;
  import org.ehcache.config.builders.ResourcePoolsBuilder;
  import org.ehcache.config.units.EntryUnit;
  import org.ehcache.config.units.MemoryUnit;
  import org.junit.jupiter.api.Test;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertNotNull;
  import static org.junit.jupiter.api.Assertions.assertTrue;
  
  public class EhcacheCacheConfigurationTest extends CamelTestSupport {
      @EndpointInject("ehcache:myProgrammaticCacheConf?configuration=#myProgrammaticConfiguration")
      private EhcacheEndpoint ehcacheProgrammaticConf;
      @EndpointInject("ehcache:myFileCacheConf?keyType=java.lang.String&valueType=java.lang.String&configurationUri=classpath:ehcache/ehcache-file-config.xml")
      private EhcacheEndpoint ehcacheFileConf;
      @EndpointInject("ehcache:myUserCacheConf")
      private EhcacheEndpoint ehcacheUserConf;
      @EndpointInject("ehcache:myCache?cacheManager=#myCacheManager&keyType=java.lang.String&valueType=java.lang.String")
      private EhcacheEndpoint ehcacheCacheManager;
  
      @BindToRegistry("myProgrammaticConfiguration")
      private CacheConfiguration<String, String> c = CacheConfigurationBuilder.newCacheConfigurationBuilder(
              String.class,
              String.class,
              ResourcePoolsBuilder.newResourcePoolsBuilder()
                      .heap(100, EntryUnit.ENTRIES)
                      .offheap(1, MemoryUnit.MB))
              .build();
  
      @BindToRegistry("myCacheManager")
      private CacheManager el = CacheManagerBuilder.newCacheManagerBuilder()
              .withCache(
                      "myCache",
                      CacheConfigurationBuilder.newCacheConfigurationBuilder(
                              String.class,
                              String.class,
                              ResourcePoolsBuilder.newResourcePoolsBuilder()
                                      .heap(100, EntryUnit.ENTRIES)
                                      .offheap(1, MemoryUnit.MB)))
              .build(true);
  
      // *****************************
      // Test
      // *****************************
  
      @Test
      void testProgrammaticConfiguration() throws Exception {
          Cache<String, String> cache = getCache(ehcacheProgrammaticConf, "myProgrammaticCacheConf");
          ResourcePools pools = cache.getRuntimeConfiguration().getResourcePools();
  
          SizedResourcePool h = pools.getPoolForResource(ResourceType.Core.HEAP);
          assertNotNull(h);
          assertEquals(100, h.getSize());
          assertEquals(EntryUnit.ENTRIES, h.getUnit());
  
          SizedResourcePool o = pools.getPoolForResource(ResourceType.Core.OFFHEAP);
          assertNotNull(o);
          assertEquals(1, o.getSize());
          assertEquals(MemoryUnit.MB, o.getUnit());
      }
  
      @Test
      void testFileConfiguration() throws Exception {
          Cache<String, String> cache = getCache(ehcacheFileConf, "myFileCacheConf");
          ResourcePools pools = cache.getRuntimeConfiguration().getResourcePools();
  
          SizedResourcePool h = pools.getPoolForResource(ResourceType.Core.HEAP);
          assertNotNull(h);
          assertEquals(150, h.getSize());
          assertEquals(EntryUnit.ENTRIES, h.getUnit());
      }
  
      @Test
      void testUserConfiguration() throws Exception {
          fluentTemplate()
                  .withHeader(EhcacheConstants.ACTION, EhcacheConstants.ACTION_PUT)
                  .withHeader(EhcacheConstants.KEY, "user-key")
                  .withBody("user-val")
                  .to("direct:ehcacheUserConf")
                  .send();
  
          Cache<Object, Object> cache = ehcacheUserConf.getManager().getCache("myUserCacheConf", Object.class, Object.class);
  
          assertTrue(cache instanceof UserManagedCache);
          assertEquals("user-val", cache.get("user-key"));
      }
  
      @Test
      void testCacheManager() throws Exception {
          assertEquals(
                  context().getRegistry().lookupByNameAndType("myCacheManager", CacheManager.class),
                  ehcacheCacheManager.getManager().getCacheManager());
  
          Cache<String, String> cache = getCache(ehcacheCacheManager, "myCache");
          ResourcePools pools = cache.getRuntimeConfiguration().getResourcePools();
  
          SizedResourcePool h = pools.getPoolForResource(ResourceType.Core.HEAP);
          assertNotNull(h);
          assertEquals(100, h.getSize());
          assertEquals(EntryUnit.ENTRIES, h.getUnit());
  
          SizedResourcePool o = pools.getPoolForResource(ResourceType.Core.OFFHEAP);
          assertNotNull(o);
          assertEquals(1, o.getSize());
          assertEquals(MemoryUnit.MB, o.getUnit());
      }
  
      protected Cache<String, String> getCache(EhcacheEndpoint endpoint, String cacheName) throws Exception {
          return endpoint.getManager().getCache(cacheName, String.class, String.class);
      }
  
      // ****************************
      // Route
      // ****************************
  
      @Override
      protected RouteBuilder createRouteBuilder() {
          return new RouteBuilder() {
              public void configure() {
                  from("direct:ehcacheProgrammaticConf")
                          .to(ehcacheProgrammaticConf);
                  from("direct:ehcacheFileConf")
                          .to(ehcacheFileConf);
                  from("direct:ehcacheUserConf")
                          .to(ehcacheUserConf);
                  from("direct:ehcacheCacheManager")
                          .to(ehcacheCacheManager);
              }
          };
      }
  }
  