diff --git a/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/discovery/TestTopology.java b/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/discovery/TestTopology.java index 0bfc09d929df..a77540ecb17f 100644 --- a/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/discovery/TestTopology.java +++ b/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/discovery/TestTopology.java @@ -50,7 +50,10 @@ public static CoreAddresses adressesForCore( int id ) public static ReadReplicaAddresses addressesForReadReplica( int id ) { - AdvertisedSocketAddress boltServerAddress = new AdvertisedSocketAddress( "localhost", (6000 + id) ); - return new ReadReplicaAddresses( wrapAsClientConnectorAddresses( boltServerAddress ) ); + AdvertisedSocketAddress advertisedSocketAddress = new AdvertisedSocketAddress( "localhost", (6000 + id) ); + ClientConnectorAddresses clientConnectorAddresses = new ClientConnectorAddresses( + singletonList( new ClientConnectorAddresses.ConnectorUri( bolt, advertisedSocketAddress ) ) ); + + return new ReadReplicaAddresses( clientConnectorAddresses, advertisedSocketAddress ); } } diff --git a/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/discovery/procedures/ClusterOverviewProcedureTest.java b/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/discovery/procedures/ClusterOverviewProcedureTest.java index f58532a08390..8347c01ccf60 100644 --- a/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/discovery/procedures/ClusterOverviewProcedureTest.java +++ b/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/discovery/procedures/ClusterOverviewProcedureTest.java @@ -19,15 +19,14 @@ */ package org.neo4j.causalclustering.discovery.procedures; +import org.hamcrest.collection.IsIterableContainingInOrder; +import org.junit.Test; + import java.util.HashMap; import java.util.List; import java.util.Map; -import java.util.Set; import java.util.UUID; -import org.hamcrest.collection.IsIterableContainingInOrder; -import org.junit.Test; - import org.neo4j.causalclustering.core.consensus.LeaderLocator; import org.neo4j.causalclustering.discovery.CoreAddresses; import org.neo4j.causalclustering.discovery.CoreTopology; @@ -40,9 +39,8 @@ import static org.hamcrest.MatcherAssert.assertThat; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; - -import static org.neo4j.causalclustering.discovery.TestTopology.addressesForReadReplicas; import static org.neo4j.causalclustering.discovery.TestTopology.adressesForCore; +import static org.neo4j.causalclustering.load_balancing.procedure.GetServersProcedureV1Test.addresses; import static org.neo4j.helpers.collection.Iterators.asList; public class ClusterOverviewProcedureTest diff --git a/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/load_balancing/procedure/GetServersProcedureV1Test.java b/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/load_balancing/procedure/GetServersProcedureV1Test.java index fd94838a3411..d5835d8bb046 100644 --- a/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/load_balancing/procedure/GetServersProcedureV1Test.java +++ b/enterprise/causal-clustering/src/test/java/org/neo4j/causalclustering/load_balancing/procedure/GetServersProcedureV1Test.java @@ -370,22 +370,15 @@ private ClusterView run( GetServersProcedureV1 proc ) throws ProcedureException return ClusterView.parse( (List>) rows[1] ); } - static Map addresses( int... ids ) + public static Map addresses( int... ids ) { return Arrays.stream( ids ).mapToObj( GetServersProcedureV1Test::readReplicaAddresses ).collect( Collectors .toMap( (p) -> new MemberId( UUID.randomUUID() ), Function.identity() ) ); } - static CoreAddresses coreAddresses( int id ) - { - AdvertisedSocketAddress advertisedSocketAddress = new AdvertisedSocketAddress( "localhost", (3000 + id) ); - return new CoreAddresses( advertisedSocketAddress, advertisedSocketAddress, - new ClientConnectorAddresses( singletonList( new ClientConnectorAddresses.ConnectorUri( bolt, advertisedSocketAddress ) ) ) ); - } - private static ReadReplicaAddresses readReplicaAddresses( int id ) { - AdvertisedSocketAddress advertisedSocketAddress = new AdvertisedSocketAddress( "localhost", (3000 + id) ); + AdvertisedSocketAddress advertisedSocketAddress = new AdvertisedSocketAddress( "localhost", (6000 + id) ); return new ReadReplicaAddresses( new ClientConnectorAddresses( singletonList( new ClientConnectorAddresses.ConnectorUri( bolt, advertisedSocketAddress ) ) ), new AdvertisedSocketAddress( "localhost", 4000 + id ));