Add server shutdown to DatabaseExtension (#1022)

neo4j · Sep 28, 2021 · c11e210 · c11e210
1 parent 16a321d
commit c11e210
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/driver/src/test/java/org/neo4j/driver/util/DatabaseExtension.java b/driver/src/test/java/org/neo4j/driver/util/DatabaseExtension.java
@@ -18,6 +18,7 @@
  */
 package org.neo4j.driver.util;
 
+import org.junit.jupiter.api.extension.AfterAllCallback;
 import org.junit.jupiter.api.extension.BeforeEachCallback;
 import org.junit.jupiter.api.extension.ExtensionContext;
 
@@ -41,7 +42,7 @@
 import static org.neo4j.driver.util.Neo4jSettings.DEFAULT_TLS_CERT_PATH;
 import static org.neo4j.driver.util.Neo4jSettings.DEFAULT_TLS_KEY_PATH;
 
-public class DatabaseExtension implements BeforeEachCallback
+public class DatabaseExtension implements BeforeEachCallback, AfterAllCallback
 {
     static final String TEST_RESOURCE_FOLDER_PATH = "src/test/resources";
 
@@ -66,6 +67,15 @@ public void beforeEach( ExtensionContext context ) throws Exception
         TestUtil.cleanDb( driver() );
     }
 
+    @Override
+    public void afterAll( ExtensionContext context )
+    {
+        if ( runner != null )
+        {
+            runner.stopNeo4j();
+        }
+    }
+
     public Driver driver()
     {
         return runner.driver();