jpa_diagram_editor/tests/org.eclipse.jpt.jpadiagrameditor.ui.tests/src/org/eclipse/jpt/jpadiagrameditor/ui/tests/internal/feature/UpdateAttributeFeatureTest.java

/*******************************************************************************
 * <copyright>
 *
 * Copyright (c) 2005, 2010 SAP AG.
 * 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:
 *    Stefan Dimov - initial API, implementation and documentation
 *
 * </copyright>
 *
 *******************************************************************************/
package org.eclipse.jpt.jpadiagrameditor.ui.tests.internal.feature;

import static org.junit.Assert.assertNotNull;

import java.util.Date;
import java.util.Iterator;

import org.easymock.EasyMock;
import org.eclipse.core.resources.IFile;
import org.eclipse.emf.common.notify.Adapter;
import org.eclipse.emf.common.util.EList;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.util.BasicInternalEList;
import org.eclipse.graphiti.features.context.ICustomContext;
import org.eclipse.graphiti.mm.algorithms.GraphicsAlgorithm;
import org.eclipse.graphiti.mm.algorithms.Rectangle;
import org.eclipse.graphiti.mm.algorithms.Text;
import org.eclipse.graphiti.mm.pictograms.ContainerShape;
import org.eclipse.graphiti.mm.pictograms.Shape;
import org.eclipse.jpt.jpa.core.JpaProject;
import org.eclipse.jpt.jpadiagrameditor.ui.internal.feature.UpdateAttributeFeature;
import org.eclipse.jpt.jpadiagrameditor.ui.internal.provider.IJPAEditorFeatureProvider;
import org.eclipse.jpt.jpadiagrameditor.ui.tests.internal.JPACreateFactory;
import org.junit.Before;
import org.junit.Test;

public class UpdateAttributeFeatureTest {
	
	private IJPAEditorFeatureProvider featureProvider;
	private ICustomContext context;
	final String TEST_PROJECT = "Test"+(new Date()).getTime();
	private JpaProject jpaProject = null;
	private JPACreateFactory factory = null;
	private IFile entity;
	
	@Before
	public void setUp() throws Exception {
		factory = JPACreateFactory.instance();
		jpaProject = factory.createJPAProject(TEST_PROJECT + "_" + System.currentTimeMillis());
		assertNotNull(jpaProject);
		entity = factory.createEntity(jpaProject, "org.eclipse.Ent");
		Thread.sleep(2000);
		factory.addAttributes(entity, "att", "java.lang.String", "", "att", false);
	}
	
	
	@SuppressWarnings({ "unchecked", "deprecation" })
	@Test
	public void testUpdateAttributeFeature(){
		Text text = EasyMock.createMock(Text.class);
		text.setValue(EasyMock.isA(java.lang.String.class));
		EasyMock.replay(text);
		
		EList<GraphicsAlgorithm> listChildren = new BasicInternalEList<GraphicsAlgorithm>(GraphicsAlgorithm.class);
		
		
		EasyMock.expect(listChildren.get(0)).andReturn(text);
		EasyMock.replay(listChildren);
		
		
		Rectangle rectangle = EasyMock.createMock(Rectangle.class);
		EasyMock.expect(rectangle.getGraphicsAlgorithmChildren()).andReturn(listChildren);
		EasyMock.expect(rectangle.getY()).andReturn(0).times(2);
		EasyMock.replay(rectangle);
		
		Iterator<Shape> iteratorShape = EasyMock.createNiceMock(Iterator.class);
		EasyMock.expect(iteratorShape.hasNext()).andReturn(false);
		EasyMock.replay(iteratorShape);
		
		EList<Shape> listShape = new BasicInternalEList<Shape>(Shape.class);
		
		ContainerShape containerShape = EasyMock.createMock(ContainerShape.class);
		EasyMock.expect(containerShape.getChildren()).andReturn(listShape);
		EasyMock.replay(containerShape);
		
		Shape shape = EasyMock.createMock(Shape.class);
		Resource res = EasyMock.createMock(Resource.class);
		ResourceSet rs = EasyMock.createMock(ResourceSet.class);
		EasyMock.expect(res.getResourceSet()).andStubReturn(rs);
		EList<Adapter> ads = new BasicInternalEList<Adapter>(Adapter.class);
		EasyMock.expect(rs.eAdapters()).andStubReturn(ads);
		EasyMock.expect(shape.eResource()).andStubReturn(res);
		

		
		EasyMock.expect(shape.getGraphicsAlgorithm()).andReturn(rectangle).times(3);
		EasyMock.expect(shape.getContainer()).andReturn(containerShape);
		
		
		EasyMock.replay(shape, res, rs);
		
		context = EasyMock.createMock(ICustomContext.class);
		EasyMock.expect(context.getInnerPictogramElement()).andReturn(shape);
		EasyMock.replay(context);
		
		featureProvider = EasyMock.createMock(IJPAEditorFeatureProvider.class);
		EasyMock.expect(featureProvider.getBusinessObjectForPictogramElement(shape)).andReturn(null);
		EasyMock.replay(featureProvider);
		
		UpdateAttributeFeature feature = new UpdateAttributeFeature(featureProvider);
		feature.execute(context);
	}
	
}