diff --git a/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/CursorsClosedPostCondition.java b/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/CursorsClosedPostCondition.java index 5a9a06f4be15..c22765c1ba7e 100644 --- a/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/CursorsClosedPostCondition.java +++ b/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/CursorsClosedPostCondition.java @@ -24,11 +24,13 @@ import org.junit.runner.Description; import org.junit.runners.model.Statement; +import java.util.function.Supplier; + public class CursorsClosedPostCondition implements TestRule { - private ManagedTestCursors cursors; + private Supplier cursors; - CursorsClosedPostCondition( ManagedTestCursors c ) + CursorsClosedPostCondition( Supplier c ) { this.cursors = c; } @@ -41,7 +43,7 @@ public Statement apply( Statement base, Description description ) public void evaluate() throws Throwable { base.evaluate(); - cursors.assertAllClosedAndReset(); // only done if test succeeds + cursors.get().assertAllClosedAndReset(); // only done if test succeeds } }; } diff --git a/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/KernelAPIReadTestBase.java b/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/KernelAPIReadTestBase.java index 743ee326d6d3..de933fdb9021 100644 --- a/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/KernelAPIReadTestBase.java +++ b/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/KernelAPIReadTestBase.java @@ -91,7 +91,7 @@ public void setupGraph() throws IOException, KernelException } @Rule - public CursorsClosedPostCondition cursorsClosedPostCondition = new CursorsClosedPostCondition( cursors ); + public CursorsClosedPostCondition cursorsClosedPostCondition = new CursorsClosedPostCondition( () -> cursors ); @After public void closeTransaction() throws Exception diff --git a/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/KernelAPIWriteTestBase.java b/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/KernelAPIWriteTestBase.java index d751a71e421a..0d89718f998d 100644 --- a/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/KernelAPIWriteTestBase.java +++ b/community/kernel-api/src/test/java/org/neo4j/internal/kernel/api/KernelAPIWriteTestBase.java @@ -54,7 +54,7 @@ public abstract class KernelAPIWriteTestBase cursors ); /** * Creates a new instance of WriteSupport, which will be used to execute the concrete test