[test infra] Add a directive for type checking utils that use @Exact

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFE10TestdataTestGenerated.java

@@ -165,6 +165,12 @@ public class DiagnosticCompilerTestFE10TestdataTestGenerated extends AbstractDia
             runTest("compiler/testData/diagnostics/tests/checkTypeTest.kt");
         }
 
+        @Test
+        @TestMetadata("checkTypeWithExactTest.kt")
+        public void testCheckTypeWithExactTest() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/checkTypeWithExactTest.kt");
+        }
+
         @Test
         @TestMetadata("CompareToWithErrorType.kt")
         public void testCompareToWithErrorType() throws Exception {

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated.java

@@ -165,6 +165,12 @@ public class LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated
             runTest("compiler/testData/diagnostics/tests/checkTypeTest.kt");
         }
 
+        @Test
+        @TestMetadata("checkTypeWithExactTest.kt")
+        public void testCheckTypeWithExactTest() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/checkTypeWithExactTest.kt");
+        }
+
         @Test
         @TestMetadata("CompareToWithErrorType.kt")
         public void testCompareToWithErrorType() throws Exception {

compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirLightTreeOldFrontendDiagnosticsTestGenerated.java

@@ -165,6 +165,12 @@ public class FirLightTreeOldFrontendDiagnosticsTestGenerated extends AbstractFir
             runTest("compiler/testData/diagnostics/tests/checkTypeTest.kt");
         }
 
+        @Test
+        @TestMetadata("checkTypeWithExactTest.kt")
+        public void testCheckTypeWithExactTest() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/checkTypeWithExactTest.kt");
+        }
+
         @Test
         @TestMetadata("CompareToWithErrorType.kt")
         public void testCompareToWithErrorType() throws Exception {

compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiOldFrontendDiagnosticsTestGenerated.java

@@ -165,6 +165,12 @@ public class FirPsiOldFrontendDiagnosticsTestGenerated extends AbstractFirPsiDia
             runTest("compiler/testData/diagnostics/tests/checkTypeTest.kt");
         }
 
+        @Test
+        @TestMetadata("checkTypeWithExactTest.kt")
+        public void testCheckTypeWithExactTest() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/checkTypeWithExactTest.kt");
+        }
+
         @Test
         @TestMetadata("CompareToWithErrorType.kt")
         public void testCompareToWithErrorType() throws Exception {

compiler/testData/diagnostics/helpers/types/checkTypeWithExact.kt

@@ -0,0 +1,4 @@
+@Suppress("INVISIBLE_MEMBER", "INVISIBLE_REFERENCE")
+fun <T> checkExactType(expr: @kotlin.internal.Exact T) {}
+@Suppress("INVISIBLE_MEMBER", "INVISIBLE_REFERENCE")
+fun <T> checkTypeEquality(reference: @kotlin.internal.Exact T, expr: @kotlin.internal.Exact T) {}

compiler/testData/diagnostics/tests/checkTypeWithExactTest.fir.kt

@@ -0,0 +1,14 @@
+// CHECK_TYPE_WITH_EXACT
+
+open class A
+open class B: A()
+class C: B()
+
+fun test(expr: B) {
+    checkExactType<A>(<!ARGUMENT_TYPE_MISMATCH!>expr<!>)
+    checkExactType<B>(expr)
+    checkExactType<C>(<!ARGUMENT_TYPE_MISMATCH!>expr<!>)
+    checkTypeEquality(A(), <!ARGUMENT_TYPE_MISMATCH!>expr<!>)
+    checkTypeEquality(B(), expr)
+    checkTypeEquality(C(), <!ARGUMENT_TYPE_MISMATCH!>expr<!>)
+}

compiler/testData/diagnostics/tests/checkTypeWithExactTest.kt

@@ -0,0 +1,14 @@
+// CHECK_TYPE_WITH_EXACT
+
+open class A
+open class B: A()
+class C: B()
+
+fun test(expr: B) {
+    checkExactType<A>(<!TYPE_MISMATCH!>expr<!>)
+    checkExactType<B>(expr)
+    checkExactType<C>(<!TYPE_MISMATCH!>expr<!>)
+    checkTypeEquality(A(), <!TYPE_MISMATCH!>expr<!>)
+    checkTypeEquality(B(), expr)
+    checkTypeEquality(C(), <!TYPE_MISMATCH!>expr<!>)
+}

compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java

@@ -165,6 +165,12 @@ public class DiagnosticTestGenerated extends AbstractDiagnosticTest {
             runTest("compiler/testData/diagnostics/tests/checkTypeTest.kt");
         }
 
+        @Test
+        @TestMetadata("checkTypeWithExactTest.kt")
+        public void testCheckTypeWithExactTest() throws Exception {
+            runTest("compiler/testData/diagnostics/tests/checkTypeWithExactTest.kt");
+        }
+
         @Test
         @TestMetadata("CompareToWithErrorType.kt")
         public void testCompareToWithErrorType() throws Exception {

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/directives/AdditionalFilesDirectives.kt

@@ -15,6 +15,13 @@ object AdditionalFilesDirectives : SimpleDirectivesContainer() {
         """.trimIndent()
     )
 
+    val CHECK_TYPE_WITH_EXACT by directive(
+        description = """
+            Adds util functions for type checking that use @kotlin.internal.Exact annotation
+            See file ./compiler/testData/diagnostics/helpers/types/checkTypeWithExact.kt
+        """.trimIndent()
+    )
+
     val WITH_COROUTINES by directive(
         description = """
             Adds util functions for checking coroutines

compiler/tests-common-new/tests/org/jetbrains/kotlin/test/services/sourceProviders/AdditionalDiagnosticsSourceFilesProvider.kt

@@ -6,6 +6,7 @@
 package org.jetbrains.kotlin.test.services.sourceProviders
 
 import org.jetbrains.kotlin.test.directives.AdditionalFilesDirectives
+import org.jetbrains.kotlin.test.directives.AdditionalFilesDirectives.CHECK_TYPE_WITH_EXACT
 import org.jetbrains.kotlin.test.directives.AdditionalFilesDirectives.CHECK_TYPE
 import org.jetbrains.kotlin.test.directives.AdditionalFilesDirectives.INFERENCE_HELPERS
 import org.jetbrains.kotlin.test.directives.model.DirectivesContainer
@@ -21,6 +22,7 @@ class AdditionalDiagnosticsSourceFilesProvider(testServices: TestServices, baseD
     private val helpersPath = "$baseDir/compiler/testData/diagnostics/helpers"
     private val directiveToFileMap: Map<SimpleDirective, String> = mapOf(
         CHECK_TYPE to "$helpersPath/types/checkType.kt",
+        CHECK_TYPE_WITH_EXACT to "$helpersPath/types/checkTypeWithExact.kt",
         INFERENCE_HELPERS to "$helpersPath/inference/inferenceUtils.kt"
     )