Spring/spring-data-examples
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

#523 - Add example for declarative MongoDB aggregations.

Browse Source 
Original pull request: #526.
closes: #523
This commit is contained in:
Christoph Strobl
2019-08-14 08:12:57 +02:00
committed by Mark Paluch
parent a44f93a9bc
commit 84ca79d97f
4 changed files with 126 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
mongodb/aggregation/src/test/java/example/springdata/mongodb/aggregation/OrderRepositoryIntegrationTests.java
View File

@@ -21,11 +21,13 @@ import static org.junit.Assert.*;
import java.util.Date;
import org.assertj.core.api.Assertions;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.domain.Sort;
import org.springframework.test.context.junit4.SpringRunner;
/**
@@ -33,6 +35,7 @@ import org.springframework.test.context.junit4.SpringRunner;
*
* @author Thomas Darimont
* @author Oliver Gierke
* @author Christoph Strobl
*/
@RunWith(SpringRunner.class)
@SpringBootTest
@@ -64,4 +67,33 @@ public class OrderRepositoryIntegrationTests {
assertThat(invoice.getTaxAmount(), is(closeTo(1.577, 000001)));
assertThat(invoice.getTotalAmount(), is(closeTo(9.877, 000001)));
}
@Test
public void declarativeAggregationWithSort() {
repository.save(new Order("c42", new Date()).addItem(product1));
repository.save(new Order("c42", new Date()).addItem(product2));
repository.save(new Order("c42", new Date()).addItem(product3));
repository.save(new Order("b12", new Date()).addItem(product1));
repository.save(new Order("b12", new Date()).addItem(product1));
Assertions.assertThat(repository.totalOrdersPerCustomer(Sort.by(Sort.Order.desc("total")))) //
.containsExactly( //
new OrdersPerCustomer("c42", 3L), new OrdersPerCustomer("b12", 2L) //
);
}
@Test
public void multiStageDeclarativeAggregation() {
repository.save(new Order("c42", new Date()).addItem(product1));
repository.save(new Order("c42", new Date()).addItem(product2));
repository.save(new Order("c42", new Date()).addItem(product3));
repository.save(new Order("b12", new Date()).addItem(product1));
repository.save(new Order("b12", new Date()).addItem(product1));
Assertions.assertThat(repository.totalOrdersForCustomer("c42")).isEqualTo(3);
}
}