com.gemstone.gemfire.cache.hdfs.internal.SignalledFlushObserverJUnitTest Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of gemfire-junit Show documentation

SnappyData store based off Pivotal GemFireXD

There is a newer version: 2.0-BETA

/*
 * Copyright (c) 2010-2015 Pivotal Software, Inc. All rights reserved.
 *
 * 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. See accompanying
 * LICENSE file.
 */
package com.gemstone.gemfire.cache.hdfs.internal;

import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicInteger;

import junit.framework.TestCase;

import com.gemstone.gemfire.cache.hdfs.internal.FlushObserver.AsyncFlushResult;
import com.gemstone.gemfire.cache.hdfs.internal.hoplog.AbstractHoplogOrganizer;

public class SignalledFlushObserverJUnitTest extends TestCase {
  private AtomicInteger events;
  private AtomicInteger delivered;
  
  private SignalledFlushObserver sfo;
  
  public void testEmpty() throws InterruptedException {
    assertFalse(sfo.shouldDrainImmediately());
    assertTrue(sfo.flush().waitForFlush(0, TimeUnit.NANOSECONDS));
    assertFalse(sfo.shouldDrainImmediately());
  }
  
  public void testSingle() throws InterruptedException {
    sfo.push();
    AsyncFlushResult result = sfo.flush();

    assertTrue(sfo.shouldDrainImmediately());
    sfo.pop(1);
    
    assertTrue(result.waitForFlush(0, TimeUnit.MILLISECONDS));
    assertFalse(sfo.shouldDrainImmediately());
  }

  public void testDouble() throws InterruptedException {
    sfo.push();
    sfo.push();

    AsyncFlushResult result = sfo.flush();
    assertTrue(sfo.shouldDrainImmediately());

    sfo.pop(1);
    assertFalse(result.waitForFlush(0, TimeUnit.MILLISECONDS));

    sfo.pop(1);
    assertTrue(result.waitForFlush(0, TimeUnit.MILLISECONDS));
    assertFalse(sfo.shouldDrainImmediately());
  }

  public void testTimeout() throws InterruptedException {
    sfo.push();
    AsyncFlushResult result = sfo.flush();

    assertTrue(sfo.shouldDrainImmediately());
    assertFalse(result.waitForFlush(100, TimeUnit.MILLISECONDS));
    sfo.pop(1);
    
    assertTrue(result.waitForFlush(0, TimeUnit.MILLISECONDS));
    assertFalse(sfo.shouldDrainImmediately());
  }
  
  @Override
  protected void setUp() {
    events = new AtomicInteger(0);
    delivered = new AtomicInteger(0);
    sfo = new SignalledFlushObserver();
    AbstractHoplogOrganizer.JUNIT_TEST_RUN = true;
  }
  
  private int push() {
    return events.incrementAndGet();
  }
  
  private int pop() {
    return delivered.incrementAndGet();
  }
}