[giraph] fixed empty input graph problem
Change-Id: I704411321ea1e562f7150f2bc6bd48285c4f89e9
diff --git a/plugins/org.eclipse.emf.henshin.tests/src/org/eclipse/emf/henshin/tests/giraph/GiraphTest.java b/plugins/org.eclipse.emf.henshin.tests/src/org/eclipse/emf/henshin/tests/giraph/GiraphTest.java
index 119f552..e22c630 100644
--- a/plugins/org.eclipse.emf.henshin.tests/src/org/eclipse/emf/henshin/tests/giraph/GiraphTest.java
+++ b/plugins/org.eclipse.emf.henshin.tests/src/org/eclipse/emf/henshin/tests/giraph/GiraphTest.java
@@ -142,7 +142,7 @@
iteratedUnit.setName(iteratedUnit.getName() + iterations);
// Run test:
- run(iteratedUnit, ((Rule) TEST_MODULE.getUnit(inputRuleName)).getLhs(), aggregateVertices, aggregateEdges);
+ run(iteratedUnit, getInputGraph(inputRuleName), aggregateVertices, aggregateEdges);
// Restore iterated unit:
iteratedUnit.setIterations(backup.getIterations());
@@ -150,9 +150,17 @@
}
+ private Graph getInputGraph(String inputRuleName) {
+ Rule rule = ((Rule) TEST_MODULE.getUnit(inputRuleName));
+ Assert.assertNotNull(rule);
+ if (!rule.getMultiRules().isEmpty()) {
+ rule = rule.getMultiRules().get(0);
+ }
+ return rule.getLhs();
+ }
+
private void run(String mainUnitName, String inputRuleName, int aggregateVertices, int aggregateEdges) {
- run(TEST_MODULE.getUnit(mainUnitName), ((Rule) TEST_MODULE.getUnit(inputRuleName)).getLhs(), aggregateVertices,
- aggregateEdges);
+ run(TEST_MODULE.getUnit(mainUnitName), getInputGraph(inputRuleName), aggregateVertices, aggregateEdges);
}
private void run(Unit mainUnit, Graph inputGraph, int aggregateVertices, int aggregateEdges) {
@@ -162,7 +170,7 @@
GiraphGenerator generator = new GiraphGenerator();
generator.setMainUnit(mainUnit);
generator.setInputGraph(inputGraph);
- generator.setInputName(inputGraph.getRule().getName());
+ generator.setInputName(inputGraph.getRule().getRootRule().getName());
if (TEST_PROJECT == null) {
System.out.println("Installing Hadoop Test Environment (may take a couple of minutes)...");
