1 files changed, 51 insertions, 0 deletions

diff --git a/test/cuchaz/enigma/TestDeobfuscator.java b/test/cuchaz/enigma/TestDeobfuscator.java
new file mode 100644
index 0000000..3310fbc
--- /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" ) );
+        }
+}