Use BeforeConvert callback result on non-versioned updates.

Previously, we used the original entity.

Closes #689
Original pull request: #690.

Author: JoseLion

src/main/java/org/springframework/data/r2dbc/core/R2dbcEntityTemplate.java

@@ -89,6 +89,7 @@
  * @author Mark Paluch
  * @author Bogdan Ilchyshyn
  * @author Jens Schauder
+ * @author Jose Luis Leon
  * @since 1.1
  */
 public class R2dbcEntityTemplate implements R2dbcEntityOperations, BeanFactoryAware, ApplicationContextAware {
@@ -686,7 +687,7 @@ private <T> Mono<T> doUpdate(T entity, SqlIdentifier tableName) {
 				entityToUse = incrementVersion(persistentEntity, it);
 			} else {
 
-				entityToUse = entity;
+				entityToUse = it;
 				matchingVersionCriteria = null;
 			}
 

src/test/java/org/springframework/data/r2dbc/core/R2dbcEntityTemplateUnitTests.java

@@ -67,6 +67,7 @@
  * Unit tests for {@link R2dbcEntityTemplate}.
  *
  * @author Mark Paluch
+ * @author Jose Luis Leon
  */
 public class R2dbcEntityTemplateUnitTests {
 
@@ -245,8 +246,8 @@ void shouldDeleteByQuery() {
 	@Test // gh-220
 	void shouldDeleteEntity() {
 
-		Person person = new Person();
-		person.id = "Walter";
+		Person person = Person.empty() //
+				.withId("Walter");
 		recorder.addStubbing(s -> s.startsWith("DELETE"), Collections.emptyList());
 
 		entityTemplate.delete(person) //
@@ -390,7 +391,7 @@ void insertShouldInvokeCallback() {
 		ValueCapturingAfterSaveCallback afterSave = new ValueCapturingAfterSaveCallback();
 
 		entityTemplate.setEntityCallbacks(ReactiveEntityCallbacks.create(beforeConvert, beforeSave, afterSave));
-		entityTemplate.insert(new Person()).as(StepVerifier::create) //
+		entityTemplate.insert(Person.empty()).as(StepVerifier::create) //
 				.assertNext(actual -> {
 					assertThat(actual.id).isEqualTo("after-save");
 					assertThat(actual.name).isEqualTo("before-convert");
@@ -439,10 +440,10 @@ void updateShouldInvokeCallback() {
 		ValueCapturingBeforeSaveCallback beforeSave = new ValueCapturingBeforeSaveCallback();
 		ValueCapturingAfterSaveCallback afterSave = new ValueCapturingAfterSaveCallback();
 
-		Person person = new Person();
-		person.id = "the-id";
-		person.name = "name";
-		person.description = "description";
+		Person person = Person.empty() //
+				.withId("the-id") //
+				.withName("name") //
+				.withDescription("description");
 
 		entityTemplate.setEntityCallbacks(ReactiveEntityCallbacks.create(beforeConvert, beforeSave, afterSave));
 		entityTemplate.update(person).as(StepVerifier::create) //
@@ -460,7 +461,8 @@ void updateShouldInvokeCallback() {
 				Parameter.from("before-save"));
 	}
 
-	@ToString
+	@Value
+	@With
 	static class Person {
 
 		@Id String id;
@@ -469,12 +471,8 @@ static class Person {
 
 		String description;
 
-		public String getName() {
-			return name;
-		}
-
-		public void setName(String name) {
-			this.name = name;
+		public static Person empty() {
+			return new Person(null, null, null);
 		}
 	}
 
@@ -548,8 +546,8 @@ static class ValueCapturingBeforeConvertCallback extends ValueCapturingEntityCal
 		public Mono<Person> onBeforeConvert(Person entity, SqlIdentifier table) {
 
 			capture(entity);
-			entity.name = "before-convert";
-			return Mono.just(entity);
+			Person person = entity.withName("before-convert");
+			return Mono.just(person);
 		}
 	}
 
@@ -573,9 +571,9 @@ public Mono<Person> onAfterSave(Person entity, OutboundRow outboundRow, SqlIdent
 
 			capture(entity);
 
-			Person person = new Person();
-			person.id = "after-save";
-			person.name = entity.name;
+			Person person = Person.empty() //
+					.withId("after-save") //
+					.withName(entity.getName());
 
 			return Mono.just(person);
 		}
@@ -588,9 +586,9 @@ static class ValueCapturingAfterConvertCallback extends ValueCapturingEntityCall
 		public Mono<Person> onAfterConvert(Person entity, SqlIdentifier table) {
 
 			capture(entity);
-			Person person = new Person();
-			person.id = "after-convert";
-			person.name = entity.name;
+			Person person = Person.empty() //
+					.withId("after-convert") //
+					.withName(entity.getName());
 
 			return Mono.just(person);
 		}