[Analysis API] Add tests for dangling files

analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/components/diagnosticProvider/AbstractDanglingFileCollectDiagnosticsTest.kt

@@ -0,0 +1,20 @@
+/*
+ * Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.analysis.api.impl.base.test.cases.components.diagnosticProvider
+
+import org.jetbrains.kotlin.psi.KtFile
+import org.jetbrains.kotlin.psi.KtPsiFactory
+import org.jetbrains.kotlin.test.model.TestModule
+import org.jetbrains.kotlin.test.services.TestServices
+
+abstract class AbstractDanglingFileCollectDiagnosticsTest : AbstractCollectDiagnosticsTest() {
+    override fun doTestByFileStructure(ktFile: KtFile, module: TestModule, testServices: TestServices) {
+        val ktPsiFactory = KtPsiFactory.contextual(ktFile, markGenerated = true, eventSystemEnabled = false)
+        val fakeKtFile = ktPsiFactory.createFile("fake.kt", ktFile.text)
+
+        super.doTestByFileStructure(fakeKtFile, module, testServices)
+    }
+}

analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/references/AbstractDanglingFileReferenceResolveTest.kt

@@ -0,0 +1,34 @@
+/*
+ * Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.analysis.api.impl.base.test.cases.references
+
+import org.jetbrains.kotlin.analysis.api.KtAnalysisSession
+import org.jetbrains.kotlin.analysis.api.symbols.KtSymbol
+import org.jetbrains.kotlin.analysis.test.framework.project.structure.ktModuleProvider
+import org.jetbrains.kotlin.analysis.test.framework.services.expressionMarkerProvider
+import org.jetbrains.kotlin.psi.KtFile
+import org.jetbrains.kotlin.psi.KtPsiFactory
+import org.jetbrains.kotlin.test.services.TestModuleStructure
+import org.jetbrains.kotlin.test.services.TestServices
+
+abstract class AbstractDanglingFileReferenceResolveTest : AbstractReferenceResolveTest() {
+    override fun KtAnalysisSession.getAdditionalSymbolInfo(symbol: KtSymbol): String? {
+        val containingFile = symbol.psi?.containingFile ?: return null
+        return containingFile.name
+    }
+
+    override fun doTestByModuleStructure(moduleStructure: TestModuleStructure, testServices: TestServices) {
+        val mainModule = findMainModule(moduleStructure)
+        val ktFiles = testServices.ktModuleProvider.getModuleFiles(mainModule).filterIsInstance<KtFile>()
+        val mainKtFile = findMainFile(ktFiles)
+        val caretPosition = testServices.expressionMarkerProvider.getCaretPosition(mainKtFile)
+
+        val ktPsiFactory = KtPsiFactory.contextual(mainKtFile, markGenerated = true, eventSystemEnabled = true)
+        val fakeKtFile = ktPsiFactory.createFile("fake.kt", mainKtFile.text)
+
+        doTestByFileStructure(fakeKtFile, caretPosition, mainModule, testServices)
+    }
+}

analysis/analysis-api-impl-base/tests/org/jetbrains/kotlin/analysis/api/impl/base/test/cases/references/AbstractReferenceResolveTest.kt

@@ -46,19 +46,30 @@ abstract class AbstractReferenceResolveTest : AbstractAnalysisApiBasedTest() {
         }
     }
 
-    final override fun doTestByModuleStructure(moduleStructure: TestModuleStructure, testServices: TestServices) {
-        val mainModule = moduleStructure.modules.singleOrNull() ?: findMainModule(moduleStructure)
+    override fun doTestByModuleStructure(moduleStructure: TestModuleStructure, testServices: TestServices) {
+        val mainModule = findMainModule(moduleStructure)
         val ktFiles = testServices.ktModuleProvider.getModuleFiles(mainModule).filterIsInstance<KtFile>()
-        doTestByFileStructure(ktFiles, mainModule, testServices)
+        val mainKtFile = findMainFile(ktFiles)
+        val caretPosition = testServices.expressionMarkerProvider.getCaretPosition(mainKtFile)
+        doTestByFileStructure(mainKtFile, caretPosition, mainModule, testServices)
     }
 
-    private fun findMainModule(moduleStructure: TestModuleStructure): TestModule =
-        moduleStructure.modules.find { it.name == "main" } ?: error("There should be a module named 'main' in the multi-module test.")
+    protected fun findMainModule(moduleStructure: TestModuleStructure): TestModule {
+        val modules = moduleStructure.modules
 
-    fun doTestByFileStructure(ktFiles: List<KtFile>, mainModule: TestModule, testServices: TestServices) {
-        val mainKtFile = ktFiles.singleOrNull() ?: ktFiles.firstOrNull { it.name == "main.kt" } ?: ktFiles.first()
-        val caretPosition = testServices.expressionMarkerProvider.getCaretPosition(mainKtFile)
-        val ktReferences = findReferencesAtCaret(mainKtFile, caretPosition)
+        return modules.singleOrNull()
+            ?: modules.find { it.name == "main" }
+            ?: error("There should be a module named 'main' in the multi-module test.")
+    }
+
+    protected fun findMainFile(ktFiles: List<KtFile>): KtFile {
+        return ktFiles.singleOrNull()
+            ?: ktFiles.firstOrNull { it.name == "main.kt" }
+            ?: ktFiles.first()
+    }
+
+    protected fun doTestByFileStructure(ktFile: KtFile, caretPosition: Int, mainModule: TestModule, testServices: TestServices) {
+        val ktReferences = findReferencesAtCaret(ktFile, caretPosition)
         if (ktReferences.isEmpty()) {
             testServices.assertions.fail { "No references at caret found" }
         }