test/cuchaz/enigma/TestTranslator.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

package cuchaz.enigma;

import static cuchaz.enigma.EntryFactory.*;
import static org.hamcrest.MatcherAssert.*;
import static org.hamcrest.Matchers.*;

import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.jar.JarFile;

import org.junit.Test;

import cuchaz.enigma.mapping.Mappings;
import cuchaz.enigma.mapping.MappingsReader;
import cuchaz.enigma.mapping.TranslationDirection;
import cuchaz.enigma.mapping.Translator;


public class TestTranslator {

	private Deobfuscator m_deobfuscator;
	private Mappings m_mappings;
	
	public TestTranslator()
	throws Exception {
		m_deobfuscator = new Deobfuscator(new JarFile("build/testTranslation.obf.jar"));
		try (InputStream in = getClass().getResourceAsStream("/cuchaz/enigma/resources/translation.mappings")) {
			m_mappings = new MappingsReader().read(new InputStreamReader(in));
			m_deobfuscator.setMappings(m_mappings);
		}
	}
	
	@Test
	public void deobfuscatingTranslations()
	throws Exception {
		Translator translator = m_deobfuscator.getTranslator(TranslationDirection.Deobfuscating);
		assertThat(translator.translateEntry(newClass("none/a")), is(newClass("deobf/A")));
	}
}