1 files changed, 82 insertions, 0 deletions
diff --git a/tests/junit/framework/org.eclipse.papyrus.junit.utils/src/org/eclipse/papyrus/junit/utils/TableUtils.java b/tests/junit/framework/org.eclipse.papyrus.junit.utils/src/org/eclipse/papyrus/junit/utils/TableUtils.java
new file mode 100644
index 00000000000..bec37f68246
--- /dev/null
+++ b/tests/junit/framework/org.eclipse.papyrus.junit.utils/src/org/eclipse/papyrus/junit/utils/TableUtils.java
@@ -0,0 +1,82 @@
+/*****************************************************************************
+ * Copyright (c) 2014 CEA LIST.
+ *
+ *    
+ * 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:
+ *  Benoit Maggi (CEA LIST) benoit.maggi@cea.fr - Initial API and implementation
+ *
+ *****************************************************************************/
+package org.eclipse.papyrus.junit.utils;
+
+import java.util.ArrayList;
+import java.util.Collection;
+
+import org.eclipse.emf.ecore.EObject;
+import org.eclipse.papyrus.infra.core.resource.AbstractBaseModel;
+import org.eclipse.papyrus.infra.core.resource.IModel;
+import org.eclipse.papyrus.infra.core.resource.ModelSet;
+import org.eclipse.papyrus.infra.gmfdiag.common.model.NotationModel;
+import org.eclipse.papyrus.infra.nattable.model.nattable.Table;
+import org.junit.Assert;
+
+/**
+ * Utility class for diagrams
+ */
+public class TableUtils {
+
+	/**
+	 * Return the first table found with the specified name
+	 * @param modelSet
+	 * @param tableName
+	 * @return
+	 */
+	public static Table getNotationFirstTable(ModelSet modelSet, String tableName) {
+		IModel notationModel = modelSet.getModel(NotationModel.MODEL_ID);
+
+		AbstractBaseModel notationBaseModel = null;
+		if (notationModel instanceof AbstractBaseModel) {
+			notationBaseModel = (AbstractBaseModel) notationModel;
+		} else {
+			Assert.fail("notation model is not an abstract base model");
+			return null;
+		}
+		Assert.assertTrue("notation resource contains nothing", notationBaseModel.getResource().getContents().size() >= 1);
+		for (EObject eObject : notationBaseModel.getResource().getContents()) {
+			if (eObject instanceof Table && tableName.equals(((Table) eObject).getName())) {
+				return (Table) eObject;
+			}
+		}
+		return null;
+	}
+
+	/**
+	 * Return the all tables found with the specified name
+	 * @param modelSet
+	 * @param tableName
+	 * @return
+	 */
+	public static Collection<Table> getAllNotationTable(ModelSet modelSet, String tableName) {
+		IModel notationModel = modelSet.getModel(NotationModel.MODEL_ID);
+		Collection<Table> tableList = new ArrayList<Table>();
+		AbstractBaseModel notationBaseModel = null;
+		if (notationModel instanceof AbstractBaseModel) {
+			notationBaseModel = (AbstractBaseModel) notationModel;
+		} else {
+			Assert.fail("notation model is not an abstract base model");
+			return null;
+		}
+		Assert.assertTrue("notation resource contains nothing", notationBaseModel.getResource().getContents().size() >= 1);
+		for (EObject eObject : notationBaseModel.getResource().getContents()) {
+			if (eObject instanceof Table && tableName.equals(((Table) eObject).getName())) {
+				tableList.add((Table) eObject);
+			}
+		}
+		return tableList;
+	}
+
+}