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.processor.aggregate;
  
  import java.util.Set;
  
  import org.apache.camel.Exchange;
  import org.apache.camel.component.ehcache.EhcacheTestSupport;
  import org.apache.camel.support.DefaultExchange;
  import org.apache.camel.support.DefaultExchangeHolder;
  import org.junit.jupiter.api.Test;
  
  import static org.junit.jupiter.api.Assertions.assertEquals;
  import static org.junit.jupiter.api.Assertions.assertFalse;
  import static org.junit.jupiter.api.Assertions.assertNotNull;
  import static org.junit.jupiter.api.Assertions.assertNull;
  import static org.junit.jupiter.api.Assertions.assertTrue;
  
  public class EhcacheAggregationRepositoryOperationTest extends EhcacheTestSupport {
      private EhcacheAggregationRepository aggregationRepository;
  
      @Override
      protected void doPreSetup() throws Exception {
          super.doPreSetup();
  
          aggregationRepository = createAggregateRepository();
          aggregationRepository.start();
      }
  
      @Override
      public void doPostTearDown() {
          aggregationRepository.stop();
      }
  
      private boolean exists(String key) {
          DefaultExchangeHolder holder = aggregationRepository.getCache().get(key);
          if (holder == null) {
              return false;
          }
          return true;
      }
  
      @Test
      void testAdd() {
          // Given
          String key = "Add";
          assertFalse(exists(key));
          Exchange exchange = new DefaultExchange(context());
          // When
          aggregationRepository.add(context(), key, exchange);
          // Then
          assertTrue(exists(key));
      }
  
      @Test
      void testGetExists() {
          // Given
          String key = "Get_Exists";
          Exchange exchange = new DefaultExchange(context());
          aggregationRepository.add(context(), key, exchange);
          assertTrue(exists(key));
  
          // When
          Exchange exchange2 = aggregationRepository.get(context(), key);
          // Then
          assertNotNull(exchange2);
          assertEquals(exchange.getExchangeId(), exchange2.getExchangeId());
      }
  
      @Test
      void testGetNotExists() {
          // Given
          String key = "Get_NotExists";
          assertFalse(exists(key));
          // When
          Exchange exchange2 = aggregationRepository.get(context(), key);
          // Then
          assertNull(exchange2);
      }
  
      @Test
      void testRemoveExists() {
          // Given
          String key = "Remove_Exists";
          Exchange exchange = new DefaultExchange(context());
          aggregationRepository.add(context(), key, exchange);
          assertTrue(exists(key));
          // When
          aggregationRepository.remove(context(), key, exchange);
          // Then
          assertFalse(exists(key));
      }
  
      @Test
      void testRemoveNotExists() {
          // Given
          String key = "RemoveNotExists";
          Exchange exchange = new DefaultExchange(context());
          assertFalse(exists(key));
          // When
          aggregationRepository.remove(context(), key, exchange);
          // Then
          assertFalse(exists(key));
      }
  
      @Test
      void testGetKeys() {
          // Given
          String[] keys = { "GetKeys1", "GetKeys2" };
          addExchanges(keys);
          // When
          Set<String> keySet = aggregationRepository.getKeys();
          // Then
          for (String key : keys) {
              assertTrue(keySet.contains(key));
          }
      }
  
      @Test
      void testConfirmExist() {
          // Given
          for (int i = 1; i < 4; i++) {
              String key = "Confirm_" + i;
              Exchange exchange = new DefaultExchange(context());
              exchange.setExchangeId("Exchange_" + i);
              aggregationRepository.add(context(), key, exchange);
              assertTrue(exists(key));
          }
          // When
          aggregationRepository.confirm(context(), "Confirm_2");
          // Then
          assertTrue(exists("Confirm_1"));
          assertFalse(exists("Confirm_2"));
          assertTrue(exists("Confirm_3"));
      }
  
      @Test
      void testConfirmNotExist() {
          // Given
          String[] keys = new String[3];
          for (int i = 1; i < 4; i++) {
              keys[i - 1] = "Confirm" + i;
          }
          addExchanges(keys);
          for (String key : keys) {
              assertTrue(exists(key));
          }
          // When
          aggregationRepository.confirm(context(), "Exchange-Confirm5");
          // Then
          for (String key : keys) {
              assertTrue(exists(key));
          }
      }
  
      private void addExchanges(String... keys) {
          for (String key : keys) {
              Exchange exchange = new DefaultExchange(context());
              exchange.setExchangeId("Exchange-" + key);
              aggregationRepository.add(context(), key, exchange);
          }
      }
  
      @Test
      void testScan() {
          // Given
          String[] keys = { "Scan1", "Scan2" };
          addExchanges(keys);
          // When
          Set<String> exchangeIdSet = aggregationRepository.scan(context());
          // Then
          for (String key : keys) {
              assertTrue(exchangeIdSet.contains(key));
          }
      }
  
      @Test
      void testRecover() {
          // Given
          String[] keys = { "Recover1", "Recover2" };
          addExchanges(keys);
          // When
          Exchange exchange2 = aggregationRepository.recover(context(), "Recover2");
          Exchange exchange3 = aggregationRepository.recover(context(), "Recover3");
          // Then
          assertNotNull(exchange2);
          assertNull(exchange3);
      }
  }
  