Fix protocol tests

neo4j · Jun 11, 2018 · 8f6ce52 · 8f6ce52
1 parent 4b22dac
commit 8f6ce52
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/...lustering/src/test/java/org/neo4j/causalclustering/core/SupportedProtocolCreatorTest.java b/...lustering/src/test/java/org/neo4j/causalclustering/core/SupportedProtocolCreatorTest.java
@@ -82,7 +82,7 @@ public void shouldFilterUnknownRaftImplementations()
         ApplicationSupportedProtocols supportedRaftProtocol = new SupportedProtocolCreator( config, log ).createSupportedRaftProtocol();
 
         // then
-        assertThat( supportedRaftProtocol.versions(), contains( 1 ) );
+        assertThat( supportedRaftProtocol.versions(), contains( 1, 2 ) );
 
     }
 

diff --git a/...ing/src/test/java/org/neo4j/causalclustering/scenarios/InstalledProtocolsProcedureIT.java b/...ing/src/test/java/org/neo4j/causalclustering/scenarios/InstalledProtocolsProcedureIT.java
@@ -30,6 +30,7 @@
 import java.util.List;
 import java.util.Objects;
 import java.util.StringJoiner;
+import java.util.stream.Stream;
 
 import org.neo4j.causalclustering.core.CausalClusteringSettings;
 import org.neo4j.causalclustering.discovery.Cluster;
@@ -87,7 +88,7 @@ public void shouldSeeOutboundInstalledProtocolsOnLeader() throws Throwable
         ProtocolInfo[] expectedProtocolInfos = cluster.coreMembers()
                 .stream()
                 .filter( member -> !member.equals( leader ) )
-                .map( member -> new ProtocolInfo( OUTBOUND, localhost( member.raftListenAddress() ), RAFT.canonicalName(), 1, modifiers ) )
+                .map( member -> new ProtocolInfo( OUTBOUND, localhost( member.raftListenAddress() ), RAFT.canonicalName(), 2, modifiers ) )
                 .toArray( ProtocolInfo[]::new );
 
         assertEventually( "should see outbound installed protocols on core " + leader.serverId(),