From a0fca062ddec8e587d05c8368c349ea779a7a969 Mon Sep 17 00:00:00 2001 From: Mark Paluch Date: Thu, 29 Mar 2018 13:35:51 +0200 Subject: [PATCH] DATALDAP-65 - Fix query execution without predicates using Querydsl. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Query execution without a predicate now no longer fails but falls back to LdapOperations.findAll(…). Previously, we did not initialize the filter of LdapQuery which caused an IllegalStateException in LdapOperations.find(…). --- .../repository/support/QueryDslLdapQuery.java | 18 ++++- .../support/QueryDslLdapQueryUnitTests.java | 78 +++++++++++++++++++ 2 files changed, 93 insertions(+), 3 deletions(-) create mode 100644 src/test/java/org/springframework/data/ldap/repository/support/QueryDslLdapQueryUnitTests.java diff --git a/src/main/java/org/springframework/data/ldap/repository/support/QueryDslLdapQuery.java b/src/main/java/org/springframework/data/ldap/repository/support/QueryDslLdapQuery.java index 1cc1831..f01ca32 100644 --- a/src/main/java/org/springframework/data/ldap/repository/support/QueryDslLdapQuery.java +++ b/src/main/java/org/springframework/data/ldap/repository/support/QueryDslLdapQuery.java @@ -20,6 +20,7 @@ import static org.springframework.ldap.query.LdapQueryBuilder.*; import java.util.List; import org.springframework.ldap.core.LdapOperations; +import org.springframework.ldap.filter.AbsoluteTrueFilter; import org.springframework.ldap.query.LdapQuery; import org.springframework.util.Assert; @@ -75,12 +76,22 @@ public class QueryDslLdapQuery implements FilteredClause */ @Override public QueryDslLdapQuery where(Predicate... o) { + + if (o == null) { + return this; + } return queryMixin.where(o); } @SuppressWarnings("unchecked") public List list() { - return (List) ldapOperations.find(buildQuery(), entityType); + + LdapQuery ldapQuery = buildQuery(); + if (ldapQuery.filter() instanceof AbsoluteTrueFilter) { + return (List) ldapOperations.findAll(entityType); + } + + return (List) ldapOperations.find(ldapQuery, entityType); } public K uniqueResult() { @@ -88,7 +99,8 @@ public class QueryDslLdapQuery implements FilteredClause } LdapQuery buildQuery() { - return query().filter(filterGenerator.handle(queryMixin.getMetadata().getWhere())); - } + Predicate where = queryMixin.getMetadata().getWhere(); + return where != null ? query().filter(filterGenerator.handle(where)) : query().filter(new AbsoluteTrueFilter()); + } } diff --git a/src/test/java/org/springframework/data/ldap/repository/support/QueryDslLdapQueryUnitTests.java b/src/test/java/org/springframework/data/ldap/repository/support/QueryDslLdapQueryUnitTests.java new file mode 100644 index 0000000..52d9869 --- /dev/null +++ b/src/test/java/org/springframework/data/ldap/repository/support/QueryDslLdapQueryUnitTests.java @@ -0,0 +1,78 @@ +/* + * Copyright 2018 the original author or authors. + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.springframework.data.ldap.repository.support; + +import static org.assertj.core.api.Assertions.*; +import static org.mockito.Mockito.*; + +import org.junit.Before; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.runners.MockitoJUnitRunner; +import org.springframework.ldap.core.LdapOperations; +import org.springframework.ldap.filter.AbsoluteTrueFilter; +import org.springframework.ldap.filter.EqualsFilter; +import org.springframework.ldap.odm.core.impl.DefaultObjectDirectoryMapper; +import org.springframework.ldap.query.LdapQuery; + +/** + * Unit tests for {@link QueryDslLdapQuery}. + * + * @author Mark Paluch + */ +@RunWith(MockitoJUnitRunner.class) +public class QueryDslLdapQueryUnitTests { + + @Mock LdapOperations ldapOperations; + + @Before + public void before() { + when(ldapOperations.getObjectDirectoryMapper()).thenReturn(new DefaultObjectDirectoryMapper()); + } + + @Test // DATALDAP-65 + public void shouldCreateFilter() { + + QueryDslLdapQuery query = new QueryDslLdapQuery(ldapOperations, + UnitTestPerson.class); + + LdapQuery ldapQuery = query.where(QPerson.person.fullName.eq("foo")).buildQuery(); + + assertThat(ldapQuery.filter()).isInstanceOf(EqualsFilter.class); + } + + @Test // DATALDAP-65 + public void shouldCreateEmptyFilter() { + + QueryDslLdapQuery query = new QueryDslLdapQuery(ldapOperations, + UnitTestPerson.class); + + LdapQuery ldapQuery = query.buildQuery(); + + assertThat(ldapQuery.filter()).isInstanceOf(AbsoluteTrueFilter.class); + } + + @Test // DATALDAP-65 + public void shouldCreateEmptyFilterFromWhereNull() { + + QueryDslLdapQuery query = new QueryDslLdapQuery(ldapOperations, QPerson.person); + + LdapQuery ldapQuery = query.where(null).buildQuery(); + + assertThat(ldapQuery.filter()).isInstanceOf(AbsoluteTrueFilter.class); + } +}