From 5627cceb6a0952e3525db60743eae4fee739ac4d Mon Sep 17 00:00:00 2001 From: Anna Kozlova Date: Fri, 13 Sep 2024 15:39:54 +0200 Subject: [PATCH] [kotlin] k2 generate tests: add command after popup chooser ^KTIJ-31310 fixed GitOrigin-RevId: ef37b0f8f751a44391de3938c97c1f23df37cacc --- .../generate/KotlinGenerateTestSupportActionBase.kt | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/plugins/kotlin/code-insight/kotlin.code-insight.k2/src/org/jetbrains/kotlin/idea/k2/codeinsight/generate/KotlinGenerateTestSupportActionBase.kt b/plugins/kotlin/code-insight/kotlin.code-insight.k2/src/org/jetbrains/kotlin/idea/k2/codeinsight/generate/KotlinGenerateTestSupportActionBase.kt index c39a26615078..e5f4845c9b9a 100644 --- a/plugins/kotlin/code-insight/kotlin.code-insight.k2/src/org/jetbrains/kotlin/idea/k2/codeinsight/generate/KotlinGenerateTestSupportActionBase.kt +++ b/plugins/kotlin/code-insight/kotlin.code-insight.k2/src/org/jetbrains/kotlin/idea/k2/codeinsight/generate/KotlinGenerateTestSupportActionBase.kt @@ -34,6 +34,7 @@ import org.jetbrains.kotlin.idea.core.insertMembersAfterAndReformat import org.jetbrains.kotlin.idea.core.overrideImplement.BodyType import org.jetbrains.kotlin.idea.core.overrideImplement.generateUnsupportedOrSuperCall import org.jetbrains.kotlin.idea.createFromUsage.setupEditorSelection +import org.jetbrains.kotlin.idea.util.application.executeCommand import org.jetbrains.kotlin.idea.util.application.isUnitTestMode import org.jetbrains.kotlin.lexer.KtTokens import org.jetbrains.kotlin.psi.* @@ -123,7 +124,11 @@ abstract class KotlinGenerateTestSupportActionBase( methodKind.getFileTemplateDescriptor(it) != null && isApplicableTo(it, klass) } - chooseAndPerform(editor, frameworks) { doGenerate(editor, file, klass, it) } + chooseAndPerform(editor, frameworks) { + project.executeCommand(KotlinBundle.message("command.generate.test.support.generate.test.function"), null) { + doGenerate(editor, file, klass, it) + } + } } }