gemfire-core/src/test/java/com/gemstone/gemfire/TXWriterOOMEJUnitTest.java

/*=========================================================================
 * Copyright (c) 2010-2014 Pivotal Software, Inc. All Rights Reserved.
 * This product is protected by U.S. and international copyright
 * and intellectual property laws. Pivotal products are covered by
 * one or more patents listed at http://www.pivotal.io/patents.
 *=========================================================================
 */
package com.gemstone.gemfire;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;

import org.junit.Test;
import org.junit.experimental.categories.Category;

import com.gemstone.gemfire.cache.CacheTransactionManager;
import com.gemstone.gemfire.cache.TransactionEvent;
import com.gemstone.gemfire.cache.TransactionWriter;
import com.gemstone.gemfire.cache.TransactionWriterException;
import com.gemstone.gemfire.internal.SystemFailureTestHook;
import com.gemstone.gemfire.test.junit.categories.IntegrationTest;

/**
 * Extracted from TXWriterJUnitTest. The OOME is problematic and causes the
 * test suite to fail due to suspect strings from SystemFailure Watchdog sending
 * output to STDERR.
 * 
 * @author Kirk Lund
 */
@Category(IntegrationTest.class)
public class TXWriterOOMEJUnitTest extends TXWriterTestCase {

  @Test
  public void testAfterCommitFailedOnThrowOOM() throws Exception {
    installCacheListenerAndWriter();
    
    // install TransactionWriter
    ((CacheTransactionManager)this.txMgr).setWriter(new TransactionWriter() {
      public void beforeCommit(TransactionEvent event) throws TransactionWriterException {
        throw new OutOfMemoryError("this is expected!");
      }
      public void close() {}
    });
    
    installTransactionListener();
    
    try {
      SystemFailureTestHook.setExpectedFailureClass(OutOfMemoryError.class);
      
      this.txMgr.begin();
      this.region.create("key1", "value1");
      this.cbCount = 0;
      try {
        this.txMgr.commit();
        fail("Commit should have thrown OOME");
      } catch(OutOfMemoryError expected) {
        // this is what we expect
      }
      
      // no callbacks were invoked
      assertEquals(0, this.cbCount);
      assertEquals(0, this.failedCommits);
      assertEquals(0, this.afterCommits);
      assertEquals(0, this.afterRollbacks);
    } finally {
      SystemFailureTestHook.setExpectedFailureClass(null);
    }
  }
}