Use closeAfterClass instead of destroy

trinodb · Mar 29, 2022 · aece856 · aece856
1 parent 7f954b5
commit aece856
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/...ino-singlestore/src/test/java/io/trino/plugin/singlestore/TestSingleStoreTypeMapping.java b/...ino-singlestore/src/test/java/io/trino/plugin/singlestore/TestSingleStoreTypeMapping.java
@@ -29,7 +29,6 @@
 import io.trino.testing.sql.SqlExecutor;
 import io.trino.testing.sql.TestTable;
 import io.trino.testing.sql.TrinoSqlExecutor;
-import org.testng.annotations.AfterClass;
 import org.testng.annotations.DataProvider;
 import org.testng.annotations.Test;
 
@@ -88,16 +87,10 @@ public class TestSingleStoreTypeMapping
     protected QueryRunner createQueryRunner()
             throws Exception
     {
-        singleStoreServer = new TestingSingleStoreServer();
+        singleStoreServer = closeAfterClass(new TestingSingleStoreServer());
         return createSingleStoreQueryRunner(singleStoreServer, ImmutableMap.of(), ImmutableMap.of(), ImmutableList.of());
     }
 
-    @AfterClass(alwaysRun = true)
-    public final void destroy()
-    {
-        singleStoreServer.close();
-    }
-
     @Test
     public void testBit()
     {