From ef7209f04fd0b1d9e06e8d09ecf7eef094f64608 Mon Sep 17 00:00:00 2001
From: Jean-Noel Rouvignac <jean-noel.rouvignac@forgerock.com>
Date: Fri, 03 Jan 2014 13:31:58 +0000
Subject: [PATCH] ImportLDIFTestCase.java: Code cleanup.

---
 opends/tests/unit-tests-testng/src/server/org/opends/server/tools/ImportLDIFTestCase.java |  173 ++++++++++++---------------------------------------------
 1 files changed, 36 insertions(+), 137 deletions(-)

diff --git a/opends/tests/unit-tests-testng/src/server/org/opends/server/tools/ImportLDIFTestCase.java b/opends/tests/unit-tests-testng/src/server/org/opends/server/tools/ImportLDIFTestCase.java
index 9784ebe..96d80dc 100644
--- a/opends/tests/unit-tests-testng/src/server/org/opends/server/tools/ImportLDIFTestCase.java
+++ b/opends/tests/unit-tests-testng/src/server/org/opends/server/tools/ImportLDIFTestCase.java
@@ -22,23 +22,13 @@
  *
  *
  *      Copyright 2006-2008 Sun Microsystems, Inc.
- *      Portions copyright 2013 ForgeRock AS.
+ *      Portions copyright 2013-2014 ForgeRock AS.
  */
 package org.opends.server.tools;
 
-
-
-import static org.opends.server.TestCaseUtils.readFile;
-import static org.testng.Assert.assertEquals;
-import static org.testng.Assert.assertFalse;
-import static org.testng.Assert.assertNotNull;
-import static org.testng.Assert.assertTrue;
-import static org.testng.Assert.fail;
-
 import java.io.File;
 import java.io.FileOutputStream;
 import java.io.PrintStream;
-import java.util.List;
 
 import org.opends.server.TestCaseUtils;
 import org.opends.server.admin.server.ServerManagementContext;
@@ -53,31 +43,22 @@
 import org.testng.annotations.BeforeClass;
 import org.testng.annotations.Test;
 
-
+import static org.opends.server.TestCaseUtils.*;
+import static org.testng.Assert.*;
 
 @SuppressWarnings("javadoc")
 public class ImportLDIFTestCase extends ToolsTestCase
 {
 
   private File tempDir;
-
   private String ldifFilePath;
-
   private String configFilePath;
-
   private String homeDirName;
-
   private String beID;
-
   private String baseDN = "dc=example,dc=com";
 
-
-
   /**
    * Ensures that the ldif file is created with the entry.
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
   @BeforeClass
   public void setUp() throws Exception
@@ -114,9 +95,6 @@
   /**
    * Tests an import of LDIF with only the operational attributes
    * included.
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
   @Test
   public void testImportIncludeOnlyOperational() throws Exception
@@ -148,10 +126,7 @@
 
 
   /**
-   * Tests an import of LDIF with only thel user attributes included.
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
+   * Tests an import of LDIF with only the user attributes included.
    */
   @Test
   public void testImportIncludeOnlyUser() throws Exception
@@ -179,13 +154,10 @@
     // Expecting an empty reject file.
     assertRejectedFile(reject, true);
 
-    Attribute[] opAttr =
-    {
-        Attributes.create("creatorsname", "cn=Import"),
-        Attributes.create("modifiersname", "cn=Import")
-    };
     // operational attributes shouldn't be present.
-    assertEntry(opAttr, false);
+    assertEntry(false,
+        Attributes.create("creatorsname", "cn=Import"),
+        Attributes.create("modifiersname", "cn=Import"));
   }
 
 
@@ -194,9 +166,6 @@
    * Tests a simple Import LDIF with none of the attributes excluded
    * or included. It is expected to import the entry(ies) with all the
    * attributes in the ldif file.
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
   @Test
   public void testImportDefault() throws Exception
@@ -222,15 +191,11 @@
     // Reject file should be empty.
     assertRejectedFile(reject, true);
     // check the presence of some random attributes.
-    Attribute[] attr =
-    {
-        Attributes.create("description",
-            "This is the description for Aaccf Amar"),
+    assertEntry(true,
+        Attributes.create("description", "This is the description for Aaccf Amar"),
         Attributes.create("mail", "user.0@example.com"),
         Attributes.create("creatorsname", "cn=Import"),
-        Attributes.create("modifiersname", "cn=Import")
-    };
-    assertEntry(attr, true);
+        Attributes.create("modifiersname", "cn=Import"));
   }
 
 
@@ -239,9 +204,6 @@
    * Tests a simple Import LDIF using base DN with none of the
    * attributes excluded or included. It is expected to import the
    * entry(ies) with all the attributes in the ldif file.
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
   @Test
   public void testImportDefaultBaseDN() throws Exception
@@ -267,15 +229,11 @@
     // Reject file should be empty.
     assertRejectedFile(reject, true);
     // check the presence of some random attributes.
-    Attribute[] attr =
-    {
-        Attributes.create("description",
-            "This is the description for Aaccf Amar"),
+    assertEntry(true,
+        Attributes.create("description", "This is the description for Aaccf Amar"),
         Attributes.create("mail", "user.0@example.com"),
         Attributes.create("creatorsname", "cn=Import"),
-        Attributes.create("modifiersname", "cn=Import")
-    };
-    assertEntry(attr, true);
+        Attributes.create("modifiersname", "cn=Import"));
   }
 
 
