From 1d8e237eebea0d4edac5947490a2804d17538095 Mon Sep 17 00:00:00 2001
From: Mikhail Pyltsin <Mikhail.Pyltsin@jetbrains.com>
Date: Tue, 21 May 2024 15:49:22 +0200
Subject: [PATCH] IJ-CR-133924 [java-test] EA-1144109 Fix
 IndexNotReadyException for other languages

- fix application run markers for scala

GitOrigin-RevId: eef28f08131202e99e8962c0fc3a50501791f3de
---
 .../src/com/intellij/testIntegration/JavaTestFramework.java     | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/java/java-impl/src/com/intellij/testIntegration/JavaTestFramework.java b/java/java-impl/src/com/intellij/testIntegration/JavaTestFramework.java
index 31d9ab3c3348..bf2f0901b6db 100644
--- a/java/java-impl/src/com/intellij/testIntegration/JavaTestFramework.java
+++ b/java/java-impl/src/com/intellij/testIntegration/JavaTestFramework.java
@@ -135,6 +135,7 @@ public abstract class JavaTestFramework implements JvmTestFramework {
 
   @Override
   public @Nullable PsiElement findSetUpMethod(@NotNull PsiElement clazz) {
+    if (DumbService.isDumb(clazz.getProject()) && !supportDumbMode(clazz)) return null;
     if (clazz instanceof PsiClass && isFrameworkAvailable(clazz)) {
       return findSetUpMethod((PsiClass)clazz);
     }
@@ -145,6 +146,7 @@ public abstract class JavaTestFramework implements JvmTestFramework {
 
   @Override
   public @Nullable PsiElement findTearDownMethod(@NotNull PsiElement clazz) {
+    if (DumbService.isDumb(clazz.getProject()) && !supportDumbMode(clazz)) return null;
     if (clazz instanceof PsiClass && isFrameworkAvailable(clazz)) {
       return findTearDownMethod((PsiClass)clazz);
     }