package org.multiverse.stms.gamma.transactionalobjects.txnref;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.multiverse.api.LockMode;
import org.multiverse.api.TxnFactory;
import org.multiverse.api.exceptions.DeadTxnException;
import org.multiverse.api.exceptions.PreparedTxnException;
import org.multiverse.api.exceptions.ReadWriteConflict;
import org.multiverse.api.functions.Function;
import org.multiverse.api.functions.Functions;
import org.multiverse.api.functions.LongFunction;
import org.multiverse.stms.gamma.GammaStm;
import org.multiverse.stms.gamma.transactionalobjects.GammaTxnRef;
import org.multiverse.stms.gamma.transactions.GammaTxn;
import org.multiverse.stms.gamma.transactions.GammaTxnFactory;
import org.multiverse.stms.gamma.transactions.fat.FatFixedLengthGammaTxnFactory;
import org.multiverse.stms.gamma.transactions.fat.FatMonoGammaTxnFactory;
import org.multiverse.stms.gamma.transactions.fat.FatVariableLengthGammaTxnFactory;
import java.util.Collection;
import static java.util.Arrays.asList;
import static org.junit.Assert.*;
import static org.mockito.Matchers.anyLong;
import static org.mockito.Mockito.*;
import static org.multiverse.TestUtils.*;
import static org.multiverse.api.TxnThreadLocal.clearThreadLocalTxn;
import static org.multiverse.api.TxnThreadLocal.getThreadLocalTxn;
import static org.multiverse.stms.gamma.GammaTestUtils.*;
@RunWith(Parameterized.class)
public class GammaTxnRef_alterAndGet2Test {
private final GammaTxnFactory transactionFactory;
private final GammaStm stm;
public GammaTxnRef_alterAndGet2Test(GammaTxnFactory transactionFactory) {
this.transactionFactory = transactionFactory;
this.stm = transactionFactory.getConfig().getStm();
}
@Before
public void setUp() {
clearThreadLocalTxn();
}
@Parameterized.Parameters
public static Collection<TxnFactory[]> configs() {
return asList(
new TxnFactory[]{new FatVariableLengthGammaTxnFactory(new GammaStm())},
new TxnFactory[]{new FatFixedLengthGammaTxnFactory(new GammaStm())},
new TxnFactory[]{new FatMonoGammaTxnFactory(new GammaStm())}
);
}
@Test
public void whenNullTransaction_thenNullPointerException() {
Long initialValue = 10L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
LongFunction function = mock(LongFunction.class);
try {
ref.alterAndGet(null, function);
fail();
} catch (NullPointerException expected) {
}
verifyZeroInteractions(function);
assertVersionAndValue(ref, initialVersion, initialValue);
}
@Test
public void whenNullFunction_thenNullPointerException() {
Long initialValue = 10L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
GammaTxn tx = transactionFactory.newTxn();
try {
ref.alterAndGet(tx, null);
fail();
} catch (NullPointerException expected) {
}
assertRefHasNoLocks(ref);
assertIsAborted(tx);
assertVersionAndValue(ref, initialVersion, initialValue);
}
@Test
public void whenCommittedTransaction_thenDeadTxnException() {
Long initialValue = 10L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
GammaTxn tx = transactionFactory.newTxn();
tx.commit();
LongFunction function = mock(LongFunction.class);
try {
ref.alterAndGet(tx, function);
fail();
} catch (DeadTxnException expected) {
}
assertIsCommitted(tx);
assertRefHasNoLocks(ref);
assertVersionAndValue(ref, initialVersion, initialValue);
}
@Test
public void whenPreparedTransaction_thenPreparedTxnException() {
Long initialValue = 10L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
GammaTxn tx = transactionFactory.newTxn();
tx.prepare();
LongFunction function = mock(LongFunction.class);
try {
ref.alterAndGet(tx, function);
fail();
} catch (PreparedTxnException expected) {
}
assertRefHasNoLocks(ref);
assertIsAborted(tx);
assertVersionAndValue(ref, initialVersion, initialValue);
}
@Test
public void whenAbortedTransaction() {
Long initialValue = 10L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
GammaTxn tx = transactionFactory.newTxn();
tx.abort();
LongFunction function = mock(LongFunction.class);
try {
ref.alterAndGet(tx, function);
fail();
} catch (DeadTxnException expected) {
}
assertRefHasNoLocks(ref);
assertIsAborted(tx);
assertVersionAndValue(ref, initialVersion, initialValue);
}
@Test
public void whenFunctionCausesException() {
Long initialValue = 10L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
LongFunction function = mock(LongFunction.class);
RuntimeException ex = new RuntimeException();
when(function.call(anyLong())).thenThrow(ex);
GammaTxn tx = transactionFactory.newTxn();
try {
ref.alterAndGet(tx, function);
fail();
} catch (RuntimeException found) {
assertSame(ex, found);
}
assertRefHasNoLocks(ref);
assertIsAborted(tx);
assertNull(getThreadLocalTxn());
assertVersionAndValue(ref, initialVersion, initialValue);
}
@Test
public void whenPrivatizedByOther() {
Long initialValue = 10L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long version = ref.getVersion();
GammaTxn otherTx = transactionFactory.newTxn();
ref.getLock().acquire(otherTx, LockMode.Exclusive);
GammaTxn tx = transactionFactory.newTxn();
LongFunction function = mock(LongFunction.class);
try {
ref.alterAndGet(tx, function);
fail();
} catch (ReadWriteConflict expected) {
}
assertSurplus(ref, 1);
assertIsAborted(tx);
assertRefHasExclusiveLock(ref, otherTx);
assertVersionAndValue(ref, version, initialValue);
}
@Test
public void whenWriteLockedByOther_thenOperationSucceedsButCommitFails() {
Long initialValue = 10L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long version = ref.getVersion();
GammaTxn otherTx = transactionFactory.newTxn();
ref.getLock().acquire(otherTx, LockMode.Write);
GammaTxn tx = transactionFactory.newTxn();
LongFunction function = Functions.incLongFunction(1);
ref.alterAndGet(tx, function);
try {
tx.commit();
fail();
} catch (ReadWriteConflict expected) {
}
assertRefHasWriteLock(ref, otherTx);
assertSurplus(ref, 1);
assertIsActive(otherTx);
assertIsAborted(tx);
assertVersionAndValue(ref, version, initialValue);
}
@Test
public void whenListenersAvailable_thenTheyAreNotified() {
Long initialValue = 10L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
final Long newValue = 11L;
TxnRefAwaitThread<Long> thread = new TxnRefAwaitThread<Long>(ref, newValue);
thread.start();
sleepMs(500);
Function<Long> function = new Function<Long>(){
@Override
public Long call(Long value) {
return newValue;
}
};
GammaTxn tx = transactionFactory.newTxn();
ref.alterAndGet(tx, function);
tx.commit();
joinAll(thread);
assertVersionAndValue(ref, initialVersion + 1, newValue);
}
@Test
public void whenSuccess() {
LongFunction function = new LongFunction() {
@Override
public long call(long current) {
return current + 1;
}
};
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, 100L);
GammaTxn tx = transactionFactory.newTxn();
long result = ref.alterAndGet(tx, function);
tx.commit();
assertEquals(new Long(101), ref.atomicGet());
assertEquals(101L, result);
}
}