IDEA hangs during test generation #1207

Author: Vassiliy-Kudryashov

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/CodeGenerationController.kt

@@ -140,14 +140,14 @@ object CodeGenerationController {
             }
         }
 
-        run(THREAD_POOL, indicator) {
+        run(THREAD_POOL, indicator, "Waiting for per-class Sarif reports") {
             waitForCountDown(latch, indicator = indicator) {
-                run(EDT_LATER, indicator) {
-                    run(WRITE_ACTION, indicator) {
+                run(EDT_LATER, indicator,"Go to EDT for utility class creation") {
+                    run(WRITE_ACTION, indicator, "Need write action for utility class creation") {
                         createUtilityClassIfNeed(utilClassListener, model, baseTestDirectory, indicator)
-                        run(EDT_LATER, indicator) {
+                        run(EDT_LATER, indicator, "Proceed test report") {
                             proceedTestReport(proc, model)
-                            run(THREAD_POOL, indicator) {
+                            run(THREAD_POOL, indicator, "Generate summary Sarif report") {
                                 val sarifReportsPath =
                                     model.testModule.getOrCreateSarifReportsPath(model.testSourceRoot)
                                 UtTestsDialogProcessor.updateIndicator(indicator, UtTestsDialogProcessor.ProgressRange.SARIF, "Merge Sarif reports", 0.75)
@@ -157,9 +157,9 @@ object CodeGenerationController {
                                     RunConfigurationHelper.runTestsWithCoverage(model, testFilesPointers)
                                 }
                                 proc.forceTermination()
-                                UtTestsDialogProcessor.updateIndicator(indicator, UtTestsDialogProcessor.ProgressRange.SARIF, "Start tests with coverage", 1.0)
+                                UtTestsDialogProcessor.updateIndicator(indicator, UtTestsDialogProcessor.ProgressRange.SARIF, "Generation finished", 1.0)
 
-                                invokeLater {
+                                run(EDT_LATER, null, "Run sarif-based inspections") {
                                     runInspectionsIfNeeded(model, srcClassPathToSarifReport)
                                 }
                             }
@@ -339,8 +339,8 @@ object CodeGenerationController {
 
         val utUtilsText = utilClassKind.getUtilClassText(model.codegenLanguage)
 
-        run(EDT_LATER, indicator) {
-            run(WRITE_ACTION, indicator) {
+        run(EDT_LATER, indicator, "Overwrite utility class") {
+            run(WRITE_ACTION, indicator, "Overwrite utility class") {
                 unblockDocument(model.project, utilsClassDocument)
                 executeCommand {
                     utilsClassDocument.setText(utUtilsText.replace("jdk.internal.misc", "sun.misc"))
@@ -510,7 +510,7 @@ object CodeGenerationController {
     private fun waitForCountDown(latch: CountDownLatch, timeout: Long = 5, timeUnit: TimeUnit = TimeUnit.SECONDS, indicator : ProgressIndicator, action: Runnable) {
         try {
             if (!latch.await(timeout, timeUnit)) {
-                run(THREAD_POOL, indicator) { waitForCountDown(latch, timeout, timeUnit, indicator, action) }
+                run(THREAD_POOL, indicator, "Waiting for ${latch.count} sarif report(s) in a loop") { waitForCountDown(latch, timeout, timeUnit, indicator, action) }
             } else {
                 action.run()
             }
@@ -633,7 +633,7 @@ object CodeGenerationController {
         val editor = CodeInsightUtil.positionCursorAtLBrace(testClass.project, filePointer.containingFile, testClass)
         //TODO: Use PsiDocumentManager.getInstance(model.project).getDocument(file)
         // if we don't want to open _all_ new files with tests in editor one-by-one
-        run(THREAD_POOL, indicator) {
+        run(THREAD_POOL, indicator, "Rendering test code") {
             val (generatedTestsCode, utilClassKind) = try {
                 proc.render(
                     testSetsId,
@@ -658,8 +658,8 @@ object CodeGenerationController {
                 return@run
             }
             utilClassListener.onTestClassGenerated(utilClassKind)
-            run(EDT_LATER, indicator) {
-                run(WRITE_ACTION, indicator) {
+            run(EDT_LATER, indicator, "Writing generation text to documents") {
+                run(WRITE_ACTION, indicator, "Writing generation text to documents") {
                     try {
                         unblockDocument(testClass.project, editor.document)
                         // TODO: JIRA:1246 - display warnings if we rewrite the file
@@ -679,9 +679,9 @@ object CodeGenerationController {
 
                     // reformatting before creating reports due to
                     // SarifReport requires the final version of the generated tests code
-                    run(THREAD_POOL, indicator) {
+//                    run(THREAD_POOL, indicator, "Intention helper") {
 //                        IntentionHelper(model.project, editor, filePointer).applyIntentions()
-                        run(EDT_LATER, indicator) {
+                        run(EDT_LATER, indicator, "Tests reformatting") {
                             try {
                                 runWriteCommandAction(filePointer.project, "UtBot tests reformatting", null, {
                                     reformat(model, filePointer, testClassUpdated)
@@ -709,7 +709,7 @@ object CodeGenerationController {
 
                             unblockDocument(testClassUpdated.project, editor.document)
                         }
-                    }
+//                    }
                 }
             }
         }