package io.eguan.nbdsrv;
/*
* #%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 io.eguan.nbdsrv.NbdExportAttributes;
import io.eguan.nbdsrv.NbdServer;
import io.eguan.nbdsrv.NbdServerMXBean;
import io.eguan.utils.Jmx;
import java.lang.management.ManagementFactory;
import java.net.InetAddress;
import javax.management.MBeanServer;
import javax.management.ObjectName;
import org.junit.After;
import org.junit.Before;
public class TestNbdServerJmx extends TestNbdServerAbstract {
private String serverMBeanObjName;
/**
* Initialize NBD server MBean.
*
* @throws Exception
*/
@Before
public void initServerMBean() throws Exception {
serverOrig = new NbdServer(InetAddress.getLoopbackAddress());
final MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
serverMBeanObjName = serverOrig.getClass().getPackage().getName() + ":type=Server";
final ObjectName mbeanName = new ObjectName(serverMBeanObjName);
mbs.registerMBean(serverOrig, mbeanName);
server = Jmx.findLocalMBean(serverMBeanObjName, NbdServerMXBean.class);
}
/**
* Unregister server. The server should be stopped.
*
* @throws Exception
*/
@After
public void finiServerMBean() throws Exception {
try {
final MBeanServer mbs = ManagementFactory.getPlatformMBeanServer();
final ObjectName mbeanName = new ObjectName(serverMBeanObjName);
mbs.unregisterMBean(mbeanName);
}
finally {
serverMBeanObjName = null;
}
}
@Override
protected final NbdExportAttributes[] getServerTargetAttributes() {
// TODO build IscsiTargetAttributes[] from composite data returned by the proxy?
return ((NbdServer) serverOrig).getExports();
}
}