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.ServiceReferenceWritableRegistry;
15 import org.opendaylight.controller.config.api.jmx.ObjectNameUtil;
16 import org.opendaylight.controller.config.manager.impl.jmx.BaseJMXRegistrator;
17 import org.opendaylight.controller.config.manager.impl.jmx.TransactionJMXRegistrator;
18 import org.opendaylight.controller.config.manager.impl.jmx.TransactionModuleJMXRegistrator;
19 import org.opendaylight.controller.config.manager.impl.runtimembean.TestingRuntimeBean;
20 import org.opendaylight.controller.config.spi.ModuleFactory;
21 import org.osgi.framework.BundleContext;
23 import javax.management.MBeanServer;
24 import javax.management.MBeanServerFactory;
25 import javax.management.ObjectName;
26 import java.lang.management.ManagementFactory;
27 import java.util.HashMap;
31 import static org.junit.Assert.assertEquals;
33 public class ConfigTransactionControllerImplTest extends
34 AbstractLockedPlatformMBeanServerTest {
36 private BaseJMXRegistrator baseJMXRegistrator;
38 private ConfigTransactionControllerImpl testedTxController;
39 private MBeanServer transactionsMBeanServer;
41 private static final String transactionName123 = "testTX1";
42 private static final String transactionName4 = "testTX2";
44 private static final String moduleName124 = "module124";
45 private static final String moduleName3 = "module3";
47 private static final String instanceName134 = "instA";
48 private static final String instanceName2 = "instB";
50 private static final ObjectName name1 = ObjectNameUtil
51 .createTransactionModuleON(transactionName123, moduleName124, instanceName134);
52 private static final ObjectName name2 = ObjectNameUtil
53 .createTransactionModuleON(transactionName123, moduleName124, instanceName2);
54 private static final ObjectName name3 = ObjectNameUtil
55 .createTransactionModuleON(transactionName123, moduleName3, instanceName134);
56 private static final ObjectName name4 = ObjectNameUtil
57 .createTransactionModuleON(transactionName4, moduleName124, instanceName134);
60 public void setUp() throws Exception {
61 baseJMXRegistrator = new BaseJMXRegistrator(
62 ManagementFactory.getPlatformMBeanServer());
63 transactionsMBeanServer = MBeanServerFactory.createMBeanServer();
64 Map<String, Map.Entry<ModuleFactory, BundleContext>> currentlyRegisteredFactories = new HashMap<>();
66 ConfigTransactionLookupRegistry txLookupRegistry = new ConfigTransactionLookupRegistry(new TransactionIdentifier(transactionName123), new TransactionJMXRegistratorFactory() {
68 public TransactionJMXRegistrator create() {
69 return baseJMXRegistrator.createTransactionJMXRegistrator(transactionName123);
71 }, currentlyRegisteredFactories);
73 ServiceReferenceWritableRegistry writableRegistry = ServiceReferenceRegistryImpl.createSRWritableRegistry(
74 ServiceReferenceRegistryImpl.createInitialSRLookupRegistry(), txLookupRegistry, currentlyRegisteredFactories);
77 testedTxController = new ConfigTransactionControllerImpl(
78 txLookupRegistry, 1, 1,
79 currentlyRegisteredFactories, transactionsMBeanServer,
80 ManagementFactory.getPlatformMBeanServer(), false, writableRegistry);
81 TransactionModuleJMXRegistrator transactionModuleJMXRegistrator123 = testedTxController
82 .getTxModuleJMXRegistrator();
83 transactionModuleJMXRegistrator123.registerMBean(
84 new TestingRuntimeBean(), name1);
85 transactionModuleJMXRegistrator123.registerMBean(
86 new TestingRuntimeBean(), name2);
87 transactionModuleJMXRegistrator123.registerMBean(
88 new TestingRuntimeBean(), name3);
89 TransactionJMXRegistrator jmxRegistrator4 = baseJMXRegistrator
90 .createTransactionJMXRegistrator(transactionName4);
91 jmxRegistrator4.createTransactionModuleJMXRegistrator().registerMBean(
92 new TestingRuntimeBean(), name4);
96 public void cleanUp() {
97 baseJMXRegistrator.close();
98 MBeanServerFactory.releaseMBeanServer(transactionsMBeanServer);
102 * Tests if lookup method returns all beans with defined transaction name
107 public void testLookupConfigBeans() {
108 Set<ObjectName> beans = testedTxController.lookupConfigBeans();
109 assertEquals(Sets.newHashSet(name1, name2, name3), beans);
113 public void testLookupConfigBeansWithModuleName() {
114 Set<ObjectName> beans = testedTxController
115 .lookupConfigBeans(moduleName124);
116 assertEquals(Sets.newHashSet(name1, name2), beans);
120 public void lookupConfigBeansWithModuleNameAndImplName() throws Exception {
121 Set<ObjectName> beans = testedTxController.lookupConfigBeans(
122 moduleName124, instanceName134);
123 assertEquals(Sets.newHashSet(name1), beans);