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 com.google.common.collect.Sets;
11 import org.junit.After;
12 import org.junit.Before;
13 import org.junit.Test;
14 import org.opendaylight.controller.config.api.jmx.ObjectNameUtil;
15 import org.opendaylight.controller.config.manager.impl.jmx.BaseJMXRegistrator;
16 import org.opendaylight.controller.config.manager.impl.jmx.TransactionJMXRegistrator;
17 import org.opendaylight.controller.config.manager.impl.jmx.TransactionModuleJMXRegistrator;
18 import org.opendaylight.controller.config.manager.impl.runtimembean.TestingRuntimeBean;
19 import org.opendaylight.controller.config.spi.ModuleFactory;
20 import org.osgi.framework.BundleContext;
22 import javax.management.MBeanServer;
23 import javax.management.MBeanServerFactory;
24 import javax.management.ObjectName;
25 import java.lang.management.ManagementFactory;
26 import java.util.HashMap;
30 import static org.junit.Assert.assertEquals;
32 public class ConfigTransactionControllerImplTest extends
33 AbstractLockedPlatformMBeanServerTest {
35 private BaseJMXRegistrator baseJMXRegistrator;
37 private ConfigTransactionControllerImpl testedTxController;
38 private MBeanServer transactionsMBeanServer;
40 private static final String transactionName123 = "testTX1";
41 private static final String transactionName4 = "testTX2";
43 private static final String moduleName124 = "module124";
44 private static final String moduleName3 = "module3";
46 private static final String instanceName134 = "instA";
47 private static final String instanceName2 = "instB";
49 private static final ObjectName name1 = ObjectNameUtil
50 .createTransactionModuleON(transactionName123, moduleName124, instanceName134);
51 private static final ObjectName name2 = ObjectNameUtil
52 .createTransactionModuleON(transactionName123, moduleName124, instanceName2);
53 private static final ObjectName name3 = ObjectNameUtil
54 .createTransactionModuleON(transactionName123, moduleName3, instanceName134);
55 private static final ObjectName name4 = ObjectNameUtil
56 .createTransactionModuleON(transactionName4, moduleName124, instanceName134);
59 public void setUp() throws Exception {
60 baseJMXRegistrator = new BaseJMXRegistrator(
61 ManagementFactory.getPlatformMBeanServer());
62 transactionsMBeanServer = MBeanServerFactory.createMBeanServer();
63 Map<String, Map.Entry<ModuleFactory, BundleContext>> currentlyRegisteredFactories = new HashMap<>();
65 ConfigTransactionLookupRegistry txLookupRegistry = new ConfigTransactionLookupRegistry(new TransactionIdentifier(transactionName123), new TransactionJMXRegistratorFactory() {
67 public TransactionJMXRegistrator create() {
68 return baseJMXRegistrator.createTransactionJMXRegistrator(transactionName123);
70 }, currentlyRegisteredFactories);
72 SearchableServiceReferenceWritableRegistry writableRegistry = ServiceReferenceRegistryImpl.createSRWritableRegistry(
73 ServiceReferenceRegistryImpl.createInitialSRLookupRegistry(), txLookupRegistry, currentlyRegisteredFactories);
76 testedTxController = new ConfigTransactionControllerImpl(
77 txLookupRegistry, 1, null, 1,
78 currentlyRegisteredFactories, transactionsMBeanServer,
79 ManagementFactory.getPlatformMBeanServer(), false, writableRegistry);
80 TransactionModuleJMXRegistrator transactionModuleJMXRegistrator123 = testedTxController
81 .getTxModuleJMXRegistrator();
82 transactionModuleJMXRegistrator123.registerMBean(
83 new TestingRuntimeBean(), name1);
84 transactionModuleJMXRegistrator123.registerMBean(
85 new TestingRuntimeBean(), name2);
86 transactionModuleJMXRegistrator123.registerMBean(
87 new TestingRuntimeBean(), name3);
88 TransactionJMXRegistrator jmxRegistrator4 = baseJMXRegistrator
89 .createTransactionJMXRegistrator(transactionName4);
90 jmxRegistrator4.createTransactionModuleJMXRegistrator().registerMBean(
91 new TestingRuntimeBean(), name4);
95 public void cleanUp() {
96 baseJMXRegistrator.close();
97 MBeanServerFactory.releaseMBeanServer(transactionsMBeanServer);
101 * Tests if lookup method returns all beans with defined transaction name
106 public void testLookupConfigBeans() {
107 Set<ObjectName> beans = testedTxController.lookupConfigBeans();
108 assertEquals(Sets.newHashSet(name1, name2, name3), beans);
112 public void testLookupConfigBeansWithModuleName() {
113 Set<ObjectName> beans = testedTxController
114 .lookupConfigBeans(moduleName124);
115 assertEquals(Sets.newHashSet(name1, name2), beans);
119 public void lookupConfigBeansWithModuleNameAndImplName() throws Exception {
120 Set<ObjectName> beans = testedTxController.lookupConfigBeans(
121 moduleName124, instanceName134);
122 assertEquals(Sets.newHashSet(name1), beans);