From 5c077f687d88188dfd5a0b88e95a8554e82b08ec Mon Sep 17 00:00:00 2001
From: IlyaMuravjov <muravjovilya@gmail.com>
Date: Fri, 21 Apr 2023 13:00:36 +0300
Subject: [PATCH] Add Spring config to files to compile

---
 .../utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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 84a578d736..d68b73bf18 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
@@ -177,7 +177,9 @@ object UtTestsDialogProcessor {
                 }
         }
 
-        val filesToCompile = model.srcClasses.map { it.containingFile.virtualFile }.toTypedArray()
+        val filesToCompile = (model.srcClasses + listOfNotNull(springConfigClass))
+            .map { it.containingFile.virtualFile }
+            .toTypedArray()
 
         val promise = compile(project, filesToCompile, springConfigClass)
         promise.onSuccess {