[jboss-cvs] JBossAS SVN: r90960 - projects/mc-tools/grapher/trunk/src/test/java/org/jboss/test/mctools/grapher/render/test.

[jboss-cvs-commits at lists.jboss.org]

Author: alesj
Date: 2009-07-08 17:31:26 -0400 (Wed, 08 Jul 2009)
New Revision: 90960

Modified:
   projects/mc-tools/grapher/trunk/src/test/java/org/jboss/test/mctools/grapher/render/test/ManualGraphWrite2FileTestCase.java
Log:
Take the graph from the example.

Modified: projects/mc-tools/grapher/trunk/src/test/java/org/jboss/test/mctools/grapher/render/test/ManualGraphWrite2FileTestCase.java
===================================================================
--- projects/mc-tools/grapher/trunk/src/test/java/org/jboss/test/mctools/grapher/render/test/ManualGraphWrite2FileTestCase.java	2009-07-08 21:29:09 UTC (rev 90959)
+++ projects/mc-tools/grapher/trunk/src/test/java/org/jboss/test/mctools/grapher/render/test/ManualGraphWrite2FileTestCase.java	2009-07-08 21:31:26 UTC (rev 90960)
@@ -21,18 +21,18 @@
  */
 package org.jboss.test.mctools.grapher.render.test;
 
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
+import java.awt.*;
+import java.awt.geom.Rectangle2D;
 
 import junit.framework.Test;
 import org.jgraph.JGraph;
-import org.jgraph.graph.ConnectionSet;
+import org.jgraph.graph.DefaultCellViewFactory;
 import org.jgraph.graph.DefaultEdge;
 import org.jgraph.graph.DefaultGraphCell;
 import org.jgraph.graph.DefaultGraphModel;
-import org.jgraph.graph.Edge;
+import org.jgraph.graph.DefaultPort;
+import org.jgraph.graph.GraphConstants;
+import org.jgraph.graph.GraphLayoutCache;
 import org.jgraph.graph.GraphModel;
 
 /**
@@ -45,7 +45,7 @@
    public ManualGraphWrite2FileTestCase(String name)
    {
       super(name);
-      //setDeleteFile(false);
+      setDeleteFile(false);
    }
 
    public static Test suite()
@@ -56,25 +56,35 @@
    @SuppressWarnings("unchecked")
    protected JGraph createGraph()
    {
-      List cells = new ArrayList();
-      ConnectionSet cs = new ConnectionSet();
-      Map attributes = new HashMap();
+      DefaultGraphCell[] cells = new DefaultGraphCell[3];
 
-      DefaultGraphCell v1 = new DefaultGraphCell("v1");
-      v1.addPort();
-      cells.add(v1);
+      cells[0] = new DefaultGraphCell("Hello");
+      GraphConstants.setBounds(cells[0].getAttributes(), new Rectangle2D.Double(20, 20, 40, 20));
+      GraphConstants.setGradientColor(cells[0].getAttributes(), Color.orange);
+      GraphConstants.setOpaque(cells[0].getAttributes(), true);
+      DefaultPort port0 = new DefaultPort();
+      cells[0].add(port0);
 
-      DefaultGraphCell v2 = new DefaultGraphCell("v2");
-      v2.addPort();
-      cells.add(v2);
+      cells[1] = new DefaultGraphCell("World");
+      GraphConstants.setBounds(cells[1].getAttributes(), new Rectangle2D.Double(140, 140, 40, 20));
+      GraphConstants.setGradientColor(cells[1].getAttributes(), Color.red);
+      GraphConstants.setOpaque(cells[1].getAttributes(), true);
+      DefaultPort port1 = new DefaultPort();
+      cells[1].add(port1);
 
-      Edge e1 = new DefaultEdge("e1");
-      cs.connect(e1, v1.getChildAt(0), v2.getChildAt(0));
-      cells.add(e1);
+      DefaultEdge edge = new DefaultEdge();
+      edge.setSource(cells[0].getChildAt(0));
+      edge.setTarget(cells[1].getChildAt(0));
+      cells[2] = edge;
+      int arrow = GraphConstants.ARROW_CLASSIC;
+      GraphConstants.setLineEnd(edge.getAttributes(), arrow);
+      GraphConstants.setEndFill(edge.getAttributes(), true);
 
-      // create simple graph
       GraphModel model = new DefaultGraphModel();
-      model.insert(cells.toArray(), attributes, cs, null, null);
-      return new JGraph(model);
+      GraphLayoutCache view = new GraphLayoutCache(model, new DefaultCellViewFactory());
+      JGraph graph = new JGraph(model, view);
+      graph.getGraphLayoutCache().insert(cells);
+
+      return graph;
    }
 }
\ No newline at end of file