1   package com.atlassian.user.search.page;
2   
3   import javax.naming.directory.InitialDirContext;
4   import javax.naming.directory.SearchControls;
5   import javax.naming.directory.SearchResult;
6   import javax.naming.NamingException;
7   import javax.naming.NamingEnumeration;
8   import java.util.List;
9   import java.util.ArrayList;
10  
11  public class MockLDAPContext extends InitialDirContext
12  {
13      private final List<? extends SearchResult> resultsToReturn;
14  
15      public MockLDAPContext(List<? extends SearchResult> resultsToReturn) throws NamingException
16      {
17          super();
18          this.resultsToReturn = resultsToReturn;
19      }
20  
21      public NamingEnumeration<SearchResult> search(String name, String filter, SearchControls cons) throws NamingException
22      {
23          return new MockNamingEnumeration<SearchResult>(new ArrayList<SearchResult>(resultsToReturn));
24      }
25  }