Skip to content

Commit 4e93977

Browse files
committed
Fix MockedDriverContextFactory for LWT routing method mock
The DefaultLoadBalancingPolicy constructor reads LOAD_BALANCING_DEFAULT_LWT_REQUEST_ROUTING_METHOD which was missing from the mocked profile, causing NPE in StartupOptionsBuilderTest.
1 parent 2451315 commit 4e93977

2 files changed

Lines changed: 6 additions & 0 deletions

File tree

core/src/test/java/com/datastax/oss/driver/internal/core/context/MockedDriverContextFactory.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,9 @@ public static DriverExecutionProfile createProfile(String name, String localDc)
119119
.thenReturn("DefaultMetricsFactory");
120120
when(defaultProfile.getString(DefaultDriverOption.LOAD_BALANCING_LOCAL_DATACENTER))
121121
.thenReturn(localDc);
122+
when(defaultProfile.getString(
123+
DefaultDriverOption.LOAD_BALANCING_DEFAULT_LWT_REQUEST_ROUTING_METHOD))
124+
.thenReturn("REGULAR");
122125
return defaultProfile;
123126
}
124127

core/src/test/java/com/datastax/oss/driver/internal/core/context/StartupOptionsBuilderTest.java

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,9 @@ private DefaultDriverContext buildMockedContext(String compression) {
4242
when(defaultProfile.getString(DefaultDriverOption.PROTOCOL_COMPRESSION, "none"))
4343
.thenReturn(compression);
4444
when(defaultProfile.getName()).thenReturn(DriverExecutionProfile.DEFAULT_NAME);
45+
when(defaultProfile.getString(
46+
DefaultDriverOption.LOAD_BALANCING_DEFAULT_LWT_REQUEST_ROUTING_METHOD))
47+
.thenReturn("REGULAR");
4548
return MockedDriverContextFactory.defaultDriverContext(defaultProfile);
4649
}
4750

0 commit comments

Comments
 (0)