Adjust move code tests for KEEP_FIRST_COLUMN_COMMENT option

#KT-4175 In Progress

idea/testData/codeInsight/moveUpDown/classBodyDeclarations/class/classWithoutBody1.kt

@@ -9,3 +9,5 @@ class D {
 // class C
 class C {
 }
+
+// SET_FALSE: KEEP_FIRST_COLUMN_COMMENT

idea/testData/codeInsight/moveUpDown/classBodyDeclarations/class/classWithoutBody1.kt.after

@@ -9,3 +9,5 @@ class D {
 // class C
 class C {
 }
+
+// SET_FALSE: KEEP_FIRST_COLUMN_COMMENT

idea/testData/codeInsight/moveUpDown/classBodyDeclarations/class/classWithoutBody3.kt.after

@@ -2,6 +2,6 @@
 class D {
 
     // MOVE: down
-    // class A
+// class A
     <caret>class A
 }

idea/tests/org/jetbrains/kotlin/idea/codeInsight/moveUpDown/AbstractCodeMoverTest.java

@@ -23,11 +23,16 @@ import com.intellij.openapi.application.ApplicationManager;
 import com.intellij.openapi.editor.actionSystem.EditorAction;
 import com.intellij.openapi.extensions.Extensions;
 import com.intellij.openapi.util.io.FileUtil;
+import com.intellij.psi.codeStyle.CodeStyleSettings;
 import com.intellij.testFramework.LightCodeInsightTestCase;
+import junit.framework.ComparisonFailure;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.kotlin.formatter.FormatSettingsUtil;
 import org.jetbrains.kotlin.idea.codeInsight.upDownMover.KotlinDeclarationMover;
 import org.jetbrains.kotlin.idea.codeInsight.upDownMover.KotlinExpressionMover;
 import org.jetbrains.kotlin.test.InTextDirectivesUtils;
+import org.jetbrains.kotlin.test.KotlinTestUtils;
+import org.jetbrains.kotlin.test.SettingsConfigurator;
 
 import java.io.File;
 
@@ -80,19 +85,35 @@ public abstract class AbstractCodeMoverTest extends LightCodeInsightTestCase {
         assertEquals("Invalid applicability", isApplicableExpected, info.toMove2 != null);
 
         if (isApplicableExpected) {
-            invokeAndCheck(path, down);
+            invokeAndCheck(fileText, path, down);
         }
     }
 
-    private void invokeAndCheck(@NotNull String path, final boolean down) {
+    private void invokeAndCheck(@NotNull String fileText, @NotNull String path, final boolean down) {
-        ApplicationManager.getApplication().runWriteAction(new Runnable() {
+        CodeStyleSettings codeStyleSettings = FormatSettingsUtil.getSettings();
-            @Override
+        SettingsConfigurator configurator = FormatSettingsUtil.createConfigurator(fileText, codeStyleSettings);
-            public void run() {
+        configurator.configureSettings();
-                EditorAction action = down ? new MoveStatementDownAction() : new MoveStatementUpAction();
+
-                action.actionPerformed(getEditor(), getCurrentEditorDataContext());
+        try {
+            ApplicationManager.getApplication().runWriteAction(new Runnable() {
+                @Override
+                public void run() {
+                    EditorAction action = down ? new MoveStatementDownAction() : new MoveStatementUpAction();
+                    action.actionPerformed(getEditor(), getCurrentEditorDataContext());
+                }
+            });
+
+            String afterFilePath = path + ".after";
+            try {
+                checkResultByFile(afterFilePath);
             }
-        });
+            catch (ComparisonFailure e) {
-        checkResultByFile(path + ".after");
+                KotlinTestUtils.assertEqualsToFile(new File(afterFilePath), getEditor());
+            }
+        }
+        finally {
+            codeStyleSettings.clearCodeStyleSettings();
+        }
     }
 
     @NotNull