IJPL-224362 Performance test for 'Go to File' for non-indexable files

GitOrigin-RevId: d964e87158b759ee7dbb29c99072730bd8436b63
2026-09-27 10:03:11 +07:00 · 2026-01-12 18:51:26 +00:00
parent 93aa188c67
commit dd0f3f54d2
1 changed files with 113 additions and 0 deletions
@@ -0,0 +1,113 @@
+// Copyright 2000-2026 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
+package com.intellij.ide.actions.searcheverywhere
+
+import com.intellij.mock.MockProgressIndicator
+import com.intellij.openapi.actionSystem.impl.SimpleDataContext
+import com.intellij.openapi.application.PathManager
+import com.intellij.openapi.application.readAction
+import com.intellij.openapi.util.Disposer
+import com.intellij.openapi.vfs.VfsUtil
+import com.intellij.openapi.vfs.newvfs.NewVirtualFile
+import com.intellij.testFramework.TestActionEvent
+import com.intellij.testFramework.junit5.RegistryKey
+import com.intellij.testFramework.junit5.StressTestApplication
+import com.intellij.testFramework.rules.ProjectModelExtension
+import com.intellij.testFramework.runInEdtAndWait
+import com.intellij.tools.ide.metrics.benchmark.Benchmark
+import com.intellij.workspaceModel.core.fileIndex.WorkspaceFileIndex
+import com.intellij.workspaceModel.ide.registerProjectRootBlocking
+import kotlinx.coroutines.runBlocking
+import org.junit.jupiter.api.Assumptions
+import org.junit.jupiter.api.BeforeEach
+import org.junit.jupiter.api.Test
+import org.junit.jupiter.api.extension.RegisterExtension
+import kotlin.io.path.Path
+
+/**
+ * Performance test for [NonIndexableFilesSEContributor].
+ */
+@StressTestApplication
+@RegistryKey(key = "se.enable.non.indexable.files.contributor", value = "true")
+class NonIndexableFileSearchPerformanceTest {
+  @RegisterExtension
+  private val projectModel: ProjectModelExtension = ProjectModelExtension()
+
+  private val project get() = projectModel.project
+  private val workspaceFileIndex get() = WorkspaceFileIndex.getInstance(project)
+
+
+  companion object {
+    private val communityPath = Path(PathManager.getCommunityHomePath())
+    private val communityVirtualFile = VfsUtil.findFile(communityPath, true)!!
+
+    private val nonIndexableFilesCount: Int = run {
+      var nonIndexableFiles = 0
+      VfsUtil.processFilesRecursively(NewVirtualFile.asCacheAvoiding(communityVirtualFile)) {
+        nonIndexableFiles++
+        true
+      }
+      nonIndexableFiles
+    }
+  }
+
+  @BeforeEach
+  fun createNonIndexableFileset(): Unit = runBlocking {
+      Assumptions.assumeTrue(project.isOpen)
+
+      runInEdtAndWait { registerProjectRootBlocking(project, communityPath) }
+
+      Assumptions.assumeTrue(readAction { workspaceFileIndex.isInContent(communityVirtualFile) }) {
+          "project root must be in content"
+      }
+      Assumptions.assumeFalse(readAction { workspaceFileIndex.isIndexable(communityVirtualFile) }) {
+          "project root must be non-indexable"
+      }
+  }
+
+  @Test
+  fun `iterate over all files`(): Unit = runBlocking {
+      val searchPattern = "ProjectRootEntity"
+      val contributor = createContributor()
+      Benchmark.newBenchmarkWithVariableInputSize("search \"$searchPattern\"", nonIndexableFilesCount) {
+          contributor.search(searchPattern, MockProgressIndicator())
+          nonIndexableFilesCount
+      }.start()
+  }
+
+  @Test
+  fun `search for one file deep inside`(): Unit = runBlocking {
+      val searchPattern = "ProjectRootEntity"
+      val contributor = createContributor()
+      Benchmark.newBenchmarkWithVariableInputSize("search \"$searchPattern\"", nonIndexableFilesCount) {
+          contributor.search(searchPattern, MockProgressIndicator(), 1)
+          nonIndexableFilesCount
+      }.start()
+  }
+
+  @Test
+  fun `search for one last root child`(): Unit = runBlocking {
+      val filename = communityVirtualFile.getChildren(true).last().name
+      val contributor = createContributor()
+      Benchmark.newBenchmarkWithVariableInputSize("search \"$filename\"", nonIndexableFilesCount) {
+          contributor.search(filename, MockProgressIndicator(), 1)
+          nonIndexableFilesCount
+      }.start()
+  }
+
+  @Test
+  fun `search for the first root child`(): Unit = runBlocking {
+      val filename = communityVirtualFile.getChildren(true).first().name
+      val contributor = createContributor()
+      Benchmark.newBenchmark("search \"$filename\"") {
+          contributor.search(filename, MockProgressIndicator(), 1)
+      }.start()
+  }
+
+
+  private fun createContributor(): NonIndexableFilesSEContributor {
+    val event = TestActionEvent.createTestEvent(SimpleDataContext.getProjectContext(project))
+    return NonIndexableFilesSEContributor(event).also { contributor ->
+      Disposer.register(projectModel.disposableRule.disposable, contributor)
+    }
+  }
+}