tests/org.eclipse.dltk.python.tests/src/org/eclipse/dltk/python/tests/model/ComplexConstructsTests.java

package org.eclipse.dltk.python.tests.model;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.List;

import org.eclipse.core.resources.IFile;
import org.eclipse.core.resources.IFolder;
import org.eclipse.core.resources.IProject;
import org.eclipse.core.runtime.NullProgressMonitor;
import org.eclipse.dltk.ast.declarations.ModuleDeclaration;
import org.eclipse.dltk.ast.expressions.Expression;
import org.eclipse.dltk.ast.statements.Block;
import org.eclipse.dltk.compiler.env.ModuleSource;
import org.eclipse.dltk.core.IModelElement;
import org.eclipse.dltk.core.IScriptProject;
import org.eclipse.dltk.core.ISourceModule;
import org.eclipse.dltk.core.tests.model.AbstractModelTests;
import org.eclipse.dltk.core.tests.util.ModelTestUtils;
import org.eclipse.dltk.python.internal.core.parser.PythonSourceParser;
import org.eclipse.dltk.python.parser.ast.PythonWithStatement;
import org.eclipse.dltk.python.tests.PythonTestsPlugin;

import junit.framework.Test;

public class ComplexConstructsTests extends AbstractModelTests {
	private static final String PROJECT_NAME = "pytests";
	private static final String FOLDER = "src";
	private static final String ASSIGNMENT_TESTS_FILE_NAME = "assignment_tests.py";

	private static final String SCRIPT_SOURCE = "a,b=1,2;c=d=1;x=y=lambda z:z;";

	public ComplexConstructsTests(String name) {
		super(PythonTestsPlugin.PLUGIN_NAME, name);
	}

	public static Test suite() {
		return new Suite(ComplexConstructsTests.class);
	}

	@Override
	public void setUpSuite() throws Exception {
		super.setUpSuite();
		IScriptProject sp = this.setUpScriptProject(PROJECT_NAME);
		IProject prj = sp.getProject();
		IFolder src = prj.getFolder(FOLDER);
		src.create(true, true, null);
		IFile testFile = src.getFile(ASSIGNMENT_TESTS_FILE_NAME);
		InputStream source = new ByteArrayInputStream(SCRIPT_SOURCE.getBytes());
		testFile.create(source, true, new NullProgressMonitor());
	}

	@Override
	public void tearDownSuite() throws Exception {
		super.tearDownSuite();
		deleteProject(PROJECT_NAME);
	}

	// this is a model test
	public void testAssignment() throws Exception {
		IScriptProject project = getScriptProject(PROJECT_NAME);
		ISourceModule module = this.getSourceModule(PROJECT_NAME, FOLDER,
				ASSIGNMENT_TESTS_FILE_NAME);
		IModelElement[] children = module.getChildren();
		ModelTestUtils.getAssertField(children, "a");
		ModelTestUtils.getAssertField(children, "b");
		ModelTestUtils.getAssertField(children, "c");
		ModelTestUtils.getAssertField(children, "d");
		ModelTestUtils.getAssertMethod(children, "x", 1);
		ModelTestUtils.getAssertMethod(children, "y", 1);
	}

	// this is a parser test
	public void testWithStatement() {
		final String script = "with a as b : pass";
		PythonSourceParser parser = new PythonSourceParser();
		ModuleDeclaration module = (ModuleDeclaration) parser.parse(new ModuleSource(script), null);
		List children = ((Block) module.getChilds().get(0)).getChilds();
		Object pws = children.get(0);
		assertTrue("with...as... statement expected.",
				pws instanceof PythonWithStatement);
		children = ((PythonWithStatement) pws).getChilds();
		assertTrue("Three children expected", 3 == children.size());
		assertTrue("Expression expected.",
				children.get(0) instanceof Expression);
		assertTrue("Expression expected.",
				children.get(1) instanceof Expression);
		assertTrue("Block expected.", children.get(2) instanceof Block);
	}
}