From 5be409c32375900b359ffabfb79cd117442eb6ea Mon Sep 17 00:00:00 2001
From: Jean-Noel Rouvignac <jean-noel.rouvignac@forgerock.com>
Date: Fri, 19 Dec 2014 15:49:17 +0000
Subject: [PATCH] OPENDJ-1602 New pluggable storage based backend

---
 opendj-sdk/opendj3-server-dev/tests/unit-tests-testng/src/server/org/opends/server/TestCaseUtils.java |   49 +++++++++++++++++++++++++++++++++++++------------
 1 files changed, 37 insertions(+), 12 deletions(-)

diff --git a/opendj-sdk/opendj3-server-dev/tests/unit-tests-testng/src/server/org/opends/server/TestCaseUtils.java b/opendj-sdk/opendj3-server-dev/tests/unit-tests-testng/src/server/org/opends/server/TestCaseUtils.java
index e3716de..cfb6dba 100644
--- a/opendj-sdk/opendj3-server-dev/tests/unit-tests-testng/src/server/org/opends/server/TestCaseUtils.java
+++ b/opendj-sdk/opendj3-server-dev/tests/unit-tests-testng/src/server/org/opends/server/TestCaseUtils.java
@@ -878,23 +878,48 @@
    */
   public static void clearJEBackend(String backendId, String baseDN) throws Exception
   {
-    final BackendImpl backend = (BackendImpl)DirectoryServer.getBackend(backendId);
-    final RootContainer rootContainer = backend.getRootContainer();
-    if (rootContainer != null)
+    Backend<?> b = DirectoryServer.getBackend(backendId);
+    if (clearBackend(b) && baseDN != null)
     {
-      for (EntryContainer ec : rootContainer.getEntryContainers())
-      {
-        ec.clear();
-        assertEquals(ec.getHighestEntryID().longValue(), 0L);
-      }
-      rootContainer.resetNextEntryID();
+      Entry e = createEntry(DN.valueOf(baseDN));
+      DirectoryServer.getBackend(backendId).addEntry(e, null);
+    }
+  }
 
-      if (baseDN != null)
+  private static boolean clearBackend(Backend<?> b)
+  {
+    if (b instanceof BackendImpl)
+    {
+      final BackendImpl backend = (BackendImpl) b;
+      final RootContainer rootContainer = backend.getRootContainer();
+      if (rootContainer != null)
       {
-        Entry e = createEntry(DN.valueOf(baseDN));
-        DirectoryServer.getBackend(backendId).addEntry(e, null);
+        for (EntryContainer ec : rootContainer.getEntryContainers())
+        {
+          ec.clear();
+          assertEquals(ec.getHighestEntryID().longValue(), 0L);
+        }
+        rootContainer.resetNextEntryID();
+        return true;
       }
     }
+    else if (b instanceof org.opends.server.backends.pluggable.BackendImpl)
+    {
+      final org.opends.server.backends.pluggable.BackendImpl backend =
+          (org.opends.server.backends.pluggable.BackendImpl) b;
+      final org.opends.server.backends.pluggable.RootContainer rootContainer = backend.getRootContainer();
+      if (rootContainer != null)
+      {
+        for (org.opends.server.backends.pluggable.EntryContainer ec : rootContainer.getEntryContainers())
+        {
+          ec.clear();
+          // assertEquals(ec.getHighestEntryID().longValue(), 0L);
+        }
+        rootContainer.resetNextEntryID();
+        return true;
+      }
+    }
+    return false;
   }
 
   /**

--
Gitblit v1.10.0