From 8c31853b3a5a51945d7356ad500b9a149ecc07aa Mon Sep 17 00:00:00 2001
From: Nikita Stroganov <stroganov.n.36@gmail.com>
Date: Wed, 7 Dec 2022 16:47:16 +0300
Subject: [PATCH] Fix IncorrectOperationException on close UtbotContext

---
 .../plugin/inspection/UnitTestBotInspectionContext.kt        | 5 +++++
 .../plugin/inspection/UnitTestBotInspectionManager.kt        | 2 ++
 .../inspection/UnitTestBotInspectionToolPresentation.kt      | 2 ++
 3 files changed, 9 insertions(+)

diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionContext.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionContext.kt
index e44cdb5de0..63769d71f9 100644
--- a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionContext.kt
+++ b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionContext.kt
@@ -1,3 +1,5 @@
+@file:Suppress("UnstableApiUsage")
+
 package org.utbot.intellij.plugin.inspection
 
 import com.intellij.codeInspection.ex.*
@@ -41,6 +43,9 @@ class UnitTestBotInspectionContext(
     }
 
     override fun close(noSuspiciousCodeFound: Boolean) {
+        if (!noSuspiciousCodeFound && (view == null || view.isRerun)) {
+            return
+        }
         myPresentationMap.clear()
         super.close(noSuspiciousCodeFound)
     }
diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionManager.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionManager.kt
index 82ed4e87f8..c942e014d9 100644
--- a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionManager.kt
+++ b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionManager.kt
@@ -1,3 +1,5 @@
+@file:Suppress("UnstableApiUsage")
+
 package org.utbot.intellij.plugin.inspection
 
 import com.intellij.codeInspection.ex.GlobalInspectionContextImpl
diff --git a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionToolPresentation.kt b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionToolPresentation.kt
index 264a37f32e..f777c2cfb6 100644
--- a/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionToolPresentation.kt
+++ b/utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/inspection/UnitTestBotInspectionToolPresentation.kt
@@ -1,3 +1,5 @@
+@file:Suppress("UnstableApiUsage")
+
 package org.utbot.intellij.plugin.inspection
 
 import com.intellij.codeInspection.CommonProblemDescriptor