DATAMONGO-1718 - Polishing.

Add test and hand over Object.class as placeholder for required domain type.

Original Pull Request: #469
This commit is contained in:
Christoph Strobl
2017-06-16 11:35:09 +02:00
parent 18355aed5e
commit e5c634c8f6
2 changed files with 44 additions and 1 deletions

View File

@@ -1499,7 +1499,7 @@ public class MongoTemplate implements MongoOperations, ApplicationContextAware {
*/
@Override
public <T> List<T> findAllAndRemove(Query query, String collectionName) {
return findAllAndRemove(query, null, collectionName);
return (List<T>) findAllAndRemove(query, Object.class, collectionName);
}
/*

View File

@@ -3264,6 +3264,28 @@ public class MongoTemplateTests {
assertThat(loaded.getValue(), instanceOf(decimal128Type));
}
/**
* @see DATAMONGO-1718
*/
@Test
public void findAndRemoveAllWithoutExplicitDomainTypeShouldRemoveAndReturnEntitiesCorrectly() {
Sample jon = new Sample("1", "jon snow");
Sample bran = new Sample("2", "bran stark");
Sample rickon = new Sample("3", "rickon stark");
template.save(jon);
template.save(bran);
template.save(rickon);
List<Sample> result = template.findAllAndRemove(query(where("field").regex(".*stark$")),
template.determineCollectionName(Sample.class));
assertThat(result, hasSize(2));
assertThat(result, containsInAnyOrder(bran, rickon));
assertThat(template.count(new BasicQuery("{}"), template.determineCollectionName(Sample.class)), is(equalTo(1L)));
}
static class TypeWithNumbers {
@Id String id;
@@ -3435,6 +3457,27 @@ public class MongoTemplateTests {
this.id = id;
this.field = field;
}
@Override
public boolean equals(Object o) {
if (this == o)
return true;
if (o == null || getClass() != o.getClass())
return false;
Sample sample = (Sample) o;
if (id != null ? !id.equals(sample.id) : sample.id != null)
return false;
return field != null ? field.equals(sample.field) : sample.field == null;
}
@Override
public int hashCode() {
int result = id != null ? id.hashCode() : 0;
result = 31 * result + (field != null ? field.hashCode() : 0);
return result;
}
}
static class TestClass {