| /*========================================================================= |
| * Copyright (c) 2010-2014 Pivotal Software, Inc. All Rights Reserved. |
| * This product is protected by U.S. and international copyright |
| * and intellectual property laws. Pivotal products are covered by |
| * one or more patents listed at http://www.pivotal.io/patents. |
| *========================================================================= |
| */ |
| package com.gemstone.gemfire.cache30; |
| |
| import com.gemstone.gemfire.cache.*; |
| import com.gemstone.gemfire.cache.util.*; |
| import com.gemstone.gemfire.distributed.*; |
| import com.gemstone.gemfire.distributed.internal.*; |
| import com.gemstone.gemfire.distributed.internal.membership.*; |
| import dunit.*; |
| import java.util.*; |
| |
| /** |
| * Tests the functionality of the {@link RegionRoleListener} class. |
| * |
| * @author Kirk Lund |
| * @since 5.0 |
| */ |
| public class RegionReliabilityListenerDUnitTest extends ReliabilityTestCase { |
| |
| protected static transient Set rolesGain = null; |
| protected static transient Set rolesLoss = null; |
| |
| public RegionReliabilityListenerDUnitTest(String name) { |
| super(name); |
| } |
| |
| /** |
| * Tests the notification of afterRoleGain and afterRoleLoss |
| */ |
| public void testRoleGainAndLoss() throws Exception { |
| final String name = this.getUniqueName(); |
| final int vm0 = 0; |
| final int vm1 = 1; |
| final int vm2 = 2; |
| final int vm3 = 3; |
| |
| final String roleA = name+"-A"; |
| final String roleC = name+"-C"; |
| final String roleD = name+"-D"; |
| |
| // assign names to 4 vms... |
| final String[] requiredRoles = {roleA,roleC,roleD}; |
| final String[] rolesProp = {"",roleA,roleA,roleC+","+roleD}; |
| final String[][] vmRoles = new String[][] {{},{roleA},{roleA},{roleC,roleD}}; |
| for (int i = 0; i < vmRoles.length; i++) { |
| final int vm = i; |
| Host.getHost(0).getVM(vm).invoke(new SerializableRunnable() { |
| public void run() { |
| Properties config = new Properties(); |
| config.setProperty(DistributionConfig.ROLES_NAME, rolesProp[vm]); |
| getSystem(config); |
| } |
| }); |
| } |
| |
| // define RegionRoleListener |
| final RegionRoleListener listener = |
| new RegionRoleListenerAdapter() { |
| public void afterRoleGain(RoleEvent event) { |
| RegionReliabilityListenerDUnitTest.rolesGain = event.getRequiredRoles(); |
| } |
| public void afterRoleLoss(RoleEvent event) { |
| RegionReliabilityListenerDUnitTest.rolesLoss = event.getRequiredRoles(); |
| } |
| }; |
| |
| // connect controller to system... |
| Properties config = new Properties(); |
| config.setProperty(DistributionConfig.ROLES_NAME, ""); |
| getSystem(config); |
| |
| // create region in controller... |
| MembershipAttributes ra = new MembershipAttributes( |
| requiredRoles, LossAction.FULL_ACCESS, ResumptionAction.NONE); |
| |
| AttributesFactory fac = new AttributesFactory(); |
| fac.addCacheListener(listener); |
| fac.setMembershipAttributes(ra); |
| fac.setScope(Scope.DISTRIBUTED_ACK); |
| |
| RegionAttributes attr = fac.create(); |
| createRootRegion(name, attr); |
| |
| // wait for memberTimeout to expire |
| waitForMemberTimeout(); |
| |
| // assert in state of role loss... test all are missing according to RequiredRoles |
| assertMissingRoles(name, requiredRoles); |
| |
| // assert gain is fired... |
| SerializableRunnable create = new CacheSerializableRunnable("Create Region") { |
| public void run2() throws CacheException { |
| AttributesFactory fac = new AttributesFactory(); |
| fac.setScope(Scope.DISTRIBUTED_ACK); |
| RegionAttributes attr = fac.create(); |
| createRootRegion(name, attr); |
| } |
| }; |
| |
| // create region in vm0... no gain for no role |
| Host.getHost(0).getVM(vm0).invoke(create); |
| assertNull(rolesGain); |
| assertNull(rolesLoss); |
| assertMissingRoles(name, requiredRoles); |
| |
| // create region in vm1... gain for 1st instance of redundant role |
| Host.getHost(0).getVM(vm1).invoke(create); |
| assertNotNull(rolesGain); |
| assertEquals(vmRoles[vm1].length, rolesGain.size()); |
| for (int i = 0; i < vmRoles[vm1].length; i++) { |
| Role role = InternalRole.getRole(vmRoles[vm1][i]); |
| assertEquals(true, rolesGain.contains(role)); |
| } |
| assertNull(rolesLoss); |
| rolesGain = null; |
| assertMissingRoles(name, vmRoles[vm3]); // only vm3 has missing roles |
| |
| // create region in vm2... no gain for 2nd instance of redundant role |
| Host.getHost(0).getVM(vm2).invoke(create); |
| assertNull(rolesGain); |
| assertNull(rolesLoss); |
| assertMissingRoles(name, vmRoles[vm3]); // only vm3 has missing roles |
| |
| // create region in vm3... gain for 2 roles |
| Host.getHost(0).getVM(vm3).invoke(create); |
| assertNotNull(rolesGain); |
| assertEquals(vmRoles[vm3].length, rolesGain.size()); |
| for (int i = 0; i < vmRoles[vm3].length; i++) { |
| Role role = InternalRole.getRole(vmRoles[vm3][i]); |
| assertEquals(true, rolesGain.contains(role)); |
| } |
| assertNull(rolesLoss); |
| rolesGain = null; |
| assertMissingRoles(name, new String[0]); // no missing roles |
| |
| // assert loss is fired... |
| SerializableRunnable destroy = new CacheSerializableRunnable("Destroy Region") { |
| public void run2() throws CacheException { |
| Region region = getRootRegion(name); |
| region.localDestroyRegion(); |
| } |
| }; |
| |
| // destroy region in vm0... no loss of any role |
| Host.getHost(0).getVM(vm0).invoke(destroy); |
| assertNull(rolesGain); |
| assertNull(rolesLoss); |
| assertMissingRoles(name, new String[0]); // no missing roles |
| |
| // destroy region in vm1... nothing happens in 1st removal of redundant role |
| Host.getHost(0).getVM(vm1).invoke(destroy); |
| assertNull(rolesGain); |
| assertNull(rolesLoss); |
| assertMissingRoles(name, new String[0]); // no missing roles |
| |
| // destroy region in vm2... 2nd removal of redundant role is loss |
| Host.getHost(0).getVM(vm2).invoke(destroy); |
| assertNull(rolesGain); |
| assertNotNull(rolesLoss); |
| assertEquals(vmRoles[vm2].length, rolesLoss.size()); |
| for (int i = 0; i < vmRoles[vm2].length; i++) { |
| Role role = InternalRole.getRole(vmRoles[vm2][i]); |
| assertEquals(true, rolesLoss.contains(role)); |
| } |
| rolesLoss = null; |
| assertMissingRoles(name, vmRoles[vm2]); // only vm2 has missing roles |
| |
| // destroy region in vm3... two more roles are in loss |
| Host.getHost(0).getVM(vm3).invoke(destroy); |
| assertNull(rolesGain); |
| assertNotNull(rolesLoss); |
| assertEquals(vmRoles[vm3].length, rolesLoss.size()); |
| for (int i = 0; i < vmRoles[vm3].length; i++) { |
| Role role = InternalRole.getRole(vmRoles[vm3][i]); |
| assertEquals(true, rolesLoss.contains(role)); |
| } |
| rolesLoss = null; |
| assertMissingRoles(name, requiredRoles); // all roles are missing |
| } |
| |
| } |
| |