package org.multiverse.stms.gamma.transactionalobjects.txnref;
import org.junit.Before;
import org.junit.Test;
import org.multiverse.api.LockMode;
import org.multiverse.api.exceptions.LockedException;
import org.multiverse.api.functions.Functions;
import org.multiverse.api.functions.LongFunction;
import org.multiverse.stms.gamma.GammaStm;
import org.multiverse.stms.gamma.GammaStmConfig;
import org.multiverse.stms.gamma.transactionalobjects.GammaTxnRef;
import org.multiverse.stms.gamma.transactions.GammaTxn;
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.*;
import static org.multiverse.api.functions.Functions.identityLongFunction;
import static org.multiverse.api.functions.Functions.incLongFunction;
import static org.multiverse.stms.gamma.GammaTestUtils.*;
public class GammaTxnRef_atomicAlterAndGetTest {
private GammaStm stm;
@Before
public void setUp() {
GammaStmConfig config = new GammaStmConfig();
config.maxRetries = 10;
stm = new GammaStm(config);
clearThreadLocalTxn();
}
@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);
long orecValue = ref.orec;
try {
ref.atomicAlterAndGet(function);
fail();
} catch (RuntimeException found) {
assertSame(ex, found);
}
assertVersionAndValue(ref, initialVersion, initialValue);
assertOrecValue(ref, orecValue);
assertNull(getThreadLocalTxn());
}
@Test
public void whenNullFunction_thenNullPointerException() {
Long initialValue = 5L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
try {
ref.atomicAlterAndGet(null);
fail();
} catch (NullPointerException expected) {
}
assertRefHasNoLocks(ref);
assertSurplus(ref, 0);
assertVersionAndValue(ref, initialVersion, initialValue);
}
@Test
public void whenSuccess() {
Long initialValue = 5L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
LongFunction function = Functions.incLongFunction(1);
long result = ref.atomicAlterAndGet(function);
assertEquals(initialValue + 1, result);
assertRefHasNoLocks(ref);
assertSurplus(ref, 0);
assertVersionAndValue(ref, initialVersion + 1, initialValue + 1);
}
@Test
public void whenNoChange() {
Long initialValue = 5L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long version = ref.getVersion();
LongFunction function = identityLongFunction();
Long result = ref.atomicAlterAndGet(function);
assertEquals(initialValue, result);
assertRefHasNoLocks(ref);
assertSurplus(ref, 0);
assertVersionAndValue(ref, version, initialValue);
}
@Test
public void whenActiveTransactionAvailable_thenIgnored() {
Long initialValue = 5L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
GammaTxn tx = stm.newDefaultTxn();
setThreadLocalTxn(tx);
ref.set(tx, 100L);
LongFunction function = incLongFunction(1);
long result = ref.atomicAlterAndGet(function);
assertEquals(initialValue + 1, result);
assertRefHasNoLocks(ref);
assertSurplus(ref, 0);
assertVersionAndValue(ref, initialVersion + 1, initialValue + 1);
assertIsActive(tx);
assertSame(tx, getThreadLocalTxn());
}
@Test
public void whenLockedByOther(){
whenLockedByOther(LockMode.Read);
whenLockedByOther(LockMode.Write);
whenLockedByOther(LockMode.Exclusive);
}
public void whenLockedByOther(LockMode lockMode) {
Long initialValue = 5L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
GammaTxn otherTx = stm.newDefaultTxn();
ref.getLock().acquire(otherTx, lockMode);
long orecValue = ref.orec;
LongFunction function = mock(LongFunction.class);
try {
ref.atomicAlterAndGet(function);
fail();
} catch (LockedException expected) {
}
assertOrecValue(ref, orecValue);
verifyZeroInteractions(function);
assertRefHasLockMode(ref, otherTx, lockMode.asInt());
assertVersionAndValue(ref, initialVersion, initialValue);
}
@Test
public void whenListenersAvailable() {
Long initialValue = 10L;
GammaTxnRef<Long> ref = new GammaTxnRef<Long>(stm, initialValue);
long initialVersion = ref.getVersion();
TxnRefAwaitThread thread = new TxnRefAwaitThread(ref, initialValue + 1);
thread.start();
sleepMs(500);
ref.atomicAlterAndGet(Functions.incLongFunction());
joinAll(thread);
assertRefHasNoLocks(ref);
assertVersionAndValue(ref, initialVersion + 1, initialValue + 1);
}
}