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(
65 new TransactionIdentifier(transactionName123), () ->
66 baseJMXRegistrator.createTransactionJMXRegistrator(transactionName123), currentlyRegisteredFactories);
68 SearchableServiceReferenceWritableRegistry writableRegistry = ServiceReferenceRegistryImpl.createSRWritableRegistry(
69 ServiceReferenceRegistryImpl.createInitialSRLookupRegistry(), txLookupRegistry, currentlyRegisteredFactories);
72 testedTxController = new ConfigTransactionControllerImpl(
73 txLookupRegistry, 1, null, 1,
74 currentlyRegisteredFactories, transactionsMBeanServer,
75 ManagementFactory.getPlatformMBeanServer(), false, writableRegistry);
76 TransactionModuleJMXRegistrator transactionModuleJMXRegistrator123 = testedTxController
77 .getTxModuleJMXRegistrator();
78 transactionModuleJMXRegistrator123.registerMBean(
79 new TestingRuntimeBean(), name1);
80 transactionModuleJMXRegistrator123.registerMBean(
81 new TestingRuntimeBean(), name2);
82 transactionModuleJMXRegistrator123.registerMBean(
83 new TestingRuntimeBean(), name3);
84 TransactionJMXRegistrator jmxRegistrator4 = baseJMXRegistrator
85 .createTransactionJMXRegistrator(transactionName4);
86 jmxRegistrator4.createTransactionModuleJMXRegistrator().registerMBean(
87 new TestingRuntimeBean(), name4);
91 public void cleanUp() {
92 baseJMXRegistrator.close();
93 MBeanServerFactory.releaseMBeanServer(transactionsMBeanServer);
97 * Tests if lookup method returns all beans with defined transaction name
100 public void testLookupConfigBeans() {
101 Set<ObjectName> beans = testedTxController.lookupConfigBeans();
102 assertEquals(Sets.newHashSet(name1, name2, name3), beans);
106 public void testLookupConfigBeansWithModuleName() {
107 Set<ObjectName> beans = testedTxController
108 .lookupConfigBeans(moduleName124);
109 assertEquals(Sets.newHashSet(name1, name2), beans);
113 public void lookupConfigBeansWithModuleNameAndImplName() throws Exception {
114 Set<ObjectName> beans = testedTxController.lookupConfigBeans(
115 moduleName124, instanceName134);
116 assertEquals(Sets.newHashSet(name1), beans);