1 /*
   2  * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
   3  * 
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * The contents of this file are subject to the terms of either the Universal Permissive License
   7  * v 1.0 as shown at http://oss.oracle.com/licenses/upl
   8  *
   9  * or the following license:
  10  *
  11  * Redistribution and use in source and binary forms, with or without modification, are permitted
  12  * provided that the following conditions are met:
  13  * 
  14  * 1. Redistributions of source code must retain the above copyright notice, this list of conditions
  15  * and the following disclaimer.
  16  * 
  17  * 2. Redistributions in binary form must reproduce the above copyright notice, this list of
  18  * conditions and the following disclaimer in the documentation and/or other materials provided with
  19  * the distribution.
  20  * 
  21  * 3. Neither the name of the copyright holder nor the names of its contributors may be used to
  22  * endorse or promote products derived from this software without specific prior written permission.
  23  * 
  24  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
  25  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
  26  * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
  27  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  29  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
  30  * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY
  31  * WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33 package org.openjdk.jmc.rjmx.test;
  34 
  35 import static org.junit.Assert.assertEquals;
  36 
  37 import java.io.IOException;
  38 
  39 import javax.management.MBeanServerConnection;
  40 import javax.management.remote.JMXServiceURL;
  41 
  42 import org.junit.Test;
  43 
  44 import org.openjdk.jmc.rjmx.ConnectionDescriptorBuilder;
  45 import org.openjdk.jmc.rjmx.ConnectionToolkit;
  46 import org.openjdk.jmc.rjmx.IConnectionDescriptor;
  47 import org.openjdk.jmc.rjmx.IConnectionHandle;
  48 import org.openjdk.jmc.rjmx.IServerHandle;
  49 import org.openjdk.jmc.rjmx.internal.JMXConnectionDescriptor;
  50 import org.openjdk.jmc.rjmx.subscription.IMRISubscription;
  51 import org.openjdk.jmc.rjmx.subscription.IMRIValueListener;
  52 import org.openjdk.jmc.rjmx.subscription.ISubscriptionService;
  53 import org.openjdk.jmc.rjmx.subscription.MRI;
  54 import org.openjdk.jmc.rjmx.subscription.MRI.Type;
  55 import org.openjdk.jmc.rjmx.subscription.MRIValueEvent;
  56 import org.openjdk.jmc.rjmx.subscription.PolicyFactory;
  57 
  58 /**
  59  * Basic tests for the new cleaned up RJMX API.
  60  */
  61 public class BasicCommunicationTest extends ServerHandleTestCase {
  62         final static int DEFAULT_PORT = 0;
  63         final static String DEFAULT_HOST = "localhost".intern();
  64         final static String ALTERNATIVE_HOST = "127.1.0.1".intern();
  65 
  66         /**
  67          * Attempts creating a {@link JMXConnectionDescriptor} with a bunch of options.
  68          */
  69         @Test
  70         public void testCreateJMXDescriptor() throws IOException {
  71                 IConnectionDescriptor descriptor = LocalRJMXTestToolkit.createDefaultDescriptor();
  72                 JMXServiceURL url = descriptor.createJMXServiceURL();
  73                 assertEquals(DEFAULT_HOST, ConnectionToolkit.getHostName(url));
  74                 assertEquals(DEFAULT_PORT, ConnectionToolkit.getPort(url));
  75         }
  76 
  77         @Test
  78         public void testIConnectionHandle() throws Exception {
  79                 IConnectionHandle handle = getDefaultServer().connect("Test");
  80                 MRI descriptor = new MRI(Type.ATTRIBUTE, "java.lang:type=Threading",
  81                                 "ThreadCount");
  82                 MBeanServerConnection connection = handle.getServiceOrThrow(MBeanServerConnection.class);
  83                 assertBetween(1L, 1000L,
  84                                 ((Number) connection.getAttribute(descriptor.getObjectName(), descriptor.getDataPath())).longValue());
  85                 handle.close();
  86         }
  87 
  88         @Test
  89         public void testServerHandle() throws Exception {
  90                 IConnectionHandle handle = getDefaultServer().connect("Test");
  91 
  92                 assertEquals(1, getDefaultServer().getConnectionHandles().length);
  93                 assertEquals(handle, getDefaultServer().getConnectionHandles()[0]);
  94 
  95                 handle.close();
  96                 assertEquals(0, getDefaultServer().getConnectionHandles().length);
  97         }
  98 
  99         protected void muppTestConnection() throws Exception {
 100                 ConnectionDescriptorBuilder builder = new ConnectionDescriptorBuilder();
 101                 IConnectionDescriptor descriptor = builder.hostName("localhost").port(0).build();
 102                 IConnectionHandle handle = IServerHandle.create(descriptor).connect("Test");
 103 
 104                 ISubscriptionService service = handle.getServiceOrThrow(ISubscriptionService.class);
 105                 MRI attribute = new MRI(Type.ATTRIBUTE, "java.lang:type=Threading",
 106                                 "ThreadCount");
 107                 service.addMRIValueListener(attribute, new IMRIValueListener() {
 108                         @Override
 109                         public void valueChanged(MRIValueEvent event) {
 110                                 System.out.println(event.getValue());
 111                         }
 112                 });
 113 
 114                 IMRISubscription subscription = service.getMRISubscription(attribute);
 115                 subscription.setUpdatePolicy(PolicyFactory.createSimpleUpdatePolicy(1500));
 116 
 117                 Thread.sleep(5000);
 118                 handle.close();
 119         }
 120 }