RemoteConnectorInfoManagerClearTests.java

/*
 * ====================
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2008-2009 Sun Microsystems, Inc. All rights reserved.
 *
 * The contents of this file are subject to the terms of the Common Development
 * and Distribution License("CDDL") (the "License").  You may not use this file
 * except in compliance with the License.
 *
 * You can obtain a copy of the License at
 * http://opensource.org/licenses/cddl1.php
 * See the License for the specific language governing permissions and limitations
 * under the License.
 *
 * When distributing the Covered Code, include this CDDL Header Notice in each file
 * and include the License file at http://opensource.org/licenses/cddl1.php.
 * If applicable, add the following below this CDDL Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyrighted [year] [name of copyright owner]"
 * ====================
 */
package org.identityconnectors.framework.impl.api;

import java.net.InetAddress;
import java.net.URL;
import java.util.List;
import java.util.concurrent.TimeUnit;

import org.identityconnectors.common.security.GuardedString;
import org.identityconnectors.common.security.SecurityUtil;
import org.identityconnectors.framework.api.APIConfiguration;
import org.identityconnectors.framework.api.ConfigurationProperties;
import org.identityconnectors.framework.api.ConfigurationProperty;
import org.identityconnectors.framework.api.ConnectorFacade;
import org.identityconnectors.framework.api.ConnectorFacadeFactory;
import org.identityconnectors.framework.api.ConnectorInfo;
import org.identityconnectors.framework.api.ConnectorInfoManager;
import org.identityconnectors.framework.api.ConnectorInfoManagerFactory;
import org.identityconnectors.framework.api.RemoteFrameworkConnectionInfo;
import org.identityconnectors.framework.impl.api.remote.RemoteConnectorInfoManagerImpl;
import org.identityconnectors.framework.impl.api.remote.messages.HelloResponse;
import org.identityconnectors.framework.server.ConnectorServer;
import org.testng.Assert;
import org.testng.annotations.Test;

public class RemoteConnectorInfoManagerClearTests extends ConnectorInfoManagerTestBase {

    private static ConnectorServer _server;

    final int PORT = 8759;

    /**
     * To be overridden by subclasses to get different ConnectorInfoManagers
     *
     * @return
     * @throws Exception
     */
    @Override
    protected ConnectorInfoManager getConnectorInfoManager() throws Exception {
        List<URL> urls = getTestBundles();

        synchronized (RemoteConnectorInfoManagerClearTests.class) {
            if (_server == null) {
                _server = ConnectorServer.newInstance();
                _server.setKeyHash(SecurityUtil.computeBase64SHA1Hash("changeit".toCharArray()));
                _server.setBundleURLs(urls);
                _server.setPort(PORT);
                _server.setIfAddress(InetAddress.getByName("127.0.0.1"));
                _server.start();
            }
        }
        ConnectorInfoManagerFactory fact = ConnectorInfoManagerFactory.getInstance();

        RemoteFrameworkConnectionInfo connInfo =
                new RemoteFrameworkConnectionInfo("127.0.0.1", PORT, new GuardedString("changeit"
                        .toCharArray()), false, null, 0);

        ConnectorInfoManager manager = fact.getRemoteManager(connInfo);

        return manager;
    }

    @Override
    protected synchronized void shutdownConnnectorInfoManager() {
        synchronized (RemoteConnectorInfoManagerClearTests.class) {
            if (_server != null) {
                _server.stop();
                _server = null;
            }
        }
        // These are initialized by the connector server.
        ConnectorFacadeFactory.getInstance().dispose();
        ConnectorInfoManagerFactory.getInstance().clearLocalCache();
    }

    @Test
    public void testRemoteHelloRequest() throws Exception {
        getConnectorInfoManager();
        RemoteConnectorInfoManagerImpl mgr =
                new RemoteConnectorInfoManagerImpl(new RemoteFrameworkConnectionInfo("127.0.0.1",
                        PORT, new GuardedString("changeit".toCharArray())));
        Assert.assertNotNull(mgr.getServerInfo().get(HelloResponse.SERVER_START_TIME));
        Assert.assertEquals(mgr.getConnectorKeys().size(), 4);
    }

    @Test
    public void testFacadeEviction() throws Exception {
        ConnectorServer server = ConnectorServer.newInstance();
        try {
            server.setKeyHash(SecurityUtil.computeBase64SHA1Hash("changeit".toCharArray()));
            server.setBundleURLs(getTestBundles());
            server.setPort(8760);
            server.setIfAddress(InetAddress.getByName("127.0.0.1"));
            server.setMaxFacadeLifeTime(1);
            server.start();

            RemoteFrameworkConnectionInfo connInfo =
                    new RemoteFrameworkConnectionInfo("127.0.0.1", 8760, new GuardedString(
                            "changeit".toCharArray()), false, null, 0);

            final ConnectorInfoManager remoteManager =
                    ConnectorInfoManagerFactory.getInstance().getRemoteManager(connInfo);

            final ConnectorInfo remoteInfo =
                    findConnectorInfo(remoteManager, "1.0.0.0",
                            "org.identityconnectors.testconnector.TstConnector");

            APIConfiguration api = remoteInfo.createDefaultAPIConfiguration();
            // api.getConfigurationProperties().getProperty("resetConnectionCount").setValue(true);
            ConnectorFacade remoteFacade =
                    ConnectorFacadeFactory.getInstance().newInstance(
                            remoteInfo.createDefaultAPIConfiguration());

            ManagedConnectorFacadeFactoryImpl managedFactory =
                    (ManagedConnectorFacadeFactoryImpl) ConnectorFacadeFactory.getManagedInstance();

            managedFactory.dispose();
            // Assert it's empty
            Assert.assertNull(managedFactory.find(remoteFacade.getConnectorFacadeKey()));
            remoteFacade.schema();
            // Assert it has one item
            Assert.assertNotNull(managedFactory.find(remoteFacade.getConnectorFacadeKey()));
            for (int i = 0; i <= 3; i++) {
                Thread.sleep(TimeUnit.SECONDS.toMillis(60));
                if (null == managedFactory.find(remoteFacade.getConnectorFacadeKey())) {
                    break;
                }
            }
            // Assert it's empty
            Assert.assertNull(managedFactory.find(remoteFacade.getConnectorFacadeKey()));
        } finally {
            server.stop();
        }
    }
}