geode-core/src/test/java/org/apache/geode/internal/cache/TombstoneServiceTest.java

/*
 * 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.geode.internal.cache;

import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import java.util.concurrent.ExecutorService;

import org.junit.Before;
import org.junit.Test;
import org.mockito.Mockito;

import org.apache.geode.CancelCriterion;
import org.apache.geode.cache.DataPolicy;
import org.apache.geode.distributed.internal.CacheTime;
import org.apache.geode.internal.cache.versions.RegionVersionVector;
import org.apache.geode.internal.cache.versions.VersionTag;

public class TombstoneServiceTest {
  CacheTime cacheTime;
  CachePerfStats stats;
  CancelCriterion cancelCriterion;
  ExecutorService executor;
  RegionMap regionMap;
  RegionEntry entry;
  DistributedRegion region;
  VersionTag destroyedVersion;
  private TombstoneService.ReplicateTombstoneSweeper replicateTombstoneSweeper;
  private TombstoneService.Tombstone tombstone1;

  private TombstoneService.Tombstone tombstone2;


  @Before
  public void setUp() throws Exception {
    cacheTime = mock(CacheTime.class);
    stats = mock(CachePerfStats.class);
    cancelCriterion = mock(CancelCriterion.class);
    executor = mock(ExecutorService.class);
    regionMap = mock(RegionMap.class);
    entry = mock(RegionEntry.class);
    region = mock(DistributedRegion.class);
    destroyedVersion = mock(VersionTag.class);
    replicateTombstoneSweeper = new TombstoneService.ReplicateTombstoneSweeper(cacheTime, stats,
        cancelCriterion, executor);
    tombstone1 = new TombstoneService.Tombstone(entry, region, destroyedVersion);
    tombstone2 = new TombstoneService.Tombstone(entry, region, destroyedVersion);
    tombstone1.entry = entry;
  }

  @Test
  public void validateThatRemoveIsNotCalledOnTombstoneInRegionThatIsNotInitialized() {
    when(region.isInitialized()).thenReturn(false);
    when(region.getRegionMap()).thenReturn(regionMap);

    replicateTombstoneSweeper.expireTombstone(tombstone1);
    replicateTombstoneSweeper.expireBatch();
    verify(regionMap, Mockito.never()).removeTombstone(tombstone1.entry, tombstone1);
  }

  @Test
  public void validateThatRemoveIsCalledOnTombstoneInRegionThatIsInitialized() {
    RegionVersionVector regionVersionVector = mock(RegionVersionVector.class);

    when(region.isInitialized()).thenReturn(true);
    when(region.getRegionMap()).thenReturn(regionMap);
    when(region.getVersionVector()).thenReturn(regionVersionVector);
    when(region.getDataPolicy()).thenReturn(DataPolicy.PERSISTENT_REPLICATE);
    when(region.getDiskRegion()).thenReturn(mock(DiskRegion.class));


    replicateTombstoneSweeper.expireTombstone(tombstone1);
    replicateTombstoneSweeper.expireBatch();
    verify(regionMap).removeTombstone(tombstone1.entry, tombstone1);
  }

  @Test
  public void validateThatTheExpiredTombstonesAreCleared() {
    when(region.getRegionMap()).thenReturn(regionMap);
    replicateTombstoneSweeper.expireTombstone(tombstone1);
    assertThat(replicateTombstoneSweeper.getScheduledTombstoneCount()).isOne();
    replicateTombstoneSweeper.unscheduleTombstones(region);
    assertThat(replicateTombstoneSweeper.getScheduledTombstoneCount()).isZero();
  }

  @Test
  public void validateThatTheNonExpiredTombstonesAreCleared() {
    when(region.getRegionMap()).thenReturn(regionMap);
    replicateTombstoneSweeper.scheduleTombstone(tombstone1);
    assertThat(replicateTombstoneSweeper.getScheduledTombstoneCount()).isOne();
    replicateTombstoneSweeper.unscheduleTombstones(region);
    assertThat(replicateTombstoneSweeper.getScheduledTombstoneCount()).isZero();
  }

  @Test
  public void validateThatTheNonExpiredAndExpiredTombstonesAreCleared() {
    when(region.getRegionMap()).thenReturn(regionMap);
    replicateTombstoneSweeper.scheduleTombstone(tombstone1);
    replicateTombstoneSweeper.expireTombstone(tombstone2);
    assertThat(replicateTombstoneSweeper.getScheduledTombstoneCount()).isEqualTo(2);
    replicateTombstoneSweeper.unscheduleTombstones(region);
    assertThat(replicateTombstoneSweeper.getScheduledTombstoneCount()).isZero();
  }
}