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

Nicolas Capponi
22.58.2015 9defd2349274d077ad120d780a6fac5f6d594c7f
opendj-server-legacy/src/test/java/org/opends/server/ServerContextBuilder.java
@@ -21,7 +21,7 @@
 * CDDL HEADER END
 *
 *
 *      Copyright 2014 ForgeRock AS.
 *      Copyright 2014-2015 ForgeRock AS.
 */
package org.opends.server;
@@ -30,6 +30,8 @@
import java.io.File;
import org.forgerock.opendj.config.server.ServerManagementContext;
import org.forgerock.opendj.ldap.schema.Schema;
import org.forgerock.opendj.ldap.schema.SchemaBuilder;
import org.opends.server.core.ConfigurationBootstrapper;
import org.opends.server.core.ServerContext;
import org.opends.server.schema.SchemaUpdater;
@@ -63,21 +65,27 @@
    return serverContext;
  }
  public ServerContextBuilder setSchemaDirectory(File path)
  public ServerContextBuilder schemaDirectory(File path)
      throws InitializationException
  {
    env.setSchemaDirectory(path);
    return this;
  }
  public ServerContextBuilder setConfigFile(File path)
  public ServerContextBuilder schemaNG(Schema schema)
  {
    when(serverContext.getSchemaNG()).thenReturn(schema);
    return this;
  }
  public ServerContextBuilder configFile(File path)
      throws InitializationException
  {
    env.setConfigFile(path);
    return this;
  }
  public ServerContextBuilder setSchemaUpdater(SchemaUpdater updater)
  public ServerContextBuilder schemaUpdater(SchemaUpdater updater)
  {
    when(serverContext.getSchemaUpdater()).thenReturn(updater);
    return this;
@@ -100,4 +108,28 @@
    return this;
  }
  /** A mock for schema updater. */
  public static final class MockSchemaUpdater implements SchemaUpdater
  {
    private Schema schema;
    public MockSchemaUpdater(Schema schema)
    {
      this.schema = schema;
    }
    @Override
    public boolean updateSchema(Schema schema)
    {
      this.schema = schema;
      return true;
    }
    @Override
    public SchemaBuilder getSchemaBuilder()
    {
      return new SchemaBuilder(schema);
    }
  }
}