18
18
*/
19
19
package org .neo4j .driver .internal ;
20
20
21
+ import java .util .Collections ;
22
+
21
23
import org .junit .Test ;
22
24
import org .mockito .InOrder ;
23
25
24
- import java .util .Collections ;
25
-
26
- import org .neo4j .driver .internal .spi .Connection ;
27
26
import org .neo4j .driver .internal .spi .Collector ;
27
+ import org .neo4j .driver .internal .spi .Connection ;
28
28
import org .neo4j .driver .v1 .Value ;
29
29
30
30
import static org .mockito .Matchers .any ;
@@ -50,11 +50,11 @@ public void shouldRollbackOnImplicitFailure() throws Throwable
50
50
51
51
// Then
52
52
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 ) );
55
55
order .verify ( conn ).isOpen ();
56
56
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 ) );
58
58
order .verify ( conn ).sync ();
59
59
verify ( cleanup ).run ();
60
60
verifyNoMoreInteractions ( conn , cleanup );
@@ -77,10 +77,10 @@ public void shouldRollbackOnExplicitFailure() throws Throwable
77
77
// Then
78
78
InOrder order = inOrder ( conn );
79
79
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 ) );
81
81
order .verify ( conn ).isOpen ();
82
82
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 ) );
84
84
order .verify ( conn ).sync ();
85
85
verify ( cleanup ).run ();
86
86
verifyNoMoreInteractions ( conn , cleanup );
@@ -103,10 +103,10 @@ public void shouldCommitOnSuccess() throws Throwable
103
103
104
104
InOrder order = inOrder ( conn );
105
105
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 ) );
107
107
order .verify ( conn ).isOpen ();
108
108
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 ) );
110
110
order .verify ( conn ).sync ();
111
111
verify ( cleanup ).run ();
112
112
verifyNoMoreInteractions ( conn , cleanup );
0 commit comments