1   package com.atlassian.user.search.page;
2   
3   import com.atlassian.user.impl.ldap.repository.DefaultLdapContextFactory;
4   import com.atlassian.user.impl.RepositoryException;
5   
6   import javax.naming.directory.DirContext;
7   import javax.naming.directory.SearchResult;
8   import javax.naming.NamingException;
9   import java.util.List;
10  
11  public class MockLdapContextFactory extends DefaultLdapContextFactory
12  {
13      private final List<? extends SearchResult> resultsToReturn;
14  
15      public MockLdapContextFactory(List<? extends SearchResult> resultsToReturn)
16      {
17          super(null);
18          this.resultsToReturn = resultsToReturn;
19      }
20  
21      public DirContext getLDAPContext() throws RepositoryException
22      {
23          DirContext ctx = null;
24          try
25          {
26              ctx = new MockLDAPContext(resultsToReturn);
27          }
28          catch (NamingException e)
29          {
30              e.printStackTrace();
31          }
32          return ctx;
33      }
34  }