IllegalStateException is thrown in IDEA together with notification th… (#1311)

IllegalStateException is thrown in IDEA together with notification that file exceeds configured limit #1295

Author: Vassiliy-Kudryashov

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

@@ -77,7 +77,7 @@ import org.utbot.intellij.plugin.models.packageName
 import org.utbot.intellij.plugin.process.EngineProcess
 import org.utbot.intellij.plugin.process.RdTestGenerationResult
 import org.utbot.intellij.plugin.sarif.SarifReportIdea
-import org.utbot.intellij.plugin.ui.CommonErrorNotifier
+import org.utbot.intellij.plugin.ui.CommonLoggingNotifier
 import org.utbot.intellij.plugin.ui.DetailsTestsReportNotifier
 import org.utbot.intellij.plugin.ui.SarifReportNotifier
 import org.utbot.intellij.plugin.ui.TestReportUrlOpeningListener
@@ -760,9 +760,12 @@ object CodeGenerationController {
  val project = model.project
  val codeStyleManager = CodeStyleManager.getInstance(project)
  val file = smartPointer.containingFile?: return
-
- if (file.virtualFile.length > UtSettings.maxTestFileSize) {
- CommonErrorNotifier.notify(
+ val fileLength = runReadAction {
+ FileDocumentManager.getInstance().getDocument(file.virtualFile)?.textLength
+ ?: file.virtualFile.length.toInt()
+ }
+ if (fileLength > UtSettings.maxTestFileSize && file.name != model.codegenLanguage.utilClassFileName) {
+ CommonLoggingNotifier().notify(
  "Size of ${file.virtualFile.presentableName} exceeds configured limit " +
  "(${FileUtil.byteCountToDisplaySize(UtSettings.maxTestFileSize.toLong())}), reformatting was skipped. " +
  "The limit can be configured in '{HOME_DIR}/.utbot/settings.properties' with 'maxTestFileSize' property",

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/ui/Notifications.kt

@@ -19,11 +19,14 @@ import com.intellij.ui.awt.RelativePoint
 import com.intellij.util.ui.JBFont
 import java.awt.Point
 import javax.swing.event.HyperlinkEvent
+import mu.KotlinLogging
 
 abstract class Notifier {
+ protected val logger = KotlinLogging.logger {}
+
  protected abstract val notificationType: NotificationType
  protected abstract val displayId: String
- protected abstract fun content(project: Project?, module: Module?, info: String): String
+ protected open fun content(project: Project?, module: Module?, info: String): String = info
 
  open fun notify(info: String, project: Project? = null, module: Module? = null) {
  notificationGroup
@@ -47,15 +50,28 @@ abstract class WarningNotifier : Notifier() {
 abstract class ErrorNotifier : Notifier() {
  final override val notificationType: NotificationType = NotificationType.ERROR
 
- final override fun notify(info: String, project: Project?, module: Module?) {
+ override fun notify(info: String, project: Project?, module: Module?) {
  super.notify(info, project, module)
  error(content(project, module, info))
  }
 }
 
 object CommonErrorNotifier : ErrorNotifier() {
  override val displayId: String = "UTBot plugin errors"
- override fun content(project: Project?, module: Module?, info: String): String = info
+}
+
+class CommonLoggingNotifier(val type :NotificationType = NotificationType.WARNING) : Notifier() {
+ override val displayId: String = "UTBot plugin errors"
+ override val notificationType = type
+
+ override fun notify(info: String, project: Project?, module: Module?) {
+ super.notify(info, project, module)
+ when (notificationType) {
+ NotificationType.WARNING -> logger.warn(content(project, module, info))
+ NotificationType.INFORMATION -> logger.info(content(project, module, info))
+ else -> logger.error(content(project, module, info))
+ }
+ }
 }
 
 object UnsupportedJdkNotifier : ErrorNotifier() {
@@ -113,8 +129,6 @@ object SarifReportNotifier : EventLogNotifier() {
  override val titleText: String = "" // no title
 
  override val urlOpeningListener: NotificationListener = NotificationListener.UrlOpeningListener(false)
-
- override fun content(project: Project?, module: Module?, info: String): String = info
 }
 
 object TestsReportNotifier : InformationUrlNotifier() {
@@ -123,8 +137,6 @@ object TestsReportNotifier : InformationUrlNotifier() {
  override val titleText: String = "UTBot: unit tests generated successfully"
 
  public override val urlOpeningListener: TestReportUrlOpeningListener = TestReportUrlOpeningListener
-
- override fun content(project: Project?, module: Module?, info: String): String = info
 }
 
 // TODO replace inheritance with decorators
@@ -134,8 +146,6 @@ object WarningTestsReportNotifier : WarningUrlNotifier() {
  override val titleText: String = "UTBot: unit tests generated with warnings"
 
  public override val urlOpeningListener: TestReportUrlOpeningListener = TestReportUrlOpeningListener
-
- override fun content(project: Project?, module: Module?, info: String): String = info
 }
 
 object DetailsTestsReportNotifier : EventLogNotifier() {
@@ -144,8 +154,6 @@ object DetailsTestsReportNotifier : EventLogNotifier() {
  override val titleText: String = "Test report details of the unit tests generation via UtBot"
 
  public override val urlOpeningListener: TestReportUrlOpeningListener = TestReportUrlOpeningListener
-
- override fun content(project: Project?, module: Module?, info: String): String = info
 }
 
 /**