@@ -192,6 +192,8 @@ void canCopyExistingOptionValuesOnTransform() {
192
192
193
193
DataLoaderInstrumentation instrumentation1 = new DataLoaderInstrumentation () {
194
194
};
195
+ DataLoaderInstrumentation instrumentation2 = new DataLoaderInstrumentation () {
196
+ };
195
197
BatchLoaderContextProvider contextProvider1 = () -> null ;
196
198
197
199
DataLoaderOptions startingOptions = DataLoaderOptions .newOptionsBuilder ().setBatchingEnabled (false )
@@ -205,7 +207,8 @@ void canCopyExistingOptionValuesOnTransform() {
205
207
assertThat (startingOptions .getInstrumentation (), equalTo (instrumentation1 ));
206
208
assertThat (startingOptions .getBatchLoaderContextProvider (), equalTo (contextProvider1 ));
207
209
208
- DataLoaderOptions newOptions = startingOptions .transform (builder -> builder .setBatchingEnabled (true ));
210
+ DataLoaderOptions newOptions = startingOptions .transform (builder ->
211
+ builder .setBatchingEnabled (true ).setInstrumentation (instrumentation2 ));
209
212
210
213
211
214
// immutable
@@ -215,10 +218,13 @@ void canCopyExistingOptionValuesOnTransform() {
215
218
assertThat (startingOptions .getInstrumentation (), equalTo (instrumentation1 ));
216
219
assertThat (startingOptions .getBatchLoaderContextProvider (), equalTo (contextProvider1 ));
217
220
218
- // copied values
219
- assertThat (newOptions .batchingEnabled (), equalTo (true ));
221
+ // stayed the same
220
222
assertThat (newOptions .cachingEnabled (), equalTo (false ));
221
- assertThat (newOptions .getInstrumentation (), equalTo (instrumentation1 ));
222
223
assertThat (newOptions .getBatchLoaderContextProvider (), equalTo (contextProvider1 ));
224
+
225
+ // was changed
226
+ assertThat (newOptions .batchingEnabled (), equalTo (true ));
227
+ assertThat (newOptions .getInstrumentation (), equalTo (instrumentation2 ));
228
+
223
229
}
224
230
}
0 commit comments