KT-14191 FormattingModelInconsistencyException Exception on applying intention Replace with count()

#KT-14191 Fixed

idea/idea-analysis/src/org/jetbrains/kotlin/idea/util/CommentSaver.kt

@@ -285,6 +285,7 @@ class CommentSaver(originalElements: PsiChildRange, private val saveLineBreaks:
             val document = psiDocumentManager.getDocument(file)
             if (document != null) {
                 psiDocumentManager.doPostponedOperationsAndUnblockDocument(document)
+                psiDocumentManager.commitDocument(document)
             }
             CodeStyleManager.getInstance(project).adjustLineIndent(file, resultElements.textRange)
         }

idea/testData/intentions/loopToCallChain/count/KT14191.kt

@@ -0,0 +1,12 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'filterNotNull().count()'"
+// INTENTION_TEXT_2: "Replace with 'asSequence().filterNotNull().count()'"
+fun f11(list: List<Any?>): Int{
+    var objs = 0
+    <caret>for (d in list) {
+        if (d != null) {
+            objs++
+        }
+    }
+    return objs
+}

idea/testData/intentions/loopToCallChain/count/KT14191.kt.after

@@ -0,0 +1,9 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'filterNotNull().count()'"
+// INTENTION_TEXT_2: "Replace with 'asSequence().filterNotNull().count()'"
+fun f11(list: List<Any?>): Int{
+    val <caret>objs = list
+            .filterNotNull()
+            .count()
+    return objs
+}

idea/testData/intentions/loopToCallChain/count/KT14191.kt.after2

@@ -0,0 +1,10 @@
+// WITH_RUNTIME
+// INTENTION_TEXT: "Replace with 'filterNotNull().count()'"
+// INTENTION_TEXT_2: "Replace with 'asSequence().filterNotNull().count()'"
+fun f11(list: List<Any?>): Int{
+    val <caret>objs = list
+            .asSequence()
+            .filterNotNull()
+            .count()
+    return objs
+}

idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTest2Generated.java

@@ -235,6 +235,12 @@ public class IntentionTest2Generated extends AbstractIntentionTest2 {
             KotlinTestUtils.assertAllTestsPresentByMetadata(this.getClass(), new File("idea/testData/intentions/loopToCallChain/count"), Pattern.compile("^([\\w\\-_]+)\\.kt$"), true);
         }
 
+        @TestMetadata("KT14191.kt")
+        public void testKT14191() throws Exception {
+            String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/loopToCallChain/count/KT14191.kt");
+            doTest(fileName);
+        }
+
         @TestMetadata("long.kt")
         public void testLong() throws Exception {
             String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/loopToCallChain/count/long.kt");

idea/tests/org/jetbrains/kotlin/idea/intentions/IntentionTestGenerated.java

@@ -8143,6 +8143,12 @@ public class IntentionTestGenerated extends AbstractIntentionTest {
                 KotlinTestUtils.assertAllTestsPresentByMetadata(this.getClass(), new File("idea/testData/intentions/loopToCallChain/count"), Pattern.compile("^([\\w\\-_]+)\\.kt$"), true);
             }
 
+            @TestMetadata("KT14191.kt")
+            public void testKT14191() throws Exception {
+                String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/loopToCallChain/count/KT14191.kt");
+                doTest(fileName);
+            }
+
             @TestMetadata("long.kt")
             public void testLong() throws Exception {
                 String fileName = KotlinTestUtils.navigationMetadata("idea/testData/intentions/loopToCallChain/count/long.kt");