CollectionFeaturesTest.java

/*******************************************************************************
 * Copyright (c) 2005, 2007 committers of openArchitectureWare and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     committers of openArchitectureWare - initial API and implementation
 *******************************************************************************/

package org.eclipse.xtend.typesystem.baseimpl;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Iterator;

import junit.framework.TestCase;

import org.eclipse.xtend.expression.EvaluationException;
import org.eclipse.xtend.expression.ExecutionContextImpl;
import org.eclipse.xtend.expression.ExpressionFacade;
import org.eclipse.xtend.type.impl.java.JavaMetaModel;
import org.eclipse.xtend.type.impl.java.beans.JavaBeansStrategy;

/**
 * @author Sven Efftinge (http://www.efftinge.de)
 * @author Arno Haase
 * @author Bernd Kolb
 */
@SuppressWarnings("unchecked")
public class CollectionFeaturesTest extends TestCase {
	ExecutionContextImpl ec;

	private ExpressionFacade ef;

	private ArrayList<String> vals;

	@Override
	protected void setUp() throws Exception {
		vals = new ArrayList<String>();
		vals.add("1");
		vals.add("2");
		vals.add("3");
		vals.add("4");
		vals.add("5");
		ec = new ExecutionContextImpl();
		ec.registerMetaModel(new JavaMetaModel("asdf", new JavaBeansStrategy()));
		ef = new ExpressionFacade(ec);
	}

	public final void testSelect() {
		final Collection result = (Collection) ef.evaluate("vals.select(s| s >= \"3\")", Collections.singletonMap(
				"vals", vals));
		assertEquals(3, result.size());
		assertTrue(result.contains("3"));
		assertTrue(result.contains("4"));
		assertTrue(result.contains("5"));
	}

	public final void testSelectFirst() {
		final Object result = ef.evaluate("vals.selectFirst(s| s >= \"3\")", Collections.singletonMap("vals", vals));
		assertEquals("3", result);
	}

	public final void testContains() {
		final Boolean result = (Boolean) ef.evaluate("vals.contains(\"3\")", Collections.singletonMap("vals", vals));
		assertTrue(result.booleanValue());
	}

	public final void testContainsAll() {
		final Boolean result = (Boolean) ef.evaluate("vals.containsAll({\"3\",\"4\"})", Collections.singletonMap(
				"vals", vals));
		assertTrue(result.booleanValue());
	}
	
	public final void testContainsAllFail() {
		try {
			ef.evaluate("vals.containsAll(\"3\")", Collections.singletonMap(
					"vals", vals));
			fail();
		}
		catch (EvaluationException e) {
			//Expected exception
		}
	}
	
	public final void testRemove() {
		final Collection result = (Collection) ef.evaluate("vals.remove(\"3\").length", Collections.singletonMap("vals", vals));
		assertEquals(4, result.size());
	}

	public final void testRemoveAll() {
		final Collection result = (Collection) ef.evaluate("vals.removeAll({\"3\",\"4\"})", Collections.singletonMap(
				"vals", vals));
		assertEquals(3, result.size());
	}
	
	public final void testRemoveAllFail() {
		try {
			ef.evaluate("vals.removeAll(\"3\")", Collections.singletonMap(
					"vals", vals));
			fail();
		}
		catch (EvaluationException e) {
			//Expected exception
		}
	}

	public final void testCollect() {
		final Collection result = (Collection) ef.evaluate("vals.collect( s| s.length)", Collections.singletonMap(
				"vals", vals));
		assertEquals(5, result.size());
		for (final Iterator iter = result.iterator(); iter.hasNext();) {
			assertEquals(new Long(1), iter.next());
		}
	}

	public final void testExists() {
		assertEquals(Boolean.TRUE, ef.evaluate("vals.exists(s| s == \"3\")", Collections.singletonMap("vals", vals)));
		assertEquals(Boolean.FALSE, ef.evaluate("vals.exists(s| s == \"33\")", Collections.singletonMap("vals", vals)));
	}

	public final void testNotExists() {
		assertEquals(Boolean.FALSE, ef
				.evaluate("vals.notExists(s| s == \"3\")", Collections.singletonMap("vals", vals)));
		assertEquals(Boolean.TRUE, ef
				.evaluate("vals.notExists(s| s == \"33\")", Collections.singletonMap("vals", vals)));
	}

	public final void testForAll() {
		assertEquals(Boolean.TRUE, ef.evaluate("{}.forAll(s | true)"));
		assertEquals(Boolean.TRUE, ef.evaluate("vals.forAll(s| s.length == 1)", Collections.singletonMap("vals", vals)));
		assertEquals(Boolean.FALSE, ef.evaluate("vals.forAll(s| s == \"3\")", Collections.singletonMap("vals", vals)));
	}

	public final void testEquals() {
		assertEquals(Boolean.TRUE, ef.evaluate("{}=={}"));
		assertEquals(Boolean.FALSE, ef.evaluate("{}=={1}"));
		assertEquals(ef.evaluate("{'test',123}"), ef.evaluate("{'test',123}"));
		assertEquals(Boolean.TRUE,ef.evaluate("{'test',123}=={'test',123}"));
	}

}