Merge pull request #469 from lutovich/1.5-more-cc-tests

Added couple routing driver integration tests

driver/src/test/java/org/neo4j/driver/internal/util/ChannelTrackingDriverFactory.java

@@ -40,6 +40,11 @@ public class ChannelTrackingDriverFactory extends DriverFactoryWithClock
     private final int eventLoopThreads;
     private ConnectionPool pool;
 
+    public ChannelTrackingDriverFactory()
+    {
+        this( 0, Clock.SYSTEM );
+    }
+
     public ChannelTrackingDriverFactory( Clock clock )
     {
         this( 0, clock );
@@ -88,13 +93,11 @@ public List<Channel> channels()
         return new ArrayList<>( channels );
     }
 
-    public void closeChannels()
+    public List<Channel> pollChannels()
     {
-        for ( Channel channel : channels )
-        {
-            channel.close().syncUninterruptibly();
-        }
+        List<Channel> result = new ArrayList<>( channels );
         channels.clear();
+        return result;
     }
 
     public int activeChannels( BoltServerAddress address )

driver/src/test/java/org/neo4j/driver/internal/util/ThrowingMessageEncoder.java

@@ -0,0 +1,55 @@
+/*
+ * Copyright (c) 2002-2018 "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.internal.util;
+
+import io.netty.channel.ChannelHandlerContext;
+import io.netty.handler.codec.MessageToMessageEncoder;
+
+import java.util.List;
+
+import org.neo4j.driver.internal.messaging.ResetMessage;
+import org.neo4j.driver.internal.messaging.RunMessage;
+
+public class ThrowingMessageEncoder<T> extends MessageToMessageEncoder<T>
+{
+    private final RuntimeException error;
+
+    private ThrowingMessageEncoder( Class<T> messageType, RuntimeException error )
+    {
+        super( messageType );
+        this.error = error;
+    }
+
+    @Override
+    protected void encode( ChannelHandlerContext ctx, T msg, List<Object> out )
+    {
+        ctx.pipeline().remove( this );
+        throw error;
+    }
+
+    public static ThrowingMessageEncoder<RunMessage> forRunMessage( RuntimeException error )
+    {
+        return new ThrowingMessageEncoder<>( RunMessage.class, error );
+    }
+
+    public static ThrowingMessageEncoder<ResetMessage> forResetMessage( RuntimeException error )
+    {
+        return new ThrowingMessageEncoder<>( ResetMessage.class, error );
+    }
+}