package io.eguan.vold.model;
/*
* #%L
* Project eguan
* %%
* Copyright (C) 2012 - 2017 Oodrive
* %%
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* #L%
*/
import java.lang.management.ManagementFactory;
import java.net.UnknownHostException;
import java.util.concurrent.atomic.AtomicBoolean;
import javax.management.AttributeChangeNotification;
import javax.management.InstanceNotFoundException;
import javax.management.ListenerNotFoundException;
import javax.management.MBeanServer;
import javax.management.Notification;
import javax.management.NotificationListener;
import javax.management.ObjectName;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
public abstract class AbstractMXBeanListener extends AbstractVoldTest {
protected AbstractMXBeanListener() throws Exception {
super();
}
/**
* Dummy listener. Does nothing.
*/
private static final NotificationListener DUMMY_LISTENER = new NotificationListener() {
@Override
public final void handleNotification(final Notification notification, final Object handback) {
// No op
}
};
protected MBeanServer server;
protected ObjectName mbeanName;
protected ObjectName vvrObjectName;
public abstract void setName(final String name);
public abstract void setDescription(final String description);
@Before
public void initVvr() throws Exception {
vvrObjectName = helper.newVvrObjectName(vvrUuid);
server = ManagementFactory.getPlatformMBeanServer();
}
/**
* Listener that expects some values.
*
* @author
*/
class NotificationListenerExpectation implements NotificationListener {
private final Object handback;
private final Object oldValue;
private final Object newValue;
final AtomicBoolean notified = new AtomicBoolean(false);
NotificationListenerExpectation(final Object handback, final Object oldValue, final Object newValue) {
super();
this.handback = handback;
this.oldValue = oldValue;
this.newValue = newValue;
}
@Override
public final void handleNotification(final Notification notification, final Object handback) {
Assert.assertEquals(mbeanName, notification.getSource());
Assert.assertEquals(this.handback, handback);
Assert.assertTrue(notification instanceof AttributeChangeNotification);
final AttributeChangeNotification attributeChangeNotification = (AttributeChangeNotification) notification;
Assert.assertEquals(this.oldValue, attributeChangeNotification.getOldValue());
Assert.assertEquals(this.newValue, attributeChangeNotification.getNewValue());
notified.set(true);
}
}
@Test(expected = ListenerNotFoundException.class)
public void removeListener1() throws ListenerNotFoundException, InstanceNotFoundException {
server.removeNotificationListener(vvrObjectName, DUMMY_LISTENER);
}
@Test(expected = ListenerNotFoundException.class)
public void removeListener3() throws ListenerNotFoundException, InstanceNotFoundException {
server.removeNotificationListener(vvrObjectName, DUMMY_LISTENER, null, new Object());
}
@Test
public void removeAddedListener1() throws ListenerNotFoundException, InstanceNotFoundException {
server.addNotificationListener(vvrObjectName, DUMMY_LISTENER, null, new Object());
server.removeNotificationListener(vvrObjectName, DUMMY_LISTENER);
}
@Test
public void removeAddedListener3() throws ListenerNotFoundException, InstanceNotFoundException {
final Object handback = new Object();
server.addNotificationListener(vvrObjectName, DUMMY_LISTENER, null, handback);
server.removeNotificationListener(vvrObjectName, DUMMY_LISTENER, null, handback);
}
@Test(expected = ListenerNotFoundException.class)
public void removeAddedListener3Failed() throws ListenerNotFoundException, InstanceNotFoundException {
server.addNotificationListener(vvrObjectName, DUMMY_LISTENER, null, new Object());
server.removeNotificationListener(vvrObjectName, DUMMY_LISTENER, null, new Object());
}
@Test
public void removeListenNameChange() throws ListenerNotFoundException, UnknownHostException,
InstanceNotFoundException, InterruptedException {
final Object handback = new Object();
final NotificationListenerExpectation listener = new NotificationListenerExpectation(handback, "name0", "name1");
server.addNotificationListener(vvrObjectName, listener, null, handback);
try {
setName("name1");
Assert.assertTrue(listener.notified.get());
}
finally {
server.removeNotificationListener(vvrObjectName, listener);
}
}
@Test
public void removeListenDescriptionChange() throws ListenerNotFoundException, UnknownHostException,
InstanceNotFoundException, InterruptedException {
final Object handback = new Object();
final NotificationListenerExpectation listener = new NotificationListenerExpectation(handback, "desc0", "desc1");
server.addNotificationListener(vvrObjectName, listener, null, handback);
try {
setDescription("desc1");
Assert.assertTrue(listener.notified.get());
}
finally {
server.removeNotificationListener(vvrObjectName, listener);
}
}
}