Fix disabled action (#1481)

* Add checking languageAssistant

* Add python test generation for file

Co-authored-by: Vyacheslav Tamarin <vyacheslav.tamarin@yandex.ru>

Author: tamarinvs19

utbot-intellij-python/src/main/kotlin/org/utbot/intellij/plugin/language/python/PythonLanguageAssistant.kt

@@ -44,13 +44,18 @@ object PythonLanguageAssistant : LanguageAssistant() {
     }
 
     private fun getPsiTargets(e: AnActionEvent): Targets? {
-        val editor = e.getData(CommonDataKeys.EDITOR) ?: return null
+        val editor = e.getData(CommonDataKeys.EDITOR)
         val file = e.getData(CommonDataKeys.PSI_FILE) as? PyFile ?: return null
-        val element = findPsiElement(file, editor) ?: return null
 
         if (file.module?.sdk?.sdkType !is PythonSdkType)
             return null
 
+        val element = if (editor != null) {
+            findPsiElement(file, editor) ?: return null
+        } else {
+            e.getData(CommonDataKeys.PSI_ELEMENT) ?: return null
+        }
+
         val containingFunction = getContainingElement<PyFunction>(element)
         val containingClass = getContainingElement<PyClass>(element)
 
@@ -67,7 +72,8 @@ object PythonLanguageAssistant : LanguageAssistant() {
         if (functions.isEmpty())
             return null
 
-        val focusedFunction = if (functions.any { it.name == containingFunction?.name }) containingFunction else null
+        val focusedFunction =
+            if (functions.any { it.name == containingFunction?.name }) containingFunction else null
         return Targets(functions.toSet(), containingClass, focusedFunction, file)
     }
 

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/actions/GenerateTestsAction.kt

@@ -10,6 +10,11 @@ class GenerateTestsAction : AnAction() {
     }
 
     override fun update(e: AnActionEvent) {
-        LanguageAssistant.get(e)?.update(e)
+        val languageAssistant = LanguageAssistant.get(e)
+        if (languageAssistant == null) {
+            e.presentation.isEnabled = false
+        } else {
+            languageAssistant.update(e)
+        }
     }
 }