From 3755acff68664ad7e32c17130dd97eae73258fc8 Mon Sep 17 00:00:00 2001
From: Andrey Cherkasov <andrey.cherkasov@jetbrains.com>
Date: Fri, 10 May 2024 13:02:35 +0400
Subject: [PATCH] [kotlin] Improve test coverage for K2

^KTIJ-23928

GitOrigin-RevId: 1c2d8ff22ddfce1acd787a3b4b4b4f48afd6d132
---
 ...ighLevelQuickFixMultiFileTestGenerated.java | 18 ++++++++++++++++++
 .../codeinsight/GenerateK2QuickFixTests.kt     |  5 +++++
 2 files changed, 23 insertions(+)

diff --git a/plugins/kotlin/code-insight/fixes-k2/tests/test/org/jetbrains/kotlin/idea/k2/codeinsight/fixes/HighLevelQuickFixMultiFileTestGenerated.java b/plugins/kotlin/code-insight/fixes-k2/tests/test/org/jetbrains/kotlin/idea/k2/codeinsight/fixes/HighLevelQuickFixMultiFileTestGenerated.java
index 08a730d14bea..63be81f75f07 100644
--- a/plugins/kotlin/code-insight/fixes-k2/tests/test/org/jetbrains/kotlin/idea/k2/codeinsight/fixes/HighLevelQuickFixMultiFileTestGenerated.java
+++ b/plugins/kotlin/code-insight/fixes-k2/tests/test/org/jetbrains/kotlin/idea/k2/codeinsight/fixes/HighLevelQuickFixMultiFileTestGenerated.java
@@ -972,4 +972,22 @@ public abstract class HighLevelQuickFixMultiFileTestGenerated extends AbstractHi
             runTest("../../../idea/tests/testData/quickfix/surroundWithNullCheck/javaReceiverNullabilityInvoke.test");
         }
     }
+
+    @RunWith(JUnit3RunnerWithInners.class)
+    @TestMetadata("../../../idea/tests/testData/quickfix/modifiers/addOpenToClassDeclaration")
+    public static class AddOpenToClassDeclaration extends AbstractHighLevelQuickFixMultiFileTest {
+        private void runTest(String testDataFilePath) throws Exception {
+            KotlinTestUtils.runTest(this::doTestWithExtraFile, this, testDataFilePath);
+        }
+
+        @TestMetadata("finalJavaSupertype.before.Main.kt")
+        public void testFinalJavaSupertype() throws Exception {
+            runTest("../../../idea/tests/testData/quickfix/modifiers/addOpenToClassDeclaration/finalJavaSupertype.before.Main.kt");
+        }
+
+        @TestMetadata("finalJavaUpperBound.before.Main.kt")
+        public void testFinalJavaUpperBound() throws Exception {
+            runTest("../../../idea/tests/testData/quickfix/modifiers/addOpenToClassDeclaration/finalJavaUpperBound.before.Main.kt");
+        }
+    }
 }
diff --git a/plugins/kotlin/util/test-generator-fir/test/org/jetbrains/kotlin/fir/testGenerator/codeinsight/GenerateK2QuickFixTests.kt b/plugins/kotlin/util/test-generator-fir/test/org/jetbrains/kotlin/fir/testGenerator/codeinsight/GenerateK2QuickFixTests.kt
index f93db8b2613d..d2d906cdc541 100644
--- a/plugins/kotlin/util/test-generator-fir/test/org/jetbrains/kotlin/fir/testGenerator/codeinsight/GenerateK2QuickFixTests.kt
+++ b/plugins/kotlin/util/test-generator-fir/test/org/jetbrains/kotlin/fir/testGenerator/codeinsight/GenerateK2QuickFixTests.kt
@@ -221,6 +221,11 @@ internal fun MutableTWorkspace.generateK2FixTests() {
                 pattern = pattern,
                 testMethodName = testMethodName,
             )
+            model(
+                "$idea/quickfix/modifiers/addOpenToClassDeclaration",
+                pattern = pattern,
+                testMethodName = testMethodName,
+            )
         }
 
         testClass<AbstractHighLevelQuickFixMultiModuleTest> {