/*
* ScanManagerTest.java
* JUnit based test
*
* Created on July 10, 2006, 5:57 PM
*
* @(#)ScanManagerTest.java 1.2 06/08/02
*
* Copyright (c) 2006 Sun Microsystems, Inc. All Rights Reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
*
* -Redistribution of source code must retain the above copyright notice, this
* list of conditions and the following disclaimer.
*
* -Redistribution in binary form must reproduce the above copyright notice,
* this list of conditions and the following disclaimer in the documentation
* and/or other materials provided with the distribution.
*
* Neither the name of Sun Microsystems, Inc. or the names of contributors may
* be used to endorse or promote products derived from this software without
* specific prior written permission.
*
* This software is provided "AS IS," without a warranty of any kind. ALL
* EXPRESS OR IMPLIED CONDITIONS, REPRESENTATIONS AND WARRANTIES, INCLUDING
* ANY IMPLIED WARRANTY OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE
* OR NON-INFRINGEMENT, ARE HEREBY EXCLUDED. SUN MICROSYSTEMS, INC. ("SUN")
* AND ITS LICENSORS SHALL NOT BE LIABLE FOR ANY DAMAGES SUFFERED BY LICENSEE
* AS A RESULT OF USING, MODIFYING OR DISTRIBUTING THIS SOFTWARE OR ITS
* DERIVATIVES. IN NO EVENT WILL SUN OR ITS LICENSORS BE LIABLE FOR ANY LOST
* REVENUE, PROFIT OR DATA, OR FOR DIRECT, INDIRECT, SPECIAL, CONSEQUENTIAL,
* INCIDENTAL OR PUNITIVE DAMAGES, HOWEVER CAUSED AND REGARDLESS OF THE THEORY
* OF LIABILITY, ARISING OUT OF THE USE OF OR INABILITY TO USE THIS SOFTWARE,
* EVEN IF SUN HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
*
* You acknowledge that this software is not designed, licensed or intended
* for use in the design, construction, operation or maintenance of any
* nuclear facility.
*/
package com.sun.jmx.examples.scandir;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;
import javax.management.InstanceNotFoundException;
import javax.management.Notification;
import junit.framework.*;
import static com.sun.jmx.examples.scandir.ScanManagerMXBean.ScanState.*;
import com.sun.jmx.examples.scandir.ScanManagerMXBean.ScanState;
import java.io.IOException;
import java.lang.management.ManagementFactory;
import java.util.EnumSet;
import java.util.HashMap;
import java.util.logging.Logger;
import javax.management.AttributeChangeNotification;
import javax.management.JMException;
import javax.management.JMX;
import javax.management.ListenerNotFoundException;
import javax.management.MBeanNotificationInfo;
import javax.management.MBeanRegistration;
import javax.management.MBeanServer;
import javax.management.MBeanServerConnection;
import javax.management.NotificationBroadcasterSupport;
import javax.management.NotificationEmitter;
import javax.management.NotificationFilter;
import javax.management.NotificationListener;
import javax.management.ObjectInstance;
import javax.management.ObjectName;
import static com.sun.jmx.examples.scandir.ScanManagerMXBean.ScanState.*;
/**
* Unit tests for {@code ScanManager}
*
* @author Sun Microsystems, 2006 - All rights reserved.
*/
public class ScanManagerTest extends TestCase {
public ScanManagerTest(String testName) {
super(testName);
}
protected void setUp() throws Exception {
}
protected void tearDown() throws Exception {
}
public static Test suite() {
TestSuite suite = new TestSuite(ScanManagerTest.class);
return suite;
}
/**
* Test of makeSingletonName method, of class com.sun.jmx.examples.scandir.ScanManager.
*/
public void testMakeSingletonName() {
System.out.println("makeSingletonName");
Class clazz = ScanManagerMXBean.class;
ObjectName expResult = ScanManager.SCAN_MANAGER_NAME;
ObjectName result = ScanManager.makeSingletonName(clazz);
assertEquals(expResult, result);
}
/**
* Test of register method, of class com.sun.jmx.examples.scandir.ScanManager.
*/
public void testRegister() throws Exception {
System.out.println("register");
MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
ScanManagerMXBean result = ScanManager.register(mbs);
try {
assertEquals(STOPPED,result.getState());
} finally {
try {
mbs.unregisterMBean(ScanManager.SCAN_MANAGER_NAME);
} catch (Exception x) {
System.err.println("Failed to cleanup: "+x);
}
}
}
public interface Call {
public void call() throws Exception;
public void cancel() throws Exception;
}
/**
* Test of addNotificationListener method, of class com.sun.jmx.examples.scandir.ScanManager.
*/
public void testAddNotificationListener() throws Exception {
System.out.println("addNotificationListener");
final ScanManagerMXBean manager = ScanManager.register();
final Call op = new Call() {
public void call() throws Exception {
manager.schedule(100000,0);
}
public void cancel() throws Exception {
manager.stop();
}
};
try {
doTestOperation(manager,op,
EnumSet.of(RUNNING,SCHEDULED),
"schedule");
} finally {
try {
ManagementFactory.getPlatformMBeanServer().
unregisterMBean(ScanManager.SCAN_MANAGER_NAME);
} catch (Exception x) {
System.err.println("Failed to cleanup: "+x);
}
}
}
/**
* Test of addNotificationListener method, of class com.sun.jmx.examples.scandir.ScanManager.
*/
private void doTestOperation(
ScanManagerMXBean proxy,
Call op,
EnumSet<ScanState> after,
String testName)
throws Exception {
System.out.println("doTestOperation: "+testName);
final LinkedBlockingQueue<Notification> queue =
new LinkedBlockingQueue<Notification>();
NotificationListener listener = new NotificationListener() {
public void handleNotification(Notification notification,
Object handback) {
try {
queue.put(notification);
} catch (Exception x) {
System.err.println("Failed to queue notif: "+x);
}
}
};
NotificationFilter filter = null;
Object handback = null;
final ScanState before;
final NotificationEmitter emitter = (NotificationEmitter)proxy;
emitter.addNotificationListener(listener, filter, handback);
before = proxy.getState();
op.call();
try {
final Notification notification =
queue.poll(3000,TimeUnit.MILLISECONDS);
assertEquals(AttributeChangeNotification.ATTRIBUTE_CHANGE,
notification.getType());
assertEquals(AttributeChangeNotification.class,
notification.getClass());
assertEquals(ScanManager.SCAN_MANAGER_NAME,
notification.getSource());
AttributeChangeNotification acn =
(AttributeChangeNotification)notification;
assertEquals("State",acn.getAttributeName());
assertEquals(ScanState.class.getName(),acn.getAttributeType());
assertEquals(before,ScanState.valueOf((String)acn.getOldValue()));
assertContained(after,ScanState.valueOf((String)acn.getNewValue()));
emitter.removeNotificationListener(listener,filter,handback);
} finally {
try {
op.cancel();
} catch (Exception x) {
System.err.println("Failed to cleanup: "+x);
}
}
}
/**
* Test of preRegister method, of class com.sun.jmx.examples.scandir.ScanManager.
*/
public void testPreRegister() throws Exception {
System.out.println("preRegister");
MBeanServer server = ManagementFactory.getPlatformMBeanServer();
ObjectName name = new ObjectName("DownUnder:type=Wombat");
ScanManager instance = new ScanManager();
ObjectName expResult = ScanManager.SCAN_MANAGER_NAME;
ObjectName result;
try {
result = instance.preRegister(server, name);
throw new RuntimeException("bad name accepted!");
} catch (IllegalArgumentException x) {
// OK!
result = instance.preRegister(server, null);
}
assertEquals(expResult, result);
result = instance.preRegister(server, ScanManager.SCAN_MANAGER_NAME);
assertEquals(expResult, result);
}
/**
* Test of getState method, of class com.sun.jmx.examples.scandir.ScanManager.
*/
public void testGetState() throws IOException, InstanceNotFoundException {
System.out.println("getState");
ScanManager instance = new ScanManager();
ScanState expResult = ScanState.STOPPED;
ScanState result = instance.getState();
assertEquals(expResult, result);
instance.start();
final ScanState afterStart = instance.getState();
assertContained(EnumSet.of(RUNNING,SCHEDULED,COMPLETED),afterStart);
instance.stop();
assertEquals(STOPPED,instance.getState());
instance.schedule(1000000L,1000000L);
assertEquals(SCHEDULED,instance.getState());
instance.stop();
assertEquals(STOPPED,instance.getState());
}
/**
* Test of schedule method, of class com.sun.jmx.examples.scandir.ScanManager.
*/
public void testSchedule() throws Exception {
System.out.println("schedule");
final long delay = 10000L;
final long interval = 10000L;
final ScanManagerMXBean manager = ScanManager.register();
final Call op = new Call() {
public void call() throws Exception {
manager.schedule(delay,interval);
assertEquals(SCHEDULED,manager.getState());
}
public void cancel() throws Exception {
manager.stop();
}
};
try {
doTestOperation(manager,op,EnumSet.of(SCHEDULED),
"schedule");
} finally {
try {
ManagementFactory.getPlatformMBeanServer().
unregisterMBean(ScanManager.SCAN_MANAGER_NAME);
} catch (Exception x) {
System.err.println("Failed to cleanup: "+x);
}
}
}
public static void assertContained(EnumSet<ScanState> allowed,
ScanState state) {
final String msg = String.valueOf(state) + " is not one of " + allowed;
assertTrue(msg,allowed.contains(state));
}
/**
* Test of stop method, of class com.sun.jmx.examples.scandir.ScanManager.
*/
public void testStop() throws Exception {
System.out.println("stop");
final ScanManagerMXBean manager = ScanManager.register();
try {
manager.schedule(1000000,0);
assertContained(EnumSet.of(SCHEDULED),manager.getState());
final Call op = new Call() {
public void call() throws Exception {
manager.stop();
assertEquals(STOPPED,manager.getState());
}
public void cancel() throws Exception {
if (manager.getState() != STOPPED)
manager.stop();
}
};
doTestOperation(manager,op,EnumSet.of(STOPPED),"stop");
} finally {
try {
ManagementFactory.getPlatformMBeanServer().
unregisterMBean(ScanManager.SCAN_MANAGER_NAME);
} catch (Exception x) {
System.err.println("Failed to cleanup: "+x);
}
}
}
/**
* Test of start method, of class com.sun.jmx.examples.scandir.ScanManager.
*/
public void testStart() throws Exception {
final ScanManagerMXBean manager = ScanManager.register();
try {
final Call op = new Call() {
public void call() throws Exception {
assertEquals(STOPPED,manager.getState());
manager.start();
assertContained(EnumSet.of(RUNNING,SCHEDULED,COMPLETED),
manager.getState());
}
public void cancel() throws Exception {
manager.stop();
}
};
doTestOperation(manager,op,EnumSet.of(RUNNING,SCHEDULED,COMPLETED),
"start");
} finally {
try {
ManagementFactory.getPlatformMBeanServer().
unregisterMBean(ScanManager.SCAN_MANAGER_NAME);
} catch (Exception x) {
System.err.println("Failed to cleanup: "+x);
}
}
}
}