116 lines
3.1 KiB
Java
116 lines
3.1 KiB
Java
package com.baeldung.repository;
|
|
|
|
import static org.hamcrest.CoreMatchers.is;
|
|
import static org.junit.Assert.assertThat;
|
|
|
|
import java.util.List;
|
|
|
|
import com.baeldung.config.MongoConfig;
|
|
import com.baeldung.model.QUser;
|
|
import com.baeldung.model.User;
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
import org.springframework.test.context.ContextConfiguration;
|
|
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
|
|
|
|
import com.querydsl.core.types.Predicate;
|
|
|
|
|
|
|
|
@RunWith(SpringJUnit4ClassRunner.class)
|
|
@ContextConfiguration(classes = MongoConfig.class)
|
|
public class DSLQueryLiveTest extends BaseQueryLiveTest {
|
|
|
|
@Test
|
|
public void givenUsersExist_whenFindingUsersByName_thenUserAreFound() {
|
|
User user = new User();
|
|
user.setName("Eric");
|
|
user.setAge(45);
|
|
mongoOps.insert(user);
|
|
user = new User();
|
|
user.setName("Antony");
|
|
user.setAge(55);
|
|
mongoOps.insert(user);
|
|
|
|
QUser qUser = new QUser("user");
|
|
|
|
Predicate predicate = qUser.name.eq("Eric");
|
|
List<User> users = (List<User>) userRepository.findAll(predicate);
|
|
|
|
assertThat(users.size(), is(1));
|
|
}
|
|
|
|
@Test
|
|
public void givenUsersExist_whenFindingUsersWithAgeCreaterThanAndLessThan_thenUsersAreFound() {
|
|
User user = new User();
|
|
user.setAge(20);
|
|
user.setName("Jon");
|
|
mongoOps.insert(user);
|
|
|
|
user = new User();
|
|
user.setAge(50);
|
|
user.setName("Jon");
|
|
mongoOps.insert(user);
|
|
|
|
user = new User();
|
|
user.setAge(33);
|
|
user.setName("Jim");
|
|
mongoOps.insert(user);
|
|
|
|
QUser qUser = new QUser("user");
|
|
|
|
Predicate predicate = qUser.age.between(26, 40);
|
|
List<User> users = (List<User>) userRepository.findAll(predicate);
|
|
|
|
assertThat(users.size(), is(1));
|
|
}
|
|
|
|
@Test
|
|
public void givenUsersExist_whenFindingUserWithNameStartWithA_thenUsersAreFound() {
|
|
User user = new User();
|
|
user.setName("Eric");
|
|
user.setAge(45);
|
|
mongoOps.insert(user);
|
|
|
|
user = new User();
|
|
user.setName("Antony");
|
|
user.setAge(33);
|
|
mongoOps.insert(user);
|
|
|
|
user = new User();
|
|
user.setName("Alice");
|
|
user.setAge(35);
|
|
mongoOps.insert(user);
|
|
|
|
QUser qUser = new QUser("user");
|
|
|
|
Predicate predicate = qUser.name.startsWith("A");
|
|
List<User> users = (List<User>) userRepository.findAll(predicate);
|
|
assertThat(users.size(), is(2));
|
|
}
|
|
|
|
@Test
|
|
public void givenUsersExist_whenFindingUserWithNameEndWithC_thenUsersAreFound() {
|
|
User user = new User();
|
|
user.setName("Eric");
|
|
user.setAge(45);
|
|
mongoOps.insert(user);
|
|
|
|
user = new User();
|
|
user.setName("Antony");
|
|
user.setAge(33);
|
|
mongoOps.insert(user);
|
|
|
|
user = new User();
|
|
user.setName("Alice");
|
|
user.setAge(35);
|
|
mongoOps.insert(user);
|
|
|
|
QUser qUser = new QUser("user");
|
|
Predicate predicate = qUser.name.endsWith("c");
|
|
List<User> users = (List<User>) userRepository.findAll(predicate);
|
|
|
|
assertThat(users.size(), is(1));
|
|
}
|
|
}
|