Added "Add remaining branches with import" quick fix #KT-16033 Fixed

idea/testData/quickfix/when/addRemainingBranchesEnumImport1.kt

@@ -0,0 +1,12 @@
+// "Add remaining branches with import" "true"
+// WITH_RUNTIME
+enum class Foo {
+    A, B, C
+}
+
+class Test {
+    fun foo(e: Foo) {
+        when<caret> (e) {
+        }
+    }
+}

idea/testData/quickfix/when/addRemainingBranchesEnumImport1.kt.after

@@ -0,0 +1,17 @@
+import Foo.*
+
+// "Add remaining branches with import" "true"
+// WITH_RUNTIME
+enum class Foo {
+    A, B, C
+}
+
+class Test {
+    fun foo(e: Foo) {
+        when<caret> (e) {
+            A -> TODO()
+            B -> TODO()
+            C -> TODO()
+        }
+    }
+}

idea/testData/quickfix/when/addRemainingBranchesEnumImport2.kt

@@ -0,0 +1,15 @@
+// "Add remaining branches with import" "true"
+// WITH_RUNTIME
+import Foo.*
+
+enum class Foo {
+    A, B, C
+}
+
+class Test {
+    fun foo(e: Foo) {
+        when<caret> (e) {
+            A, Foo.B -> TODO()
+        }
+    }
+}

idea/testData/quickfix/when/addRemainingBranchesEnumImport2.kt.after

@@ -0,0 +1,16 @@
+// "Add remaining branches with import" "true"
+// WITH_RUNTIME
+import Foo.*
+
+enum class Foo {
+    A, B, C
+}
+
+class Test {
+    fun foo(e: Foo) {
+        when<caret> (e) {
+            A, B -> TODO()
+            C -> TODO()
+        }
+    }
+}

idea/testData/quickfix/when/addRemainingBranchesEnumImport3.kt

@@ -0,0 +1,25 @@
+// "Add remaining branches with import" "true"
+// WITH_RUNTIME
+import Foo.*
+
+enum class Foo {
+    A, B, C
+}
+
+enum class Bar {
+    A, B, C
+}
+
+class Test {
+    fun foo(e: Foo) {
+        when (e) {
+            A -> TODO()
+            B -> TODO()
+            C -> TODO()
+        }
+    }
+    fun bar(e: Bar) {
+        when<caret> (e) {
+        }
+    }
+}

idea/testData/quickfix/when/addRemainingBranchesEnumImport3.kt.after

@@ -0,0 +1,28 @@
+// "Add remaining branches with import" "true"
+// WITH_RUNTIME
+import Foo.*
+
+enum class Foo {
+    A, B, C
+}
+
+enum class Bar {
+    A, B, C
+}
+
+class Test {
+    fun foo(e: Foo) {
+        when (e) {
+            A -> TODO()
+            B -> TODO()
+            C -> TODO()
+        }
+    }
+    fun bar(e: Bar) {
+        when<caret> (e) {
+            Bar.A -> TODO()
+            Bar.B -> TODO()
+            Bar.C -> TODO()
+        }
+    }
+}

idea/testData/quickfix/when/addRemainingBranchesEnumImport4.kt

@@ -0,0 +1,25 @@
+// "Add remaining branches with import" "true"
+// WITH_RUNTIME
+import Foo.*
+
+enum class Foo {
+    A, B, C
+}
+
+enum class Baz {
+    AA, B, CC
+}
+
+class Test {
+    fun foo(e: Foo) {
+        when (e) {
+            A -> TODO()
+            B -> TODO()
+            C -> TODO()
+        }
+    }
+    fun baz(e: Baz) {
+        when<caret> (e) {
+        }
+    }
+}

idea/testData/quickfix/when/addRemainingBranchesEnumImport4.kt.after

@@ -0,0 +1,30 @@
+// "Add remaining branches with import" "true"
+// WITH_RUNTIME
+import Baz.*
+import Foo.*
+import Foo.B
+
+enum class Foo {
+    A, B, C
+}
+
+enum class Baz {
+    AA, B, CC
+}
+
+class Test {
+    fun foo(e: Foo) {
+        when (e) {
+            A -> TODO()
+            B -> TODO()
+            C -> TODO()
+        }
+    }
+    fun baz(e: Baz) {
+        when<caret> (e) {
+            AA -> TODO()
+            Baz.B -> TODO()
+            CC -> TODO()
+        }
+    }
+}