addendum - add test

Author: virajjasani

hadoop-common-project/hadoop-common/src/test/java/org/apache/hadoop/util/TestClassUtil.java

@@ -20,21 +20,41 @@
 
 import java.io.File;
 
-import org.junit.Assert;
-
+import org.apache.hadoop.fs.viewfs.ViewFileSystem;
 import org.apache.log4j.Logger;
 import org.junit.Test;
 
+import static org.assertj.core.api.Assertions.assertThat;
+
 public class TestClassUtil {
   @Test(timeout=10000)
   public void testFindContainingJar() {
     String containingJar = ClassUtil.findContainingJar(Logger.class);
-    Assert.assertNotNull("Containing jar not found for Logger", 
-        containingJar);
+    assertThat(containingJar)
+        .describedAs("Containing jar for %s", Logger.class)
+        .isNotNull();
     File jarFile = new File(containingJar);
-    Assert.assertTrue("Containing jar does not exist on file system ",
-        jarFile.exists());
-    Assert.assertTrue("Incorrect jar file " + containingJar,
-        jarFile.getName().matches("log4j.*[.]jar"));
+    assertThat(jarFile)
+        .describedAs("Containing jar %s", jarFile)
+        .exists();
+    assertThat(jarFile.getName())
+        .describedAs("Containing jar name %s", jarFile.getName())
+        .matches("log4j.*[.]jar");
   }
+
+  @Test(timeout = 10000)
+  public void testFindContainingClass() {
+    String classFileLocation = ClassUtil.findClassLocation(ViewFileSystem.class);
+    assertThat(classFileLocation)
+        .describedAs("Class path for %s", ViewFileSystem.class)
+        .isNotNull();
+    File classFile = new File(classFileLocation);
+    assertThat(classFile)
+        .describedAs("Containing class file %s", classFile)
+        .exists();
+    assertThat(classFile.getName())
+        .describedAs("Containing class file name %s", classFile.getName())
+        .matches("ViewFileSystem.class");
+  }
+
 }