Implement members: fix it works correctly for data class

#KT-36686 Fixed

idea/idea-core/src/org/jetbrains/kotlin/idea/core/overrideImplement/OverrideImplementMembersHandler.kt

@@ -146,9 +146,9 @@ abstract class OverrideImplementMembersHandler : LanguageCodeInsightActionHandle
 
             fun ClassDescriptor.findElement(memberDescriptor: CallableMemberDescriptor): KtDeclaration? {
                 return implementedElements[memberDescriptor]
-                    ?: (findCallableMemberBySignature(memberDescriptor)?.source?.getPsi() as? KtDeclaration)?.also {
+                    ?: (findCallableMemberBySignature(memberDescriptor)?.source?.getPsi() as? KtDeclaration)
-                        implementedElements[memberDescriptor] = it
+                        ?.takeIf { it != classOrObject }
-                    }
+                        ?.also { implementedElements[memberDescriptor] = it }
             }
 
             fun getAnchor(selectedElement: KtDeclaration): PsiElement? {

idea/testData/quickfix/implement/dataClass.kt

@@ -0,0 +1,7 @@
+// "Implement members" "true"
+// WITH_RUNTIME
+interface I {
+    fun foo()
+}
+
+data <caret>class C(val i: Int) : I

idea/testData/quickfix/implement/dataClass.kt.after

@@ -0,0 +1,11 @@
+// "Implement members" "true"
+// WITH_RUNTIME
+interface I {
+    fun foo()
+}
+
+data class C(val i: Int) : I {
+    override fun foo() {
+        TODO("Not yet implemented")
+    }
+}

idea/testData/quickfix/implement/dataClass2.kt

@@ -0,0 +1,7 @@
+// "Implement members" "true"
+// WITH_RUNTIME
+interface I {
+    fun foo()
+}
+
+data <caret>class C(val i: Int) : I {}

idea/testData/quickfix/implement/dataClass2.kt.after

@@ -0,0 +1,11 @@
+// "Implement members" "true"
+// WITH_RUNTIME
+interface I {
+    fun foo()
+}
+
+data class C(val i: Int) : I {
+    override fun foo() {
+        TODO("Not yet implemented")
+    }
+}

idea/testData/quickfix/implement/dataClass3.kt

@@ -0,0 +1,9 @@
+// "Implement members" "true"
+// WITH_RUNTIME
+interface I {
+    fun foo()
+}
+
+data <caret>class C(val i: Int) : I {
+    fun bar() {}
+}

idea/testData/quickfix/implement/dataClass3.kt.after

@@ -0,0 +1,12 @@
+// "Implement members" "true"
+// WITH_RUNTIME
+interface I {
+    fun foo()
+}
+
+data class C(val i: Int) : I {
+    fun bar() {}
+    override fun foo() {
+        TODO("Not yet implemented")
+    }
+}

idea/tests/org/jetbrains/kotlin/idea/quickfix/QuickFixTestGenerated.java

@@ -7669,6 +7669,21 @@ public class QuickFixTestGenerated extends AbstractQuickFixTest {
             runTest("idea/testData/quickfix/implement/annotation.kt");
         }
 
+        @TestMetadata("dataClass.kt")
+        public void testDataClass() throws Exception {
+            runTest("idea/testData/quickfix/implement/dataClass.kt");
+        }
+
+        @TestMetadata("dataClass2.kt")
+        public void testDataClass2() throws Exception {
+            runTest("idea/testData/quickfix/implement/dataClass2.kt");
+        }
+
+        @TestMetadata("dataClass3.kt")
+        public void testDataClass3() throws Exception {
+            runTest("idea/testData/quickfix/implement/dataClass3.kt");
+        }
+
         @TestMetadata("doNotAddExpectForVal.kt")
         public void testDoNotAddExpectForVal() throws Exception {
             runTest("idea/testData/quickfix/implement/doNotAddExpectForVal.kt");