use of org.springframework.data.annotation.Id in project spring-data-jdbc by spring-projects.
the class EntityRowMapperUnitTests method listReferenceGetsLoadedWithAdditionalSelect.
// DATAJDBC-130
@Test
public void listReferenceGetsLoadedWithAdditionalSelect() throws SQLException {
ResultSet rs = mockResultSet(//
asList("id", "name"), ID_FOR_ENTITY_REFERENCING_LIST, "alpha");
rs.next();
OneToMap extracted = createRowMapper(OneToMap.class).mapRow(rs, 1);
//
assertThat(extracted).isNotNull().extracting(e -> e.id, e -> e.name, //
e -> e.children.size()).containsExactly(ID_FOR_ENTITY_REFERENCING_LIST, "alpha", 2);
}
use of org.springframework.data.annotation.Id in project spring-data-jdbc by spring-projects.
the class SimpleJdbcRepositoryEventsUnitTests method createIdGeneratingOperations.
private static NamedParameterJdbcOperations createIdGeneratingOperations() {
Answer<Integer> setIdInKeyHolder = invocation -> {
HashMap<String, Object> keys = new HashMap<>();
keys.put("id", 4711L);
KeyHolder keyHolder = invocation.getArgument(2);
keyHolder.getKeyList().add(keys);
return 1;
};
NamedParameterJdbcOperations operations = mock(NamedParameterJdbcOperations.class);
when(operations.update(anyString(), any(SqlParameterSource.class), any(KeyHolder.class))).thenAnswer(setIdInKeyHolder);
return operations;
}
use of org.springframework.data.annotation.Id in project spring-data-jdbc by spring-projects.
the class EntityRowMapperUnitTests method mapReferenceGetsLoadedWithAdditionalSelect.
// DATAJDBC-131
@Test
public void mapReferenceGetsLoadedWithAdditionalSelect() throws SQLException {
ResultSet rs = mockResultSet(//
asList("id", "name"), ID_FOR_ENTITY_REFERENCING_MAP, "alpha");
rs.next();
OneToMap extracted = createRowMapper(OneToMap.class).mapRow(rs, 1);
//
assertThat(extracted).isNotNull().extracting(e -> e.id, e -> e.name, //
e -> e.children.size()).containsExactly(ID_FOR_ENTITY_REFERENCING_MAP, "alpha", 2);
}
use of org.springframework.data.annotation.Id in project spring-data-mongodb by spring-projects.
the class MongoTemplateTests method removeShouldConsiderSkipAndSort.
// DATAMONGO-1870
@Test
public void removeShouldConsiderSkipAndSort() {
List<Sample> samples = //
IntStream.range(0, 100).mapToObj(//
i -> new Sample("id-" + i, i % 2 == 0 ? "stark" : "lannister")).collect(Collectors.toList());
template.insertAll(samples);
DeleteResult wr = template.remove(new Query().skip(25).with(Sort.by("field")), Sample.class);
assertThat(wr.getDeletedCount()).isEqualTo(75L);
assertThat(template.count(new Query(), Sample.class)).isEqualTo(25L);
assertThat(template.count(query(where("field").is("lannister")), Sample.class)).isEqualTo(25L);
assertThat(template.count(query(where("field").is("stark")), Sample.class)).isEqualTo(0L);
}
use of org.springframework.data.annotation.Id in project spring-data-mongodb by spring-projects.
the class ReactiveMongoTemplateTests method testAddingToList.
// DATAMONGO-1444
@Test
public void testAddingToList() {
PersonWithAList person = createPersonWithAList("Sven", 22);
StepVerifier.create(template.insert(person)).expectNextCount(1).verifyComplete();
Query query = new Query(where("id").is(person.getId()));
StepVerifier.create(template.findOne(query, PersonWithAList.class)).consumeNextWith(actual -> {
assertThat(actual.getWishList().size(), is(0));
}).verifyComplete();
person.addToWishList("please work!");
StepVerifier.create(template.save(person)).expectNextCount(1).verifyComplete();
StepVerifier.create(template.findOne(query, PersonWithAList.class)).consumeNextWith(actual -> {
assertThat(actual.getWishList().size(), is(1));
}).verifyComplete();
Friend friend = new Friend();
person.setFirstName("Erik");
person.setAge(21);
person.addFriend(friend);
StepVerifier.create(template.save(person)).expectNextCount(1).verifyComplete();
StepVerifier.create(template.findOne(query, PersonWithAList.class)).consumeNextWith(actual -> {
assertThat(actual.getWishList().size(), is(1));
assertThat(actual.getFriends().size(), is(1));
}).verifyComplete();
}
Aggregations