33import org .junit .Test ;
44
55import java .util .ArrayList ;
6- import java .util .Collections ;
76import java .util .HashMap ;
87import java .util .List ;
98import java .util .Map ;
@@ -39,7 +38,7 @@ public void context_is_passed_to_batch_loader_function() throws Exception {
3938 return CompletableFuture .completedFuture (list );
4039 };
4140 DataLoaderOptions options = DataLoaderOptions .newOptions ()
42- .setBatchLoaderEnvironmentProvider (() -> "ctx" );
41+ .setBatchLoaderContextProvider (() -> "ctx" );
4342 DataLoader <String , String > loader = DataLoader .newDataLoader (batchLoader , options );
4443
4544 loader .load ("A" );
@@ -55,7 +54,7 @@ public void context_is_passed_to_batch_loader_function() throws Exception {
5554 public void key_contexts_are_passed_to_batch_loader_function () throws Exception {
5655 BatchLoaderWithContext <String , String > batchLoader = contextBatchLoader ();
5756 DataLoaderOptions options = DataLoaderOptions .newOptions ()
58- .setBatchLoaderEnvironmentProvider (() -> "ctx" );
57+ .setBatchLoaderContextProvider (() -> "ctx" );
5958 DataLoader <String , String > loader = DataLoader .newDataLoader (batchLoader , options );
6059
6160 loader .load ("A" , "aCtx" );
@@ -72,7 +71,7 @@ public void key_contexts_are_passed_to_batch_loader_function_when_batching_disab
7271 BatchLoaderWithContext <String , String > batchLoader = contextBatchLoader ();
7372 DataLoaderOptions options = DataLoaderOptions .newOptions ()
7473 .setBatchingEnabled (false )
75- .setBatchLoaderEnvironmentProvider (() -> "ctx" );
74+ .setBatchLoaderContextProvider (() -> "ctx" );
7675 DataLoader <String , String > loader = DataLoader .newDataLoader (batchLoader , options );
7776
7877 CompletableFuture <String > aLoad = loader .load ("A" , "aCtx" );
@@ -89,7 +88,7 @@ public void key_contexts_are_passed_to_batch_loader_function_when_batching_disab
8988 public void missing_key_contexts_are_passed_to_batch_loader_function () throws Exception {
9089 BatchLoaderWithContext <String , String > batchLoader = contextBatchLoader ();
9190 DataLoaderOptions options = DataLoaderOptions .newOptions ()
92- .setBatchLoaderEnvironmentProvider (() -> "ctx" );
91+ .setBatchLoaderContextProvider (() -> "ctx" );
9392 DataLoader <String , String > loader = DataLoader .newDataLoader (batchLoader , options );
9493
9594 loader .load ("A" , "aCtx" );
@@ -113,7 +112,7 @@ public void context_is_passed_to_map_batch_loader_function() throws Exception {
113112 return CompletableFuture .completedFuture (map );
114113 };
115114 DataLoaderOptions options = DataLoaderOptions .newOptions ()
116- .setBatchLoaderEnvironmentProvider (() -> "ctx" );
115+ .setBatchLoaderContextProvider (() -> "ctx" );
117116 DataLoader <String , String > loader = DataLoader .newMappedDataLoader (mapBatchLoader , options );
118117
119118 loader .load ("A" , "aCtx" );
0 commit comments