From 9defd2349274d077ad120d780a6fac5f6d594c7f Mon Sep 17 00:00:00 2001
From: Nicolas Capponi <nicolas.capponi@forgerock.com>
Date: Fri, 22 May 2015 15:58:46 +0000
Subject: [PATCH] OPENDJ-1631 CR-6864 Migrate server to SDK syntaxes

---
 opendj-server-legacy/src/test/java/org/opends/server/ServerContextBuilder.java |   40 ++++++++++++++++++++++++++++++++++++----
 1 files changed, 36 insertions(+), 4 deletions(-)

diff --git a/opendj-server-legacy/src/test/java/org/opends/server/ServerContextBuilder.java b/opendj-server-legacy/src/test/java/org/opends/server/ServerContextBuilder.java
index b1db2ac..a1d513b 100644
--- a/opendj-server-legacy/src/test/java/org/opends/server/ServerContextBuilder.java
+++ b/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);
+    }
+  }
+
 }

--
Gitblit v1.10.0