From 58c0aeb15a65324de08a914dfa62cc68a516a4e3 Mon Sep 17 00:00:00 2001
From: Runemoro
Date: Mon, 9 Mar 2020 06:04:08 -0400
Subject: CFR support (#192)

* Add decompiler API

* Add CFR support---
 src/test/java/cuchaz/enigma/TestDeobfuscator.java | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'src/test/java/cuchaz/enigma/TestDeobfuscator.java')

diff --git a/src/test/java/cuchaz/enigma/TestDeobfuscator.java b/src/test/java/cuchaz/enigma/TestDeobfuscator.java
index 2a6fec4..6619d26 100644
--- a/src/test/java/cuchaz/enigma/TestDeobfuscator.java
+++ b/src/test/java/cuchaz/enigma/TestDeobfuscator.java
@@ -11,13 +11,15 @@
 
 package cuchaz.enigma;
 
+import cuchaz.enigma.source.Decompiler;
+import cuchaz.enigma.source.Decompilers;
+import cuchaz.enigma.source.SourceSettings;
 import org.junit.Test;
 
 import java.io.IOException;
 import java.nio.file.Paths;
 
 public class TestDeobfuscator {
-
 	private EnigmaProject openProject() throws IOException {
 		Enigma enigma = Enigma.create();
 		return enigma.openJar(Paths.get("build/test-obf/loneClass.jar"), ProgressListener.none());
@@ -32,10 +34,8 @@ public class TestDeobfuscator {
 	@Test
 	public void decompileClass() throws Exception {
 		EnigmaProject project = openProject();
+		Decompiler decompiler = Decompilers.PROCYON.create(project.getClassCache(), new SourceSettings(false, false));
 
-		CompiledSourceTypeLoader typeLoader = new CompiledSourceTypeLoader(project.getClassCache());
-		SourceProvider sourceProvider = new SourceProvider(SourceProvider.createSettings(), typeLoader);
-
-		sourceProvider.writeSourceToString(sourceProvider.getSources("a"));
+		decompiler.getSource("a").asString();
 	}
 }
-- 
cgit v1.2.3