01: /*
02: * JBoss, Home of Professional Open Source.
03: * Copyright 2006, Red Hat Middleware LLC, and individual contributors
04: * as indicated by the @author tags. See the copyright.txt file in the
05: * distribution for a full listing of individual contributors.
06: *
07: * This is free software; you can redistribute it and/or modify it
08: * under the terms of the GNU Lesser General Public License as
09: * published by the Free Software Foundation; either version 2.1 of
10: * the License, or (at your option) any later version.
11: *
12: * This software is distributed in the hope that it will be useful,
13: * but WITHOUT ANY WARRANTY; without even the implied warranty of
14: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15: * Lesser General Public License for more details.
16: *
17: * You should have received a copy of the GNU Lesser General Public
18: * License along with this software; if not, write to the Free
19: * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
20: * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
21: */
22: package test.compliance.server.support;
23:
24: import java.lang.reflect.InvocationHandler;
25: import java.lang.reflect.Method;
26: import java.lang.reflect.Proxy;
27:
28: import javax.management.MBeanServer;
29:
30: /**
31: * A wrapper for an MBeanServer
32: *
33: * @author <a href="mailto:Adrian.Brock@HappeningTimes.com">Adrian Brock</a>.
34: * @version $Revision: 57200 $
35: */
36: public class MBeanServerWrapper implements InvocationHandler {
37: public MBeanServer server;
38:
39: public boolean invoked = false;
40:
41: private static Method EQUALS;
42:
43: static {
44: try {
45: EQUALS = Object.class.getClass().getMethod("equals",
46: new Class[] { Object.class });
47: } catch (Exception e) {
48: throw new RuntimeException(e.toString());
49: }
50: }
51:
52: public static MBeanServer getWrapper() {
53: return (MBeanServer) Proxy.newProxyInstance(Thread
54: .currentThread().getContextClassLoader(),
55: new Class[] { MBeanServer.class },
56: new MBeanServerWrapper());
57: }
58:
59: public static MBeanServerWrapper getHandler(MBeanServer proxy) {
60: return (MBeanServerWrapper) Proxy.getInvocationHandler(proxy);
61: }
62:
63: public Object invoke(Object proxy, Method method, Object[] args)
64: throws Throwable {
65: invoked = true;
66: if (method.equals(EQUALS))
67: return new Boolean(proxy == args[0]);
68: if (method.getName().equals("queryMBeans"))
69: throw new MBeanServerReplaced();
70: return method.invoke(server, args);
71: }
72: }
|