18
18
import static org .assertj .core .api .Assertions .*;
19
19
import static org .mockito .Mockito .*;
20
20
21
+ import io .r2dbc .spi .R2dbcType ;
21
22
import io .r2dbc .spi .test .MockColumnMetadata ;
22
23
import io .r2dbc .spi .test .MockResult ;
23
24
import io .r2dbc .spi .test .MockRow ;
@@ -87,8 +88,8 @@ void before() {
87
88
void shouldCountBy () {
88
89
89
90
MockRowMetadata metadata = MockRowMetadata .builder ()
90
- .columnMetadata (MockColumnMetadata .builder ().name ("name" ).build ()).build ();
91
- MockResult result = MockResult .builder (). rowMetadata ( metadata )
91
+ .columnMetadata (MockColumnMetadata .builder ().name ("name" ).type ( R2dbcType . VARCHAR ). build ()).build ();
92
+ MockResult result = MockResult .builder ()
92
93
.row (MockRow .builder ().identified (0 , Long .class , 1L ).build ()).build ();
93
94
94
95
recorder .addStubbing (s -> s .startsWith ("SELECT" ), result );
@@ -108,8 +109,8 @@ void shouldCountBy() {
108
109
void shouldProjectExistsResult () {
109
110
110
111
MockRowMetadata metadata = MockRowMetadata .builder ()
111
- .columnMetadata (MockColumnMetadata .builder ().name ("name" ).build ()).build ();
112
- MockResult result = MockResult .builder (). rowMetadata ( metadata )
112
+ .columnMetadata (MockColumnMetadata .builder ().name ("name" ).type ( R2dbcType . VARCHAR ). build ()).build ();
113
+ MockResult result = MockResult .builder ()
113
114
.row (MockRow .builder ().identified (0 , Object .class , null ).build ()).build ();
114
115
115
116
recorder .addStubbing (s -> s .startsWith ("SELECT" ), result );
@@ -126,8 +127,8 @@ void shouldProjectExistsResult() {
126
127
void shouldExistsByCriteria () {
127
128
128
129
MockRowMetadata metadata = MockRowMetadata .builder ()
129
- .columnMetadata (MockColumnMetadata .builder ().name ("name" ).build ()).build ();
130
- MockResult result = MockResult .builder (). rowMetadata ( metadata )
130
+ .columnMetadata (MockColumnMetadata .builder ().name ("name" ).type ( R2dbcType . VARCHAR ). build ()).build ();
131
+ MockResult result = MockResult .builder ()
131
132
.row (MockRow .builder ().identified (0 , Long .class , 1L ).build ()).build ();
132
133
133
134
recorder .addStubbing (s -> s .startsWith ("SELECT" ), result );
@@ -162,10 +163,11 @@ void shouldSelectByCriteria() {
162
163
@ Test // gh-215
163
164
void selectShouldInvokeCallback () {
164
165
165
- MockRowMetadata metadata = MockRowMetadata .builder ().columnMetadata (MockColumnMetadata .builder ().name ("id" ).build ())
166
- .columnMetadata (MockColumnMetadata .builder ().name ("THE_NAME" ).build ()).build ();
167
- MockResult result = MockResult .builder ().rowMetadata (metadata ).row (MockRow .builder ()
168
- .identified ("id" , Object .class , "Walter" ).identified ("THE_NAME" , Object .class , "some-name" ).build ()).build ();
166
+ MockRowMetadata metadata = MockRowMetadata .builder ()
167
+ .columnMetadata (MockColumnMetadata .builder ().name ("id" ).type (R2dbcType .INTEGER ).build ())
168
+ .columnMetadata (MockColumnMetadata .builder ().name ("THE_NAME" ).type (R2dbcType .VARCHAR ).build ()).build ();
169
+ MockResult result = MockResult .builder ().row (MockRow .builder ().identified ("id" , Object .class , "Walter" )
170
+ .identified ("THE_NAME" , Object .class , "some-name" ).metadata (metadata ).build ()).build ();
169
171
170
172
recorder .addStubbing (s -> s .startsWith ("SELECT" ), result );
171
173
@@ -204,8 +206,8 @@ void shouldSelectOne() {
204
206
void shouldUpdateByQuery () {
205
207
206
208
MockRowMetadata metadata = MockRowMetadata .builder ()
207
- .columnMetadata (MockColumnMetadata .builder ().name ("name" ).build ()).build ();
208
- MockResult result = MockResult .builder ().rowMetadata ( metadata ). rowsUpdated (1 ).build ();
209
+ .columnMetadata (MockColumnMetadata .builder ().name ("name" ).type ( R2dbcType . VARCHAR ). build ()).build ();
210
+ MockResult result = MockResult .builder ().rowsUpdated (1 ).build ();
209
211
210
212
recorder .addStubbing (s -> s .startsWith ("UPDATE" ), result );
211
213
@@ -226,8 +228,8 @@ void shouldUpdateByQuery() {
226
228
void shouldDeleteByQuery () {
227
229
228
230
MockRowMetadata metadata = MockRowMetadata .builder ()
229
- .columnMetadata (MockColumnMetadata .builder ().name ("name" ).build ()).build ();
230
- MockResult result = MockResult .builder ().rowMetadata ( metadata ). rowsUpdated (1 ).build ();
231
+ .columnMetadata (MockColumnMetadata .builder ().name ("name" ).type ( R2dbcType . VARCHAR ). build ()).build ();
232
+ MockResult result = MockResult .builder ().rowsUpdated (1 ).build ();
231
233
232
234
recorder .addStubbing (s -> s .startsWith ("DELETE" ), result );
233
235
0 commit comments