1818 */
1919package org .neo4j .driver .internal ;
2020
21+ import java .util .Collections ;
22+
2123import org .junit .Test ;
2224import org .mockito .InOrder ;
2325
24- import java .util .Collections ;
25-
26- import org .neo4j .driver .internal .spi .Connection ;
2726import org .neo4j .driver .internal .spi .Collector ;
27+ import org .neo4j .driver .internal .spi .Connection ;
2828import org .neo4j .driver .v1 .Value ;
2929
3030import static org .mockito .Matchers .any ;
@@ -50,11 +50,11 @@ public void shouldRollbackOnImplicitFailure() throws Throwable
5050
5151 // Then
5252 InOrder order = inOrder ( conn );
53- order .verify ( conn ).run ( "BEGIN" , Collections .<String , Value >emptyMap (), Collector .NO_OP );
54- order .verify ( conn ).discardAll ( any ( BookmarkCollector .class ) );
53+ order .verify ( conn ).run ( "BEGIN" , Collections .<String , Value >emptyMap (),Collector .NO_OP );
54+ order .verify ( conn ).pullAll ( any ( Collector .class ) );
5555 order .verify ( conn ).isOpen ();
5656 order .verify ( conn ).run ( "ROLLBACK" , Collections .<String , Value >emptyMap (), Collector .NO_OP );
57- order .verify ( conn ).discardAll ( any ( BookmarkCollector .class ) );
57+ order .verify ( conn ).pullAll ( any ( Collector .class ) );
5858 order .verify ( conn ).sync ();
5959 verify ( cleanup ).run ();
6060 verifyNoMoreInteractions ( conn , cleanup );
@@ -77,10 +77,10 @@ public void shouldRollbackOnExplicitFailure() throws Throwable
7777 // Then
7878 InOrder order = inOrder ( conn );
7979 order .verify ( conn ).run ( "BEGIN" , Collections .<String , Value >emptyMap (), Collector .NO_OP );
80- order .verify ( conn ).discardAll ( any ( BookmarkCollector .class ) );
80+ order .verify ( conn ).pullAll ( any ( BookmarkCollector .class ) );
8181 order .verify ( conn ).isOpen ();
8282 order .verify ( conn ).run ( "ROLLBACK" , Collections .<String , Value >emptyMap (), Collector .NO_OP );
83- order .verify ( conn ).discardAll ( any ( BookmarkCollector .class ) );
83+ order .verify ( conn ).pullAll ( any ( BookmarkCollector .class ) );
8484 order .verify ( conn ).sync ();
8585 verify ( cleanup ).run ();
8686 verifyNoMoreInteractions ( conn , cleanup );
@@ -103,10 +103,10 @@ public void shouldCommitOnSuccess() throws Throwable
103103
104104 InOrder order = inOrder ( conn );
105105 order .verify ( conn ).run ( "BEGIN" , Collections .<String , Value >emptyMap (), Collector .NO_OP );
106- order .verify ( conn ).discardAll ( any ( BookmarkCollector .class ) );
106+ order .verify ( conn ).pullAll ( any ( BookmarkCollector .class ) );
107107 order .verify ( conn ).isOpen ();
108108 order .verify ( conn ).run ( "COMMIT" , Collections .<String , Value >emptyMap (), Collector .NO_OP );
109- order .verify ( conn ).discardAll ( any ( BookmarkCollector .class ) );
109+ order .verify ( conn ).pullAll ( any ( BookmarkCollector .class ) );
110110 order .verify ( conn ).sync ();
111111 verify ( cleanup ).run ();
112112 verifyNoMoreInteractions ( conn , cleanup );
0 commit comments