Skip to content

Commit 156725f

Browse files
author
Zhen Li
committed
Merge branch '1.7' into 4.0
2 parents e7c4027 + 3765540 commit 156725f

File tree

4 files changed

+34
-4
lines changed

4 files changed

+34
-4
lines changed

driver/src/test/java/org/neo4j/driver/stress/AbstractStressTestBase.java

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -111,14 +111,13 @@ void setUp()
111111
{
112112
logging = new LoggerNameTrackingLogging();
113113

114-
Config config = Config.builder()
114+
Config.ConfigBuilder builder = Config.builder()
115115
.withLogging( logging )
116116
.withMaxConnectionPoolSize( 100 )
117117
.withConnectionAcquisitionTimeout( 1, MINUTES )
118-
.withDriverMetrics()
119-
.build();
118+
.withDriverMetrics();
120119

121-
driver = (InternalDriver) GraphDatabase.driver( databaseUri(), authToken(), config );
120+
driver = (InternalDriver) GraphDatabase.driver( databaseUri(), authToken(), config( builder ) );
122121

123122
ThreadFactory threadFactory = new DaemonThreadFactory( getClass().getSimpleName() + "-worker-" );
124123
executor = Executors.newCachedThreadPool( threadFactory );
@@ -217,6 +216,8 @@ private void runStressTest( Function<C,List<Future<?>>> threadLauncher ) throws
217216

218217
abstract AuthToken authToken();
219218

219+
abstract Config config( Config.ConfigBuilder builder );
220+
220221
abstract C createContext();
221222

222223
abstract List<BlockingCommand<C>> createTestSpecificBlockingCommands();

driver/src/test/java/org/neo4j/driver/stress/CausalClusteringStressIT.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
import java.util.concurrent.atomic.AtomicLong;
3434

3535
import org.neo4j.driver.AuthToken;
36+
import org.neo4j.driver.Config;
3637
import org.neo4j.driver.Driver;
3738
import org.neo4j.driver.exceptions.SessionExpiredException;
3839
import org.neo4j.driver.internal.BoltServerAddress;
@@ -65,6 +66,12 @@ AuthToken authToken()
6566
return clusterRule.getAuthToken();
6667
}
6768

69+
@Override
70+
Config config( Config.ConfigBuilder builder )
71+
{
72+
return clusterRule.getDriverConfig( builder );
73+
}
74+
6875
@Override
6976
Context createContext()
7077
{

driver/src/test/java/org/neo4j/driver/stress/SingleInstanceStressIT.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
import java.util.concurrent.atomic.AtomicLong;
2727

2828
import org.neo4j.driver.AuthToken;
29+
import org.neo4j.driver.Config;
2930
import org.neo4j.driver.summary.ResultSummary;
3031
import org.neo4j.driver.util.DatabaseExtension;
3132
import org.neo4j.driver.util.ParallelizableIT;
@@ -52,6 +53,12 @@ AuthToken authToken()
5253
return neo4j.authToken();
5354
}
5455

56+
@Override
57+
Config config( Config.ConfigBuilder builder )
58+
{
59+
return builder.build();
60+
}
61+
5562
@Override
5663
Context createContext()
5764
{

driver/src/test/java/org/neo4j/driver/util/cc/LocalOrRemoteClusterExtension.java

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727

2828
import org.neo4j.driver.AuthToken;
2929
import org.neo4j.driver.AuthTokens;
30+
import org.neo4j.driver.Config;
3031
import org.neo4j.driver.Driver;
3132
import org.neo4j.driver.internal.util.DriverFactoryWithOneEventLoopThread;
3233
import org.neo4j.driver.util.TestUtil;
@@ -61,6 +62,20 @@ public AuthToken getAuthToken()
6162
return localClusterExtension.getDefaultAuthToken();
6263
}
6364

65+
public Config getDriverConfig( Config.ConfigBuilder builder )
66+
{
67+
if ( remoteClusterExists() )
68+
{
69+
builder.withEncryption();
70+
}
71+
else
72+
{
73+
builder.withoutEncryption();
74+
}
75+
76+
return builder.build();
77+
}
78+
6479
@Override
6580
public void beforeAll( ExtensionContext context ) throws Exception
6681
{

0 commit comments

Comments
 (0)