/*
* JBoss, Home of Professional Open Source
* Copyright 2010 Red Hat Inc. and/or its affiliates and other
* contributors as indicated by the @author tags. All rights reserved.
* See the copyright.txt in the distribution for a full listing of
* individual contributors.
*
* This 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 2.1 of
* the License, or (at your option) any later version.
*
* This software 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 this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/
package org.infinispan.affinity;
import org.infinispan.manager.EmbeddedCacheManager;
import org.testng.annotations.Test;
import static junit.framework.Assert.assertEquals;
/**
* @author Mircea.Markus@jboss.com
* @since 4.1
*/
@Test (groups = "functional", testName = "affinity.KeyAffinityServiceShutdownTest")
public class KeyAffinityServiceShutdownTest extends BaseKeyAffinityServiceTest {
private EmbeddedCacheManager cacheManager;
@Override
protected void createCacheManagers() throws Throwable {
super.createCacheManagers();
cacheManager = manager(0);
keyAffinityService = (KeyAffinityServiceImpl<Object>) KeyAffinityServiceFactory.newKeyAffinityService(cacheManager.getCache(cacheName),
executor, new RndKeyGenerator(), 100);
}
public void testSimpleShutdown() throws Exception {
assertListenerRegistered(true);
assertEventualFullCapacity();
assert keyAffinityService.isKeyGeneratorThreadAlive();
keyAffinityService.stop();
for (int i = 0; i < 10; i++) {
if (!keyAffinityService.isKeyGeneratorThreadAlive())
break;
Thread.sleep(1000);
}
assert !keyAffinityService.isKeyGeneratorThreadAlive();
assert !executor.isShutdown();
}
@Test(dependsOnMethods = "testSimpleShutdown")
public void testServiceCannotBeUsedAfterShutdown() {
try {
keyAffinityService.getKeyForAddress(topology().get(0));
assert false : "Exception expected!";
} catch (IllegalStateException e) {
//expected
}
try {
keyAffinityService.getCollocatedKey("a");
assert false : "Exception expected!";
} catch (IllegalStateException e) {
//expected
}
}
@Test (dependsOnMethods = "testServiceCannotBeUsedAfterShutdown")
public void testViewChaneListenerUnregistered() {
assertListenerRegistered(false);
}
@Test (dependsOnMethods = "testViewChaneListenerUnregistered")
public void testRestart() throws InterruptedException {
keyAffinityService.start();
assertEventualFullCapacity();
}
private void assertListenerRegistered(boolean registered) {
boolean isRegistered = false;
for (Object o : cacheManager.getListeners()) {
if (o instanceof ListenerRegistration) {
isRegistered = true;
break;
}
}
assertEquals(registered, isRegistered);
}
}