code: /*
   * Copyright 2012-2018 Chronicle Map Contributors
   *
   * 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
   *
   *     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 net.openhft.chronicle.set;
  
  import org.jetbrains.annotations.NotNull;
  import org.junit.Assert;
  import org.junit.Test;
  
  import java.util.concurrent.atomic.AtomicInteger;
  
  public class SetEntryOperationsTest {
  
      @Test
      public void setEntryOperationsTest() {
          AtomicInteger insertCounter = new AtomicInteger();
          AtomicInteger remoteCounter = new AtomicInteger();
          try (ChronicleSet<String> fruits = ChronicleSet
                  .of(String.class)
                  .entries(3)
                  .averageKey("apple")
                  .entryOperations(new SetEntryOperations<String, Void>() {
                      @Override
                      public Void remove(@NotNull SetEntry<String> entry) {
                          remoteCounter.addAndGet(1);
                          entry.doRemove();
                          return null;
                      }
  
                      @Override
                      public Void insert(@NotNull SetAbsentEntry<String> absentEntry) {
                          insertCounter.addAndGet(1);
                          absentEntry.doInsert();
                          return null;
                      }
                  })
                  .create()) {
              fruits.add("apple");
              fruits.add("banana");
              fruits.remove("banana");
              fruits.remove("grapes");
  
              Assert.assertEquals(2, insertCounter.get());
              Assert.assertEquals(1, remoteCounter.get());
          }
      }
  }
  