opendaylight/config/config-manager/src/test/java/org/opendaylight/controller/config/manager/impl/ConfigTransactionControllerImplTest.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   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
   7  */
   8 package org.opendaylight.controller.config.manager.impl;
   9
  10 import static org.junit.Assert.assertEquals;
  11
  12 import java.lang.management.ManagementFactory;
  13 import java.util.ArrayList;
  14 import java.util.List;
  15 import java.util.Set;
  16
  17 import javax.management.MBeanServer;
  18 import javax.management.MBeanServerFactory;
  19 import javax.management.ObjectName;
  20
  21 import org.junit.After;
  22 import org.junit.Before;
  23 import org.junit.Test;
  24 import org.opendaylight.controller.config.api.jmx.ObjectNameUtil;
  25 import org.opendaylight.controller.config.manager.impl.jmx.BaseJMXRegistrator;
  26 import org.opendaylight.controller.config.manager.impl.jmx.TransactionJMXRegistrator;
  27 import org.opendaylight.controller.config.manager.impl.jmx.TransactionModuleJMXRegistrator;
  28 import org.opendaylight.controller.config.manager.impl.runtimembean.TestingRuntimeBean;
  29 import org.opendaylight.controller.config.spi.ModuleFactory;
  30
  31 import com.google.common.collect.Sets;
  32
  33 public class ConfigTransactionControllerImplTest extends
  34         AbstractLockedPlatformMBeanServerTest {
  35
  36     private BaseJMXRegistrator baseJMXRegistrator;
  37
  38     private ConfigTransactionControllerImpl testedTxController;
  39     private MBeanServer transactionsMBeanServer;
  40
  41     private static final String transactionName123 = "testTX1";
  42     private static final String transactionName4 = "testTX2";
  43
  44     private static final String moduleName124 = "module124";
  45     private static final String moduleName3 = "module3";
  46
  47     private static final String instanceName134 = "instA";
  48     private static final String instanceName2 = "instB";
  49
  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);
  58
  59     @Before
  60     public void setUp() throws Exception {
  61         baseJMXRegistrator = new BaseJMXRegistrator(
  62                 ManagementFactory.getPlatformMBeanServer());
  63         transactionsMBeanServer = MBeanServerFactory.createMBeanServer();
  64         List<ModuleFactory> currentlyRegisteredFactories = new ArrayList<>();
  65         TransactionJMXRegistrator jmxRegistrator123 = baseJMXRegistrator
  66                 .createTransactionJMXRegistrator(transactionName123);
  67
  68         testedTxController = new ConfigTransactionControllerImpl(
  69                 transactionName123, jmxRegistrator123, 1, 1,
  70                 currentlyRegisteredFactories, transactionsMBeanServer,
  71                 ManagementFactory.getPlatformMBeanServer());
  72         TransactionModuleJMXRegistrator transactionModuleJMXRegistrator123 = testedTxController
  73                 .getTxModuleJMXRegistrator();
  74         transactionModuleJMXRegistrator123.registerMBean(
  75                 new TestingRuntimeBean(), name1);
  76         transactionModuleJMXRegistrator123.registerMBean(
  77                 new TestingRuntimeBean(), name2);
  78         transactionModuleJMXRegistrator123.registerMBean(
  79                 new TestingRuntimeBean(), name3);
  80         TransactionJMXRegistrator jmxRegistrator4 = baseJMXRegistrator
  81                 .createTransactionJMXRegistrator(transactionName4);
  82         jmxRegistrator4.createTransactionModuleJMXRegistrator().registerMBean(
  83                 new TestingRuntimeBean(), name4);
  84     }
  85
  86     @After
  87     public void cleanUp() {
  88         baseJMXRegistrator.close();
  89         MBeanServerFactory.releaseMBeanServer(transactionsMBeanServer);
  90     }
  91
  92     /**
  93      * Tests if lookup method returns all beans with defined transaction name
  94      *
  95      * @throws Exception
  96      */
  97     @Test
  98     public void testLookupConfigBeans() {
  99         Set<ObjectName> beans = testedTxController.lookupConfigBeans();
 100         assertEquals(Sets.newHashSet(name1, name2, name3), beans);
 101     }
 102
 103     @Test
 104     public void testLookupConfigBeansWithModuleName() {
 105         Set<ObjectName> beans = testedTxController
 106                 .lookupConfigBeans(moduleName124);
 107         assertEquals(Sets.newHashSet(name1, name2), beans);
 108     }
 109
 110     @Test
 111     public void lookupConfigBeansWithModuleNameAndImplName() throws Exception {
 112         Set<ObjectName> beans = testedTxController.lookupConfigBeans(
 113                 moduleName124, instanceName134);
 114         assertEquals(Sets.newHashSet(name1), beans);
 115     }
 116
 117 }