fix names

1 files changed, 6 insertions, 5 deletions

diff --git a/ex1/src/main/java/reflection/api/ReflectionInvestigator.java b/ex1/src/main/java/reflection/api/ReflectionInvestigator.java
index 160102a..03e6dab 100644
--- a/ex1/src/main/java/reflection/api/ReflectionInvestigator.java
+++ b/ex1/src/main/java/reflection/api/ReflectionInvestigator.java
@@ -2,6 +2,7 @@ package reflection.api;
 
 import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Modifier;
+import java.util.HashSet;
 import java.util.Set;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
@@ -36,7 +37,7 @@ public class ReflectionInvestigator implements Investigator {
   @Override
   public Set<String> getAllImplementedInterfaces() {
     return Stream.of(classInfo.getInterfaces())
-        .map(info -> info.getName())
+        .map(info -> info.getSimpleName())
         .collect(Collectors.toSet());
   }
 
@@ -116,13 +117,13 @@ public class ReflectionInvestigator implements Investigator {
 
   @Override
   public String getInheritanceChain(String delimiter) {
-    var sb = new StringBuilder();
-    sb.append(classInfo.getName());
+    var nameStack = new HashSet<String>();
+    nameStack.add(classInfo.getSimpleName());
     var cursor = classInfo;
     while (cursor != null) {
-      sb.append(cursor.getName() + " -> ");
+      nameStack.add(cursor.getSimpleName());
       cursor = cursor.getSuperclass();
     }
-    return sb.toString();
+    return String.join(" >> ", nameStack);
   }
 }