DATAMONGO-2189 - Polishing.

Assert returned object is not the same as the saved one and move helper method.

Original Pull Request: #638

Author: christophstrobl

spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/MongoTemplateTests.java

@@ -45,6 +45,7 @@
 
 import org.bson.types.ObjectId;
 import org.hamcrest.collection.IsMapContaining;
+import org.hamcrest.core.IsSame;
 import org.joda.time.DateTime;
 import org.junit.After;
 import org.junit.Before;
@@ -3520,39 +3521,30 @@ public void onBeforeSave(BeforeSaveEvent<Document> event) {
 	public void afterSaveEventContainsSavedObjectUsingInsertAll() {
 
 		AtomicReference<ImmutableVersioned> saved = createAfterSaveReference();
+		ImmutableVersioned source = new ImmutableVersioned();
 
-		template.insertAll(Collections.singletonList(new ImmutableVersioned()));
+		template.insertAll(Collections.singletonList(source));
 
 		assertThat(saved.get(), is(notNullValue()));
+		assertThat(saved.get(), is(not(sameInstance(source))));
 		assertThat(saved.get().id, is(notNullValue()));
+
 	}
 
 	@Test // DATAMONGO-2189
 	@DirtiesContext
 	public void afterSaveEventContainsSavedObjectUsingInsert() {
 
 		AtomicReference<ImmutableVersioned> saved = createAfterSaveReference();
+		ImmutableVersioned source = new ImmutableVersioned();
 
-		template.insert(new ImmutableVersioned());
+		template.insert(source);
 
 		assertThat(saved.get(), is(notNullValue()));
+		assertThat(saved.get(), is(not(sameInstance(source))));
 		assertThat(saved.get().id, is(notNullValue()));
 	}
 
-	private AtomicReference<ImmutableVersioned> createAfterSaveReference() {
-
-		AtomicReference<ImmutableVersioned> saved = new AtomicReference<>();
-		context.addApplicationListener(new AbstractMongoEventListener<ImmutableVersioned>() {
-
-			@Override
-			public void onAfterSave(AfterSaveEvent<ImmutableVersioned> event) {
-				saved.set(event.getSource());
-			}
-		});
-
-		return saved;
-	}
-
 	@Test // DATAMONGO-1509
 	public void findsByGenericNestedListElements() {
 
@@ -3695,6 +3687,20 @@ public void saveAndLoadStringThatIsAnObjectIdAsString() {
 		assertThat(target).isEqualTo(source);
 	}
 
+	private AtomicReference<ImmutableVersioned> createAfterSaveReference() {
+
+		AtomicReference<ImmutableVersioned> saved = new AtomicReference<>();
+		context.addApplicationListener(new AbstractMongoEventListener<ImmutableVersioned>() {
+
+			@Override
+			public void onAfterSave(AfterSaveEvent<ImmutableVersioned> event) {
+				saved.set(event.getSource());
+			}
+		});
+
+		return saved;
+	}
+
 	static class TypeWithNumbers {
 
 		@Id String id;

spring-data-mongodb/src/test/java/org/springframework/data/mongodb/core/ReactiveMongoTemplateTests.java

@@ -1330,13 +1330,14 @@ public void removeShouldConsiderSkipAndSort() {
 	public void afterSaveEventContainsSavedObjectUsingInsert() {
 
 		AtomicReference<ImmutableVersioned> saved = createAfterSaveReference();
+		ImmutableVersioned source = new ImmutableVersioned();
 
-		template.insert(new ImmutableVersioned()) //
+		template.insert(source) //
 				.as(StepVerifier::create) //
 				.expectNextCount(1) //
 				.verifyComplete();
 
-		assertThat(saved.get()).isNotNull();
+		assertThat(saved.get()).isNotNull().isNotSameAs(source);
 		assertThat(saved.get().id).isNotNull();
 	}
 
@@ -1345,30 +1346,17 @@ public void afterSaveEventContainsSavedObjectUsingInsert() {
 	public void afterSaveEventContainsSavedObjectUsingInsertAll() {
 
 		AtomicReference<ImmutableVersioned> saved = createAfterSaveReference();
+		ImmutableVersioned source = new ImmutableVersioned();
 
 		template.insertAll(Collections.singleton(new ImmutableVersioned())) //
 				.as(StepVerifier::create) //
 				.expectNextCount(1) //
 				.verifyComplete();
 
-		assertThat(saved.get()).isNotNull();
+		assertThat(saved.get()).isNotNull().isNotSameAs(source);
 		assertThat(saved.get().id).isNotNull();
 	}
 
-	private AtomicReference<ImmutableVersioned> createAfterSaveReference() {
-
-		AtomicReference<ImmutableVersioned> saved = new AtomicReference<>();
-		context.addApplicationListener(new AbstractMongoEventListener<ImmutableVersioned>() {
-
-			@Override
-			public void onAfterSave(AfterSaveEvent<ImmutableVersioned> event) {
-				saved.set(event.getSource());
-			}
-		});
-
-		return saved;
-	}
-
 	@Test // DATAMONGO-2012
 	public void watchesDatabaseCorrectly() throws InterruptedException {
 
@@ -1490,6 +1478,20 @@ private PersonWithAList createPersonWithAList(String firstname, int age) {
 		return p;
 	}
 
+	private AtomicReference<ImmutableVersioned> createAfterSaveReference() {
+
+		AtomicReference<ImmutableVersioned> saved = new AtomicReference<>();
+		context.addApplicationListener(new AbstractMongoEventListener<ImmutableVersioned>() {
+
+			@Override
+			public void onAfterSave(AfterSaveEvent<ImmutableVersioned> event) {
+				saved.set(event.getSource());
+			}
+		});
+
+		return saved;
+	}
+
 	@AllArgsConstructor
 	@Wither
 	static class ImmutableVersioned {