//------------------------------------------------------------------------------
// Copyright (c) 2005, 2006 IBM Corporation 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:
// IBM Corporation - initial implementation
//------------------------------------------------------------------------------
package org.eclipse.epf.diagram.tests;

import java.util.List;

import org.eclipse.epf.diagram.ui.service.DiagramUIService;
import org.eclipse.epf.diagram.ui.service.IDiagramUIService;
import org.eclipse.epf.library.tests.LibraryTestHelper;
import org.eclipse.epf.uma.Activity;
import org.eclipse.epf.uma.CapabilityPattern;
import org.eclipse.epf.uma.ContentPackage;
import org.eclipse.epf.uma.MethodConfiguration;
import org.eclipse.epf.uma.MethodPlugin;
import org.eclipse.epf.uma.ProcessComponent;
import org.eclipse.epf.uma.Task;
import org.eclipse.gmf.runtime.notation.Diagram;
import org.eclipse.gmf.runtime.notation.Node;

/**
 * JUnit tests for the org.eclipse.epf.diagram.ui.DiagramUIService classes.
 * 
 * @author Shilpa Toraskar
 * @since 1.2
 */
public class DiagramUIServiceTest extends DiagramTestCase {

	private IDiagramUIService diagramUIService = null;

	/**
	 * Creates a new instance.
	 */
	public DiagramUIServiceTest(String name) {
		super(name);
	}

	/**
	 * Tests the DiagamUIService.createDiagram method.
	 */
	public void testCreateDiagram() throws Exception {

		// create method plugin
		MethodPlugin plugin = LibraryTestHelper
				.createMethodPlugin("Test Plugin");
		ContentPackage pkg = LibraryTestHelper.createContentPackage(plugin,
				"p1");

		// create configuration
		MethodConfiguration config = LibraryTestHelper
				.createConfiguration("Test Config");

		// create process
		CapabilityPattern cp1 = LibraryTestHelper.createCapabilityPattern(
				plugin, "cp1", config);

		// update the configuration
		config.getMethodPluginSelection().add(plugin);
		config.getMethodPackageSelection().add(pkg);
		config.getMethodPackageSelection().add(
				(ProcessComponent) cp1.eContainer());

		// create task
		Task task1 = LibraryTestHelper.createTask(pkg, "Task1");
		Task task2 = LibraryTestHelper.createTask(pkg, "Task2");
		Task[] tasks = new Task[] { task1, task2 };

		Activity act1 = LibraryTestHelper.createActivity(plugin, cp1, "A1",
				tasks);
		Activity act2 = LibraryTestHelper.createActivity(plugin, cp1, "A1",
				tasks);

		diagramUIService = new DiagramUIService();
		
		Diagram diagram = diagramUIService.createActivityDiagram(cp1);

		assertTrue(diagram != null);
	
		
		Node act1Node = diagramUIService.createNode(diagram,  act1);
		Node initialNode = diagramUIService.createNode(diagram, IDiagramUIService.INITIAL_NODE, "initial");
		Node act2Node = diagramUIService.createNode(diagram, act2);
		Node decisionNode = diagramUIService.createNode(diagram, IDiagramUIService.DECISION_NODE, "decision");
		Node endNode = diagramUIService.createNode(diagram, IDiagramUIService.FINAL_NODE, "end");
		
		diagramUIService.createEdge(diagram, initialNode, act1Node, "link1");
		diagramUIService.createEdge(diagram, act1Node, decisionNode, "link2");
		diagramUIService.createEdge(diagram, decisionNode, act2Node, "link3");
		diagramUIService.createEdge(diagram, act2Node, endNode, "link4");
		
		
		List nodes = diagramUIService.getNodes(diagram, cp1, IDiagramUIService.ACTIVITY_NODE);
		if (!nodes.isEmpty()) {
			System.out.println("****** ACTIVITY Nodes - ");
			for (int i=0; i < nodes.size(); i++)
				System.out.println(nodes.get(i));
		}
		
		nodes = diagramUIService.getNodes(diagram, cp1, IDiagramUIService.INITIAL_NODE);
		if (!nodes.isEmpty()) {
			System.out.println("****** Initial Nodes - ");
			for (int i=0; i < nodes.size(); i++)
				System.out.println(nodes.get(i));
		}
		
		nodes = diagramUIService.getNodes(diagram, cp1, IDiagramUIService.DECISION_NODE);
		if (!nodes.isEmpty()) {
			System.out.println("****** Decision Nodes - ");
			for (int i=0; i < nodes.size(); i++)
				System.out.println(nodes.get(i));
		}
		
		nodes = diagramUIService.getAllNodes(diagram);
		if (!nodes.isEmpty()) {
			System.out.println("****** ALL Nodes - ");
			for (int i=0; i < nodes.size(); i++)
				System.out.println(nodes.get(i));
		}
		
		assertTrue(!nodes.isEmpty());
		
		List edges = diagramUIService.getEdges(diagram);
		if (!edges.isEmpty()) {
			System.out.println("****** ALL Edges - ");
			for (int i=0; i < edges.size(); i++)
				System.out.println(edges.get(i));
		}
		
		assertTrue(!edges.isEmpty());
		
	}

}