@Test
public void testStart() throws Exception {
RoundRobinSchedulerStats stats =
EasyMock.createMockBuilder(RoundRobinSchedulerStats.class)
.withConstructor()
.addMockedMethod("registerMBean")
.addMockedMethod("createMovingAverage")
.createStrictMock();
MovingAverage mavg = EasyMock.createStrictMock(MovingAverage.class);
EasyMock.expect(stats.createMovingAverage()).andReturn(mavg);
mavg.startTimer("RoundRobinAddMavg", stats.getAddMavgPeriod(), TimeUnit.MILLISECONDS);
EasyMock.expectLastCall();
stats.registerMBean();
EasyMock.expectLastCall();
EasyMock.replay(stats, mavg);
stats.start();
EasyMock.verify(stats, mavg);
assertEquals("Mavg not set.", mavg, stats.m_addMavg);
// with already called
EasyMock.reset(stats, mavg);
EasyMock.replay(stats, mavg);
stats.start();
EasyMock.verify(stats, mavg);
}
@Test
public void testRegisterMBean() throws Exception {
RoundRobinSchedulerStats stats =
EasyMock.createMockBuilder(RoundRobinSchedulerStats.class)
.withConstructor()
.addMockedMethod("getMBeanServer")
.addMockedMethod("createObjectName")
.createStrictMock();
// do nothing
stats.setUseJMX(false);
EasyMock.replay(stats);
stats.registerMBean();
EasyMock.verify(stats);
assertNull("Object name should still be null.", stats.m_mbeanName);
// now work
EasyMock.reset(stats);
stats.setUseJMX(true);
MBeanServer mbs = EasyMock.createStrictMock(MBeanServer.class);
ObjectName objectName = EasyMock.createStrictMock(ObjectName.class);
EasyMock.expect(stats.getMBeanServer()).andReturn(mbs);
EasyMock.expect(stats.createObjectName()).andReturn(objectName);
EasyMock.expect(mbs.registerMBean(stats, objectName)).andReturn(null);
EasyMock.replay(stats, mbs, objectName);
stats.registerMBean();
EasyMock.verify(stats, mbs, objectName);
assertEquals("Object name should not be null.", objectName, stats.m_mbeanName);
// with exception that is just logged
EasyMock.reset(stats, mbs, objectName);
stats.m_mbeanName = null;
EasyMock.expect(stats.getMBeanServer()).andReturn(mbs);
EasyMock.expect(stats.createObjectName()).andReturn(objectName);
EasyMock.expect(mbs.registerMBean(stats, objectName))
.andThrow(new NotCompliantMBeanException("blah"));
EasyMock.replay(stats, mbs, objectName);
stats.registerMBean();
EasyMock.verify(stats, mbs, objectName);
assertNull("Object name should still be null.", stats.m_mbeanName);
}
