gemfire-core/src/test/java/com/gemstone/gemfire/management/CompositeTypeTestDUnitTest.java

/*=========================================================================
 * 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.management;


import javax.management.MalformedObjectNameException;
import javax.management.ObjectName;

import com.gemstone.gemfire.internal.cache.GemFireCacheImpl;
import com.gemstone.gemfire.management.internal.MBeanJMXAdapter;
import com.gemstone.gemfire.management.internal.ManagementConstants;
import com.gemstone.gemfire.management.internal.SystemManagementService;

import dunit.SerializableRunnable;
import dunit.VM;

public class CompositeTypeTestDUnitTest extends ManagementTestBase {

  public CompositeTypeTestDUnitTest(String name) {
    super(name);
    // TODO Auto-generated constructor stub
  }

  /**
   * 
   */
  private static final long serialVersionUID = 1L;
  
  private static ObjectName objectName;

  public void setUp() throws Exception {
    super.setUp();
    
  }

  public void tearDown2() throws Exception {
    super.tearDown2();
    
  }
  
  public void testCompositeTypeGetters() throws Exception{
    
    initManagement(false);
    String member = getMemberId(managedNode1);
    member = MBeanJMXAdapter.makeCompliantName(member);
    
    registerMBeanWithCompositeTypeGetters(managedNode1,member);

    
    checkMBeanWithCompositeTypeGetters(managingNode,member);
    
  }

  
  /**
   * Creates a Local region
   *
   * @param vm
   *          reference to VM
   * @param localRegionName
   *          name of the local region
   * @throws Throwable
   */
  protected void registerMBeanWithCompositeTypeGetters(VM vm,final String memberID)
      throws Exception {
    SerializableRunnable regMBean = new SerializableRunnable(
        "Register CustomMBean with composite Type") {
      public void run() {
        GemFireCacheImpl cache = GemFireCacheImpl.getInstance();
        SystemManagementService service = (SystemManagementService) getManagementService();
 
        try {
          ObjectName objectName = new ObjectName("GemFire:service=custom,type=composite");
          CompositeTestMXBean mbean =  new CompositeTestMBean();
          objectName = service.registerMBean(mbean, objectName);
          service.federate(objectName, CompositeTestMXBean.class, false);
        } catch (MalformedObjectNameException e) {
          // TODO Auto-generated catch block
          e.printStackTrace();
        } catch (NullPointerException e) {
          // TODO Auto-generated catch block
          e.printStackTrace();
        }

        

      }
    };
    vm.invoke(regMBean);
  }
  
  
  /**
   * Creates a Local region
   *
   * @param vm
   *          reference to VM
   * @param localRegionName
   *          name of the local region
   * @throws Throwable
   */
  protected void checkMBeanWithCompositeTypeGetters(VM vm,final String memberID)
      throws Exception {
    SerializableRunnable checkMBean = new SerializableRunnable(
        "Check CustomMBean with composite Type") {
      public void run() {
        GemFireCacheImpl cache = GemFireCacheImpl.getInstance();
        final SystemManagementService service = (SystemManagementService) getManagementService();

        try {
          final ObjectName objectName = new ObjectName("GemFire:service=custom,type=composite,member="+memberID);
          
          waitForCriterion(new WaitCriterion() {
            public String description() {
              return "Waiting for Composite Type MBean";
            }

            public boolean done() {
              CompositeTestMXBean bean = service.getMBeanInstance(objectName, CompositeTestMXBean.class);
              boolean done = (bean != null);
              return done;
            }

          },  ManagementConstants.REFRESH_TIME*4, 500, true);

          
          CompositeTestMXBean bean = service.getMBeanInstance(objectName, CompositeTestMXBean.class);
          
          CompositeStats listData = bean.listCompositeStats();
          
          System.out.println("connectionStatsType = "+listData.getConnectionStatsType());
          System.out.println("connectionsOpened = "+listData.getConnectionsOpened());
          System.out.println("connectionsClosed = "+listData.getConnectionsClosed());
          System.out.println("connectionsAttempted = "+listData.getConnectionsAttempted());
          System.out.println("connectionsFailed = "+listData.getConnectionsFailed());
          
          CompositeStats getsData = bean.getCompositeStats();
          System.out.println("connectionStatsType = "+getsData.getConnectionStatsType());
          System.out.println("connectionsOpened = "+getsData.getConnectionsOpened());
          System.out.println("connectionsClosed = "+getsData.getConnectionsClosed());
          System.out.println("connectionsAttempted = "+getsData.getConnectionsAttempted());
          System.out.println("connectionsFailed = "+getsData.getConnectionsFailed());
          
          CompositeStats[] arrayData = bean.getCompositeArray();
          Integer[] intArrayData = bean.getIntegerArray();
          Thread.sleep(2*60*1000);
        } catch (MalformedObjectNameException e) {
          // TODO Auto-generated catch block
          e.printStackTrace();
        } catch (NullPointerException e) {
          // TODO Auto-generated catch block
          e.printStackTrace();
        } catch (InterruptedException e) {
          // TODO Auto-generated catch block
          e.printStackTrace();
        }
        
        

      }
    };
    vm.invoke(checkMBean);
  }

  
}