core/tests/org.hawk.integration.tests/src/org/hawk/integration/tests/emf/CountInstancesTest.java

/*******************************************************************************
 * Copyright (c) 2015-2017 The University of York, Aston University.
 * 
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License 2.0 which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the Eclipse
 * Public License, v. 2.0 are satisfied: GNU General Public License, version 3.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-3.0
 *
 * Contributors:
 *     Antonio Garcia-Dominguez - initial API and implementation
 ******************************************************************************/
package org.hawk.integration.tests.emf;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.io.File;
import java.util.Arrays;
import java.util.Collection;

import org.hawk.backend.tests.factories.IGraphDatabaseFactory;
import org.hawk.graph.syncValidationListener.SyncValidationListener;
import org.hawk.integration.tests.IntegrationTestSuite;
import org.hawk.integration.tests.ModelIndexingTest;
import org.hawk.localfolder.LocalFolder;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runners.Parameterized.Parameters;

/**
 * Indexes models and counts the instances of a certain type.
 */
public class CountInstancesTest extends ModelIndexingTest {

	@Rule
	public GraphChangeListenerRule<SyncValidationListener> syncValidation
		= new GraphChangeListenerRule<>(new SyncValidationListener());

	@Parameters(name="{1}")
	public static Iterable<Object[]> params() {
		return IntegrationTestSuite.caseParams();
	}

	public CountInstancesTest(File baseDir, IGraphDatabaseFactory dbf) {
		super(baseDir, dbf, new EMFModelSupportFactory());
	}

	@Test
	public void tree() throws Throwable {
		indexer.registerMetamodels(
			new File(baseDir, "resources/metamodels/Ecore.ecore"),
			new File(baseDir, "resources/metamodels/Tree.ecore"));
		requestFolderIndex(new File(baseDir, "resources/models/tree"));

		scheduleAndWait(() -> {
			assertNoErrors(syncValidation.getListener());
			assertEquals(2, eol("return Tree.all.size;"));
			assertEquals("t3", eol("return Tree.all.selectOne(t|t.label='t9000').eContainer.label;"));
			assertEquals(0, eol("return Tree.all.selectOne(t|t.label='t3').eContainers.size;"));
			assertEquals(1, eol("return Tree.all.selectOne(t|t.label='t9000').eContainers.size;"));
			return null;
		});
	}

	@SuppressWarnings("unchecked")
	@Test
	public void treeCrossResourceContainment() throws Throwable {
		indexer.registerMetamodels(
			new File(baseDir, "resources/metamodels/Ecore.ecore"),
			new File(baseDir, "resources/metamodels/Tree.ecore"));
		requestFolderIndex(new File(baseDir, "resources/models/tree-xres"));

		scheduleAndWait(() -> {
			assertNoErrors(syncValidation.getListener());
		
			// Test for bug #56: select(t:Type|xyz) does not work
			assertEquals(3, eol("return Model.allContents.select(t:Tree|true).size;"));
		
			final Collection<String> labels = (Collection<String>) eol("return Model.allContents.collect(t:Tree|t.label);");
			assertEquals(3, labels.size());
			for (String e : Arrays.asList("xyz", "root", "abc")) {
				assertTrue(labels.contains(e));
			}
		
			assertEquals(3, eol("return Tree.all.size;"));
			assertEquals(2, eol("return Tree.all.selectOne(t|t.label='root').children.size;"));
			assertEquals("root", eol("return Tree.all.selectOne(t|t.label='xyz').eContainer.label;"));
			assertEquals("root", eol("return Tree.all.selectOne(t|t.label='abc').eContainer.label;"));
			return null;
		});
	}

	@Test
	public void set0() throws Throwable {
		indexer.registerMetamodels(
			new File(baseDir, "resources/metamodels/Ecore.ecore"),
			new File(baseDir, "resources/metamodels/JDTAST.ecore"));

		requestFolderIndex(new File(baseDir, "resources/models/set0"));
		scheduleAndWait(() -> {
			assertNoErrors(syncValidation.getListener());
			assertEquals(1, eol("return IJavaProject.all.size;"));
		
			final int reportedSize = (Integer) eol("return TypeDeclaration.all.size;");
			final Collection<?> actualList = (Collection<?>) eol("return TypeDeclaration.all;");
			assertEquals(reportedSize, actualList.size());

			return null;
		});
	}

	@Test
	public void treeWithSpaces() throws Throwable {
		indexer.registerMetamodels(
			new File(baseDir, "resources/metamodels/Ecore.ecore"),
			new File(baseDir, "resources/metamodels/Tree.ecore"));

		final LocalFolder vcs = new LocalFolder();
		final File folder = new File(baseDir, "resources/models");
		vcs.init(folder.getAbsolutePath(), indexer);
		vcs.setFileFilter(f -> {
			return f.getPath().contains("tree with spaces");
		});
		vcs.run();
		indexer.addVCSManager(vcs, true);

		scheduleAndWait(() -> {
			assertNoErrors(syncValidation.getListener());
			assertEquals(1, eol("return Tree.all.size;"));
			assertEquals(1, eol("return Model.getAllOf('Tree', 'Tree', '/tree with spaces/space tree.model').size;"));
			assertEquals(1, eol("return Model.getAllOf('Tree', 'Tree', '/tree%20with%20spaces/space%20tree.model').size;"));
			return null;
		});
	}
}