/*
* Copyright (c) 2008-2017, Hazelcast, Inc. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.hazelcast.spring.cache;
import com.hazelcast.core.DistributedObject;
import com.hazelcast.core.DistributedObjectEvent;
import com.hazelcast.core.DistributedObjectListener;
import com.hazelcast.core.Hazelcast;
import com.hazelcast.core.HazelcastInstance;
import com.hazelcast.core.IMap;
import com.hazelcast.spring.CustomSpringJUnit4ClassRunner;
import com.hazelcast.test.HazelcastTestSupport;
import com.hazelcast.test.annotation.QuickTest;
import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.experimental.categories.Category;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cache.CacheManager;
import org.springframework.test.context.ContextConfiguration;
import javax.annotation.Resource;
import java.util.Collection;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.atomic.AtomicBoolean;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.fail;
@RunWith(CustomSpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = {"cacheManager-applicationContext-hazelcast.xml"})
@Category(QuickTest.class)
public class TestCacheManager extends HazelcastTestSupport {
@Resource(name = "instance")
private HazelcastInstance instance;
@Autowired
private IDummyBean bean;
@Autowired
private CacheManager cacheManager;
@BeforeClass
@AfterClass
public static void start() {
Hazelcast.shutdownAll();
}
@Test
public void testBean_withValue() {
for (int i = 0; i < 100; i++) {
assertEquals("name:" + i, bean.getName(i));
assertEquals("city:" + i, bean.getCity(i));
}
}
@Test
public void testBean_withNull() {
for (int i = 0; i < 100; i++) {
assertNull(bean.getNull());
}
}
@Test
public void testBean_withTTL() {
String name = bean.getNameWithTTL();
assertEquals("ali", name);
String nameFromCache = bean.getNameWithTTL();
assertEquals("ali", nameFromCache);
sleepSeconds(3);
String nameFromCacheAfterTTL = bean.getNameWithTTL();
assertNull(nameFromCacheAfterTTL);
}
@Test
public void testCacheNames() {
// create a test instance, to reproduce the behavior described in the GitHub issue
// https://github.com/hazelcast/hazelcast/issues/492
final String testMap = "test-map";
final CountDownLatch distributionSignal = new CountDownLatch(1);
instance.addDistributedObjectListener(new DistributedObjectListener() {
@Override
public void distributedObjectCreated(DistributedObjectEvent event) {
DistributedObject distributedObject = event.getDistributedObject();
if (distributedObject instanceof IMap) {
IMap<?, ?> map = (IMap) distributedObject;
if (testMap.equals(map.getName())) {
distributionSignal.countDown();
}
}
}
@Override
public void distributedObjectDestroyed(DistributedObjectEvent event) {
}
});
HazelcastInstance testInstance = Hazelcast.newHazelcastInstance();
testInstance.getMap(testMap);
// be sure that test-map is distributed
HazelcastTestSupport.assertOpenEventually(distributionSignal);
Collection<String> test = cacheManager.getCacheNames();
assertContains(test, testMap);
testInstance.shutdown();
}
public static class DummyBean implements IDummyBean {
final AtomicBoolean nullCall = new AtomicBoolean(false);
final AtomicBoolean firstCall = new AtomicBoolean(false);
@Override
public String getName(int k) {
fail("should not call this method!");
return null;
}
@Override
public String getCity(int k) {
fail("should not call this method!");
return null;
}
@Override
public Object getNull() {
if (nullCall.compareAndSet(false, true)) {
return null;
}
fail("should not call this method!");
return null;
}
@Override
public String getNameWithTTL() {
if (firstCall.compareAndSet(false, true)) {
return "ali";
}
return null;
}
}
}