Merge pull request #12 from cawolf/fixes-8-object-prophecy

Fixes #8: add ObjectProphecy doc

Author: cawolf

src/main/kotlin/de/cawolf/quickmock/intention/QuickMockCreator.kt

@@ -55,8 +55,9 @@ class QuickMockCreator : PsiElementBaseIntentionAction(), IntentionAction {
 
         // actually create mocks
         var currentAnchor = beginningOfClass
+        addMissingUseStatements.invoke(namespace, "\\Prophecy\\Prophecy\\ObjectProphecy")
         for (parameter in parameters) {
-            addMissingUseStatements.invoke(namespace, parameter)
+            addMissingUseStatements.invoke(namespace, parameter.type.toString())
             addMockAssignment.invoke(project, constructStatement, parameter)
 
             if (!PRIMITIVES_NOT_TO_ADD_OR_MOCK.contains(parameter.type.toString())) {

src/main/kotlin/de/cawolf/quickmock/intention/service/AddMissingUseStatements.kt

@@ -2,18 +2,16 @@ package de.cawolf.quickmock.intention.service
 
 import com.intellij.psi.PsiElement
 import com.jetbrains.php.codeInsight.PhpCodeInsightUtil
-import com.jetbrains.php.lang.psi.elements.Parameter
 import com.jetbrains.php.refactoring.PhpAliasImporter
 import de.cawolf.quickmock.intention.PRIMITIVES_NOT_TO_USE
 
 class AddMissingUseStatements {
-    fun invoke(namespace: PsiElement, parameter: Parameter) {
-        if (PRIMITIVES_NOT_TO_USE.contains(parameter.type.toString())) return
+    fun invoke(namespace: PsiElement, fqcn: String) {
+        if (PRIMITIVES_NOT_TO_USE.contains(fqcn)) return
 
-        val interfaceFqn = parameter.type.toString()
         val scopeHolder = PhpCodeInsightUtil.findScopeForUseOperator(namespace)
-        if (scopeHolder != null && PhpCodeInsightUtil.alreadyImported(scopeHolder, interfaceFqn) == null && PhpCodeInsightUtil.canImport(scopeHolder, interfaceFqn)) {
-            PhpAliasImporter.insertUseStatement(interfaceFqn, scopeHolder)
+        if (scopeHolder != null && PhpCodeInsightUtil.alreadyImported(scopeHolder, fqcn) == null && PhpCodeInsightUtil.canImport(scopeHolder, fqcn)) {
+            PhpAliasImporter.insertUseStatement(fqcn, scopeHolder)
         }
     }
 }

src/main/kotlin/de/cawolf/quickmock/intention/service/AddProperty.kt

@@ -13,7 +13,7 @@ class AddProperty {
         val docCommentType = IElementType.enumerate { it -> it.toString() == "PhpDocComment" }.first()
 
         val field = PhpPsiElementFactory.createFromText(project, classFieldsType, "class C { private \$${parameter.name}; }")
-        val docBlock = PhpPsiElementFactory.createFromText(project, docCommentType, "/** @var ${parameter.node.firstChildNode.text} */\n\$foo = null;")
+        val docBlock = PhpPsiElementFactory.createFromText(project, docCommentType, "/** @var ${parameter.node.firstChildNode.text}|ObjectProphecy */\n\$foo = null;")
 
         val docBlockAnchor = clazz.addAfter(docBlock, currentAnchor)
         return clazz.addAfter(field, docBlockAnchor)