3 files changed, 76 insertions, 0 deletions

diff --git a/test/cuchaz/enigma/TestDeobfuscator.java b/test/cuchaz/enigma/TestDeobfuscator.java
new file mode 100644
index 00000000..3310fbcc
--- /dev/null
+++ b/test/cuchaz/enigma/TestDeobfuscator.java
@@ -0,0 +1,51 @@
+package cuchaz.enigma;
+
+import static org.junit.Assert.*;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.List;
+
+import org.junit.Test;
+
+import com.beust.jcommander.internal.Lists;
+
+import cuchaz.enigma.mapping.ClassEntry;
+
+public class TestDeobfuscator
+{
+        private Deobfuscator getDeobfuscator( )
+        throws IOException
+        {
+                return new Deobfuscator( new File( "build/libs/testCases.obf.jar" ) );
+        }
+        
+        @Test
+        public void loadJar( )
+        throws Exception
+        {
+                getDeobfuscator();
+        }
+        
+        @Test
+        public void getClasses( )
+        throws Exception
+        {
+                Deobfuscator deobfuscator = getDeobfuscator();
+                List<ClassEntry> obfClasses = Lists.newArrayList();
+                List<ClassEntry> deobfClasses = Lists.newArrayList();
+                deobfuscator.getSeparatedClasses( obfClasses, deobfClasses );
+                assertEquals( 1, obfClasses.size() );
+                assertEquals( "none/a", obfClasses.get( 0 ).getName() );
+                assertEquals( 1, deobfClasses.size() );
+                assertEquals( "cuchaz/enigma/inputs/Keep", deobfClasses.get( 0 ).getName() );
+        }
+        
+        @Test
+        public void decompileClass( )
+        throws Exception
+        {
+                Deobfuscator deobfuscator = getDeobfuscator();
+                deobfuscator.getSource( deobfuscator.getSourceTree( "none/a" ) );
+        }
+}
diff --git a/test/cuchaz/enigma/inputs/Keep.java b/test/cuchaz/enigma/inputs/Keep.java
new file mode 100644
index 00000000..3c12baea
--- /dev/null
+++ b/test/cuchaz/enigma/inputs/Keep.java
@@ -0,0 +1,9 @@
+package cuchaz.enigma.inputs;
+
+public class Keep
+{
+        public static void main( String[] args )
+        {
+                System.out.println( "Keep me!" );
+        }
+}
diff --git a/test/cuchaz/enigma/inputs/LoneClass.java b/test/cuchaz/enigma/inputs/LoneClass.java
new file mode 100644
index 00000000..a3d8cded
--- /dev/null
+++ b/test/cuchaz/enigma/inputs/LoneClass.java
@@ -0,0 +1,16 @@
+package cuchaz.enigma.inputs;
+
+public class LoneClass
+{
+        private String m_name;
+        
+        public LoneClass( String name )
+        {
+                m_name = name;
+        }
+        
+        public String getName( )
+        {
+                return m_name;
+        }
+}