Refactor ClusterCompressionIT

neo4j · Aug 4, 2018 · 9dcfee6 · 9dcfee6
1 parent 84dc81d
commit 9dcfee6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/...l-clustering/src/test/java/org/neo4j/causalclustering/scenarios/ClusterCompressionIT.java b/...l-clustering/src/test/java/org/neo4j/causalclustering/scenarios/ClusterCompressionIT.java
@@ -30,10 +30,10 @@
 import java.util.Arrays;
 import java.util.Collection;
 import java.util.UUID;
-import java.util.stream.Collectors;
 
 import org.neo4j.causalclustering.discovery.Cluster;
 import org.neo4j.causalclustering.discovery.CoreClusterMember;
+import org.neo4j.causalclustering.protocol.Protocol;
 import org.neo4j.causalclustering.protocol.Protocol.ModifierProtocols;
 import org.neo4j.helpers.collection.Pair;
 import org.neo4j.test.causalclustering.ClusterRule;
@@ -51,12 +51,12 @@
 public class ClusterCompressionIT
 {
     @Parameterized.Parameter
-    public ModifierProtocols modifierProtocol;
+    public Protocol.ModifierProtocol modifierProtocol;
 
     @Parameterized.Parameters( name = "{0}" )
-    public static Collection<Object[]> params()
+    public static Collection<Protocol.ModifierProtocol> params()
     {
-        return Arrays.stream( ModifierProtocols.values() ).map( mp -> new Object[]{mp} ).collect( Collectors.toList() );
+        return Arrays.asList( ModifierProtocols.values() );
     }
 
     @Rule