split codegen tests a little bit

2011-04-15 11:57:46 +02:00
parent 3a203302dc
commit b30f6db5e6
3 changed files with 145 additions and 121 deletions
@@ -0,0 +1,90 @@
+package org.jetbrains.jet.codegen;
+
+import com.intellij.testFramework.LightProjectDescriptor;
+import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.jet.JetLightProjectDescriptor;
+import org.jetbrains.jet.lang.JetFileType;
+import org.jetbrains.jet.lang.psi.JetFile;
+import org.jetbrains.jet.lang.psi.JetNamespace;
+import org.jetbrains.jet.parsing.JetParsingTest;
+
+import java.lang.reflect.Method;
+import java.util.List;
+
+/**
+ * @author yole
+ */
+public abstract class CodegenTestCase extends LightCodeInsightFixtureTestCase {
+    protected void loadText(final String text) {
+        myFixture.configureByText(JetFileType.INSTANCE, text);
+    }
+
+    protected void loadFile(final String name) {
+        myFixture.configureByFile(JetParsingTest.getTestDataDir() + "/codegen/" + name);
+    }
+
+    protected String generateToText() {
+        Codegens state = new Codegens(getProject(), true);
+        JetFile jetFile = (JetFile) myFixture.getFile();
+        JetNamespace namespace = jetFile.getRootNamespace();
+        NamespaceCodegen codegen = state.forNamespace(namespace);
+        codegen.generate(namespace);
+
+        List<String> files = state.files();
+        assertEquals("This test only supposed to generate single class file", 1, files.size());
+
+        return state.asText(files.get(0));
+    }
+
+    private Class generateToClass() {
+        Codegens state = new Codegens(getProject(), false);
+        JetFile jetFile = (JetFile) myFixture.getFile();
+        final JetNamespace namespace = jetFile.getRootNamespace();
+
+        NamespaceCodegen codegen = state.forNamespace(namespace);
+        codegen.generate(namespace);
+
+        List<String> files = state.files();
+        assertEquals("This test only supposed to generate single class file", 1, files.size());
+        final byte[] data = state.asBytes(files.get(0));
+        MyClassLoader classLoader = new MyClassLoader(NamespaceGenTest.class.getClassLoader());
+        return classLoader.doDefineClass(NamespaceCodegen.getJVMClassName(namespace.getFQName()).replace("/", "."), data);
+    }
+
+    protected Method generateFunction() {
+        Class aClass = generateToClass();
+        return aClass.getMethods()[0];
+    }
+
+    protected Method generateFunction(String name) {
+        Class aClass = generateToClass();
+        for (Method method : aClass.getMethods()) {
+            if (method.getName().equals(name)) {
+                return method;
+            }
+        }
+        throw new IllegalArgumentException("couldn't find method " + name);
+    }
+
+    private static class MyClassLoader extends ClassLoader {
+      public MyClassLoader(ClassLoader parent) {
+        super(parent);
+      }
+
+      public Class doDefineClass(String name, byte[] data) {
+        return defineClass(name, data, 0, data.length);
+      }
+
+      @Override
+      public Class<?> loadClass(String name) throws ClassNotFoundException {
+        return super.loadClass(name);
+      }
+    }
+
+    @NotNull
+    @Override
+    protected LightProjectDescriptor getProjectDescriptor() {
+        return JetLightProjectDescriptor.INSTANCE;
+    }
+}