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 java.util.Collections;
  
  import org.apache.camel.BindToRegistry;
  import org.apache.camel.Component;
  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.config.CacheRuntimeConfiguration;
  import org.ehcache.config.ResourceType;
  import org.ehcache.config.builders.CacheConfigurationBuilder;
  import org.ehcache.config.builders.ResourcePoolsBuilder;
  import org.ehcache.config.units.EntryUnit;
  import org.ehcache.config.units.MemoryUnit;
  import org.ehcache.core.Ehcache;
  import org.junit.jupiter.api.Test;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertTrue;
  
  public class EhcacheConfigurationTest extends CamelTestSupport {
      @EndpointInject("ehcache:globalConfig")
      EhcacheEndpoint globalConfig;
      @EndpointInject("ehcache:customConfig")
      EhcacheEndpoint customConfig;
  
      @BindToRegistry("ehcache")
      public Component createEhcacheComponent() {
          EhcacheComponent component = new EhcacheComponent();
          component.setCacheConfiguration(
                  CacheConfigurationBuilder.newCacheConfigurationBuilder(
                          String.class,
                          String.class,
                          ResourcePoolsBuilder.newResourcePoolsBuilder()
                                  .heap(100, EntryUnit.ENTRIES)
                                  .offheap(1, MemoryUnit.MB))
                          .build());
          component.setCachesConfigurations(
                  Collections.singletonMap(
                          "customConfig",
                          CacheConfigurationBuilder.newCacheConfigurationBuilder(
                                  String.class,
                                  String.class,
                                  ResourcePoolsBuilder.newResourcePoolsBuilder()
                                          .heap(200, EntryUnit.ENTRIES)
                                          .offheap(2, MemoryUnit.MB))
                                  .build()));
  
          return component;
      }
  
      // *****************************
      // Test
      // *****************************
  
      @Test
      void testComponentConfiguration() throws Exception {
          Cache<String, String> globalConfigCache
                  = globalConfig.getManager().getCache("globalConfig", String.class, String.class);
          Cache<String, String> customConfigCache
                  = customConfig.getManager().getCache("customConfig", String.class, String.class);
  
          assertTrue(globalConfigCache instanceof Ehcache);
          assertTrue(customConfigCache instanceof Ehcache);
  
          CacheRuntimeConfiguration<String, String> gc = globalConfigCache.getRuntimeConfiguration();
          assertEquals(100, gc.getResourcePools().getPoolForResource(ResourceType.Core.HEAP).getSize());
          assertEquals(EntryUnit.ENTRIES, gc.getResourcePools().getPoolForResource(ResourceType.Core.HEAP).getUnit());
          assertEquals(1, gc.getResourcePools().getPoolForResource(ResourceType.Core.OFFHEAP).getSize());
          assertEquals(MemoryUnit.MB, gc.getResourcePools().getPoolForResource(ResourceType.Core.OFFHEAP).getUnit());
  
          CacheRuntimeConfiguration<String, String> cc = customConfigCache.getRuntimeConfiguration();
          assertEquals(200, cc.getResourcePools().getPoolForResource(ResourceType.Core.HEAP).getSize());
          assertEquals(EntryUnit.ENTRIES, cc.getResourcePools().getPoolForResource(ResourceType.Core.HEAP).getUnit());
          assertEquals(2, cc.getResourcePools().getPoolForResource(ResourceType.Core.OFFHEAP).getSize());
          assertEquals(MemoryUnit.MB, cc.getResourcePools().getPoolForResource(ResourceType.Core.OFFHEAP).getUnit());
      }
  
      // ****************************
      // Route
      // ****************************
  
      @Override
      protected RouteBuilder createRouteBuilder() {
          return new RouteBuilder() {
              public void configure() {
                  from("direct:ehcache")
                          .to(globalConfig)
                          .to(customConfig);
              }
          };
      }
  }
  