Fix autocomplete on mock objects

META-INF/plugin.xml

@@ -38,6 +38,10 @@
     <!-- Add your extensions here -->
   </extensions>
 
+  <extensions defaultExtensionNs="com.jetbrains.php">
+    <typeProvider3 implementation="org.atoum.intellij.plugin.atoum.MockTypeProvider" />
+  </extensions>
+
   <application-components>
     <!-- Add your application components here -->
   </application-components>
@@ -79,4 +83,4 @@
   <depends>com.jetbrains.php</depends>
   <depends>com.intellij.modules.platform</depends>
 
-</idea-plugin>
+</idea-plugin>

src/org/atoum/intellij/plugin/atoum/MockTypeProvider.java

@@ -0,0 +1,59 @@
+package org.atoum.intellij.plugin.atoum;
+
+import com.intellij.openapi.project.Project;
+import com.intellij.psi.PsiElement;
+import com.jetbrains.php.lang.psi.elements.*;
+import com.jetbrains.php.lang.psi.resolve.types.PhpType;
+import com.jetbrains.php.lang.psi.resolve.types.PhpTypeProvider3;
+import org.jetbrains.annotations.Nullable;
+
+import java.util.Collection;
+import java.util.Set;
+
+public class MockTypeProvider implements PhpTypeProvider3 {
+
+    @Override
+    public char getKey() {
+        return '\u0102';
+    }
+
+    @Nullable
+    @Override
+    public PhpType getType(PsiElement psiElement) {
+        if (psiElement instanceof NewExpression) {
+            NewExpression expr = (NewExpression) psiElement;
+            ClassReference ref = expr.getClassReference();
+
+            if (ref != null && ref.getFQN() != null && ref.getFQN().startsWith("\\mock\\")) {
+                return new PhpType().add(ref.getFQN().substring("\\mock".length()));
+            }
+        } else if (psiElement instanceof MethodReference) {
+            MethodReference expr = (MethodReference) psiElement;
+            if (expr.getName() == null || !expr.getName().equals("newMockInstance")) {
+                return null;
+            }
+
+            if (expr.getParameters().length == 0) {
+                return null;
+            }
+
+            PsiElement param = expr.getParameters()[0];
+
+            if (param instanceof ClassConstantReference) {
+                PhpExpression ref = ((ClassConstantReference) param).getClassReference();
+                if (ref instanceof ClassReference) {
+                    return new PhpType().add(ref);
+                }
+            } else if (param instanceof StringLiteralExpression) {
+                return new PhpType().add(((StringLiteralExpression) param).getContents());
+            }
+        }
+
+        return null;
+    }
+
+    @Override
+    public Collection<? extends PhpNamedElement> getBySignature(String s, Set<String> set, int i, Project project) {
+        return null;
+    }
+}