1 /*
   2  * Copyright (c) 2008, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25  * @test
  26  * @bug 6689505
  27  * @summary Checks that MBeanServerNotification.toString contains the
  28  *          MBean name.
  29  * @author Daniel Fuchs
  30  * @compile MBeanServerNotificationTest.java
  31  * @run main MBeanServerNotificationTest
  32  */
  33 
  34 import com.sun.jmx.mbeanserver.Util;
  35 import javax.management.*;
  36 import java.util.concurrent.*;
  37 
  38 public class MBeanServerNotificationTest {
  39     final static String[] names = {
  40         ":type=Wombat", "wombat:type=Wombat",null,
  41     };
  42     public static void main(String[] args) throws Exception {
  43         System.out.println("Test that MBeanServerNotification.toString " +
  44                 "contains the name of the MBean being registered " +
  45                 "or unregistered.");
  46         int failures = 0;
  47         final MBeanServer mbs = MBeanServerFactory.createMBeanServer();
  48         for (String str:names) {
  49             try {
  50                 final ObjectName name = (str==null)?null:new ObjectName(str);
  51                 failures+=test(mbs, name, name!=null);
  52             } catch(Exception x) {
  53                 x.printStackTrace(System.out);
  54                 System.out.println("Test failed for: "+str);
  55                 failures++;
  56             }
  57         }
  58         if (failures == 0)
  59             System.out.println("Test passed");
  60         else {
  61             System.out.println("TEST FAILED: " + failures + " failure(s)");
  62             System.exit(1);
  63         }
  64     }
  65 
  66     private static enum Registration {
  67         REGISTER(MBeanServerNotification.REGISTRATION_NOTIFICATION),
  68         UNREGISTER(MBeanServerNotification.UNREGISTRATION_NOTIFICATION);
  69         final String type;
  70         private Registration(String type) {this.type = type;}
  71         public int test(MBeanServerNotification n, ObjectName name) {
  72             int failures = 0;
  73             System.out.println("Testing: "+n);
  74             if (!n.toString().endsWith("[type="+type+
  75                 "][message="+n.getMessage()+
  76                 "][mbeanName="+name+"]")) {
  77                 System.err.println("Test failed for "+ type+
  78                    " ["+name+"]: "+n);
  79                 failures++;
  80             }
  81             return failures;
  82         }
  83         public MBeanServerNotification create(ObjectName name) {
  84             return new MBeanServerNotification(type,
  85                 MBeanServerDelegate.DELEGATE_NAME, next(), name);
  86         }
  87         private static long next = 0;
  88         private static synchronized long next() {return next++;}
  89 
  90     }
  91 
  92     private static int test(MBeanServer mbs, ObjectName name,
  93                             boolean register)
  94             throws Exception {
  95         System.out.println("--------" + name + "--------");
  96 
  97         int failures = 0;
  98         for (Registration reg : Registration.values()) {
  99             failures = reg.test(reg.create(name), name);
 100         }
 101         if (!register) return failures;
 102 
 103         final ArrayBlockingQueue<Notification> queue =
 104                 new ArrayBlockingQueue<Notification>(10);
 105         final NotificationListener listener = new NotificationListener() {
 106             public void handleNotification(Notification notification,
 107                     Object handback) {
 108                 try {
 109                     queue.put(notification);
 110                 } catch(Exception x) {
 111                     x.printStackTrace(System.out);
 112                 }
 113             }
 114         };
 115         mbs.addNotificationListener(MBeanServerDelegate.DELEGATE_NAME,
 116                 listener, null, name);
 117         final ObjectInstance oi = mbs.registerMBean(new Wombat(), name);
 118         try {
 119             failures+=Registration.REGISTER.test((MBeanServerNotification)
 120                 queue.poll(2, TimeUnit.SECONDS), oi.getObjectName());
 121         } finally {
 122             mbs.unregisterMBean(oi.getObjectName());
 123             failures+=Registration.UNREGISTER.test((MBeanServerNotification)
 124                 queue.poll(2, TimeUnit.SECONDS), oi.getObjectName());
 125         }
 126         return failures;
 127     }
 128 
 129     public static interface WombatMBean {}
 130     public static class Wombat implements WombatMBean {}
 131 
 132 }