close lettuce client in unit test

alibaba · Mar 28, 2018 · 3b9cd1f · 3b9cd1f
1 parent c701379
commit 3b9cd1f
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 0 deletions.
diff --git a/...edis-lettuce/src/main/java/com/alicp/jetcache/redis/lettuce/LettuceConnectionManager.java b/...edis-lettuce/src/main/java/com/alicp/jetcache/redis/lettuce/LettuceConnectionManager.java
@@ -121,5 +121,6 @@ public void removeAndClose(AbstractRedisClient redisClient) {
         if (lo.connection != null) {
             lo.connection.close();
         }
+        redisClient.shutdown();
     }
 }
diff --git a/...is-lettuce4/src/main/java/com/alicp/jetcache/redis/lettuce4/LettuceConnectionManager.java b/...is-lettuce4/src/main/java/com/alicp/jetcache/redis/lettuce4/LettuceConnectionManager.java
@@ -121,5 +121,6 @@ public void removeAndClose(AbstractRedisClient redisClient) {
         if (lo.connection != null) {
             lo.connection.close();
         }
+        redisClient.shutdown();
     }
 }
diff --git a/jetcache-test/src/test/java/com/alicp/jetcache/redis/lettuce/RedisLettuceCacheTest.java b/jetcache-test/src/test/java/com/alicp/jetcache/redis/lettuce/RedisLettuceCacheTest.java
@@ -94,6 +94,8 @@ public void testWithMultiLevelCache() throws Exception {
         LoadingCacheTest.loadingCacheTest(MultiLevelCacheBuilder.createMultiLevelCacheBuilder()
                 .expireAfterWrite(5000, TimeUnit.MILLISECONDS)
                 .addCache(l1Cache, l2Cache), 50);
+
+        LettuceConnectionManager.defaultManager().removeAndClose(client);
     }
 
     private void test(AbstractRedisClient client) throws Exception {

diff --git a/jetcache-test/src/test/java/com/alicp/jetcache/redis/lettuce4/RedisLettuceCacheTest.java b/jetcache-test/src/test/java/com/alicp/jetcache/redis/lettuce4/RedisLettuceCacheTest.java
@@ -94,6 +94,8 @@ public void testWithMultiLevelCache() throws Exception {
         LoadingCacheTest.loadingCacheTest(MultiLevelCacheBuilder.createMultiLevelCacheBuilder()
                 .expireAfterWrite(5000, TimeUnit.MILLISECONDS)
                 .addCache(l1Cache, l2Cache), 50);
+
+        LettuceConnectionManager.defaultManager().removeAndClose(client);
     }
 
     private void test(AbstractRedisClient client) throws Exception {