2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
4 * This program and the accompanying materials are made available under the
5 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6 * and is available at http://www.eclipse.org/legal/epl-v10.html
8 package org.opendaylight.controller.config.manager.impl;
10 import static org.junit.Assert.assertEquals;
12 import com.google.common.collect.Sets;
13 import java.lang.management.ManagementFactory;
14 import java.util.HashMap;
17 import javax.management.MBeanServer;
18 import javax.management.MBeanServerFactory;
19 import javax.management.ObjectName;
20 import org.junit.After;
21 import org.junit.Before;
22 import org.junit.Test;
23 import org.opendaylight.controller.config.api.jmx.ObjectNameUtil;
24 import org.opendaylight.controller.config.manager.impl.jmx.BaseJMXRegistrator;
25 import org.opendaylight.controller.config.manager.impl.jmx.TransactionJMXRegistrator;
26 import org.opendaylight.controller.config.manager.impl.jmx.TransactionModuleJMXRegistrator;
27 import org.opendaylight.controller.config.manager.impl.runtimembean.TestingRuntimeBean;
28 import org.opendaylight.controller.config.spi.ModuleFactory;
29 import org.osgi.framework.BundleContext;
31 public class ConfigTransactionControllerImplTest extends
32 AbstractLockedPlatformMBeanServerTest {
34 private BaseJMXRegistrator baseJMXRegistrator;
36 private ConfigTransactionControllerImpl testedTxController;
37 private MBeanServer transactionsMBeanServer;
39 private static final String transactionName123 = "testTX1";
40 private static final String transactionName4 = "testTX2";
42 private static final String moduleName124 = "module124";
43 private static final String moduleName3 = "module3";
45 private static final String instanceName134 = "instA";
46 private static final String instanceName2 = "instB";
48 private static final ObjectName name1 = ObjectNameUtil
49 .createTransactionModuleON(transactionName123, moduleName124, instanceName134);
50 private static final ObjectName name2 = ObjectNameUtil
51 .createTransactionModuleON(transactionName123, moduleName124, instanceName2);
52 private static final ObjectName name3 = ObjectNameUtil
53 .createTransactionModuleON(transactionName123, moduleName3, instanceName134);
54 private static final ObjectName name4 = ObjectNameUtil
55 .createTransactionModuleON(transactionName4, moduleName124, instanceName134);
58 public void setUp() throws Exception {
59 baseJMXRegistrator = new BaseJMXRegistrator(
60 ManagementFactory.getPlatformMBeanServer());
61 transactionsMBeanServer = MBeanServerFactory.createMBeanServer();
62 Map<String, Map.Entry<ModuleFactory, BundleContext>> currentlyRegisteredFactories = new HashMap<>();
64 ConfigTransactionLookupRegistry txLookupRegistry = new ConfigTransactionLookupRegistry(new TransactionIdentifier(transactionName123), new TransactionJMXRegistratorFactory() {
66 public TransactionJMXRegistrator create() {
67 return baseJMXRegistrator.createTransactionJMXRegistrator(transactionName123);
69 }, currentlyRegisteredFactories);
71 SearchableServiceReferenceWritableRegistry writableRegistry = ServiceReferenceRegistryImpl.createSRWritableRegistry(
72 ServiceReferenceRegistryImpl.createInitialSRLookupRegistry(), txLookupRegistry, currentlyRegisteredFactories);
75 testedTxController = new ConfigTransactionControllerImpl(
76 txLookupRegistry, 1, null, 1,
77 currentlyRegisteredFactories, transactionsMBeanServer,
78 ManagementFactory.getPlatformMBeanServer(), false, writableRegistry);
79 TransactionModuleJMXRegistrator transactionModuleJMXRegistrator123 = testedTxController
80 .getTxModuleJMXRegistrator();
81 transactionModuleJMXRegistrator123.registerMBean(
82 new TestingRuntimeBean(), name1);
83 transactionModuleJMXRegistrator123.registerMBean(
84 new TestingRuntimeBean(), name2);
85 transactionModuleJMXRegistrator123.registerMBean(
86 new TestingRuntimeBean(), name3);
87 TransactionJMXRegistrator jmxRegistrator4 = baseJMXRegistrator
88 .createTransactionJMXRegistrator(transactionName4);
89 jmxRegistrator4.createTransactionModuleJMXRegistrator().registerMBean(
90 new TestingRuntimeBean(), name4);
94 public void cleanUp() {
95 baseJMXRegistrator.close();
96 MBeanServerFactory.releaseMBeanServer(transactionsMBeanServer);
100 * Tests if lookup method returns all beans with defined transaction name
105 public void testLookupConfigBeans() {
106 Set<ObjectName> beans = testedTxController.lookupConfigBeans();
107 assertEquals(Sets.newHashSet(name1, name2, name3), beans);
111 public void testLookupConfigBeansWithModuleName() {
112 Set<ObjectName> beans = testedTxController
113 .lookupConfigBeans(moduleName124);
114 assertEquals(Sets.newHashSet(name1, name2), beans);
118 public void lookupConfigBeansWithModuleNameAndImplName() throws Exception {
119 Set<ObjectName> beans = testedTxController.lookupConfigBeans(
120 moduleName124, instanceName134);
121 assertEquals(Sets.newHashSet(name1), beans);