1 files changed, 189 insertions, 0 deletions
diff --git a/eclipse/plugins/com.android.ide.eclipse.tests/unittests/com/android/ide/eclipse/mock/Mocks.java b/eclipse/plugins/com.android.ide.eclipse.tests/unittests/com/android/ide/eclipse/mock/Mocks.java
new file mode 100644
index 000000000..65e2144be
--- /dev/null
+++ b/eclipse/plugins/com.android.ide.eclipse.tests/unittests/com/android/ide/eclipse/mock/Mocks.java
@@ -0,0 +1,189 @@
+/*
+ * Copyright (C) 2010 The Android Open Source Project
+ *
+ * Licensed under the Eclipse Public License, Version 1.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.eclipse.org/org/documents/epl-v10.php
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.android.ide.eclipse.mock;
+
+import static org.easymock.EasyMock.capture;
+import static org.easymock.EasyMock.createMock;
+import static org.easymock.EasyMock.createNiceMock;
+import static org.easymock.EasyMock.eq;
+import static org.easymock.EasyMock.expect;
+import static org.easymock.EasyMock.expectLastCall;
+import static org.easymock.EasyMock.isA;
+import static org.easymock.EasyMock.replay;
+
+import com.android.io.IAbstractFolder;
+import com.android.io.IAbstractResource;
+
+import org.easymock.Capture;
+import org.easymock.EasyMock;
+import org.easymock.IAnswer;
+import org.eclipse.core.resources.IFile;
+import org.eclipse.core.resources.IFolder;
+import org.eclipse.core.resources.IProject;
+import org.eclipse.core.resources.IResource;
+import org.eclipse.core.runtime.IPath;
+import org.eclipse.core.runtime.IProgressMonitor;
+import org.eclipse.core.runtime.Path;
+import org.eclipse.jdt.core.IClasspathEntry;
+import org.eclipse.jdt.core.IJavaProject;
+import org.eclipse.jdt.core.JavaCore;
+
+import java.util.Map;
+
+public class Mocks {
+    public static IJavaProject createProject(IClasspathEntry[] entries, IPath outputLocation)
+            throws Exception {
+        IJavaProject javaProject = createMock(IJavaProject.class);
+        final Capture<IClasspathEntry[]> capturedEntries = new Capture<IClasspathEntry[]>();
+        Capture<IPath> capturedOutput = new Capture<IPath>();
+        capturedEntries.setValue(entries);
+        capturedOutput.setValue(outputLocation);
+
+        IProject project = createProject();
+        expect(javaProject.getProject()).andReturn(project).anyTimes();
+        expect(javaProject.getOutputLocation()).andReturn(capturedOutput.getValue()).anyTimes();
+
+        expect(javaProject.getRawClasspath()).andAnswer(new IAnswer<IClasspathEntry[]>() {
+            @Override
+            public IClasspathEntry[] answer() throws Throwable {
+                return capturedEntries.getValue();
+            }
+        }).anyTimes();
+
+        javaProject.setRawClasspath(capture(capturedEntries), isA(IProgressMonitor.class));
+        expectLastCall().anyTimes();
+
+        javaProject.setRawClasspath(capture(capturedEntries), capture(capturedOutput),
+                isA(IProgressMonitor.class));
+        expectLastCall().anyTimes();
+
+        final Capture<String> capturedCompliance = new Capture<String>();
+        capturedCompliance.setValue("1.4");
+        final Capture<String> capturedSource = new Capture<String>();
+        capturedSource.setValue("1.4");
+        final Capture<String> capturedTarget = new Capture<String>();
+        capturedTarget.setValue("1.4");
+
+        expect(javaProject.getOption(JavaCore.COMPILER_COMPLIANCE, true)).andAnswer(
+                new IAnswer<String>() {
+                    @Override
+                    public String answer() throws Throwable {
+                        return capturedCompliance.getValue();
+                    }
+                });
+        expect(javaProject.getOption(JavaCore.COMPILER_SOURCE, true)).andAnswer(
+                new IAnswer<String>() {
+                    @Override
+                    public String answer() throws Throwable {
+                        return capturedSource.getValue();
+                    }
+                });
+        expect(javaProject.getOption(JavaCore.COMPILER_CODEGEN_TARGET_PLATFORM, true)).andAnswer(
+                new IAnswer<String>() {
+                    @Override
+                    public String answer() throws Throwable {
+                        return capturedTarget.getValue();
+                    }
+                });
+
+        javaProject.setOption(eq(JavaCore.COMPILER_COMPLIANCE), capture(capturedCompliance));
+        expectLastCall().anyTimes();
+        javaProject.setOption(eq(JavaCore.COMPILER_SOURCE), capture(capturedSource));
+        expectLastCall().anyTimes();
+        javaProject.setOption(eq(JavaCore.COMPILER_CODEGEN_TARGET_PLATFORM),
+                capture(capturedTarget));
+        expectLastCall().anyTimes();
+
+        replay(javaProject);
+
+        return javaProject;
+    }
+
+    /**
+     * Creates a mock implementation of {@link IFile}.
+     * <p/>
+     * Supported methods:
+     * <ul>
+     * <li>IFile#getName()</li>
+     * <li>IFile#getLocation()</li>
+     * </ul>
+     */
+    public static IFile createFile(String fileName) {
+        IFile file = createNiceMock(IFile.class);
+        expect(file.getName()).andReturn(fileName).anyTimes();
+        expect(file.getLocation()).andReturn(new Path(fileName)).anyTimes();
+        replay(file);
+        return file;
+    }
+
+    /**
+     * Creates a mock implementation of {@link IFolder}.
+     * <p/>
+     * Supported methods:
+     * <ul>
+     * <li>{@link IFolder#getName()}</li>
+     * <li>{@link IFolder#members()}</li>
+     * </ul>
+     */
+    public static IFolder createFolder(String name, IResource[] members) throws Exception {
+        IFolder file = createNiceMock(IFolder.class);
+        expect(file.getName()).andReturn(name).anyTimes();
+        // expect(file.getLocation()).andReturn(new Path(name)).anyTimes();
+        expect(file.members()).andReturn(members).anyTimes();
+        replay(file);
+        return file;
+    }
+
+    public static IAbstractFolder createAbstractFolder(String name, IAbstractResource[] members) {
+        IAbstractFolder folder = createNiceMock(IAbstractFolder.class);
+        expect(folder.getName()).andReturn(name).anyTimes();
+        // expect(file.getLocation()).andReturn(new Path(name)).anyTimes();
+        expect(folder.listMembers()).andReturn(members).anyTimes();
+        replay(folder);
+
+        return folder;
+    }
+
+    /**
+     * Mock implementation of {@link IProject}.
+     * <p/>
+     * Supported methods:
+     * <ul>
+     * <li>{@link IProject#build(int kind, IProgressMonitor monitor)}</li>
+     * <li>
+     * {@link IProject#build(int kind, String builderName, Map args, IProgressMonitor monitor)}
+     * </li>
+     * </ul>
+     */
+    public static IProject createProject() {
+        IProject project = EasyMock.createNiceMock(IProject.class);
+        replay(project);
+        return project;
+    }
+
+    /**
+     * Creates a mock implementation of an {@link IClasspathEntry}, which supports
+     * {@link IClasspathEntry#getEntryKind} and {@link IClasspathEntry#getPath}.
+     */
+    public static IClasspathEntry createClasspathEntry(IPath path, int kind) {
+        IClasspathEntry entry = createNiceMock(IClasspathEntry.class);
+        expect(entry.getEntryKind()).andReturn(kind).anyTimes();
+        expect(entry.getPath()).andReturn(path).anyTimes();
+        replay(entry);
+        return entry;
+    }
+}