lean4-htt/tests/bench/identifier_completion_runner.lean
Marc Huisinga dfd3d18530
test: improve language server test coverage (#10574)
This PR significantly improves the test coverage of the language server,
providing at least a single basic test for every request that is used by
the client. It also implements infrastructure for testing all of these
requests, e.g. the ability to run interactive tests in a project context
and refactors the interactive test runner to be more maintainable.
Finally, it also fixes a small bug with the recently implemented unknown
identifier code actions for auto-implicits (#10442) that was discovered
in testing, where the "import all unambiguous unknown identifiers" code
action didn't work correctly on auto-implicit identifiers.
2025-09-30 11:15:03 +00:00

29 lines
1.2 KiB
Text

import Lean.Data.Lsp
open IO Lean Lsp
def mkCompletionRequest (id : Nat) : JsonRpc.Request Json :=
let param := Json.parse r#"{"textDocument":{"uri":"file:///home/sebastian/lean4/tests/bench/identifier_completion.lean"},"position":{"line":24,"character":40},"context":{"triggerKind":1}}"#
let param := param.toOption.get!
{ id, method := "textDocument/completion", param }
def main : IO Unit := do
Ipc.runWith "lean" #["--server"] do
let hIn ← Ipc.stdin
hIn.write (← FS.readBinFile "identifier_completion_initialization.log")
hIn.flush
let _ ← Ipc.readResponseAs 0 InitializeResult
Ipc.writeNotification {
method := "initialized"
param := InitializedParams.mk
}
hIn.write (← FS.readBinFile "identifier_completion_didOpen.log")
-- Let file progress proceed to the point of the completion that we want to benchmark
Ipc.writeRequest <| mkCompletionRequest 1
let _ ← Ipc.readResponseAs 1 CompletionList
let startTime ← IO.monoMsNow
for i in 2...(5 : Nat) do
Ipc.writeRequest <| mkCompletionRequest i
let _ ← Ipc.readResponseAs i CompletionList
let endTime ← IO.monoMsNow
IO.println s!"completion: {(endTime - startTime).toFloat / 1000.0}"
Ipc.shutdown 5