Shut down lettuce RedisClient (#9537)

Author: laurit

instrumentation/lettuce/lettuce-4.0/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/lettuce/v4_0/LettuceAsyncClientTest.java

@@ -103,12 +103,14 @@ static void setUp() {
     syncCommands.set("TESTKEY", "TESTVAL");
 
     // 1 set + 1 connect trace
+    testing.waitForTraces(2);
     testing.clearData();
   }
 
   @AfterAll
   static void cleanUp() {
     connection.close();
+    redisClient.shutdown();
     redisServer.stop();
   }
 
@@ -121,6 +123,7 @@ void testConnectUsingGetOnConnectionFuture() {
         testConnectionClient.connect(
             new Utf8StringCodec(), new RedisURI(host, port, 3, TimeUnit.SECONDS));
     cleanup.deferCleanup(connection1);
+    cleanup.deferCleanup(testConnectionClient::shutdown);
 
     assertThat(connection1).isNotNull();
 
@@ -140,6 +143,7 @@ void testConnectUsingGetOnConnectionFuture() {
   void testExceptionInsideTheConnectionFuture() {
     RedisClient testConnectionClient = RedisClient.create(dbUriNonExistent);
     testConnectionClient.setOptions(CLIENT_OPTIONS);
+    cleanup.deferCleanup(testConnectionClient::shutdown);
 
     Exception exception =
         catchException(
@@ -460,11 +464,14 @@ void testDebugSegfaultCommandWithNoArgumentShouldProduceSpan() {
 
     long serverPort = server.getMappedPort(6379);
     RedisClient client = RedisClient.create("redis://" + host + ":" + serverPort + "/" + DB_INDEX);
+    client.setOptions(CLIENT_OPTIONS);
     StatefulRedisConnection<String, String> connection1 = client.connect();
     cleanup.deferCleanup(connection1);
+    cleanup.deferCleanup(client::shutdown);
 
     RedisAsyncCommands<String, String> commands = connection1.async();
     // 1 connect trace
+    testing.waitForTraces(1);
     testing.clearData();
 
     commands.debugSegfault();
@@ -491,11 +498,14 @@ void testShutdownCommandShouldProduceSpan() {
 
     RedisClient client =
         RedisClient.create("redis://" + host + ":" + shutdownServerPort + "/" + DB_INDEX);
+    client.setOptions(CLIENT_OPTIONS);
     StatefulRedisConnection<String, String> connection1 = client.connect();
     cleanup.deferCleanup(connection1);
+    cleanup.deferCleanup(client::shutdown);
 
     RedisAsyncCommands<String, String> commands = connection1.async();
     // 1 connect trace
+    testing.waitForTraces(1);
     testing.clearData();
 
     commands.shutdown(false);

instrumentation/lettuce/lettuce-4.0/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/lettuce/v4_0/LettuceSyncClientTest.java

@@ -92,6 +92,7 @@ static void setUp() {
   @AfterAll
   static void cleanUp() {
     connection.close();
+    redisClient.shutdown();
     redisServer.stop();
   }
 
@@ -102,6 +103,7 @@ void testConnect() {
 
     StatefulRedisConnection<String, String> testConnection = testConnectionClient.connect();
     cleanup.deferCleanup(() -> testConnection.close());
+    cleanup.deferCleanup(testConnectionClient::shutdown);
 
     testing.waitAndAssertTraces(
         trace ->
@@ -119,6 +121,7 @@ void testConnect() {
   void testConnectException() {
     RedisClient testConnectionClient = RedisClient.create(dbUriNonExistent);
     testConnectionClient.setOptions(CLIENT_OPTIONS);
+    cleanup.deferCleanup(testConnectionClient::shutdown);
 
     Exception exception = catchException(testConnectionClient::connect);
 
@@ -258,11 +261,14 @@ void testDebugSegfaultCommandWithNoArgumentShouldProduceSpan() {
 
     long serverPort = server.getMappedPort(6379);
     RedisClient client = RedisClient.create("redis://" + host + ":" + serverPort + "/" + DB_INDEX);
+    client.setOptions(CLIENT_OPTIONS);
     StatefulRedisConnection<String, String> connection1 = client.connect();
     cleanup.deferCleanup(connection1);
+    cleanup.deferCleanup(client::shutdown);
 
     RedisCommands<String, String> commands = connection1.sync();
     // 1 connect trace
+    testing.waitForTraces(1);
     testing.clearData();
 
     commands.debugSegfault();
@@ -289,11 +295,14 @@ void testShutdownCommandShouldProduceSpan() {
 
     RedisClient client =
         RedisClient.create("redis://" + host + ":" + shutdownServerPort + "/" + DB_INDEX);
+    client.setOptions(CLIENT_OPTIONS);
     StatefulRedisConnection<String, String> connection1 = client.connect();
     cleanup.deferCleanup(connection1);
+    cleanup.deferCleanup(client::shutdown);
 
     RedisCommands<String, String> commands = connection1.sync();
     // 1 connect trace
+    testing.waitForTraces(1);
     testing.clearData();
 
     commands.shutdown(false);