com.graphaware.test.integration.cluster.CausalClusterDatabasesintegrationTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tests Show documentation
Tools for testing Neo4j-related and GraphAware-related code
There is a newer version: 4.2.0.58
/*
 * Copyright (c) 2013-2019 GraphAware
 *
 * This file is part of the GraphAware Framework.
 *
 * GraphAware Framework is free software: you can redistribute it and/or modify it under
 * the terms of the GNU General Public License as published by the Free Software Foundation, either
 * version 3 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of
 * the GNU General Public License along with this program.  If not, see
 * .
 */
package com.graphaware.test.integration.cluster;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import org.neo4j.causalclustering.core.CausalClusteringSettings;
import org.neo4j.causalclustering.core.CoreGraphDatabase;
import org.neo4j.causalclustering.readreplica.ReadReplicaGraphDatabase;
import org.neo4j.graphdb.GraphDatabaseService;

/**
 * Superclass for Causal Cluster integration testing
 */
public abstract class CausalClusterDatabasesintegrationTest extends ClusterDatabasesIntegrationTest {

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * com.graphaware.test.integration.cluster.ClusterDatabasesIntegrationTest#
	 * getTopology()
	 */
	@Override
	protected List> getTopology() {
		return Arrays.asList(CoreGraphDatabase.class, CoreGraphDatabase.class, CoreGraphDatabase.class,
				ReadReplicaGraphDatabase.class, ReadReplicaGraphDatabase.class);
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see
	 * com.graphaware.test.integration.cluster.ClusterDatabasesIntegrationTest#
	 * addictionalParams(int, java.lang.Class)
	 */
	@Override
	protected Map addictionalParams(int i, Class instanceClass) {
		Map params = new HashMap<>();

		int coreClusterSize = getTopology().stream().filter(topologyClass -> {
			if (topologyClass.equals(CoreGraphDatabase.class)) {
				return true;
			}
			return false;
		}).collect(Collectors.toList()).size();
		
		params.put(CausalClusteringSettings.initial_discovery_members.name(),
				buildDiscoveryAddresses(5000, coreClusterSize));

		params.put("dbms.connector.bolt.enabled", "true");
		params.put("dbms.connector.bolt.listen_address", "localhost:" + String.valueOf(7687 + i));

		params.put(CausalClusteringSettings.transaction_listen_address.name(), "localhost:601" + i);
		params.put(CausalClusteringSettings.transaction_advertised_address.name(), "localhost:601" + i);

		if (instanceClass.equals(CoreGraphDatabase.class)) {
			params.put(CausalClusteringSettings.expected_core_cluster_size.name(), String.valueOf(coreClusterSize));
			params.put(CausalClusteringSettings.discovery_listen_address.name(), "127.0.0.1:510" + i);
			params.put(CausalClusteringSettings.transaction_listen_address.name(), "127.0.0.1:610" + i);
			params.put(CausalClusteringSettings.raft_listen_address.name(), "127.0.0.1:710" + i);
		}

		return params;
	}

	/**
	 * Get the leader instance
	 * 
	 * @return the leader instance
	 */
	protected GraphDatabaseService getLeaderDatabase() {
		return getMainDatabase();
	}

	/**
	 * Get the first follower
	 * 
	 * @return a follower instance
	 */
	protected GraphDatabaseService getOneFollowerDatabase() {
		return getDatabases().get(1);
	}

	/**
	 * Get the first replica
	 * 
	 * @return a replica instance
	 */
	protected GraphDatabaseService getOneReplicaDatabase() {
		return getReplicas().get(0);
	}

	/**
	 * Get the followers nodes
	 * 
	 * @return a final list containing all the followers
	 */
	protected List getFollowers() {
		final List followers = getDatabases().stream().filter(database -> {
			if (database instanceof CoreGraphDatabase) {
				return true;
			}
			return false;
		}).collect(Collectors.toList());

		return followers.subList(1, followers.size());
	}

	/**
	 * Get the replica nodes
	 * 
	 * @return a final list containing all the replicas
	 */
	protected List getReplicas() {
		final List replicas = getDatabases().stream().filter(database -> {
			if (database instanceof ReadReplicaGraphDatabase) {
				return true;
			}
			return false;
		}).collect(Collectors.toList());

		return replicas;
	}
}