mirror of
https://github.com/Alfresco/alfresco-community-repo.git
synced 2025-08-07 17:49:17 +00:00
Merged HEAD-QA to HEAD (4.2) (including moving test classes into separate folders)
51903 to 54309 git-svn-id: https://svn.alfresco.com/repos/alfresco-enterprise/alfresco/HEAD/root@54310 c4b6b30b-aa2e-2d43-bbcb-ca4b014f7261
This commit is contained in:
Samuel Langlois
460
source/test-java/org/alfresco/repo/lock/JobLockServiceTest.java
Normal file
460
source/test-java/org/alfresco/repo/lock/JobLockServiceTest.java
Normal file
@@ -0,0 +1,460 @@
|
||||
/*
|
||||
* Copyright (C) 2005-2010 Alfresco Software Limited.
|
||||
*
|
||||
* This file is part of Alfresco
|
||||
*
|
||||
* Alfresco is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Lesser General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* Alfresco is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Lesser General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU Lesser General Public License
|
||||
* along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
package org.alfresco.repo.lock;
|
||||
|
||||
import junit.framework.TestCase;
|
||||
|
||||
import org.alfresco.repo.domain.locks.LockDAO;
|
||||
import org.alfresco.repo.lock.JobLockService.JobLockRefreshCallback;
|
||||
import org.alfresco.repo.transaction.RetryingTransactionHelper;
|
||||
import org.alfresco.repo.transaction.RetryingTransactionHelper.RetryingTransactionCallback;
|
||||
import org.alfresco.service.ServiceRegistry;
|
||||
import org.alfresco.service.namespace.QName;
|
||||
import org.alfresco.service.transaction.TransactionService;
|
||||
import org.alfresco.util.ApplicationContextHelper;
|
||||
import org.springframework.context.ApplicationContext;
|
||||
|
||||
/**
|
||||
* Tests the high-level capabilities provided by the service implementation. The DAO tests
|
||||
* stress the underlying database work, so we only need to deal with deadlock resolution, etc.
|
||||
*
|
||||
* @see JobLockService the service being tested
|
||||
* @see LockDAO the DAO being indirectly tested
|
||||
*
|
||||
* @author Derek Hulley
|
||||
* @since 3.2
|
||||
*/
|
||||
@SuppressWarnings("unused")
|
||||
public class JobLockServiceTest extends TestCase
|
||||
{
|
||||
public static final String NAMESPACE = "http://www.alfresco.org/test/JobLockServiceTest";
|
||||
|
||||
private ApplicationContext ctx = ApplicationContextHelper.getApplicationContext();
|
||||
|
||||
private TransactionService transactionService;
|
||||
private RetryingTransactionHelper txnHelper;
|
||||
private JobLockService jobLockService;
|
||||
// Lock names for the tests
|
||||
private QName lockA;
|
||||
private QName lockAA;
|
||||
private QName lockAAA;
|
||||
private QName lockAAB;
|
||||
private QName lockAAC;
|
||||
private QName lockAB;
|
||||
private QName lockABA;
|
||||
private QName lockABB;
|
||||
private QName lockABC;
|
||||
|
||||
@Override
|
||||
public void setUp() throws Exception
|
||||
{
|
||||
ServiceRegistry serviceRegistry = (ServiceRegistry) ctx.getBean(ServiceRegistry.SERVICE_REGISTRY);
|
||||
transactionService = serviceRegistry.getTransactionService();
|
||||
txnHelper = transactionService.getRetryingTransactionHelper();
|
||||
|
||||
jobLockService = (JobLockService) ctx.getBean("jobLockService");
|
||||
|
||||
// Get the test name
|
||||
String testName = getName();
|
||||
// Build lock names for the test
|
||||
lockA = QName.createQName(NAMESPACE, "a-" + testName);
|
||||
lockAA = QName.createQName(NAMESPACE, "a-" + testName + ".a-" + testName);
|
||||
lockAAA = QName.createQName(NAMESPACE, "a-" + testName + ".a-" + testName + ".a-" + testName);
|
||||
lockAAB = QName.createQName(NAMESPACE, "a-" + testName + ".a-" + testName + ".b-" + testName);
|
||||
lockAAC = QName.createQName(NAMESPACE, "a-" + testName + ".a-" + testName + ".c-" + testName);
|
||||
lockAB = QName.createQName(NAMESPACE, "a-" + testName + ".b-" + testName);
|
||||
lockABA = QName.createQName(NAMESPACE, "a-" + testName + ".b-" + testName + ".a-" + testName);
|
||||
lockABB = QName.createQName(NAMESPACE, "a-" + testName + ".b-" + testName + ".b-" + testName);
|
||||
lockABC = QName.createQName(NAMESPACE, "a-" + testName + ".b-" + testName + ".c-" + testName);
|
||||
}
|
||||
|
||||
public void testSetUp()
|
||||
{
|
||||
assertNotNull(jobLockService);
|
||||
}
|
||||
|
||||
public void testSimpleLock()
|
||||
{
|
||||
String lockToken = jobLockService.getLock(lockAAA, 20L);
|
||||
jobLockService.refreshLock(lockToken, lockAAA, 20L);
|
||||
jobLockService.releaseLock(lockToken, lockAAA);
|
||||
try
|
||||
{
|
||||
jobLockService.refreshLock(lockToken, lockAAA, 20L);
|
||||
fail("Lock refresh should have failed after release");
|
||||
}
|
||||
catch (LockAcquisitionException e)
|
||||
{
|
||||
// Expected
|
||||
}
|
||||
lockToken = jobLockService.getLock(lockAAA, 20L, 5L, 0); // No retries
|
||||
jobLockService.refreshLock(lockToken, lockAAA, 20L);
|
||||
jobLockService.releaseLock(lockToken, lockAAA);
|
||||
}
|
||||
|
||||
public void testEnforceTxn()
|
||||
{
|
||||
try
|
||||
{
|
||||
jobLockService.getTransactionalLock(lockAAA, 50L);
|
||||
fail("Service did not enforce the presence of a transaction");
|
||||
}
|
||||
catch (IllegalStateException e)
|
||||
{
|
||||
// Expected
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks that the lock can be aquired by a read-only transaction i.e. that locking is
|
||||
* independent of the outer transaction.
|
||||
*/
|
||||
public void testLockInReadOnly() throws Exception
|
||||
{
|
||||
RetryingTransactionCallback<Object> lockCallback = new RetryingTransactionCallback<Object>()
|
||||
{
|
||||
public Object execute() throws Throwable
|
||||
{
|
||||
jobLockService.getTransactionalLock(lockAAA, 500);
|
||||
return null;
|
||||
}
|
||||
};
|
||||
txnHelper.doInTransaction(lockCallback, true, true);
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks that locks are released on commit
|
||||
*/
|
||||
public void testLockReleaseOnCommit() throws Exception
|
||||
{
|
||||
RetryingTransactionCallback<Object> lockCallback = new RetryingTransactionCallback<Object>()
|
||||
{
|
||||
public Object execute() throws Throwable
|
||||
{
|
||||
jobLockService.getTransactionalLock(lockAAA, 5000);
|
||||
return null;
|
||||
}
|
||||
};
|
||||
txnHelper.doInTransaction(lockCallback, true, true);
|
||||
// The lock should be free now, even though the TTL was high
|
||||
RetryingTransactionCallback<Object> lockCheckCallback = new RetryingTransactionCallback<Object>()
|
||||
{
|
||||
public Object execute() throws Throwable
|
||||
{
|
||||
jobLockService.getTransactionalLock(lockAAA, 50);
|
||||
return null;
|
||||
}
|
||||
};
|
||||
txnHelper.doInTransaction(lockCheckCallback, true, true);
|
||||
}
|
||||
|
||||
/**
|
||||
* Checks that locks are released on rollback
|
||||
*/
|
||||
public void testLockReleaseOnRollback() throws Exception
|
||||
{
|
||||
RetryingTransactionCallback<Object> lockCallback = new RetryingTransactionCallback<Object>()
|
||||
{
|
||||
public Object execute() throws Throwable
|
||||
{
|
||||
jobLockService.getTransactionalLock(lockAAA, 5000);
|
||||
throw new UnsupportedOperationException("ALERT!");
|
||||
}
|
||||
};
|
||||
try
|
||||
{
|
||||
txnHelper.doInTransaction(lockCallback, true, true);
|
||||
fail("Expected transaction failure");
|
||||
}
|
||||
catch (UnsupportedOperationException e)
|
||||
{
|
||||
// Expected
|
||||
}
|
||||
// The lock should be free now, even though the TTL was high
|
||||
RetryingTransactionCallback<Object> lockCheckCallback = new RetryingTransactionCallback<Object>()
|
||||
{
|
||||
public Object execute() throws Throwable
|
||||
{
|
||||
jobLockService.getTransactionalLock(lockAAA, 50);
|
||||
return null;
|
||||
}
|
||||
};
|
||||
txnHelper.doInTransaction(lockCheckCallback, true, true);
|
||||
}
|
||||
|
||||
/**
|
||||
* Sets up two threads in a deadlock scenario. Each of the threads has a long wait timeout
|
||||
* for the required locks. If there were a deadlock, the shorter of the the wait times would
|
||||
* be how long it would take before one of them is thrown out. Firstly, we check that one
|
||||
* of the threads <i>is</i> thrown out. Then we check that the thread is thrown out quickly.
|
||||
*/
|
||||
public synchronized void testDeadlockPrevention() throws Throwable
|
||||
{
|
||||
DeadlockingThread t1 = new DeadlockingThread(lockAAA, lockAAB);
|
||||
DeadlockingThread t2 = new DeadlockingThread(lockAAB, lockAAA);
|
||||
// Start them
|
||||
t1.start();
|
||||
t2.start();
|
||||
// They can take their first locks (there should be no contention)
|
||||
t1.incrementNextLock();
|
||||
t2.incrementNextLock();
|
||||
// Wait for them to do this
|
||||
try { this.wait(2000L); } catch (InterruptedException e) {}
|
||||
// Advance again
|
||||
t1.incrementNextLock();
|
||||
t2.incrementNextLock();
|
||||
// Wait for them to do this
|
||||
try { this.wait(2000L); } catch (InterruptedException e) {}
|
||||
// Advance again, to end threads
|
||||
t1.incrementNextLock();
|
||||
t2.incrementNextLock();
|
||||
// Wait for them to end (commit/rollback)
|
||||
try { this.wait(2000L); } catch (InterruptedException e) {}
|
||||
|
||||
if (t1.otherFailure != null)
|
||||
{
|
||||
throw t1.otherFailure;
|
||||
}
|
||||
if (t2.otherFailure != null)
|
||||
{
|
||||
throw t2.otherFailure;
|
||||
}
|
||||
assertNull("T1 should have succeeded as the ordered locker: " + t1.lockFailure, t1.lockFailure);
|
||||
assertNotNull("T2 should have failed as the unordered locker.", t2.lockFailure);
|
||||
}
|
||||
|
||||
private class DeadlockingThread extends Thread
|
||||
{
|
||||
private final QName[] lockQNames;
|
||||
private volatile int nextLock = -1;
|
||||
private LockAcquisitionException lockFailure;
|
||||
private Throwable otherFailure;
|
||||
|
||||
private DeadlockingThread(QName ... lockQNames)
|
||||
{
|
||||
super("DeadlockingThread");
|
||||
this.lockQNames = lockQNames;
|
||||
setDaemon(true);
|
||||
}
|
||||
|
||||
private void incrementNextLock()
|
||||
{
|
||||
nextLock++;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run()
|
||||
{
|
||||
RetryingTransactionCallback<Object> runCallback = new RetryingTransactionCallback<Object>()
|
||||
{
|
||||
public synchronized Object execute() throws Throwable
|
||||
{
|
||||
int currentLock = -1;
|
||||
// Take the locks in turn, quitting when told to take a lock that's not there
|
||||
while (currentLock < lockQNames.length - 1)
|
||||
{
|
||||
// Check if we have been instructed to take a lock
|
||||
if (nextLock > currentLock)
|
||||
{
|
||||
// Advance and grab the lock
|
||||
currentLock++;
|
||||
jobLockService.getTransactionalLock(lockQNames[currentLock], 5000L);
|
||||
}
|
||||
else
|
||||
{
|
||||
// No advance, so wait a bit more
|
||||
try { this.wait(20L); } catch (InterruptedException e) {}
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
};
|
||||
try
|
||||
{
|
||||
txnHelper.doInTransaction(runCallback, true);
|
||||
}
|
||||
catch (LockAcquisitionException e)
|
||||
{
|
||||
lockFailure = e;
|
||||
}
|
||||
catch (Throwable e)
|
||||
{
|
||||
otherFailure = e;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public synchronized void testLockCallbackReleaseInactive() throws Exception
|
||||
{
|
||||
final QName lockQName = QName.createQName(NAMESPACE, getName());
|
||||
final long lockTTL = 1000L;
|
||||
final String lockToken = jobLockService.getLock(lockQName, lockTTL);
|
||||
|
||||
final int[] checked = new int[1];
|
||||
final int[] released = new int[1];
|
||||
// Immediately-inactive job
|
||||
JobLockRefreshCallback callback = new JobLockRefreshCallback()
|
||||
{
|
||||
@Override
|
||||
public boolean isActive()
|
||||
{
|
||||
checked[0]++;
|
||||
return false;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void lockReleased()
|
||||
{
|
||||
released[0]++;
|
||||
}
|
||||
};
|
||||
|
||||
jobLockService.refreshLock(lockToken, lockQName, lockTTL, callback);
|
||||
// The first refresh will occur in 500ms
|
||||
wait(1000L);
|
||||
// Should have been released by now
|
||||
assertTrue("Expected lockReleased to have been called", released[0] > 0);
|
||||
try
|
||||
{
|
||||
jobLockService.getLock(lockQName, lockTTL);
|
||||
}
|
||||
catch (LockAcquisitionException e)
|
||||
{
|
||||
fail("Lock should have been released by callback infrastructure");
|
||||
}
|
||||
|
||||
// Check that the timed callback is killed properly
|
||||
int checkedCount = checked[0];
|
||||
int releasedCount = released[0];
|
||||
wait(2000L);
|
||||
assertEquals("Lock callback timer was not terminated", checkedCount, checked[0]);
|
||||
assertEquals("Lock callback timer was not terminated", releasedCount, released[0]);
|
||||
}
|
||||
|
||||
public synchronized void testLockCallbackReleaseSelf() throws Exception
|
||||
{
|
||||
final QName lockQName = QName.createQName(NAMESPACE, getName());
|
||||
final long lockTTL = 1000L;
|
||||
final String lockToken = jobLockService.getLock(lockQName, lockTTL);
|
||||
|
||||
final int[] checked = new int[1];
|
||||
final int[] released = new int[1];
|
||||
// Immediately-inactive job, releasing the lock
|
||||
JobLockRefreshCallback callback = new JobLockRefreshCallback()
|
||||
{
|
||||
@Override
|
||||
public boolean isActive()
|
||||
{
|
||||
checked[0]++;
|
||||
jobLockService.releaseLock(lockToken, lockQName);
|
||||
return false;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void lockReleased()
|
||||
{
|
||||
released[0]++;
|
||||
}
|
||||
};
|
||||
|
||||
jobLockService.refreshLock(lockToken, lockQName, lockTTL, callback);
|
||||
// The first refresh will occur in 500ms
|
||||
wait(1000L);
|
||||
// Should NOT get a callback saying that the lock has been released
|
||||
assertTrue("Lock should be optimistically released", released[0] > 0);
|
||||
try
|
||||
{
|
||||
jobLockService.getLock(lockQName, lockTTL);
|
||||
}
|
||||
catch (LockAcquisitionException e)
|
||||
{
|
||||
fail("Lock should have been released by callback infrastructure");
|
||||
}
|
||||
|
||||
// Check that the timed callback is killed properly
|
||||
int checkedCount = checked[0];
|
||||
int releasedCount = released[0];
|
||||
wait(2000L);
|
||||
assertEquals("Lock callback timer was not terminated", checkedCount, checked[0]);
|
||||
assertEquals("Lock callback timer was not terminated", releasedCount, released[0]);
|
||||
}
|
||||
|
||||
/**
|
||||
* Lets job "run" for 3 seconds and checks at 2s and 4s.
|
||||
*/
|
||||
public synchronized void testLockCallbackReleaseTimed() throws Exception
|
||||
{
|
||||
final QName lockQName = QName.createQName(NAMESPACE, getName());
|
||||
final long lockTTL = 1000L;
|
||||
final long lockNow = System.currentTimeMillis();
|
||||
final String lockToken = jobLockService.getLock(lockQName, lockTTL);
|
||||
|
||||
final int[] checked = new int[1];
|
||||
final int[] released = new int[1];
|
||||
// Do not release and remain active
|
||||
JobLockRefreshCallback callback = new JobLockRefreshCallback()
|
||||
{
|
||||
@Override
|
||||
public boolean isActive()
|
||||
{
|
||||
checked[0]++;
|
||||
if (System.currentTimeMillis() - lockNow > 3000L)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
else
|
||||
{
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void lockReleased()
|
||||
{
|
||||
released[0]++;
|
||||
}
|
||||
};
|
||||
|
||||
jobLockService.refreshLock(lockToken, lockQName, lockTTL, callback);
|
||||
// The first refresh will occur in 500ms
|
||||
wait(2000L);
|
||||
|
||||
assertTrue("Expected at least 2 active checks; only got " + checked[0], checked[0] >= 2);
|
||||
assertFalse("lockReleased should NOT have been called", released[0] > 0);
|
||||
try
|
||||
{
|
||||
jobLockService.getLock(lockQName, lockTTL);
|
||||
fail("Lock should still be held");
|
||||
}
|
||||
catch (LockAcquisitionException e)
|
||||
{
|
||||
// Expected
|
||||
}
|
||||
|
||||
// Wait for another 2s to be sure that the lock is run to completion
|
||||
wait(2000L);
|
||||
|
||||
// Check that the timed callback is killed properly
|
||||
int checkedCount = checked[0];
|
||||
int releasedCount = released[0];
|
||||
wait(2000L);
|
||||
assertEquals("Lock callback timer was not terminated", checkedCount, checked[0]);
|
||||
assertEquals("Lock callback timer was not terminated", releasedCount, released[0]);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user