1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
/*******************************************************************************
* <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.easymock.EasyMock.expect;
import static org.easymock.EasyMock.isA;
import static org.easymock.EasyMock.replay;
import static org.junit.Assert.assertNotNull;
import org.easymock.EasyMock;
import org.eclipse.core.resources.IFile;
import org.eclipse.graphiti.dt.IDiagramTypeProvider;
import org.eclipse.graphiti.features.IDirectEditingInfo;
import org.eclipse.graphiti.features.context.IAddContext;
import org.eclipse.graphiti.features.context.ICreateContext;
import org.eclipse.graphiti.mm.pictograms.ContainerShape;
import org.eclipse.jdt.core.ICompilationUnit;
import org.eclipse.jdt.core.JavaCore;
import org.eclipse.jpt.jpa.core.JpaProject;
import org.eclipse.jpt.jpa.core.context.PersistentType;
import org.eclipse.jpt.jpadiagrameditor.ui.internal.feature.ClickAddAttributeButtonFeature;
import org.eclipse.jpt.jpadiagrameditor.ui.internal.provider.IJPAEditorFeatureProvider;
import org.eclipse.jpt.jpadiagrameditor.ui.internal.util.IJPAEditorUtil;
import org.eclipse.jpt.jpadiagrameditor.ui.tests.internal.JPACreateFactory;
import org.eclipse.jpt.jpadiagrameditor.ui.tests.internal.util.IEditor;
import org.eclipse.ui.IWorkbenchPartSite;
import org.junit.Before;
import org.junit.Test;
@SuppressWarnings("nls")
public class ClickAddFeatureTest {
private IJPAEditorFeatureProvider featureProvider;
private ICreateContext context;
final String TEST_PROJECT = "Test";
private JpaProject jpaProject = null;
private JPACreateFactory factory = null;
PersistentType jpt = null;
ICompilationUnit cu = null;
@Before
public void setUp() throws Exception {
System.setProperty(JPACreateFactory.JPA_JAR_NAME_SYSTEM_PROPERTY, "C:\\lib\\persistence-api-1.0.jar");
factory = JPACreateFactory.instance();
jpaProject = factory.createJPAProject(TEST_PROJECT + "_" + System.currentTimeMillis());
assertNotNull(jpaProject);
IFile entity = factory.createEntity(jpaProject, "org.eclipse.Entity1");
ContainerShape cs = EasyMock.createMock(ContainerShape.class);
context = EasyMock.createMock(ICreateContext.class);
expect(context.getTargetContainer()).andStubReturn(cs);
expect(context.getX()).andStubReturn(0);
expect(context.getY()).andStubReturn(0);
expect(context.getWidth()).andStubReturn(100);
expect(context.getHeight()).andStubReturn(100);
expect(context.getTargetConnection()).andStubReturn(null);
IDirectEditingInfo dei = EasyMock.createMock(IDirectEditingInfo.class);
dei.setActive(true);
Thread.sleep(2000);
jpt = (PersistentType)JPACreateFactory.getPersistentType(entity);
int cnt = 0;
while ((cnt < 50) && (jpt == null)) {
Thread.sleep(200);
jpt = (PersistentType)JPACreateFactory.getPersistentType(entity);
cnt++;
}
featureProvider = EasyMock.createMock(IJPAEditorFeatureProvider.class);
expect(featureProvider.getBusinessObjectForPictogramElement(cs)).andStubReturn(jpt);
expect(featureProvider.getPictogramElementForBusinessObject(jpt)).andStubReturn(null);
cu = JavaCore.createCompilationUnitFrom(entity);
cnt = 0;
while ((cnt < 50) && (cu == null)) {
Thread.sleep(200);
cu = JavaCore.createCompilationUnitFrom(entity);
cnt++;
}
expect(featureProvider.getCompilationUnit((PersistentType) EasyMock.anyObject())).andReturn(cu).anyTimes();
expect(featureProvider.addIfPossible(isA(IAddContext.class))).andStubReturn(null);
expect(featureProvider.getDirectEditingInfo()).andStubReturn(dei);
IJPAEditorUtil ut = EasyMock.createMock(IJPAEditorUtil.class);
expect(featureProvider.getJPAEditorUtil()).andStubReturn(ut);
IDiagramTypeProvider diagramTypeProvider = EasyMock.createMock(IDiagramTypeProvider.class);
expect(featureProvider.getDiagramTypeProvider()).andStubReturn(diagramTypeProvider);
IEditor ed = EasyMock.createMock(IEditor.class);
expect(diagramTypeProvider.getDiagramEditor()).andStubReturn(ed);
IWorkbenchPartSite ws = EasyMock.createMock(IWorkbenchPartSite.class);
expect(ed.getSite()).andStubReturn(ws);
ut.formatCode((ICompilationUnit)EasyMock.anyObject(), (IWorkbenchPartSite)EasyMock.anyObject());
featureProvider.addAddIgnore(jpt, "attribute1");
replay(featureProvider, cs, context, dei, diagramTypeProvider, ed, ws, ut);
}
@Test
public void testClickAddAttributeButtonFeature(){
if ((jpt == null) || (cu == null))
return; // The test wasn't setup properly
ClickAddAttributeButtonFeature feature = new ClickAddAttributeButtonFeature(featureProvider);
Object[] created = feature.create(context);
assertNotNull(created[0]);
}
}
|
