src/test/java/org/apache/aries/jmx/CompendiumHandlerTest.java

/**
 *  Licensed to the Apache Software Foundation (ASF) under one or more
 *  contributor license agreements.  See the NOTICE file distributed with
 *  this work for additional information regarding copyright ownership.
 *  The ASF licenses this file to You under the Apache License, Version 2.0
 *  (the "License"); you may not use this file except in compliance with
 *  the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.apache.aries.jmx;

import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import javax.management.StandardMBean;

import org.apache.aries.jmx.agent.JMXAgentContext;
import org.junit.After;
import org.junit.Test;
import org.osgi.framework.BundleContext;
import org.osgi.framework.Filter;
import org.osgi.framework.ServiceReference;

import static org.mockito.Mockito.*;

public class CompendiumHandlerTest {

    protected AbstractCompendiumHandler target;
    
    @After
    public void tearDown(){
        target = null;
    }
    
    
    @Test
    public void testAddingServiceWillInitiateMBeanRegistration() throws Exception {
        
        Object service = new Object();
        
        ServiceReference reference = mock(ServiceReference.class);
        BundleContext bundleContext = mock(BundleContext.class);
        when(bundleContext.getService(reference)).thenReturn(service);
        
        Logger agentLogger = mock(Logger.class);
        JMXAgentContext agentContext = mock(JMXAgentContext.class);
        when(agentContext.getBundleContext()).thenReturn(bundleContext);
        when(agentContext.getLogger()).thenReturn(agentLogger);
        ExecutorService executor = Executors.newSingleThreadExecutor();
        when(agentContext.getRegistrationExecutor()).thenReturn(executor);
        
        AbstractCompendiumHandler concreteHandler = new CompendiumHandler(agentContext, "org.osgi.service.Xxx");
        target = spy(concreteHandler);
        
        target.addingService(reference);

        executor.shutdown();
        executor.awaitTermination(2, TimeUnit.SECONDS);
        
        //service only got once
        verify(bundleContext).getService(reference);
        //template method is invoked
        verify(target).constructInjectMBean(service);
        //registration is invoked on context
        verify(agentContext).registerMBean(target);
        
    }

    @Test
    public void testRemovedServiceWillUnregisterMBean() throws Exception{
        
        Object service = new Object();
        ServiceReference reference = mock(ServiceReference.class);
        
        BundleContext bundleContext = mock(BundleContext.class);
        Logger agentLogger = mock(Logger.class);
        JMXAgentContext agentContext = mock(JMXAgentContext.class);
        when(agentContext.getBundleContext()).thenReturn(bundleContext);
        when(agentContext.getLogger()).thenReturn(agentLogger);
        ExecutorService executor = Executors.newSingleThreadExecutor();
        when(agentContext.getRegistrationExecutor()).thenReturn(executor);
        
        AbstractCompendiumHandler concreteHandler = new CompendiumHandler(agentContext, "org.osgi.service.Xxx");
        target = spy(concreteHandler);
        
        String name = "osgi.compendium:service=xxx,version=1.0";
        doReturn(name).when(target).getName();
        
        target.removedService(reference, service);

        executor.shutdown();
        executor.awaitTermination(2, TimeUnit.SECONDS);
        
        //service unget
        verify(bundleContext).ungetService(reference);
        //unregister is invoked on context
        verify(agentContext).unregisterMBean(target);
        
    }

   
    
    /*
     * Concrete implementation used for test
     */
    class CompendiumHandler extends AbstractCompendiumHandler {

        protected CompendiumHandler(JMXAgentContext agentContext, Filter filter) {
            super(agentContext, filter);
        }

        protected CompendiumHandler(JMXAgentContext agentContext, String clazz) {
            super(agentContext, clazz);
        }
        
        protected StandardMBean constructInjectMBean(Object targetService) {
            return null;
        }

        public String getName() {
            return null;
        }
        
    }
}