@@ -283,9 +241,6 @@
   /**
    * Tests an import of LDIF with all the user attributes included but
    * "description"
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
   @Test
   public void testImportLDIFIncludeUserExcludeDescription() throws Exception
@@ -314,21 +269,14 @@
     assertEquals(
         ImportLDIF.mainImportLDIF(args, false, System.out, System.err), 0);
     assertRejectedFile(reject, true);
-    Attribute[] attr =
-    {
-      Attributes.create("description",
-          "This is the description for Aaccf Amar")
-    };
-    assertEntry(attr, false);
+    assertEntry(false,
+        Attributes.create("description", "This is the description for Aaccf Amar"));
   }
 
 
 
   /**
    * Tests an import of LDIF with all user attributes excluded option.
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
   @Test
   public void testImportLDIFExcludeUser() throws Exception
@@ -361,9 +309,6 @@
   /**
    * Tests an import of LDIF with all the operational attributes
    * excluded option.
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
   @Test
   public void testImportLDIFExcludeOperational() throws Exception
@@ -389,12 +334,9 @@
     assertEquals(
         ImportLDIF.mainImportLDIF(args, false, System.out, System.err), 0);
     assertRejectedFile(reject, true);
-    Attribute[] attrs =
-    {
+    assertEntry(false,
         Attributes.create("creatorsname", "cn=Import"),
-        Attributes.create("modifiersname", "cn=Import")
-    };
-    assertEntry(attrs, false);
+        Attributes.create("modifiersname", "cn=Import"));
   }
 
 
@@ -402,9 +344,6 @@
   /**
    * Tests an import of LDIF with all user attributes and one
    * operational attribute included..
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
   @Test
   public void testImportLDIFUserAndOperational() throws Exception
@@ -432,11 +371,7 @@
     assertEquals(
         ImportLDIF.mainImportLDIF(args, false, System.out, System.err), 0);
     assertRejectedFile(reject, true);
-    Attribute[] attrs =
-    {
-      Attributes.create("creatorsname", "cn=Import")
-    };
-    assertEntry(attrs, true);
+    assertEntry(true, Attributes.create("creatorsname", "cn=Import"));
   }
 
 
@@ -444,9 +379,6 @@
   /**
    * Tests an import of LDIF with select user and operational
    * attributes included..
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
   @Test
   public void testImportLDIFSelectiveIncludeAttributes() throws Exception
@@ -480,27 +412,18 @@
     assertEquals(
         ImportLDIF.mainImportLDIF(args, false, System.out, System.err), 0);
     assertRejectedFile(reject, true);
-    Attribute[] attrsPr =
-    {
-      Attributes.create("creatorsname", "cn=Import")
-    };
-    assertEntry(attrsPr, true);
-    Attribute[] attrsAb =
-    {
+    assertEntry(true,
+        Attributes.create("creatorsname", "cn=Import"));
+    assertEntry(false,
         Attributes.create("givenname", "Aaccf"),
-        Attributes.create("employeenumber", "0")
-    };
-    assertEntry(attrsAb, false);
+        Attributes.create("employeenumber", "0"));
   }
 
 
 
   /**
-   * Tests an import of LDIF with select user and operational
-   * attributes encluded..
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
+   * Tests an import of LDIF with select user and operational attributes
+   * included.
    */
   @Test
   public void testImportLDIFSelectiveExcludeAttributes() throws Exception
@@ -527,18 +450,12 @@
     assertEquals(
         ImportLDIF.mainImportLDIF(args, false, System.out, System.err), 0);
     assertRejectedFile(reject, true);
-    Attribute[] attrsPr =
-    {
+    assertEntry(true,
         Attributes.create("modifiersname", "cn=Import"),
-        Attributes.create("employeenumber", "0")
-    };
-    assertEntry(attrsPr, true);
-    Attribute[] attrsAb =
-    {
+        Attributes.create("employeenumber", "0"));
+    assertEntry(false,
         Attributes.create("creatorsname", "cn=Import"),
-        Attributes.create("givenname", "Aaccf")
-    };
-    assertEntry(attrsAb, false);
+        Attributes.create("givenname", "Aaccf"));
   }
 
 
@@ -551,8 +468,6 @@
    *          The file to be asserted
    * @param shouldBeEmpty
    *          whether the file should be empty.
-   * @throws Exception
-   *           If an unexpected error occurred.
    */
   private void assertRejectedFile(File reject, boolean shouldBeEmpty) throws Exception
   {
@@ -583,48 +498,32 @@
 
 
   /**
-   * Utility method which is called by the testcase for asserting the
-   * imported entry.
-   *
+   * Utility method which is called by the testcase for asserting the imported
+   * entry.
+   * @param attrsShouldExistInEntry
+   *          the boolean flag for assert type.
    * @param attrs
    *          The array of attributes to be asserted.
-   * @param assertType
-   *          the boolean flag for assert type.
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
-  private void assertEntry(Attribute[] attrs, boolean assertType)
+  private void assertEntry(boolean attrsShouldExistInEntry, Attribute... attrs)
       throws Exception
   {
     if (attrs != null && attrs.length > 0)
     {
       TaskUtils.enableBackend(beID);
-      Entry entry = DirectoryServer.getEntry(DN
-          .decode(" uid=user.0,dc=example,dc=com"));
+      Entry entry = DirectoryServer.getEntry(
+          DN.decode(" uid=user.0,dc=example,dc=com"));
       TaskUtils.disableBackend(beID);
       assertNotNull(entry);
-      List<Attribute> list = entry.getAttributes();
       for (Attribute a : attrs)
       {
-        if (assertType)
-        {
-          assertTrue(list.contains(a));
-        }
-        else
-        {
-          assertFalse(list.contains(a));
-        }
+        assertEquals(entry.getAttributes().contains(a), attrsShouldExistInEntry);
       }
     }
   }
 
-
-
   /**
    * Clean up method.
-   *
-   * @throws Exception
-   *           If an unexpected problem occurs.
    */
   @AfterClass
   public void cleanUp() throws Exception

--
Gitblit v1.10.0