mirror of https://github.com/OpenIdentityPlatform/OpenDJ.git

Matthew Swift
24.34.2011 a94cb8c8073c31b911020e8565bbd4bd60b37bad
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at
 * trunk/opendj3/legal-notices/CDDLv1_0.txt
 * or http://forgerock.org/license/CDDLv1.0.html.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at
 * trunk/opendj3/legal-notices/CDDLv1_0.txt.  If applicable,
 * add the following below this CDDL HEADER, with the fields enclosed
 * by brackets "[]" replaced with your own identifying information:
 *      Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 *
 *
 *      Copyright 2009-2010 Sun Microsystems, Inc.
 */
 
package org.forgerock.opendj.ldap;
 
 
 
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
 
import org.forgerock.opendj.ldap.Connection;
import org.forgerock.opendj.ldap.ConnectionFactory;
import org.forgerock.opendj.ldap.Connections;
 
 
 
/**
 * This class defines some utility functions which can be used by test cases.
 */
public final class TestCaseUtils
{
  /**
   * The name of the system property that specifies the ldap port. Set this
   * property when running the server if you want to use a given port number,
   * otherwise a port is chosen randomly at test startup time.
   */
  public static final String PROPERTY_LDAP_PORT = "org.forgerock.opendj.test.LdapPort";
 
  /**
   * Port number that's used by the server. Need to be used by the test cases to
   * create connections.
   */
  public static final int port;
 
  static
  {
    final String ldapPort = System.getProperty(PROPERTY_LDAP_PORT);
    if (ldapPort != null)
    {
      port = Integer.valueOf(ldapPort);
    }
    else
    {
      port = findFreePort();
    }
  }
 
 
 
  /**
   * Creates a temporary text file with the specified contents. It will be
   * marked for automatic deletion when the JVM exits.
   *
   * @param lines
   *          The file contents.
   * @return The absolute path to the file that was created.
   * @throws Exception
   *           If an unexpected problem occurs.
   */
  public static String createTempFile(final String... lines) throws Exception
  {
    final File f = File.createTempFile("LDIFBasedTestCase", ".txt");
    f.deleteOnExit();
 
    final FileWriter w = new FileWriter(f);
    for (final String s : lines)
    {
      w.write(s + System.getProperty("line.separator"));
    }
 
    w.close();
 
    return f.getAbsolutePath();
  }
 
 
 
  /**
   * Finds a free server socket port on the local host.
   *
   * @return The free port.
   */
  public static int findFreePort()
  {
    try
    {
      ServerSocket serverLdapSocket = new ServerSocket();
      serverLdapSocket.setReuseAddress(true);
      serverLdapSocket.bind(new InetSocketAddress("127.0.0.1", 0));
      final int port = serverLdapSocket.getLocalPort();
      serverLdapSocket.close();
      return port;
    }
    catch (IOException e)
    {
      throw new RuntimeException(e);
    }
  }
 
 
 
  /**
   * Returns an internal client connection to the running ldap server.
   *
   * @return The internal client connection.
   * @throws Exception
   *           When an error occurs.
   */
  public static Connection getInternalConnection() throws Exception
  {
    startServer();
    final ConnectionFactory factory = Connections.newInternalConnectionFactory(
        LDAPServer.getInstance(), null);
    return factory.getConnection();
  }
 
 
 
  /**
   * Returns the port which the test server listens on.
   *
   * @return The LDAP port.
   */
  public static int getLdapPort()
  {
    return port;
  }
 
 
 
  /**
   * Starts the test ldap server.
   *
   * @throws Exception
   *           If an error occurs when starting the server.
   */
  public static void startServer() throws Exception
  {
    LDAPServer.getInstance().start(port);
  }
 
 
 
  /**
   * Stops the test ldap server.
   */
  public static void stopServer()
  {
    LDAPServer.getInstance().stop();
  }
}