[lambda tests] AT-3596 Invoke IDE application cleanup

GitOrigin-RevId: 00df5fbd0272fe24eebb820643772fcd178fd1f4

Author: Nikita-Kudrin

tools/intellij.lambda.testFramework/src/com/intellij/lambda/testFramework/junit/BackgroundLambdaCleanupAfterEach.kt

@@ -1,9 +1,16 @@
 package com.intellij.lambda.testFramework.junit
 
+import com.intellij.ide.starter.coroutine.perTestSupervisorScope
+import com.intellij.ide.starter.utils.catchAll
 import com.intellij.lambda.testFramework.starter.IdeInstance
+import com.intellij.openapi.application.ApplicationManager
+import com.intellij.testFramework.common.cleanApplicationState
 import com.intellij.tools.ide.util.common.starterLogger
+import kotlinx.coroutines.runBlocking
+import kotlinx.coroutines.withTimeout
 import org.junit.jupiter.api.extension.AfterEachCallback
 import org.junit.jupiter.api.extension.ExtensionContext
+import kotlin.time.Duration.Companion.seconds
 
 /**
  * Ensures `BackgroundRunWithLambda.cleanUp()` is executed automatically after each test.
@@ -14,6 +21,26 @@ import org.junit.jupiter.api.extension.ExtensionContext
 class BackgroundLambdaCleanupAfterEach : AfterEachCallback {
   override fun afterEach(context: ExtensionContext) {
     starterLogger<BackgroundLambdaCleanupAfterEach>().info("Cleaning up Lambda test session(s) after test: ${context.displayName}")
+
+    invokeIdeApplicationCleanup()
     IdeInstance.cleanup()
   }
+
+  private fun invokeIdeApplicationCleanup() {
+    @Suppress("RAW_RUN_BLOCKING")
+    runBlocking(perTestSupervisorScope.coroutineContext) {
+      withTimeout(5.seconds) {
+        catchAll("Invoking test application cleanup") {
+          IdeInstance.ide.apply {
+            run("IDE test application cleanup") {
+              ApplicationManager.getApplication().cleanApplicationState()
+            }
+            runInBackend("IDE test application cleanup") {
+              ApplicationManager.getApplication().cleanApplicationState()
+            }
+          }
+        }
+      }
+    }
+  }
 }

tools/intellij.lambda.testFramework/src/com/intellij/lambda/testFramework/starter/IdeInstance.kt

@@ -12,6 +12,8 @@ import com.intellij.lambda.testFramework.utils.IdeLambdaStarter.runIdeWithLambda
 import com.intellij.lambda.testFramework.utils.LambdaTestPluginHolder
 import com.intellij.tools.ide.util.common.starterLogger
 import kotlinx.coroutines.runBlocking
+import kotlinx.coroutines.withTimeout
+import kotlin.time.Duration.Companion.seconds
 
 private val LOG = starterLogger<IdeInstance>()
 
@@ -66,8 +68,10 @@ object IdeInstance {
   fun cleanup() = synchronized(this) {
     @Suppress("RAW_RUN_BLOCKING")
     runBlocking(testSuiteSupervisorScope.coroutineContext) {
-      catchAll("IDE instance cleanup") {
-        ide.cleanUp()
+      withTimeout(5.seconds) {
+        catchAll("IDE instance cleanup") {
+          ide.cleanUp()
+        }
       }
     }
   }