Close Connection in test

trinodb · Feb 7, 2020 · c0e4b0b · c0e4b0b
1 parent 42ff57a
commit c0e4b0b
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/presto-jdbc/src/test/java/io/prestosql/jdbc/TestJdbcConnection.java b/presto-jdbc/src/test/java/io/prestosql/jdbc/TestJdbcConnection.java
@@ -272,11 +272,12 @@ public void testApplicationName()
     public void testExtraCredentials()
             throws SQLException
     {
-        Map<String, String> credentials = ImmutableMap.of("test.token.foo", "bar", "test.token.abc", "xyz");
-        Connection connection = createConnection("extraCredentials=test.token.foo:bar;test.token.abc:xyz");
-        PrestoConnection prestoConnection = connection.unwrap(PrestoConnection.class);
-        assertEquals(prestoConnection.getExtraCredentials(), credentials);
-        assertEquals(listExtraCredentials(connection), credentials);
+        try (Connection connection = createConnection("extraCredentials=test.token.foo:bar;test.token.abc:xyz")) {
+            Map<String, String> expectedCredentials = ImmutableMap.of("test.token.foo", "bar", "test.token.abc", "xyz");
+            PrestoConnection prestoConnection = connection.unwrap(PrestoConnection.class);
+            assertEquals(prestoConnection.getExtraCredentials(), expectedCredentials);
+            assertEquals(listExtraCredentials(connection), expectedCredentials);
+        }
     }
 
     private Connection createConnection()