blob: 45d5456bd66950fdb083d7acc81511c8bcf311a8 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
// Copyright 2000-2022 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
package org.jetbrains.kotlin.idea.codeInsight.gradle
import com.intellij.codeInsight.daemon.quickFix.ActionHint
import com.intellij.openapi.application.runReadAction
import com.intellij.openapi.vfs.LocalFileSystem
import com.intellij.openapi.vfs.VirtualFile
import com.intellij.testFramework.PlatformTestUtil
import com.intellij.testFramework.runInEdtAndWait
import com.intellij.util.io.readText
import org.jetbrains.kotlin.idea.core.util.toPsiFile
import org.jetbrains.kotlin.idea.test.DirectiveBasedActionUtils
import org.jetbrains.kotlin.psi.KtFile
import java.io.File
import java.nio.file.Files
import kotlin.io.path.isRegularFile
import kotlin.io.path.pathString
import kotlin.streams.asSequence
abstract class AbstractGradleMultiFileQuickFixTest : MultiplePluginVersionGradleImportingCodeInsightTestCase() {
override fun testDataDirName() = "fixes"
final override fun testDataDirectory(): File = super.testDataDirectory().resolve("before")
open val afterDirectory get() = testDataDirectory().parentFile.resolve("after")
protected fun doMultiFileQuickFixTest(
ignoreChangesInBuildScriptFiles: Boolean = true,
additionalResultFileFilter: (VirtualFile) -> Boolean = { true },
) {
configureByFiles()
val projectPath = myProjectRoot.toNioPath()
val (mainFilePath, mainFileText) = Files.walk(projectPath).asSequence()
.filter { it.isRegularFile() }
.firstNotNullOfOrNull {
val text = kotlin.runCatching { it.readText() }.getOrNull()
if (text?.startsWith("// \"") == true) it to text else null
} ?: error("file with action is not found")
importProject()
val ktFile = runReadAction {
LocalFileSystem.getInstance().findFileByNioFile(mainFilePath)?.toPsiFile(myProject)
} as KtFile
val actionHint = ActionHint.parse(ktFile, mainFileText)
codeInsightTestFixture.configureFromExistingVirtualFile(ktFile.virtualFile)
runInEdtAndWait {
val actions = codeInsightTestFixture.availableIntentions
val action = actionHint.findAndCheck(actions) { "Test file: ${projectPath.relativize(mainFilePath).pathString}" }
if (action != null) {
action.invoke(myProject, null, ktFile)
val expected = LocalFileSystem.getInstance().findFileByIoFile(afterDirectory)?.apply {
refreshRecursively(this)
} ?: error("Expected directory is not found")
val projectVFile = (LocalFileSystem.getInstance().findFileByIoFile(File("$projectPath"))
?: error("VirtualFile is not found for project path"))
PlatformTestUtil.assertDirectoriesEqual(
expected,
projectVFile,
fun(vFile: VirtualFile): Boolean {
if (vFile.parent == projectVFile) {
when (vFile.name) {
".gradle", "gradle", "build" -> return false
}
}
if (ignoreChangesInBuildScriptFiles && ".gradle" in vFile.name) return false
return additionalResultFileFilter(vFile)
},
)
}
codeInsightTestFixture.doHighlighting()
DirectiveBasedActionUtils.checkAvailableActionsAreExpected(ktFile, action?.let { actions - it } ?: actions)
DirectiveBasedActionUtils.checkForUnexpectedErrors(ktFile)
}
}
}
|