add rsql parser test

2015-04-09 20:31:05 +02:00
parent 7a61d8e790
commit 6e34100ae6
7 changed files with 340 additions and 1 deletions
@@ -0,0 +1,105 @@
+package org.baeldung.persistence.query;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hamcrest.collection.IsIn.isIn;
+import static org.hamcrest.core.IsNot.not;
+
+import java.util.List;
+
+import org.baeldung.persistence.dao.UserRepository;
+import org.baeldung.persistence.dao.rsql.CustomRsqlVisitor;
+import org.baeldung.persistence.model.User;
+import org.baeldung.spring.PersistenceConfig;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.data.jpa.domain.Specification;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+import org.springframework.test.context.transaction.TransactionConfiguration;
+import org.springframework.transaction.annotation.Transactional;
+
+import cz.jirutka.rsql.parser.RSQLParser;
+import cz.jirutka.rsql.parser.ast.Node;
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration(classes = { PersistenceConfig.class })
+@Transactional
+@TransactionConfiguration
+public class RsqlTest {
+
+    @Autowired
+    private UserRepository repository;
+
+    private User userJohn;
+
+    private User userTom;
+
+    @Before
+    public void init() {
+        userJohn = new User();
+        userJohn.setFirstName("john");
+        userJohn.setLastName("doe");
+        userJohn.setEmail("john@doe.com");
+        userJohn.setAge(22);
+        repository.save(userJohn);
+
+        userTom = new User();
+        userTom.setFirstName("tom");
+        userTom.setLastName("doe");
+        userTom.setEmail("tom@doe.com");
+        userTom.setAge(26);
+        repository.save(userTom);
+    }
+
+    @Test
+    public void givenFirstAndLastName_whenGettingListOfUsers_thenCorrect() {
+        final Node rootNode = new RSQLParser().parse("firstName==john;lastName==doe");
+        final Specification<User> spec = rootNode.accept(new CustomRsqlVisitor<User>());
+        final List<User> results = repository.findAll(spec);
+
+        assertThat(userJohn, isIn(results));
+        assertThat(userTom, not(isIn(results)));
+    }
+
+    @Test
+    public void givenFirstNameInverse_whenGettingListOfUsers_thenCorrect() {
+        final Node rootNode = new RSQLParser().parse("firstName!=john");
+        final Specification<User> spec = rootNode.accept(new CustomRsqlVisitor<User>());
+        final List<User> results = repository.findAll(spec);
+
+        assertThat(userTom, isIn(results));
+        assertThat(userJohn, not(isIn(results)));
+    }
+
+    @Test
+    public void givenMinAge_whenGettingListOfUsers_thenCorrect() {
+        final Node rootNode = new RSQLParser().parse("age>25");
+        final Specification<User> spec = rootNode.accept(new CustomRsqlVisitor<User>());
+        final List<User> results = repository.findAll(spec);
+
+        assertThat(userTom, isIn(results));
+        assertThat(userJohn, not(isIn(results)));
+    }
+
+    @Test
+    public void givenFirstNamePrefix_whenGettingListOfUsers_thenCorrect() {
+        final Node rootNode = new RSQLParser().parse("firstName==jo*");
+        final Specification<User> spec = rootNode.accept(new CustomRsqlVisitor<User>());
+        final List<User> results = repository.findAll(spec);
+
+        assertThat(userJohn, isIn(results));
+        assertThat(userTom, not(isIn(results)));
+    }
+
+    @Test
+    public void givenListOfFirstName_whenGettingListOfUsers_thenCorrect() {
+        final Node rootNode = new RSQLParser().parse("firstName=in=(john,jack)");
+        final Specification<User> spec = rootNode.accept(new CustomRsqlVisitor<User>());
+        final List<User> results = repository.findAll(spec);
+
+        assertThat(userJohn, isIn(results));
+        assertThat(userTom, not(isIn(results)));
+    }
+}