package org.infinispan.distribution;
import java.util.List;
import org.infinispan.Cache;
import org.infinispan.configuration.cache.CacheMode;
import org.infinispan.remoting.transport.Address;
import org.infinispan.test.MultipleCacheManagersTest;
import org.testng.annotations.Test;
@Test(groups = "functional", testName = "distribution.RemoteGetTest")
public class RemoteGetTest extends MultipleCacheManagersTest {
@Override
protected void createCacheManagers() throws Throwable {
createCluster(getDefaultClusteredCacheConfig(CacheMode.DIST_SYNC, false), 3);
// make sure all caches are started...
cache(0);
cache(1);
cache(2);
waitForClusterToForm();
}
@SuppressWarnings("unchecked")
private Cache<MagicKey, String> getCacheForAddress(Address a) {
for (Cache<?, ?> c: caches())
if (c.getAdvancedCache().getRpcManager().getAddress().equals(a)) return (Cache<MagicKey, String>) c;
return null;
}
@SuppressWarnings("unchecked")
private Cache<MagicKey, String> getNonOwner(List<Address> a) {
for (Cache<?, ?> c: caches())
if (!a.contains(c.getAdvancedCache().getRpcManager().getAddress())) return (Cache<MagicKey, String>) c;
return null;
}
public void testRemoteGet() {
Cache<MagicKey, String> c1 = cache(0);
Cache<MagicKey, String> c2 = cache(1);
Cache<MagicKey, String> c3 = cache(2);
MagicKey k = new MagicKey(c1, c2);
List<Address> owners = c1.getAdvancedCache().getDistributionManager().locate(k);
assert owners.size() == 2: "Key should have 2 owners";
Cache<MagicKey, String> owner1 = getCacheForAddress(owners.get(0));
assert owner1 == c1;
Cache<MagicKey, String> owner2 = getCacheForAddress(owners.get(1));
assert owner2 == c2;
Cache<MagicKey, String> nonOwner = getNonOwner(owners);
assert nonOwner == c3;
owner1.put(k, "value");
assert "value".equals(nonOwner.get(k));
}
public void testGetOfNonexistentKey() {
Object v = cache(0).get("__ doesn't exist ___");
assert v == null : "Should get a null response";
}
public void testGetOfNonexistentKeyOnOwner() {
MagicKey mk = new MagicKey("does not exist", cache(0));
Object v = cache(0).get(mk);
assert v == null : "Should get a null response";
}
}