started unit testing for inner/anonymous class detection

1 files changed, 18 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/mapping/SignatureUpdater.java b/src/cuchaz/enigma/mapping/SignatureUpdater.java
index 4c0dbac..528a743 100644
--- a/src/cuchaz/enigma/mapping/SignatureUpdater.java
+++ b/src/cuchaz/enigma/mapping/SignatureUpdater.java
@@ -12,6 +12,9 @@ package cuchaz.enigma.mapping;
 
 import java.io.IOException;
 import java.io.StringReader;
+import java.util.List;
+
+import com.beust.jcommander.internal.Lists;
 
 public class SignatureUpdater
 {
@@ -84,4 +87,19 @@ public class SignatureUpdater
                 
                 return null;
         }
+        
+        public static List<String> getClasses( String signature )
+        {
+                final List<String> classNames = Lists.newArrayList();
+                update( signature, new ClassNameUpdater( )
+                {
+                        @Override
+                        public String update( String className )
+                        {
+                                classNames.add( className );
+                                return className;
+                        }
+                } );
+                return classNames;
+        }
 }