diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt
index 7717f7d2af..ba4c3e4f58 100644
--- a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt
+++ b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt
@@ -28,7 +28,6 @@ import org.jetbrains.kotlin.idea.util.module
import org.utbot.engine.util.mockListeners.ForceMockListener
import org.utbot.framework.JdkPathService
import org.utbot.framework.UtSettings
-import org.utbot.framework.codegen.ParametrizedTestSource
import org.utbot.framework.plugin.api.TestCaseGenerator
import org.utbot.framework.plugin.api.UtMethod
import org.utbot.framework.plugin.api.UtMethodTestSet
@@ -38,9 +37,6 @@ import org.utbot.framework.plugin.api.util.withUtContext
import org.utbot.intellij.plugin.generator.CodeGenerationController.generateTests
import org.utbot.intellij.plugin.models.GenerateTestsModel
import org.utbot.intellij.plugin.ui.GenerateTestsDialogWindow
-import org.utbot.intellij.plugin.ui.utils.jdkVersion
-import org.utbot.intellij.plugin.ui.utils.showErrorDialogLater
-import org.utbot.intellij.plugin.ui.utils.testModule
import org.utbot.intellij.plugin.util.IntelliJApiHelper
import org.utbot.intellij.plugin.util.PluginJdkPathProvider
import org.utbot.intellij.plugin.util.signature
@@ -54,6 +50,11 @@ import org.utbot.common.filterWhen
import org.utbot.engine.util.mockListeners.ForceStaticMockListener
import org.utbot.framework.plugin.api.testFlow
import org.utbot.intellij.plugin.settings.Settings
+import org.utbot.intellij.plugin.ui.utils.isGradle
+import org.utbot.intellij.plugin.ui.utils.jdkVersion
+import org.utbot.intellij.plugin.ui.utils.showErrorDialogLater
+import org.utbot.intellij.plugin.ui.utils.suitableTestSourceRoots
+import org.utbot.intellij.plugin.ui.utils.testModule
import org.utbot.intellij.plugin.util.isAbstract
import kotlin.reflect.KClass
import kotlin.reflect.full.functions
@@ -89,6 +90,15 @@ object UtTestsDialogProcessor {
return null
}
+ if (project.isGradle() && testModule.suitableTestSourceRoots().isEmpty()) {
+ val errorMessage = """
+ No test source roots found in the project.
+ Please, create or configure at least one test source root.
+ """.trimIndent()
+ showErrorDialogLater(project, errorMessage, "Test source roots not found")
+ return null
+ }
+
return GenerateTestsDialogWindow(
GenerateTestsModel(
project,