From 806faf86b730bf3054c60e965357ea571e479e62 Mon Sep 17 00:00:00 2001
From: Natalia Selezneva <natalia.selezneva@jetbrains.com>
Date: Wed, 21 Feb 2018 13:22:02 +0300
Subject: [PATCH] Tests: allow file modification during highlighting for
 scripts

---
 .../jetbrains/kotlin/checkers/AbstractPsiCheckerTest.java   | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/idea/tests/org/jetbrains/kotlin/checkers/AbstractPsiCheckerTest.java b/idea/tests/org/jetbrains/kotlin/checkers/AbstractPsiCheckerTest.java
index 3eaae801751..4a0b1598d0d 100644
--- a/idea/tests/org/jetbrains/kotlin/checkers/AbstractPsiCheckerTest.java
+++ b/idea/tests/org/jetbrains/kotlin/checkers/AbstractPsiCheckerTest.java
@@ -17,8 +17,10 @@
 package org.jetbrains.kotlin.checkers;
 
 import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.psi.PsiFile;
 import com.intellij.rt.execution.junit.FileComparisonFailure;
 import com.intellij.spellchecker.inspections.SpellCheckingInspection;
+import com.intellij.testFramework.fixtures.impl.JavaCodeInsightTestFixtureImpl;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.kotlin.idea.caches.resolve.ResolutionUtils;
 import org.jetbrains.kotlin.idea.highlighter.NameHighlighter;
@@ -70,6 +72,10 @@ public abstract class AbstractPsiCheckerTest extends KotlinLightCodeInsightFixtu
 
     protected long checkHighlighting(boolean checkWarnings, boolean checkInfos, boolean checkWeakWarnings) {
         try {
+            PsiFile file = getFile();
+            if (file instanceof KtFile && ((KtFile) file).isScript() && myFixture instanceof JavaCodeInsightTestFixtureImpl) {
+                ((JavaCodeInsightTestFixtureImpl) myFixture).canChangeDocumentDuringHighlighting(true);
+            }
             return myFixture.checkHighlighting(checkWarnings, checkInfos, checkWeakWarnings);
         }
         catch (FileComparisonFailure e) {