4242import org .springframework .data .r2dbc .convert .MappingR2dbcConverter ;
4343import org .springframework .data .r2dbc .core .DefaultReactiveDataAccessStrategy ;
4444import org .springframework .data .r2dbc .core .R2dbcEntityOperations ;
45+ import org .springframework .data .r2dbc .core .R2dbcEntityTemplate ;
4546import org .springframework .data .r2dbc .core .ReactiveDataAccessStrategy ;
4647import org .springframework .data .r2dbc .dialect .PostgresDialect ;
4748import org .springframework .data .r2dbc .mapping .R2dbcMappingContext ;
5455import org .springframework .data .repository .query .Param ;
5556import org .springframework .data .repository .query .ReactiveQueryMethodEvaluationContextProvider ;
5657import org .springframework .expression .spel .standard .SpelExpressionParser ;
58+ import org .springframework .r2dbc .core .DatabaseClient ;
5759import org .springframework .r2dbc .core .Parameter ;
5860import org .springframework .r2dbc .core .PreparedOperation ;
5961import org .springframework .r2dbc .core .binding .BindTarget ;
@@ -216,8 +218,7 @@ void bindsPositionalNamedSpelQuery() {
216218
217219 PreparedOperation <?> stringQuery = query .createQuery (accessor ).block ();
218220
219- assertThat (stringQuery .get ())
220- .isEqualTo ("SELECT * FROM person WHERE lastname = $1 and firstname = $2" );
221+ assertThat (stringQuery .get ()).isEqualTo ("SELECT * FROM person WHERE lastname = $1 and firstname = $2" );
221222 stringQuery .bindTo (bindTarget );
222223
223224 verify (bindTarget ).bind (0 , "White" );
@@ -284,7 +285,7 @@ void usesDtoTypeForDtoResultMapping() {
284285 assertThat (query .resolveResultType (query .getQueryMethod ().getResultProcessor ())).isEqualTo (PersonDto .class );
285286 }
286287
287- @ Test // gh-475
288+ @ Test // gh-612
288289 void selectsSimpleType () {
289290
290291 MockRowMetadata metadata = MockRowMetadata .builder ()
@@ -296,9 +297,10 @@ void selectsSimpleType() {
296297 StatementRecorder recorder = StatementRecorder .newInstance ();
297298 recorder .addStubbing (s -> s .equals ("SELECT MAX(DATE)" ), result );
298299
299- databaseClient = DatabaseClient .builder () //
300+ DatabaseClient databaseClient = DatabaseClient .builder () //
300301 .connectionFactory (recorder ) //
301302 .bindMarkers (PostgresDialect .INSTANCE .getBindMarkersFactory ()).build ();
303+ entityOperations = new R2dbcEntityTemplate (databaseClient , PostgresDialect .INSTANCE , converter );
302304
303305 StringBasedR2dbcQuery query = getQueryMethod ("findAllLocalDates" );
304306
0 commit comments