AT-1811 Add test for move class to package refactoring on Hadoop

GitOrigin-RevId: 9d11a7b9cb9c48139e12c067786aff40b5f855d6

java/performancePlugin/src/com/intellij/java/performancePlugin/JavaCommandProvider.java

@@ -16,7 +16,8 @@ final class JavaCommandProvider implements CommandProvider {
       CreateJavaFileCommand.PREFIX, CreateJavaFileCommand::new,
       RenameDirectoryAsPackageCommand.PREFIX, RenameDirectoryAsPackageCommand::new,
       ChangeJavaSignatureCommand.PREFIX, ChangeJavaSignatureCommand::new,
-      InlineJavaMethodCommand.PREFIX, InlineJavaMethodCommand::new
+      InlineJavaMethodCommand.PREFIX, InlineJavaMethodCommand::new,
+      MoveClassToPackageCommand.PREFIX, MoveClassToPackageCommand::new
     );
   }
 }

java/performancePlugin/src/com/intellij/java/performancePlugin/MoveClassToPackageCommand.kt

@@ -0,0 +1,71 @@
+// Copyright 2000-2025 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
+package com.intellij.java.performancePlugin
+
+import com.intellij.openapi.application.EDT
+import com.intellij.openapi.application.readAction
+import com.intellij.openapi.application.runReadAction
+import com.intellij.openapi.application.writeIntentReadAction
+import com.intellij.openapi.fileEditor.FileEditorManager
+import com.intellij.openapi.roots.ProjectRootManager
+import com.intellij.openapi.ui.playback.PlaybackContext
+import com.intellij.openapi.ui.playback.commands.PlaybackCommandCoroutineAdapter
+import com.intellij.psi.PsiClass
+import com.intellij.psi.PsiDocumentManager
+import com.intellij.psi.PsiManager
+import com.intellij.psi.util.PsiTreeUtil
+import com.intellij.refactoring.BaseRefactoringProcessor
+import com.intellij.refactoring.PackageWrapper
+import com.intellij.refactoring.move.moveClassesOrPackages.AutocreatingSingleSourceRootMoveDestination
+import com.intellij.refactoring.move.moveClassesOrPackages.MoveClassesOrPackagesProcessor
+import com.jetbrains.performancePlugin.PerformanceTestSpan
+import kotlinx.coroutines.Dispatchers
+import kotlinx.coroutines.withContext
+
+/**
+ * Move the class under the caret to the given package in the same source root.
+ * Syntax: %moveClassToPackage packageName
+ */
+@Suppress("DuplicatedCode")
+class MoveClassToPackageCommand(text: String, line: Int) : PlaybackCommandCoroutineAdapter(text, line) {
+
+  companion object {
+    const val PREFIX: String = "${CMD_PREFIX}moveClassToPackage"
+    const val SPAN_NAME: String = "moveClassToPackage"
+  }
+
+  override suspend fun doExecute(context: PlaybackContext) {
+    val project = context.project
+    val targetPackageName: String = extractCommandArgument(PREFIX)
+
+    val editor = readAction { FileEditorManager.getInstance(context.project).selectedTextEditor }
+                 ?: throw IllegalArgumentException("There is no selected editor")
+
+    val elementUnderCaret = readAction {
+      PsiDocumentManager.getInstance(project).getPsiFile(editor.document)?.findElementAt(editor.caretModel.offset)
+    } ?: throw IllegalStateException("Unable to locate the element under the caret")
+
+    val psiClass = PsiTreeUtil.getParentOfType(elementUnderCaret, PsiClass::class.java)
+                   ?: throw IllegalStateException("Caret is not inside a class")
+
+    val targetPackage = PackageWrapper(PsiManager.getInstance(project), targetPackageName)
+    readAction { if (!targetPackage.exists()) throw IllegalArgumentException("Package $targetPackageName does not exist") }
+
+    val sourceRoot = readAction { ProjectRootManager.getInstance(project).fileIndex.getSourceRootForFile(editor.virtualFile) }
+                     ?: throw IllegalArgumentException("No source root found for ${editor.virtualFile.path}")
+
+    val destination = AutocreatingSingleSourceRootMoveDestination(targetPackage, sourceRoot)
+
+    val span = PerformanceTestSpan.TRACER.spanBuilder(SPAN_NAME).startSpan()
+
+    val processor: BaseRefactoringProcessor = runReadAction {
+      MoveClassesOrPackagesProcessor(project, arrayOf(psiClass), destination, false, false, null)
+    }
+
+    withContext(Dispatchers.EDT) {
+      writeIntentReadAction {
+        processor.run()
+        span.end()
+      }
+    }
+  }
+}

plugins/performanceTesting/commands-model/src/com/intellij/tools/ide/performanceTesting/commands/generalCommandChain.kt

@@ -1280,3 +1280,9 @@ fun <T : CommandChain> T.changeJavaSignature(action: ChangeJavaSignatureAction,
 fun <T : CommandChain> T.inlineJavaMethod(): T = apply {
   addCommand("${CMD_PREFIX}inlineJavaMethod")
 }
+
+/** @see com.intellij.java.performancePlugin.MoveClassToPackageCommand */
+@Suppress("KDocUnresolvedReference")
+fun <T : CommandChain> T.moveClassToPackage(targetPackage: String): T = apply {
+  addCommand("${CMD_PREFIX}moveClassToPackage $targetPackage")
+}