Skip to content

Commit 637be40

Browse files
author
Zhen
committed
Fix after code review
Changed to use encryption level to make the code easier to read
1 parent 05d9d40 commit 637be40

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

driver/src/test/java/org/neo4j/driver/v1/integration/ServerKilledIT.java

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -57,24 +57,24 @@ public class ServerKilledIT
5757
@Parameters(name = "{0} connections")
5858
public static Collection<Object[]> data() {
5959
return Arrays.asList(new Object[][] {
60-
{ "plaintext", Config.build().withEncryptionLevel( Config.EncryptionLevel.NONE ) },
61-
{ "tls encrypted", Config.build().withEncryptionLevel( Config.EncryptionLevel.REQUIRED ) }
60+
{ "plaintext", Config.EncryptionLevel.NONE },
61+
{ "tls encrypted", Config.EncryptionLevel.REQUIRED }
6262
});
6363
}
6464

65-
private Config.ConfigBuilder configBuilder;
65+
private Config.EncryptionLevel encryptionLevel;
6666

67-
public ServerKilledIT( String testName, Config.ConfigBuilder configBuilder )
67+
public ServerKilledIT( String testName, Config.EncryptionLevel encryptionLevel )
6868
{
69-
this.configBuilder = configBuilder;
69+
this.encryptionLevel = encryptionLevel;
7070
}
7171

7272
@Test
7373
public void shouldRecoverFromServerRestart() throws Throwable
7474
{
7575
// Given
7676
// config with sessionLivenessCheckTimeout not set, i.e. turned off
77-
Config config = configBuilder.toConfig();
77+
Config config = Config.build().withEncryptionLevel( encryptionLevel ).toConfig();
7878

7979
try ( Driver driver = GraphDatabase.driver( Neo4jRunner.DEFAULT_URI, config ) )
8080
{
@@ -113,7 +113,7 @@ public void shouldDropBrokenOldSessions() throws Throwable
113113
{
114114
// config with set liveness check timeout
115115
int livenessCheckTimeoutMinutes = 10;
116-
Config config = configBuilder
116+
Config config = Config.build().withEncryptionLevel( encryptionLevel )
117117
.withConnectionLivenessCheckTimeout( livenessCheckTimeoutMinutes, TimeUnit.MINUTES )
118118
.toConfig();
119119

0 commit comments

Comments
 (0)