Created quires are doing the same and with same db with different types

2015-08-09 18:21:14 +02:00
parent cf52efcc5b
commit 759298a2bc
7 changed files with 501 additions and 200 deletions
--- a/spring-data-mongodb/src/test/java/org/baeldung/mongotemplate/DocumentQueryIntegrationTest.java
+++ b/spring-data-mongodb/src/test/java/org/baeldung/mongotemplate/DocumentQueryIntegrationTest.java
@@ -0,0 +1,179 @@
+package org.baeldung.mongotemplate;
+
+import static org.hamcrest.CoreMatchers.is;
+import static org.junit.Assert.assertThat;
+
+import java.util.List;
+
+import org.baeldung.config.MongoConfig;
+import org.baeldung.model.User;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.domain.PageRequest;
+import org.springframework.data.domain.Pageable;
+import org.springframework.data.domain.Sort;
+import org.springframework.data.mongodb.core.MongoTemplate;
+import org.springframework.data.mongodb.core.query.Criteria;
+import org.springframework.data.mongodb.core.query.Query;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration(classes = MongoConfig.class)
+public class DocumentQueryIntegrationTest {
+
+    @Autowired
+    private MongoTemplate mongoTemplate;
+
+    @Before
+    public void testSetup() {
+        mongoTemplate.createCollection(User.class);
+    }
+
+    @After
+    public void tearDown() {
+        mongoTemplate.dropCollection(User.class);
+    }
+    
+    @Test
+    public void givenUsersExist_whenFindingUsersByName_thenUsersAreFound() {
+        User user = new User();
+        user.setName("Eric");
+        user.setAge(45);
+        mongoTemplate.insert(user);
+        user = new User();
+        user.setName("Antony");
+        user.setAge(55);
+        mongoTemplate.insert(user);
+
+        Query query = new Query();
+        query.addCriteria(Criteria.where("name").is("Eric"));
+        List<User> users = mongoTemplate.find(query, User.class);
+
+        assertThat(users.size(), is(1));
+    }
+
+    @Test
+    public void givenUsersExist_whenFindingUserWithAgeLessThan50AndGreateThan20_thenUsersAreFound() {
+        User user = new User();
+        user.setAge(20);
+        user.setName("Jon");
+        mongoTemplate.insert(user);
+
+        user = new User();
+        user.setAge(50);
+        user.setName("Jon");
+        mongoTemplate.insert(user);
+
+        user = new User();
+        user.setAge(33);
+        user.setName("Jim");
+        mongoTemplate.insert(user);
+
+        Query query = new Query();
+        query.addCriteria(Criteria.where("age").lt(40).gt(26));
+        List<User> users = mongoTemplate.find(query, User.class);
+
+        assertThat(users.size(), is(1));
+    }
+
+    @Test
+    public void givenUsersExist_whenFindingUserWithNameStartWithA_thenUsersAreFound() {
+        User user = new User();
+        user.setName("Eric");
+        user.setAge(45);
+        mongoTemplate.insert(user);
+
+        user = new User();
+        user.setName("Antony");
+        user.setAge(33);
+        mongoTemplate.insert(user);
+
+        user = new User();
+        user.setName("Alice");
+        user.setAge(35);
+        mongoTemplate.insert(user);
+
+        Query query = new Query();
+        query.addCriteria(Criteria.where("name").regex("^A"));
+
+        List<User> users = mongoTemplate.find(query, User.class);
+        assertThat(users.size(), is(2));
+    }
+    
+    @Test
+    public void givenUsersExist_whenFindingUserWithNameEndWithC_thenUsersAreFound() {
+        User user = new User();
+        user.setName("Eric");
+        user.setAge(45);
+        mongoTemplate.insert(user);
+
+        user = new User();
+        user.setName("Antony");
+        user.setAge(33);
+        mongoTemplate.insert(user);
+
+        user = new User();
+        user.setName("Alice");
+        user.setAge(35);
+        mongoTemplate.insert(user);
+
+        Query query = new Query();
+        query.addCriteria(Criteria.where("name").regex("c$"));
+
+        List<User> users = mongoTemplate.find(query, User.class);
+        assertThat(users.size(), is(1));
+    }
+
+    @Test
+    public void givenUsersExist_whenFindingByPage_thenUsersAreFoundByPage() {
+        User user = new User();
+        user.setName("Eric");
+        user.setAge(45);
+        mongoTemplate.insert(user);
+
+        user = new User();
+        user.setName("Antony");
+        user.setAge(33);
+        mongoTemplate.insert(user);
+
+        user = new User();
+        user.setName("Alice");
+        user.setAge(35);
+        mongoTemplate.insert(user);
+
+        final Pageable pageableRequest = new PageRequest(0, 2);
+        Query query = new Query();
+        query.with(pageableRequest);
+
+        List<User> users = mongoTemplate.find(query, User.class);
+        assertThat(users.size(), is(2));
+    }
+
+    @Test
+    public void givenUsersExist_whenFindingUsersAndSortThem_thenUsersAreFoundAndSorted() {
+        User user = new User();
+        user.setName("Eric");
+        user.setAge(45);
+        mongoTemplate.insert(user);
+
+        user = new User();
+        user.setName("Antony");
+        user.setAge(33);
+        mongoTemplate.insert(user);
+
+        user = new User();
+        user.setName("Alice");
+        user.setAge(35);
+        mongoTemplate.insert(user);
+
+        Query query = new Query();
+        query.with(new Sort(Sort.Direction.ASC, "age"));
+
+        List<User> users = mongoTemplate.find(query, User.class);
+        assertThat(users.size(), is(3));
+    }
+}