View Javadoc
1   /*
2    * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
3    * The software in this package is published under the terms of the CPAL v1.0
4    * license, a copy of which has been included with this distribution in the
5    * LICENSE.txt file.
6    */
7   package org.mule.management.mbeans;
8   
9   import org.mule.api.config.ThreadingProfile;
10  import org.mule.component.DefaultJavaComponent;
11  import org.mule.management.AbstractMuleJmxTestCase;
12  import org.mule.model.seda.SedaModel;
13  import org.mule.model.seda.SedaService;
14  import org.mule.module.management.mbean.ServiceService;
15  import org.mule.object.SingletonObjectFactory;
16  
17  import java.util.Set;
18  
19  import javax.management.ObjectName;
20  
21  import org.junit.Test;
22  
23  import static org.junit.Assert.assertEquals;
24  
25  public class ServiceServiceTestCase extends AbstractMuleJmxTestCase
26  {
27      @Test
28      public void testUndeploy() throws Exception
29      {
30          final String domainOriginal = "TEST_DOMAIN_1";
31  
32          final SedaService service = new SedaService(muleContext);
33          service.setName("TEST_SERVICE");
34          SingletonObjectFactory factory = new SingletonObjectFactory(Object.class);
35          final DefaultJavaComponent component = new DefaultJavaComponent(factory);
36          component.setMuleContext(muleContext);
37          service.setComponent(component);
38  
39          service.setThreadingProfile(ThreadingProfile.DEFAULT_THREADING_PROFILE);
40          SedaModel model = new SedaModel();
41          model.setMuleContext(muleContext);
42          service.setModel(model);
43          muleContext.getRegistry().registerModel(model);
44          muleContext.getRegistry().registerService(service);
45          muleContext.start();
46  
47          final ServiceService jmxService = new ServiceService("TEST_SERVICE", muleContext);
48          final ObjectName name = ObjectName.getInstance(domainOriginal + ":type=TEST_SERVICE");
49          mBeanServer.registerMBean(jmxService, name);
50          Set mbeans = mBeanServer.queryMBeans(ObjectName.getInstance(domainOriginal + ":*"), null);
51  
52          // Expecting following mbeans to be registered:
53          // 1) org.mule.management.mbeans.ServiceService@TEST_DOMAIN_1:type=TEST_SERVICE
54          // 2) org.mule.management.mbeans.ServiceStats@TEST_DOMAIN_1:type=org.mule.Statistics,service=TEST_SERVICE
55          // 3) org.mule.management.mbeans.RouterStats@TEST_DOMAIN_1:type=org.mule.Statistics,service=TEST_SERVICE,router=inbound
56          // 4) org.mule.management.mbeans.RouterStats@TEST_DOMAIN_1:type=org.mule.Statistics,service=TEST_SERVICE,router=outbound
57          assertEquals("Unexpected number of components registered in the domain.", 4, mbeans.size());
58  
59          mBeanServer.unregisterMBean(name);
60  
61          mbeans = mBeanServer.queryMBeans(ObjectName.getInstance(domainOriginal + ":*"), null);
62  
63          assertEquals("There should be no MBeans left in the domain", 0, mbeans.size());
64      }
65  }