org.walkmod.javalang.test.SemanticTest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javalang-compiler Show documentation
Library of compiler components to processs Java code.
There is a newer version: 2.3.10
/*
 Copyright (C) 2015 Raquel Pau and Albert Coroleu.
 
Walkmod is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

Walkmod is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License
along with Walkmod.  If not, see .*/
package org.walkmod.javalang.test;

import java.io.File;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.HashMap;

import org.junit.After;
import org.junit.Before;
import org.walkmod.javalang.ASTManager;
import org.walkmod.javalang.ast.CompilationUnit;
import org.walkmod.javalang.compiler.Compiler;
import org.walkmod.javalang.compiler.symbols.SymbolTable;
import org.walkmod.javalang.compiler.symbols.SymbolVisitorAdapter;
import org.walkmod.javalang.compiler.types.TypesLoaderVisitor;

public abstract class SemanticTest {

	public static String SOURCES_DIR = "./src/test/resources/tmp/";
	public static String CLASSES_DIR = "./src/test/resources/tmp/classes";
	private TypesLoaderVisitor tt = null;
	private CompilationUnit cu = null;
	protected ClassLoader cl = null;
	private SymbolTable symTable = null;

	@Before
	public void prepare() throws Exception {
		File compilerDir = new File(CLASSES_DIR);
		compilerDir.mkdir();
	}

	@After
	public void clean() throws Exception {
		File sourcesDir = new File(SOURCES_DIR);

		removeRecursively(sourcesDir);

		tt = null;
		cu = null;
		cl = null;
		symTable = null;
	}

	public void removeRecursively(File file) {
		if (file.isDirectory()) {
			File[] files = file.listFiles();
			if (files != null) {
				for (File aux : files) {
					removeRecursively(aux);
				}
			}
		}
		file.delete();
	}

	public CompilationUnit compile(String... sources) throws Exception {
		Compiler compiler = new Compiler();
		compiler.compile(new File(CLASSES_DIR), new File(SOURCES_DIR), sources);
		if (sources != null) {
			cu = ASTManager.parse(sources[0]);
			populateSemantics();
		}
		return cu;
	}

	public void populateSemantics() throws Exception {
		SymbolVisitorAdapter> visitor = new SymbolVisitorAdapter>();
		visitor.setClassLoader(getClassLoader());
		visitor.visit(cu, new HashMap());
	}

	public ClassLoader getClassLoader() throws Exception {
		if (cl == null) {
			File aux = new File(CLASSES_DIR);
			cl = new URLClassLoader(new URL[] { aux.toURI().toURL() });
		}
		return cl;
	}

	public TypesLoaderVisitor getTypeTable() throws Exception {
		if (tt == null) {
			getSymbolTable().pushScope();
			tt = new TypesLoaderVisitor(getSymbolTable(), null, null);
			tt.setClassLoader(getClassLoader());

			cu.accept(tt, null);
		}
		return tt;
	}

	public void initTypes() throws Exception {
		getTypeTable();
	}

	public SymbolTable getSymbolTable() {
		if (symTable == null) {
			symTable = new SymbolTable();
		}
		return symTable;
	}
}