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.subscription;
  34 
  35 import static org.junit.Assert.assertNotNull;
  36 import static org.junit.Assert.fail;
  37 
  38 import javax.management.MBeanServerConnection;
  39 
  40 import org.junit.Test;
  41 
  42 import org.openjdk.jmc.rjmx.ConnectionToolkit;
  43 import org.openjdk.jmc.rjmx.IConnectionHandle;
  44 import org.openjdk.jmc.rjmx.subscription.IMRIValueListener;
  45 import org.openjdk.jmc.rjmx.subscription.MRI;
  46 import org.openjdk.jmc.rjmx.subscription.MRIValueEvent;
  47 import org.openjdk.jmc.rjmx.subscription.MRI.Type;
  48 import org.openjdk.jmc.rjmx.test.RjmxTestCase;
  49 import org.openjdk.jmc.rjmx.test.testutil.TestToolkit;
  50 
  51 /**
  52  * Tests attribute subscriptions based on JMX notification.
  53  */
  54 public class NotificationAttributeSubscriptionTest extends RjmxTestCase implements IMRIValueListener {
  55         private final static int SLEEP_TIME = 6000;
  56 
  57         /**
  58          * Tests the normal one shot subscription to the strategy.
  59          */
  60         @Test
  61         public void testGetGCSubscription() {
  62                 try {
  63                         // RMP does not support notifications
  64                         MRI descriptor = null;
  65                         synchronized (this) {
  66                                 // Starting up a subscription on a one shot attribute.
  67                                 descriptor = new MRI(Type.NOTIFICATION, "com.sun.management:type=GarbageCollectionAggregator", //$NON-NLS-1$
  68                                                 "com.sun.management.gc.notification"); //$NON-NLS-1$
  69                                 getAttributeSubscriptionService().addMRIValueListener(descriptor, this);
  70 
  71                                 secondThreadException = null;
  72                                 final IConnectionHandle threadModel = m_connectionHandle;
  73                                 new Thread(new Runnable() {
  74                                         @Override
  75                                         public void run() {
  76                                                 synchronized (NotificationAttributeSubscriptionTest.this) {
  77                                                         try {
  78                                                                 ConnectionToolkit
  79                                                                                 .getMemoryBean(threadModel.getServiceOrThrow(MBeanServerConnection.class)).gc();
  80                                                         } catch (Exception e) {
  81                                                                 secondThreadException = e;
  82                                                         }
  83                                                 }
  84                                         }
  85                                 }).start();
  86                                 wait(SLEEP_TIME);
  87                                 if (secondThreadException != null) {
  88                                         throw secondThreadException;
  89                                 }
  90                         }
  91                         assertNotNull(getAttributeSubscriptionService().getLastMRIValueEvent(descriptor));
  92                         getAttributeSubscriptionService().removeMRIValueListener(this);
  93                 } catch (Exception e) {
  94                         e.printStackTrace();
  95                         fail(e.getMessage());
  96                 }
  97         }
  98 
  99         private Exception secondThreadException = null;
 100 
 101         /**
 102          * @param event
 103          * @see IMRIValueListener#valueChanged(MRIValueEvent)
 104          */
 105         @Override
 106         public synchronized void valueChanged(MRIValueEvent event) {
 107                 TestToolkit.println(event);
 108                 // Only notify if we got a real server side event
 109                 if (event.getValue() != null) {
 110                         notifyAll();
 111                 }
 112         }
 113 }