From 781c565ec9b4b13af0dd7277ed339ed8a2f960de Mon Sep 17 00:00:00 2001
From: Yan Zhulanow <yan.zhulanow@jetbrains.com>
Date: Fri, 19 Dec 2014 14:41:56 +0300
Subject: [PATCH] Revert completion settings back on Android completion test
 finish

---
 .../android/AbstractAndroidCompletionTest.kt  | 28 +++++++++++--------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/plugins/android-idea-plugin/tests/org/jetbrains/jet/android/AbstractAndroidCompletionTest.kt b/plugins/android-idea-plugin/tests/org/jetbrains/jet/android/AbstractAndroidCompletionTest.kt
index 14a429abe37..15bb4e44d79 100644
--- a/plugins/android-idea-plugin/tests/org/jetbrains/jet/android/AbstractAndroidCompletionTest.kt
+++ b/plugins/android-idea-plugin/tests/org/jetbrains/jet/android/AbstractAndroidCompletionTest.kt
@@ -37,23 +37,20 @@ import com.intellij.codeInsight.CodeInsightSettings
 public abstract class AbstractAndroidCompletionTest : KotlinAndroidTestCase() {
     private var kotlinInternalModeOriginalValue: Boolean = false
 
+    private var codeCompletionOldValue: Boolean = false
+    private var smartTypeCompletionOldValue: Boolean = false
+
     override fun setUp() {
         super.setUp()
-        setAutoCompleteSetting(false)
-    }
 
-    private fun setAutoCompleteSetting(value: Boolean): Boolean {
         val settings = CodeInsightSettings.getInstance()
-        val oldValue: Boolean
-        if (completionType() == CompletionType.SMART) {
-            oldValue = settings.AUTOCOMPLETE_ON_SMART_TYPE_COMPLETION
-            settings.AUTOCOMPLETE_ON_SMART_TYPE_COMPLETION = value
+        codeCompletionOldValue = settings.AUTOCOMPLETE_ON_CODE_COMPLETION
+        smartTypeCompletionOldValue = settings.AUTOCOMPLETE_ON_SMART_TYPE_COMPLETION
+
+        when (completionType()) {
+            CompletionType.SMART -> settings.AUTOCOMPLETE_ON_SMART_TYPE_COMPLETION = false
+            CompletionType.BASIC -> settings.AUTOCOMPLETE_ON_CODE_COMPLETION = false
         }
-        else {
-            oldValue = settings.AUTOCOMPLETE_COMMON_PREFIX
-            settings.AUTOCOMPLETE_ON_CODE_COMPLETION = value
-        }
-        return oldValue
     }
 
     private fun completionType() = CompletionType.BASIC
@@ -72,4 +69,11 @@ public abstract class AbstractAndroidCompletionTest : KotlinAndroidTestCase() {
         return KotlinAndroidTestCaseBase.getPluginTestDataPathBase() + "/completion/" + getTestName(true) + "/"
     }
 
+    override fun tearDown() {
+        val settings = CodeInsightSettings.getInstance()
+        settings.AUTOCOMPLETE_ON_CODE_COMPLETION = codeCompletionOldValue
+        settings.AUTOCOMPLETE_ON_SMART_TYPE_COMPLETION = smartTypeCompletionOldValue
+
+        super<KotlinAndroidTestCase>.tearDown()
+    }
 }