|
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114 |
- /*
- * SonarQube
- * Copyright (C) 2009-2021 SonarSource SA
- * mailto:info AT sonarsource DOT com
- *
- * This program is free software; you can redistribute it and/or
- * modify it under the terms of the GNU Lesser General Public
- * License as published by the Free Software Foundation; either
- * version 3 of the License, or (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- * Lesser General Public License for more details.
- *
- * You should have received a copy of the GNU Lesser General Public License
- * along with this program; if not, write to the Free Software Foundation,
- * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
- */
- package org.sonar.auth.ldap;
-
- import java.util.ArrayList;
- import java.util.Enumeration;
- import java.util.Map;
- import javax.naming.NamingException;
- import javax.naming.directory.SearchControls;
- import org.junit.BeforeClass;
- import org.junit.ClassRule;
- import org.junit.Test;
- import org.sonar.auth.ldap.server.LdapServer;
-
- import static org.assertj.core.api.Assertions.assertThat;
- import static org.assertj.core.api.Assertions.assertThatThrownBy;
-
- public class LdapSearchTest {
-
- @ClassRule
- public static LdapServer server = new LdapServer("/users.example.org.ldif");
- private static Map<String, LdapContextFactory> contextFactories;
-
- @BeforeClass
- public static void init() {
- contextFactories = new LdapSettingsManager(LdapSettingsFactory.generateSimpleAnonymousAccessSettings(server, null).asConfig(), new LdapAutodiscovery()).getContextFactories();
- }
-
- @Test
- public void subtreeSearch() throws Exception {
- LdapSearch search = new LdapSearch(contextFactories.values().iterator().next())
- .setBaseDn("dc=example,dc=org")
- .setRequest("(objectClass={0})")
- .setParameters("inetOrgPerson")
- .returns("objectClass");
-
- assertThat(search.getBaseDn()).isEqualTo("dc=example,dc=org");
- assertThat(search.getScope()).isEqualTo(SearchControls.SUBTREE_SCOPE);
- assertThat(search.getRequest()).isEqualTo("(objectClass={0})");
- assertThat(search.getParameters()).isEqualTo(new String[] {"inetOrgPerson"});
- assertThat(search.getReturningAttributes()).isEqualTo(new String[] {"objectClass"});
- assertThat(search.toString()).isEqualTo("LdapSearch{baseDn=dc=example,dc=org, scope=subtree, request=(objectClass={0}), parameters=[inetOrgPerson], attributes=[objectClass]}");
- assertThat(enumerationToArrayList(search.find()).size()).isEqualTo(3);
-
- assertThatThrownBy(() -> search.findUnique())
- .isInstanceOf(NamingException.class)
- .hasMessage("Non unique result for " + search.toString());
- }
-
- @Test
- public void oneLevelSearch() throws Exception {
- LdapSearch search = new LdapSearch(contextFactories.values().iterator().next())
- .setBaseDn("dc=example,dc=org")
- .setScope(SearchControls.ONELEVEL_SCOPE)
- .setRequest("(objectClass={0})")
- .setParameters("inetOrgPerson")
- .returns("cn");
-
- assertThat(search.getBaseDn()).isEqualTo("dc=example,dc=org");
- assertThat(search.getScope()).isEqualTo(SearchControls.ONELEVEL_SCOPE);
- assertThat(search.getRequest()).isEqualTo("(objectClass={0})");
- assertThat(search.getParameters()).isEqualTo(new String[] {"inetOrgPerson"});
- assertThat(search.getReturningAttributes()).isEqualTo(new String[] {"cn"});
- assertThat(search.toString()).isEqualTo("LdapSearch{baseDn=dc=example,dc=org, scope=onelevel, request=(objectClass={0}), parameters=[inetOrgPerson], attributes=[cn]}");
- assertThat(enumerationToArrayList(search.find()).size()).isZero();
- assertThat(search.findUnique()).isNull();
- }
-
- @Test
- public void objectSearch() throws Exception {
- LdapSearch search = new LdapSearch(contextFactories.values().iterator().next())
- .setBaseDn("cn=bind,ou=users,dc=example,dc=org")
- .setScope(SearchControls.OBJECT_SCOPE)
- .setRequest("(objectClass={0})")
- .setParameters("uidObject")
- .returns("uid");
-
- assertThat(search.getBaseDn()).isEqualTo("cn=bind,ou=users,dc=example,dc=org");
- assertThat(search.getScope()).isEqualTo(SearchControls.OBJECT_SCOPE);
- assertThat(search.getRequest()).isEqualTo("(objectClass={0})");
- assertThat(search.getParameters()).isEqualTo(new String[] {"uidObject"});
- assertThat(search.getReturningAttributes()).isEqualTo(new String[] {"uid"});
- assertThat(search.toString()).isEqualTo(
- "LdapSearch{baseDn=cn=bind,ou=users,dc=example,dc=org, scope=object, request=(objectClass={0}), parameters=[uidObject], attributes=[uid]}");
- assertThat(enumerationToArrayList(search.find()).size()).isEqualTo(1);
- assertThat(search.findUnique()).isNotNull();
- }
-
- private static <E> ArrayList<E> enumerationToArrayList(Enumeration<E> enumeration) {
- ArrayList<E> result = new ArrayList<>();
- while (enumeration.hasMoreElements()) {
- result.add(enumeration.nextElement());
- }
- return result;
- }
-
- }
|