GraphControllerTests.java

package controller;

import de.danielsenff.imageflow.controller.GraphController;
import de.danielsenff.imageflow.models.unit.UnitElement;
import junit.framework.TestCase;

public class GraphControllerTests extends TestCase {

	
	public void testRemoveUnit() {
		GraphController controller = new GraphController();
		controller.setupExample1();
		
		assertEquals("number of connections", 4, controller.getConnections().size());
		
		UnitElement merge = (UnitElement) controller.getUnitElements().get(3);
		assertTrue("has removed sucessfully", controller.removeNode(merge));
	}
	
	
	
}