Improve stress testing

Split `CausalClusteringStressIT` into two separate tests - one for
single instance and one for CC. Added testing of async API. Extracted
all infra classes to top level.

Author: lutovich

driver/src/test/java/org/neo4j/driver/v1/stress/AbstractAsyncQuery.java

@@ -0,0 +1,44 @@
+/*
+ * Copyright (c) 2002-2017 "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.v1.stress;
+
+import org.neo4j.driver.v1.AccessMode;
+import org.neo4j.driver.v1.Driver;
+import org.neo4j.driver.v1.Session;
+
+public abstract class AbstractAsyncQuery<C extends AbstractContext> implements AsyncCommand<C>
+{
+    protected final Driver driver;
+    protected final boolean useBookmark;
+
+    public AbstractAsyncQuery( Driver driver, boolean useBookmark )
+    {
+        this.driver = driver;
+        this.useBookmark = useBookmark;
+    }
+
+    public Session newSession( AccessMode mode, C context )
+    {
+        if ( useBookmark )
+        {
+            return driver.session( mode, context.getBookmark() );
+        }
+        return driver.session( mode );
+    }
+}

driver/src/test/java/org/neo4j/driver/v1/stress/AbstractBlockingQuery.java

@@ -0,0 +1,66 @@
+/*
+ * Copyright (c) 2002-2017 "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.v1.stress;
+
+import org.neo4j.driver.v1.AccessMode;
+import org.neo4j.driver.v1.Driver;
+import org.neo4j.driver.v1.Session;
+import org.neo4j.driver.v1.Transaction;
+import org.neo4j.driver.v1.exceptions.TransientException;
+
+public abstract class AbstractBlockingQuery<C extends AbstractContext> implements BlockingCommand<C>
+{
+    protected final Driver driver;
+    protected final boolean useBookmark;
+
+    public AbstractBlockingQuery( Driver driver, boolean useBookmark )
+    {
+        this.driver = driver;
+        this.useBookmark = useBookmark;
+    }
+
+    public Session newSession( AccessMode mode, C context )
+    {
+        if ( useBookmark )
+        {
+            return driver.session( mode, context.getBookmark() );
+        }
+        return driver.session( mode );
+    }
+
+    public Transaction beginTransaction( Session session, C context )
+    {
+        if ( useBookmark )
+        {
+            while ( true )
+            {
+                try
+                {
+                    return session.beginTransaction();
+                }
+                catch ( TransientException e )
+                {
+                    context.bookmarkFailed();
+                }
+            }
+        }
+
+        return session.beginTransaction();
+    }
+}

driver/src/test/java/org/neo4j/driver/v1/stress/AbstractContext.java

@@ -0,0 +1,86 @@
+/*
+ * Copyright (c) 2002-2017 "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.v1.stress;
+
+import java.util.concurrent.atomic.AtomicInteger;
+import java.util.concurrent.atomic.AtomicLong;
+
+import org.neo4j.driver.v1.summary.ResultSummary;
+
+public abstract class AbstractContext
+{
+    private volatile boolean stopped;
+    private volatile String bookmark;
+    private final AtomicLong readNodesCount = new AtomicLong();
+    private final AtomicLong createdNodesCount = new AtomicLong();
+    private final AtomicInteger bookmarkFailures = new AtomicInteger();
+
+    public final boolean isStopped()
+    {
+        return stopped;
+    }
+
+    public final void stop()
+    {
+        this.stopped = true;
+    }
+
+    public final String getBookmark()
+    {
+        return bookmark;
+    }
+
+    public final void setBookmark( String bookmark )
+    {
+        this.bookmark = bookmark;
+    }
+
+    public final void nodeCreated()
+    {
+        createdNodesCount.incrementAndGet();
+    }
+
+    public final long getCreatedNodesCount()
+    {
+        return createdNodesCount.get();
+    }
+
+    public final void readCompleted( ResultSummary summary )
+    {
+        readNodesCount.incrementAndGet();
+        processSummary( summary );
+    }
+
+    public abstract void processSummary( ResultSummary summary );
+
+    public long getReadNodesCount()
+    {
+        return readNodesCount.get();
+    }
+
+    public final void bookmarkFailed()
+    {
+        bookmarkFailures.incrementAndGet();
+    }
+
+    public final int getBookmarkFailures()
+    {
+        return bookmarkFailures.get();
+    }
+}