Enable mute for KotlinCompletionTestCase

compiler/tests-common/tests/org/jetbrains/kotlin/test/KotlinTestUtils.java

@@ -717,6 +717,18 @@ public class KotlinTestUtils {
         MuteWithDatabaseKt.runTest(testCase, test);
     }
 
+    public static void runTestWithThrowable(@NotNull TestCase testCase, @NotNull RunnableWithThrowable test) {
+        MuteWithDatabaseKt.runTest(testCase, () -> {
+            try {
+                test.run();
+            }
+            catch (Throwable throwable) {
+                throw new IllegalStateException(throwable);
+            }
+            return null;
+        });
+    }
+
     // In this test runner version the `testDataFile` parameter is annotated by `TestDataFile`.
     // So only file paths passed to this parameter will be used in navigation actions, like "Navigate to testdata" and "Related Symbol..."
     public static void runTest(DoTest test, TargetBackend targetBackend, @TestDataFile String testDataFile) throws Exception {

compiler/tests-common/tests/org/jetbrains/kotlin/test/RunnableWithThrowable.java

@@ -0,0 +1,10 @@
+/*
+ * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.test;
+
+public interface RunnableWithThrowable {
+    void run() throws Throwable;
+}

idea/idea-completion/tests/org/jetbrains/kotlin/idea/completion/test/KotlinCompletionTestCase.java

@@ -25,4 +25,10 @@ abstract public class KotlinCompletionTestCase extends CompletionTestCase {
         VfsRootAccess.disallowRootAccess(KotlinTestUtils.getHomeDirectory());
         super.tearDown();
     }
+
+    @Override
+    protected void runTest() throws Throwable {
+        //noinspection Convert2MethodRef
+        KotlinTestUtils.runTestWithThrowable(this, () -> super.runTest());
+    }
 }