LocalOrRemoteClusterRule.java

/*
 * Copyright (c) 2002-2017 "Neo Technology,"
 * Network Engine for Objects in Lund AB [http://neotechnology.com]
 *
 * This file is part of Neo4j.
 *
 * Licensed 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.neo4j.driver.v1.util.cc;

import org.junit.rules.ExternalResource;

import java.net.URI;

import org.neo4j.driver.v1.AuthToken;
import org.neo4j.driver.v1.AuthTokens;

public class LocalOrRemoteClusterRule extends ExternalResource
{
    private static final String CLUSTER_URI_SYSTEM_PROPERTY_NAME = "externalClusterUri";
    private static final String NEO4J_USER_PASSWORD_PROPERTY_NAME = "neo4jUserPassword";

    private ClusterRule localClusterRule;
    private URI clusterUri;

    public LocalOrRemoteClusterRule()
    {
        assertValidSystemPropertiesDefined();
    }

    public URI getClusterUri()
    {
        return clusterUri;
    }

    public AuthToken getAuthToken()
    {
        if ( externalClusterExists() )
        {
            return AuthTokens.basic( "neo4j", neo4jUserPasswordFromSystemProperty() );
        }
        return localClusterRule.getDefaultAuthToken();
    }

    @Override
    protected void before() throws Throwable
    {
        if ( externalClusterExists() )
        {
            clusterUri = externalClusterUriFromSystemProperty();
        }
        else
        {
            localClusterRule = new ClusterRule();
            localClusterRule.before();
            clusterUri = localClusterRule.getCluster().leader().getRoutingUri();
        }
    }

    @Override
    protected void after()
    {
        if ( !externalClusterExists() )
        {
            localClusterRule.after();
            localClusterRule = null;
        }
        clusterUri = null;
    }

    private static void assertValidSystemPropertiesDefined()
    {
        URI uri = externalClusterUriFromSystemProperty();
        String password = neo4jUserPasswordFromSystemProperty();
        if ( (uri != null && password == null) || (uri == null && password != null) )
        {
            throw new IllegalStateException(
                    "Both cluster uri and 'neo4j' user password system properties should be set. " +
                    "Uri: '" + uri + "', Password: '" + password + "'" );
        }
        if ( uri != null && !"bolt+routing".equals( uri.getScheme() ) )
        {
            throw new IllegalStateException( "CLuster uri should have bolt+routing scheme: '" + uri + "'" );
        }
    }

    private static boolean externalClusterExists()
    {
        return externalClusterUriFromSystemProperty() != null;
    }

    private static URI externalClusterUriFromSystemProperty()
    {
        String uri = System.getProperty( CLUSTER_URI_SYSTEM_PROPERTY_NAME );
        return uri == null ? null : URI.create( uri );
    }

    private static String neo4jUserPasswordFromSystemProperty()
    {
        return System.getProperty( NEO4J_USER_PASSWORD_PROPERTY_NAME );
    }
}