Skip to content

Commit 78cd89e

Browse files
GH-2640 - Assert Transient attributes.
1 parent a41fbad commit 78cd89e

File tree

2 files changed

+15
-1
lines changed

2 files changed

+15
-1
lines changed

src/test/java/org/springframework/data/neo4j/integration/imperative/CallbacksIT.java

+12-1
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
import java.util.Optional;
2525
// tag::faq.entities.auditing.callbacks[]
2626
import java.util.UUID;
27+
import java.util.stream.StreamSupport;
2728

2829
// end::faq.entities.auditing.callbacks[]
2930
import org.junit.jupiter.api.Test;
@@ -102,14 +103,24 @@ void onBeforeBindShouldBeCalledForAllEntities(@Autowired ThingRepository reposit
102103
thing1.setRandomValue("a");
103104
ThingWithAssignedId thing2 = new ThingWithAssignedId("id2", "Another name");
104105
thing2.setRandomValue("b");
105-
Iterable<ThingWithAssignedId> savedThings = repository.saveAll(Arrays.asList(thing1, thing2));
106+
107+
var unsaved = Arrays.asList(thing1, thing2);
108+
Iterable<ThingWithAssignedId> savedThings = repository.saveAll(unsaved);
109+
110+
assertThat(unsaved).allMatch(v -> v.getRandomValue() != null);
111+
assertThat(unsaved).noneMatch(v -> v.getAnotherRandomValue() != null);
106112

107113
assertThat(savedThings).extracting(ThingWithAssignedId::getName).containsExactlyInAnyOrder("A name (Edited)",
108114
"Another name (Edited)");
109115
assertThat(savedThings).hasSize(2)
110116
.extracting(ThingWithAssignedId::getRandomValue)
111117
.allMatch(Objects::isNull);
112118

119+
// Assert the onAfterConvert
120+
var ids = StreamSupport.stream(savedThings.spliterator(), false).map(ThingWithAssignedId::getTheId).toList();
121+
var reloaded = repository.findAllById(ids);
122+
assertThat(reloaded).allMatch(v -> v.getRandomValue() != null);
123+
113124
verifyDatabase(savedThings);
114125
}
115126

src/test/java/org/springframework/data/neo4j/integration/shared/common/CallbacksITBase.java

+3
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,9 @@ protected void verifyDatabase(Iterable<ThingWithAssignedId> expectedValues) {
7878
List<Node> nodes = record.get("things").asList(Value::asNode);
7979
assertThat(nodes).extracting(n -> n.get("theId").asString()).containsAll(ids);
8080
assertThat(nodes).extracting(n -> n.get("name").asString()).containsAll(names);
81+
assertThat(nodes).allMatch(n -> n.get("randomValue").isNull());
82+
assertThat(nodes).allMatch(n -> n.get("anotherRandomValue").isNull());
83+
8184
bookmarkCapture.seedWith(session.lastBookmarks());
8285
}
8386
}

0 commit comments

Comments
 (0)