Use assertThatClasses(..) instead of assertThat(namesOf(..))

TNG · May 19, 2017 · 0df1aa7 · 0df1aa7
1 parent 40e3301
commit 0df1aa7
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/archunit/src/test/java/com/tngtech/archunit/core/importer/ClassFileImporterTest.java b/archunit/src/test/java/com/tngtech/archunit/core/importer/ClassFileImporterTest.java
@@ -231,18 +231,15 @@ public void imports_simple_class_details() throws Exception {
 
     @Test
     public void imports_simple_enum() throws Exception {
-        ImportedClasses classes = classesIn("testexamples/simpleimport");
-        JavaClass javaClass = classes.get(EnumToImport.class);
+        JavaClass javaClass = classesIn("testexamples/simpleimport").get(EnumToImport.class);
 
         assertThat(javaClass.getName()).as("full name").isEqualTo(EnumToImport.class.getName());
         assertThat(javaClass.getSimpleName()).as("simple name").isEqualTo(EnumToImport.class.getSimpleName());
         assertThat(javaClass.getPackage()).as("package").isEqualTo(EnumToImport.class.getPackage().getName());
         assertThat(javaClass.getModifiers()).as("modifiers").containsOnly(JavaModifier.PUBLIC, JavaModifier.FINAL);
         assertThat(javaClass.getSuperClass().get()).as("super class").matches(Enum.class);
         assertThat(javaClass.getInterfaces()).as("interfaces").isEmpty();
-        Set<JavaClass> allInterfaces = javaClass.getAllInterfaces();
-        assertThat(namesOf(allInterfaces)).as("all interfaces names")
-            .containsExactlyInAnyOrder(Serializable.class.getName(), Comparable.class.getName());
+        assertThatClasses(javaClass.getAllInterfaces()).matchInAnyOrder(Serializable.class, Comparable.class);
         assertThat(javaClass.isInterface()).as("is interface").isFalse();
         assertThat(javaClass.isEnum()).as("is enum").isTrue();
     }