@@ -178,7 +178,7 @@ void reactivePingWithTrace() {
178178
179179 StatefulRedisConnection <String , String > connect = client .connect ();
180180 connect .reactive ().ping () //
181- .subscriberContext (it -> it .put (TraceContext .class , trace .context ())) //
181+ .contextWrite (it -> it .put (TraceContext .class , trace .context ())) //
182182 .as (StepVerifier ::create ) //
183183 .expectNext ("PONG" ).verifyComplete ();
184184
@@ -200,7 +200,7 @@ void reactiveGetAndSetWithTrace() {
200200 StatefulRedisConnection <String , String > connect = client .connect ();
201201 connect .reactive ().set ("foo" , "bar" ) //
202202 .then (connect .reactive ().get ("foo" )) //
203- .subscriberContext (it -> it .put (TraceContext .class , trace .context ())) //
203+ .contextWrite (it -> it .put (TraceContext .class , trace .context ())) //
204204 .as (StepVerifier ::create ) //
205205 .expectNext ("bar" ).verifyComplete ();
206206
@@ -224,8 +224,7 @@ void reactiveGetAndSetWithTraceProvider() {
224224
225225 StatefulRedisConnection <String , String > connect = client .connect ();
226226 connect .reactive ().set ("foo" , "bar" ).then (connect .reactive ().get ("foo" ))
227- .subscriberContext (io .lettuce .core .tracing .Tracing
228- .withTraceContextProvider (() -> BraveTracing .BraveTraceContext .create (trace .context ()))) //
227+ .contextWrite (io .lettuce .core .tracing .Tracing .withTraceContextProvider (() -> BraveTracing .BraveTraceContext .create (trace .context ()))) //
229228 .as (StepVerifier ::create ) //
230229 .expectNext ("bar" ).verifyComplete ();
231230
0 commit comments