X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=org.argeo.slc.core%2Fsrc%2Fmain%2Fjava%2Forg%2Fargeo%2Fslc%2Fcore%2Ftest%2Ftree%2FCompositeTreeTestDefinition.java;h=f26c085807285b0e2ba487364ea039a19da111bf;hb=56a8f70a587c793620d1bc2e73c847d935188001;hp=25479069b70e66ab9578609e3528ec76a91c814f;hpb=faf680e212bf3e18837c4f798587856e061273b3;p=gpl%2Fargeo-slc.git

diff --git a/org.argeo.slc.core/src/main/java/org/argeo/slc/core/test/tree/CompositeTreeTestDefinition.java b/org.argeo.slc.core/src/main/java/org/argeo/slc/core/test/tree/CompositeTreeTestDefinition.java
index 25479069b..f26c08580 100644
--- a/org.argeo.slc.core/src/main/java/org/argeo/slc/core/test/tree/CompositeTreeTestDefinition.java
+++ b/org.argeo.slc.core/src/main/java/org/argeo/slc/core/test/tree/CompositeTreeTestDefinition.java
@@ -9,6 +9,7 @@ import org.apache.commons.logging.LogFactory;
 import org.argeo.slc.core.structure.SimpleSElement;
 import org.argeo.slc.core.structure.StructureAware;
 import org.argeo.slc.core.structure.StructureElement;
+import org.argeo.slc.core.structure.StructureElementProvider;
 import org.argeo.slc.core.structure.StructurePath;
 import org.argeo.slc.core.structure.StructureRegistry;
 import org.argeo.slc.core.structure.tree.TreeSPath;
@@ -21,13 +22,13 @@ import org.argeo.slc.core.test.TestRun;
  * children.
  */
 public class CompositeTreeTestDefinition implements TestDefinition,
-		StructureAware {
+		StructureAware<TreeSPath> {
 	private Log log = LogFactory.getLog(CompositeTreeTestDefinition.class);
 
 	private List<TestDefinition> tasks = null;
 	private List<TreeSPath> taskPaths = null;
 	private TreeSPath path;
-	private StructureRegistry registry;
+	private StructureRegistry<TreeSPath> registry;
 
 	public void execute(TestRun testRun) {
 		log.info("Execute sequence of test definitions...");
@@ -58,18 +59,20 @@ public class CompositeTreeTestDefinition implements TestDefinition,
 		}
 	}
 
-	public void notifyCurrentPath(StructureRegistry registry, StructurePath path) {
-		this.path = (TreeSPath) path;
+	public void notifyCurrentPath(StructureRegistry<TreeSPath> registry,
+			TreeSPath path) {
+		this.path = path;
 		this.registry = registry;
-		
+
 		// clear task paths
 		taskPaths.clear();
 
 		Integer count = 0;
 		for (TestDefinition task : tasks) {
 			final StructureElement element;
-			if (task instanceof StructureElement) {
-				element = (StructureElement) task;
+			if (task instanceof StructureElementProvider) {
+				element = ((StructureElementProvider) task)
+						.createStructureElement();
 			} else {
 				element = new SimpleSElement("[no desc]");
 			}
@@ -77,7 +80,8 @@ public class CompositeTreeTestDefinition implements TestDefinition,
 			registry.register(taskPath, element);
 			taskPaths.add(taskPath);
 			if (task instanceof StructureAware) {
-				((StructureAware) task).notifyCurrentPath(registry, taskPath);
+				((StructureAware<TreeSPath>) task).notifyCurrentPath(registry,
+						taskPath);
 			}
 			count++;
 		}