@@ -156,21 +156,20 @@ void testSetCommandUsingFutureGetWithTimeout() throws Exception {
156156 .waitAndAssertTraces (
157157 trace ->
158158 trace .hasSpansSatisfyingExactly (
159- span ->
160- span .hasName ("SET" )
161- .hasKind (SpanKind .CLIENT )
162- .hasAttributesSatisfyingExactly (
163- addExtraAttributes (
164- equalTo (NETWORK_TYPE , "ipv4" ),
165- equalTo (NETWORK_PEER_ADDRESS , ip ),
166- equalTo (NETWORK_PEER_PORT , port ),
167- equalTo (SERVER_ADDRESS , host ),
168- equalTo (SERVER_PORT , port ),
169- equalTo (maybeStable (DB_SYSTEM ), "redis" ),
170- equalTo (maybeStable (DB_STATEMENT ), "SET TESTSETKEY ?" )))
171- .hasEventsSatisfyingExactly (
172- event -> event .hasName ("redis.encode.start" ),
173- event -> event .hasName ("redis.encode.end" ))));
159+ span -> {
160+ span .hasName ("SET" )
161+ .hasKind (SpanKind .CLIENT )
162+ .hasAttributesSatisfyingExactly (
163+ addExtraAttributes (
164+ equalTo (NETWORK_TYPE , "ipv4" ),
165+ equalTo (NETWORK_PEER_ADDRESS , ip ),
166+ equalTo (NETWORK_PEER_PORT , port ),
167+ equalTo (SERVER_ADDRESS , host ),
168+ equalTo (SERVER_PORT , port ),
169+ equalTo (maybeStable (DB_SYSTEM ), "redis" ),
170+ equalTo (maybeStable (DB_STATEMENT ), "SET TESTSETKEY ?" )));
171+ assertCommandEncodeEvents (span );
172+ }));
174173 }
175174
176175 @ Test
@@ -201,22 +200,21 @@ void testGetCommandChainedWithThenAccept() throws Exception {
201200 new ArrayList <>(
202201 Arrays .asList (
203202 span -> span .hasName ("parent" ).hasKind (SpanKind .INTERNAL ).hasNoParent (),
204- span ->
205- span .hasName ("GET" )
206- .hasKind (SpanKind .CLIENT )
207- .hasParent (trace .getSpan (0 ))
208- .hasAttributesSatisfyingExactly (
209- addExtraAttributes (
210- equalTo (NETWORK_TYPE , "ipv4" ),
211- equalTo (NETWORK_PEER_ADDRESS , ip ),
212- equalTo (NETWORK_PEER_PORT , port ),
213- equalTo (SERVER_ADDRESS , host ),
214- equalTo (SERVER_PORT , port ),
215- equalTo (maybeStable (DB_SYSTEM ), "redis" ),
216- equalTo (maybeStable (DB_STATEMENT ), "GET TESTKEY" )))
217- .hasEventsSatisfyingExactly (
218- event -> event .hasName ("redis.encode.start" ),
219- event -> event .hasName ("redis.encode.end" ))));
203+ span -> {
204+ span .hasName ("GET" )
205+ .hasKind (SpanKind .CLIENT )
206+ .hasParent (trace .getSpan (0 ))
207+ .hasAttributesSatisfyingExactly (
208+ addExtraAttributes (
209+ equalTo (NETWORK_TYPE , "ipv4" ),
210+ equalTo (NETWORK_PEER_ADDRESS , ip ),
211+ equalTo (NETWORK_PEER_PORT , port ),
212+ equalTo (SERVER_ADDRESS , host ),
213+ equalTo (SERVER_PORT , port ),
214+ equalTo (maybeStable (DB_SYSTEM ), "redis" ),
215+ equalTo (maybeStable (DB_STATEMENT ), "GET TESTKEY" )));
216+ assertCommandEncodeEvents (span );
217+ }));
220218
221219 if (testCallback ()) {
222220 spanAsserts .add (
@@ -280,23 +278,23 @@ void testGetNonExistentKeyCommandWithHandleAsyncAndChainedWithThenApply() throws
280278 new ArrayList <>(
281279 Arrays .asList (
282280 span -> span .hasName ("parent" ).hasKind (SpanKind .INTERNAL ).hasNoParent (),
283- span ->
284- span .hasName ("GET" )
285- .hasKind (SpanKind .CLIENT )
286- .hasParent (trace .getSpan (0 ))
287- .hasAttributesSatisfyingExactly (
288- addExtraAttributes (
289- equalTo (NETWORK_TYPE , "ipv4" ),
290- equalTo (NETWORK_PEER_ADDRESS , ip ),
291- equalTo (NETWORK_PEER_PORT , port ),
292- equalTo (SERVER_ADDRESS , host ),
293- equalTo (SERVER_PORT , port ),
294- equalTo (maybeStable (DB_SYSTEM ), "redis" ),
295- equalTo (
296- maybeStable (DB_STATEMENT ), "GET NON_EXISTENT_KEY" )))
297- . hasEventsSatisfyingExactly (
298- event -> event . hasName ( "redis.encode.start" ),
299- event -> event . hasName ( "redis.encode.end" )) ));
281+ span -> {
282+ span .hasName ("GET" )
283+ .hasKind (SpanKind .CLIENT )
284+ .hasParent (trace .getSpan (0 ))
285+ .hasAttributesSatisfyingExactly (
286+ addExtraAttributes (
287+ equalTo (NETWORK_TYPE , "ipv4" ),
288+ equalTo (NETWORK_PEER_ADDRESS , ip ),
289+ equalTo (NETWORK_PEER_PORT , port ),
290+ equalTo (SERVER_ADDRESS , host ),
291+ equalTo (SERVER_PORT , port ),
292+ equalTo (maybeStable (DB_SYSTEM ), "redis" ),
293+ equalTo (
294+ maybeStable (DB_STATEMENT ), "GET NON_EXISTENT_KEY" )));
295+
296+ assertCommandEncodeEvents ( span );
297+ } ));
300298
301299 if (testCallback ()) {
302300 spanAsserts .addAll (
@@ -347,22 +345,22 @@ void testCommandWithNoArgumentsUsingBiconsumer() throws Exception {
347345 new ArrayList <>(
348346 Arrays .asList (
349347 span -> span .hasName ("parent" ).hasKind (SpanKind .INTERNAL ).hasNoParent (),
350- span ->
351- span .hasName ("RANDOMKEY" )
352- .hasKind (SpanKind .CLIENT )
353- .hasParent (trace .getSpan (0 ))
354- .hasAttributesSatisfyingExactly (
355- addExtraAttributes (
356- equalTo (NETWORK_TYPE , "ipv4" ),
357- equalTo (NETWORK_PEER_ADDRESS , ip ),
358- equalTo (NETWORK_PEER_PORT , port ),
359- equalTo (SERVER_ADDRESS , host ),
360- equalTo (SERVER_PORT , port ),
361- equalTo (maybeStable (DB_SYSTEM ), "redis" ),
362- equalTo (maybeStable (DB_STATEMENT ), "RANDOMKEY" )))
363- . hasEventsSatisfyingExactly (
364- event -> event . hasName ( "redis.encode.start" ),
365- event -> event . hasName ( "redis.encode.end" )) ));
348+ span -> {
349+ span .hasName ("RANDOMKEY" )
350+ .hasKind (SpanKind .CLIENT )
351+ .hasParent (trace .getSpan (0 ))
352+ .hasAttributesSatisfyingExactly (
353+ addExtraAttributes (
354+ equalTo (NETWORK_TYPE , "ipv4" ),
355+ equalTo (NETWORK_PEER_ADDRESS , ip ),
356+ equalTo (NETWORK_PEER_PORT , port ),
357+ equalTo (SERVER_ADDRESS , host ),
358+ equalTo (SERVER_PORT , port ),
359+ equalTo (maybeStable (DB_SYSTEM ), "redis" ),
360+ equalTo (maybeStable (DB_STATEMENT ), "RANDOMKEY" )));
361+
362+ assertCommandEncodeEvents ( span );
363+ } ));
366364
367365 if (testCallback ()) {
368366 spanAsserts .add (
@@ -408,39 +406,38 @@ void testHashSetAndThenNestApplyToHashGetall() throws Exception {
408406 .waitAndAssertTraces (
409407 trace ->
410408 trace .hasSpansSatisfyingExactly (
411- span ->
412- span .hasName ("HMSET" )
413- .hasKind (SpanKind .CLIENT )
414- .hasAttributesSatisfyingExactly (
415- addExtraAttributes (
416- equalTo (NETWORK_TYPE , "ipv4" ),
417- equalTo (NETWORK_PEER_ADDRESS , ip ),
418- equalTo (NETWORK_PEER_PORT , port ),
419- equalTo (SERVER_ADDRESS , host ),
420- equalTo (SERVER_PORT , port ),
421- equalTo (maybeStable (DB_SYSTEM ), "redis" ),
422- equalTo (
423- maybeStable (DB_STATEMENT ),
424- "HMSET TESTHM firstname ? lastname ? age ?" )))
425- .hasEventsSatisfyingExactly (
426- event -> event .hasName ("redis.encode.start" ),
427- event -> event .hasName ("redis.encode.end" ))),
409+ span -> {
410+ span .hasName ("HMSET" )
411+ .hasKind (SpanKind .CLIENT )
412+ .hasAttributesSatisfyingExactly (
413+ addExtraAttributes (
414+ equalTo (NETWORK_TYPE , "ipv4" ),
415+ equalTo (NETWORK_PEER_ADDRESS , ip ),
416+ equalTo (NETWORK_PEER_PORT , port ),
417+ equalTo (SERVER_ADDRESS , host ),
418+ equalTo (SERVER_PORT , port ),
419+ equalTo (maybeStable (DB_SYSTEM ), "redis" ),
420+ equalTo (
421+ maybeStable (DB_STATEMENT ),
422+ "HMSET TESTHM firstname ? lastname ? age ?" )));
423+ assertCommandEncodeEvents (span );
424+ }),
428425 trace ->
429426 trace .hasSpansSatisfyingExactly (
430- span ->
431- span .hasName ("HGETALL" )
432- .hasKind (SpanKind .CLIENT )
433- .hasAttributesSatisfyingExactly (
434- addExtraAttributes (
435- equalTo (NETWORK_TYPE , "ipv4" ),
436- equalTo (NETWORK_PEER_ADDRESS , ip ),
437- equalTo (NETWORK_PEER_PORT , port ),
438- equalTo (SERVER_ADDRESS , host ),
439- equalTo (SERVER_PORT , port ),
440- equalTo (maybeStable (DB_SYSTEM ), "redis" ),
441- equalTo (maybeStable (DB_STATEMENT ), "HGETALL TESTHM" )))
442- . hasEventsSatisfyingExactly (
443- event -> event . hasName ( "redis.encode.start" ),
444- event -> event . hasName ( "redis.encode.end" )) ));
427+ span -> {
428+ span .hasName ("HGETALL" )
429+ .hasKind (SpanKind .CLIENT )
430+ .hasAttributesSatisfyingExactly (
431+ addExtraAttributes (
432+ equalTo (NETWORK_TYPE , "ipv4" ),
433+ equalTo (NETWORK_PEER_ADDRESS , ip ),
434+ equalTo (NETWORK_PEER_PORT , port ),
435+ equalTo (SERVER_ADDRESS , host ),
436+ equalTo (SERVER_PORT , port ),
437+ equalTo (maybeStable (DB_SYSTEM ), "redis" ),
438+ equalTo (maybeStable (DB_STATEMENT ), "HGETALL TESTHM" )));
439+
440+ assertCommandEncodeEvents ( span );
441+ } ));
445442 }
446443}
0 commit comments