added some basic tests for the deobufscator and the jar index

1 files changed, 9 insertions, 5 deletions

diff --git a/src/cuchaz/enigma/convert/ClassIdentity.java b/src/cuchaz/enigma/convert/ClassIdentity.java
index bd2824b..b3b043e 100644
--- a/src/cuchaz/enigma/convert/ClassIdentity.java
+++ b/src/cuchaz/enigma/convert/ClassIdentity.java
@@ -111,12 +111,16 @@ public class ClassIdentity
                 // stuff from the jar index
                 
                 m_implementations = HashMultiset.create();
-                @SuppressWarnings( "unchecked" )
-                Enumeration<ClassImplementationsTreeNode> implementations = index.getClassImplementations( null, m_classEntry ).children();
-                while( implementations.hasMoreElements() )
+                ClassImplementationsTreeNode implementationsNode = index.getClassImplementations( null, m_classEntry );
+                if( implementationsNode != null )
                 {
-                        ClassImplementationsTreeNode node = implementations.nextElement();
-                        m_implementations.add( scrubClassName( node.getClassEntry().getName() ) );
+                        @SuppressWarnings( "unchecked" )
+                        Enumeration<ClassImplementationsTreeNode> implementations = implementationsNode.children();
+                        while( implementations.hasMoreElements() )
+                        {
+                                ClassImplementationsTreeNode node = implementations.nextElement();
+                                m_implementations.add( scrubClassName( node.getClassEntry().getName() ) );
+                        }
                 }
                 
                 m_references = HashMultiset.create();