simon/linol
1
0
Fork 0
mirror of https://github.com/c-cube/linol.git synced 2025-12-08 20:25:35 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions
linol/thirdparty/lsp/ocaml-lsp-server/test/e2e-new/semantic_hl_tests.ml

754 lines
20 KiB
OCaml
Raw Blame History

open Test.Import
let semantic_tokens_full_debug = "ocamllsp/textDocument/semanticTokens/full"
let client_capabilities =
let textDocument =
let semanticTokens =
(* copied from vscode v1.69.2 client capabilities for semantic tokens;
it's easier to read in this form *)
SemanticTokensClientCapabilities.t_of_yojson
@@ Yojson.Safe.from_string
{|
{
"dynamicRegistration": true,
"tokenTypes": [
"namespace",
"type",
"class",
"enum",
"interface",
"struct",
"typeParameter",
"parameter",
"variable",
"property",
"enumMember",
"event",
"function",
"method",
"macro",
"keyword",
"modifier",
"comment",
"string",
"number",
"regexp",
"operator",
"decorator"
],
"tokenModifiers": [
"declaration",
"definition",
"readonly",
"static",
"deprecated",
"abstract",
"async",
"modification",
"documentation",
"defaultLibrary"
],
"formats": [
"relative"
],
"requests": {
"range": true,
"full": {
"delta": true
}
},
"multilineTokenSupport": false,
"overlappingTokenSupport": false,
"serverCancelSupport": true,
"augmentsSyntaxTokens": true
}
|}
in
TextDocumentClientCapabilities.create ~semanticTokens ()
in
ClientCapabilities.create ~textDocument ()
;;
type 'resp req_ctx =
{ initializeResult : InitializeResult.t
; resp : 'resp
}
let test
: type resp.
src:string
-> (SemanticTokensParams.t -> resp Client.out_request)
-> (resp req_ctx -> unit Fiber.t)
-> unit
=
fun ~src req consume_resp ->
let wait_for_diagnostics = Fiber.Ivar.create () in
let handler =
Client.Handler.make
~on_notification:(fun client -> function
| Lsp.Server_notification.PublishDiagnostics _ ->
(* we don't want to close the connection from client-side before we
process diagnostics arrived on the channel. TODO: would a better
solution be to simply flush on closing the connection because now
semantic tokens tests is coupled to diagnostics *)
let+ () = Fiber.Ivar.fill wait_for_diagnostics () in
Client.state client
| _ -> Fiber.return ())
()
in
Test.run ~handler (fun client ->
let run_client () =
Client.start client (InitializeParams.create ~capabilities:client_capabilities ())
in
let run () =
let* (initializeResult : InitializeResult.t) = Client.initialized client in
let uri = DocumentUri.of_path "test.ml" in
let textDocument =
TextDocumentItem.create ~uri ~languageId:"ocaml" ~version:0 ~text:src
in
let* () =
Client.notification
client
(TextDocumentDidOpen (DidOpenTextDocumentParams.create ~textDocument))
in
let* resp =
let textDocument = TextDocumentIdentifier.create ~uri in
let params = SemanticTokensParams.create ~textDocument () in
Client.request client (req params)
in
let* () = consume_resp { initializeResult; resp } in
let* () =
Fiber.fork_and_join_unit
(fun () -> Fiber.Ivar.read wait_for_diagnostics)
(fun () -> Client.request client Shutdown)
in
Client.stop client
in
Fiber.fork_and_join_unit run_client run)
;;
let test_semantic_tokens_full src =
let print_resp { initializeResult; resp } =
Fiber.return
@@
match resp with
| None -> print_endline "empty response"
| Some { SemanticTokens.data; _ } ->
let legend =
match
initializeResult.InitializeResult.capabilities
.ServerCapabilities.semanticTokensProvider
with
| None -> failwith "no server capabilities for semantic tokens"
| Some (`SemanticTokensOptions { legend; _ }) -> legend
| Some (`SemanticTokensRegistrationOptions { legend; _ }) -> legend
in
print_endline
@@ Semantic_hl_helpers.annotate_src_with_tokens
~legend
~encoded_tokens:data
~annot_mods:true
src
in
test ~src (fun p -> SemanticTokensFull p) print_resp
;;
let%expect_test "tokens for ocaml_lsp_server.ml" =
test_semantic_tokens_full Semantic_hl_data.src0;
[%expect
{|
module <namespace|definition-0>Moo</0> : sig
type <type|definition-1>t</1>
type <enum|definition-2>koo</2> =
| <enumMember|definition-3>Foo</3> of <type|-4>string</4>
| <enumMember|definition-5>Bar</5> of [ `Int of <type|-6>int</6> | `String of <type|-7>string</7> ]
val <variable|definition-8>u</8> : <type|-9>unit</9>
val <function|definition-10>f</10> : <type|-11>unit</11> -> <type|-12>t</12>
end = struct
type <type|definition-13>t</13> = <type|-14>int</14>
type <enum|definition-15>koo</15> =
| <enumMember|definition-16>Foo</16> of <type|-17>string</17>
| <enumMember|definition-18>Bar</18> of [ `Int of <type|-19>int</19> | `String of <type|-20>string</20> ]
let <variable|-21>u</21> = ()
let <function|definition-22>f</22> () = <number|-23>0</23>
end
module type <interface|-24>Bar</24> = sig
type <struct|definition-25>t</25> =
{ <property|-26>foo</26> : <namespace|-27>Moo</27>.<type|-28>t</28>
; <property|-29>bar</29> : <type|-30>int</30>
}
end
type <enum|definition-31>t</31> = <namespace|-32>Moo</32>.<type|-33>koo</33> =
| <enumMember|definition-34>Foo</34> of <type|-35>string</35>
| <enumMember|definition-36>Bar</36> of [ `BarInt of <type|-37>int</37> | `BarString of <type|-38>string</38> ]
let <function|definition-39>f</39> (<variable|-40>foo</40> : <type|-41>t</41>) =
match <variable|-42>foo</42> with
| <namespace|-43>Moo</43>.<enumMember|-44>Foo</44> <variable|-45>s</45> -> <variable|-46>s</46> <function|-47>^</47> <function|-48>string_of_int</48> <number|-49>0</49>
| <namespace|-50>Moo</50>.<enumMember|-51>Bar</51> (`BarInt <variable|-52>i</52>) -> <function|-53>string_of_int</53> <variable|-54>i</54>
| <namespace|-55>Moo</55>.<enumMember|-56>Bar</56> (`BarString <variable|-57>s</57>) -> <variable|-58>s</58>
module <namespace|definition-59>Foo</59> (<namespace|-60>Arg</60> : <interface|-61>Bar</61>) = struct
module <namespace|definition-62>Inner_foo</62> = struct
type <type|definition-63>t</63> = <type|-64>string</64>
end
end
module <namespace|definition-65>Foo_inst</65> = <namespace|-66>Foo</66> (struct
type <struct|definition-67>t</67> =
{ <property|-68>foo</68> : <namespace|-69>Moo</69>.<type|-70>t</70>
; <property|-71>bar</71> : <type|-72>int</72>
}
end) |}]
;;
let test_semantic_tokens_full_debug src =
test
~src
(fun p ->
UnknownRequest
{ meth = semantic_tokens_full_debug
; params =
Some (SemanticTokensParams.yojson_of_t p |> Jsonrpc.Structured.t_of_yojson)
})
(fun { resp; _ } ->
resp |> Yojson.Safe.pretty_to_string ~std:false |> print_endline |> Fiber.return)
;;
let%expect_test "tokens for ocaml_lsp_server.ml" =
test_semantic_tokens_full_debug Semantic_hl_data.src0;
[%expect
{|
[
{
"start_pos": { "character": 7, "line": 1 },
"length": 3,
"type": "namespace",
"modifiers": [ "definition" ]
},
{
"start_pos": { "character": 7, "line": 2 },
"length": 1,
"type": "type",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 7, "line": 4 },
"length": 3,
"type": "enum",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 6, "line": 5 },
"length": 3,
"type": "enumMember",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 13, "line": 5 },
"length": 6,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 6, "line": 6 },
"length": 3,
"type": "enumMember",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 23, "line": 6 },
"length": 3,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 40, "line": 6 },
"length": 6,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 6, "line": 8 },
"length": 1,
"type": "variable",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 10, "line": 8 },
"length": 4,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 6, "line": 10 },
"length": 1,
"type": "function",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 10, "line": 10 },
"length": 4,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 18, "line": 10 },
"length": 1,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 7, "line": 12 },
"length": 1,
"type": "type",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 11, "line": 12 },
"length": 3,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 7, "line": 14 },
"length": 3,
"type": "enum",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 6, "line": 15 },
"length": 3,
"type": "enumMember",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 13, "line": 15 },
"length": 6,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 6, "line": 16 },
"length": 3,
"type": "enumMember",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 23, "line": 16 },
"length": 3,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 40, "line": 16 },
"length": 6,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 6, "line": 18 },
"length": 1,
"type": "variable",
"modifiers": []
},
{
"start_pos": { "character": 6, "line": 20 },
"length": 1,
"type": "function",
"modifiers": [ "definition" ]
},
{
"start_pos": { "character": 13, "line": 20 },
"length": 1,
"type": "number",
"modifiers": []
},
{
"start_pos": { "character": 12, "line": 23 },
"length": 3,
"type": "interface",
"modifiers": []
},
{
"start_pos": { "character": 7, "line": 24 },
"length": 1,
"type": "struct",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 6, "line": 25 },
"length": 3,
"type": "property",
"modifiers": []
},
{
"start_pos": { "character": 12, "line": 25 },
"length": 3,
"type": "namespace",
"modifiers": []
},
{
"start_pos": { "character": 16, "line": 25 },
"length": 1,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 6, "line": 26 },
"length": 3,
"type": "property",
"modifiers": []
},
{
"start_pos": { "character": 12, "line": 26 },
"length": 3,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 5, "line": 30 },
"length": 1,
"type": "enum",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 9, "line": 30 },
"length": 3,
"type": "namespace",
"modifiers": []
},
{
"start_pos": { "character": 13, "line": 30 },
"length": 3,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 4, "line": 31 },
"length": 3,
"type": "enumMember",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 11, "line": 31 },
"length": 6,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 4, "line": 32 },
"length": 3,
"type": "enumMember",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 24, "line": 32 },
"length": 3,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 44, "line": 32 },
"length": 6,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 4, "line": 34 },
"length": 1,
"type": "function",
"modifiers": [ "definition" ]
},
{
"start_pos": { "character": 7, "line": 34 },
"length": 3,
"type": "variable",
"modifiers": []
},
{
"start_pos": { "character": 13, "line": 34 },
"length": 1,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 8, "line": 35 },
"length": 3,
"type": "variable",
"modifiers": []
},
{
"start_pos": { "character": 4, "line": 36 },
"length": 3,
"type": "namespace",
"modifiers": []
},
{
"start_pos": { "character": 8, "line": 36 },
"length": 3,
"type": "enumMember",
"modifiers": []
},
{
"start_pos": { "character": 12, "line": 36 },
"length": 1,
"type": "variable",
"modifiers": []
},
{
"start_pos": { "character": 17, "line": 36 },
"length": 1,
"type": "variable",
"modifiers": []
},
{
"start_pos": { "character": 19, "line": 36 },
"length": 1,
"type": "function",
"modifiers": []
},
{
"start_pos": { "character": 21, "line": 36 },
"length": 13,
"type": "function",
"modifiers": []
},
{
"start_pos": { "character": 35, "line": 36 },
"length": 1,
"type": "number",
"modifiers": []
},
{
"start_pos": { "character": 4, "line": 37 },
"length": 3,
"type": "namespace",
"modifiers": []
},
{
"start_pos": { "character": 8, "line": 37 },
"length": 3,
"type": "enumMember",
"modifiers": []
},
{
"start_pos": { "character": 21, "line": 37 },
"length": 1,
"type": "variable",
"modifiers": []
},
{
"start_pos": { "character": 27, "line": 37 },
"length": 13,
"type": "function",
"modifiers": []
},
{
"start_pos": { "character": 41, "line": 37 },
"length": 1,
"type": "variable",
"modifiers": []
},
{
"start_pos": { "character": 4, "line": 38 },
"length": 3,
"type": "namespace",
"modifiers": []
},
{
"start_pos": { "character": 8, "line": 38 },
"length": 3,
"type": "enumMember",
"modifiers": []
},
{
"start_pos": { "character": 24, "line": 38 },
"length": 1,
"type": "variable",
"modifiers": []
},
{
"start_pos": { "character": 30, "line": 38 },
"length": 1,
"type": "variable",
"modifiers": []
},
{
"start_pos": { "character": 7, "line": 40 },
"length": 3,
"type": "namespace",
"modifiers": [ "definition" ]
},
{
"start_pos": { "character": 12, "line": 40 },
"length": 3,
"type": "namespace",
"modifiers": []
},
{
"start_pos": { "character": 18, "line": 40 },
"length": 3,
"type": "interface",
"modifiers": []
},
{
"start_pos": { "character": 9, "line": 41 },
"length": 9,
"type": "namespace",
"modifiers": [ "definition" ]
},
{
"start_pos": { "character": 9, "line": 42 },
"length": 1,
"type": "type",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 13, "line": 42 },
"length": 6,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 7, "line": 46 },
"length": 8,
"type": "namespace",
"modifiers": [ "definition" ]
},
{
"start_pos": { "character": 18, "line": 46 },
"length": 3,
"type": "namespace",
"modifiers": []
},
{
"start_pos": { "character": 7, "line": 47 },
"length": 1,
"type": "struct",
"modifiers": [ "declaration" ]
},
{
"start_pos": { "character": 6, "line": 48 },
"length": 3,
"type": "property",
"modifiers": []
},
{
"start_pos": { "character": 12, "line": 48 },
"length": 3,
"type": "namespace",
"modifiers": []
},
{
"start_pos": { "character": 16, "line": 48 },
"length": 1,
"type": "type",
"modifiers": []
},
{
"start_pos": { "character": 6, "line": 49 },
"length": 3,
"type": "property",
"modifiers": []
},
{
"start_pos": { "character": 12, "line": 49 },
"length": 3,
"type": "type",
"modifiers": []
}
] |}]
;;
let%expect_test "highlighting longidents with space between identifiers" =
test_semantic_tokens_full
@@ String.trim
{|
let foo = Bar.jar
let joo = Bar. jar
|};
[%expect
{|
let <variable|-0>foo</0> = <namespace|-1>Bar</1>.<variable|-2>jar</2>
let <variable|-3>joo</3> = <namespace|-4>Bar</4>. <variable|-5>jar</5> |}]
;;
let%expect_test "highlighting longidents with space between identifiers and infix fns" =
test_semantic_tokens_full
@@ String.trim
{|
Bar.(+) ;;
Bar.( + ) ;;
Bar. (+) ;;
Bar. ( + ) ;;
|};
[%expect
{|
<namespace|-0>Bar</0>.<variable|-1>(+)</1> ;;
<namespace|-2>Bar</2>.<namespace|-3>(</3> <namespace|-4>+</4> <variable|-5>)</5> ;;
<namespace|-6>Bar</6>. <variable|-7>(+)</7> ;;
<namespace|-8>Bar</8>. <namespace|-9>(</9> <namespace|-10>+</10> <variable|-11>)</11> ;; |}]
;;
let%expect_test "longidents in records" =
test_semantic_tokens_full
@@ String.trim
{|
module M = struct type r = { foo : int ; bar : string } end
let x = { M . foo = 0 ; bar = "bar"}
|};
[%expect
{|
module <namespace|definition-0>M</0> = struct type <struct|definition-1>r</1> = { <property|-2>foo</2> : <type|-3>int</3> ; <property|-4>bar</4> : <type|-5>string</5> } end
let <variable|-6>x</6> = { <namespace|-7>M</7> . <property|-8>foo</8> = <number|-9>0</9> ; <property|-10>bar</10> = <string|-11>"bar"</11>} |}]
;;
let%expect_test "operators" =
test_semantic_tokens_full
@@ String.trim
{|
let x = 1.0 *. 2.0
let y = 1 * 2
let z = 0 >>= 1
|};
[%expect
{|
let <variable|-0>x</0> = <number|-1>1.0</1> <function|-2>*.</2> <number|-3>2.0</3>
let <variable|-4>y</4> = <number|-5>1</5> <function|-6>*</6> <number|-7>2</7>
let <variable|-8>z</8> = <number|-9>0</9> <function|-10>>>=</10> <number|-11>1</11> |}]
;;
let%expect_test "comment in unit" =
test_semantic_tokens_full
@@ String.trim
{|
let y = (* comment *) 0
let x = ((* comment *))
let ((*comment*)) = ()
|};
[%expect
{|
let <variable|-0>y</0> = (* comment *) <number|-1>0</1>
let <variable|-2>x</2> = ((* comment *))
let ((*comment*)) = () |}]
;;
Reference in a new issue View git blame Copy permalink