topology-manager/src/test/java/org/apache/aries/rsa/topologymanager/importer/local/EndpointListenerImplTest.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.rsa.topologymanager.importer.local;

import java.util.Dictionary;
import java.util.List;

import org.apache.aries.rsa.topologymanager.importer.TopologyManagerImport;
import org.apache.aries.rsa.topologymanager.importer.local.EndpointListenerManager;
import org.easymock.EasyMock;
import org.easymock.IAnswer;
import org.easymock.IMocksControl;
import org.junit.Assert;
import org.junit.Test;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceRegistration;
import org.osgi.service.remoteserviceadmin.EndpointEventListener;
import org.osgi.service.remoteserviceadmin.EndpointListener;

@SuppressWarnings("deprecation")
public class EndpointListenerImplTest extends Assert {

    int testCase;

    @SuppressWarnings({ "rawtypes", "unchecked" })
    @Test
    public void testScopeChange() {
        IMocksControl c = EasyMock.createNiceControl();
        BundleContext bc = c.createMock(BundleContext.class);
        TopologyManagerImport tm = c.createMock(TopologyManagerImport.class);
        ServiceRegistration sr = c.createMock(ServiceRegistration.class);

        String[] ifAr = new String[] {EndpointListener.class.getName(), EndpointEventListener.class.getName()};
        // expect Listener registration
        EasyMock.expect(bc.registerService(EasyMock.aryEq(ifAr),
                                           EasyMock.anyObject(),
                                           (Dictionary)EasyMock.anyObject())).andReturn(sr).atLeastOnce();

        sr.setProperties(EasyMock.anyObject());

        // expect property changes based on later calls
        EasyMock.expectLastCall().andAnswer(new IAnswer<Object>() {

            public Object answer() throws Throwable {
                Object[] args = EasyMock.getCurrentArguments();
                Dictionary props = (Dictionary)args[0];
                List<String> scope = (List<String>)props.get(EndpointListener.ENDPOINT_LISTENER_SCOPE);
                switch (testCase) {
                case 1:
                    assertEquals(1, scope.size());
                    assertEquals("(a=b)", scope.get(0));
                    break;
                case 2:
                    assertEquals(0, scope.size());
                    break;
                case 3:
                    assertEquals("adding entry to empty list failed", 1, scope.size());
                    assertEquals("(a=b)", scope.get(0));
                    break;
                case 4:
                    assertEquals("adding second entry failed", 2, scope.size());
                    assertNotNull(scope.contains("(a=b)"));
                    assertNotNull(scope.contains("(c=d)"));
                    break;
                case 5:
                    assertEquals("remove failed", 1, scope.size());
                    assertEquals("(c=d)", scope.get(0));
                    break;
                default:
                    assertTrue("This should not happen!", false);
                }
                return null;
            }
        }).atLeastOnce();

        c.replay();

        EndpointListenerManager endpointListener = new EndpointListenerManager(bc, tm);

        endpointListener.start();

        testCase = 1;
        endpointListener.extendScope("(a=b)");
        testCase = 2;
        endpointListener.reduceScope("(a=b)");

        testCase = 3;
        endpointListener.extendScope("(a=b)");
        testCase = 4;
        endpointListener.extendScope("(c=d)");
        testCase = 5;
        endpointListener.reduceScope("(a=b)");

        endpointListener.stop();

        c.verify();
    }
}