diff --git a/integration-tests/src/test/java/com/datastax/oss/driver/core/PeersV2NodeRefreshIT.java b/integration-tests/src/test/java/com/datastax/oss/driver/core/PeersV2NodeRefreshIT.java index 52a5b6eef53..06ac146018c 100644 --- a/integration-tests/src/test/java/com/datastax/oss/driver/core/PeersV2NodeRefreshIT.java +++ b/integration-tests/src/test/java/com/datastax/oss/driver/core/PeersV2NodeRefreshIT.java @@ -56,17 +56,18 @@ public static void tearDown() { @Test public void should_successfully_send_peers_v2_node_refresh_query() throws InterruptedException, ExecutionException { - CqlSession session = - CqlSession.builder().addContactPoint(cluster.node(1).inetSocketAddress()).build(); - Node node = findNonControlNode(session); - ((InternalDriverContext) session.getContext()) - .getMetadataManager() - .refreshNode(node) - .toCompletableFuture() - .get(); - assertThat(hasNodeRefreshQuery()) - .describedAs("Expecting peers_v2 node refresh query to be present but it wasn't") - .isTrue(); + try (CqlSession session = + CqlSession.builder().addContactPoint(cluster.node(1).inetSocketAddress()).build()) { + Node node = findNonControlNode(session); + ((InternalDriverContext) session.getContext()) + .getMetadataManager() + .refreshNode(node) + .toCompletableFuture() + .get(); + assertThat(hasNodeRefreshQuery()) + .describedAs("Expecting peers_v2 node refresh query to be present but it wasn't") + .isTrue(); + } } private Node findNonControlNode(CqlSession session) {