goto_definition tests

src/handlers/completion.rs

@@ -165,7 +165,7 @@ mod test {
         &[]
     )]
     #[tokio::test(flavor = "current_thread")]
-    async fn test_server_completions(
+    async fn server_completions(
         #[case] source: &str,
         #[case] position: Position,
         #[case] symbols: &[SymbolInfo],

src/handlers/did_change.rs

@@ -122,7 +122,7 @@ mod test {
         ]
     )]
     #[tokio::test(flavor = "current_thread")]
-    async fn test_server_did_change(
+    async fn server_did_change(
         #[case] original: &str,
         #[case] expected: &str,
         #[case] edits: &[TestEdit],

src/handlers/did_change_configuration.rs

@@ -29,7 +29,7 @@ mod test {
     };
 
     #[tokio::test(flavor = "current_thread")]
-    async fn test_server_did_change_configuration() {
+    async fn server_did_change_configuration() {
         // Arrange
         let mut service = initialize_server(&[]).await.0;
 

src/handlers/did_open.rs

@@ -137,7 +137,7 @@ mod test {
     };
 
     #[tokio::test(flavor = "current_thread")]
-    async fn test_server_did_open_document() {
+    async fn server_did_open_document() {
         // Arrange
         let mut service = initialize_server(&[]).await.0;
         let source = r#""[" @cap"#;

src/handlers/formatting.rs

@@ -350,7 +350,7 @@ mod test {
     };
 
     #[tokio::test(flavor = "current_thread")]
-    async fn test_server_formatting() {
+    async fn server_formatting() {
         // Arrange
         let mut service = initialize_server(&[(
             TEST_URI.clone(),

src/handlers/goto_definition.rs

@@ -57,3 +57,100 @@ pub async fn goto_definition(
 
     Ok(Some(GotoDefinitionResponse::Array(defs)))
 }
+
+#[cfg(test)]
+mod test {
+    use pretty_assertions::assert_eq;
+    use rstest::rstest;
+    use tower::{Service, ServiceExt};
+    use tower_lsp::lsp_types::{
+        request::GotoDefinition, GotoDefinitionParams, GotoDefinitionResponse, Location,
+        PartialResultParams, Position, Range, TextDocumentIdentifier, TextDocumentPositionParams,
+        WorkDoneProgressParams,
+    };
+
+    use crate::test_helpers::helpers::{
+        initialize_server, lsp_request_to_jsonrpc_request, lsp_response_to_jsonrpc_response,
+        COMPLEX_FILE, SIMPLE_FILE, TEST_URI,
+    };
+
+    type Coordinate = ((u32, u32), (u32, u32));
+
+    #[rstest]
+    #[case(
+        &SIMPLE_FILE,
+        Position { line: 0, character: 4 },
+        &[]
+    )]
+    #[case(
+        &SIMPLE_FILE,
+        Position { line: 0, character: 20 },
+        &[((0, 14), (0, 23))]
+    )]
+    #[case(
+        &COMPLEX_FILE,
+        Position { line: 12, character: 14 },
+        &[((8, 24), (8, 42)), ((9, 22), (9, 40))]
+    )]
+    #[tokio::test(flavor = "current_thread")]
+    async fn goto_definition(
+        #[case] input: &str,
+        #[case] position: Position,
+        #[case] locations: &[Coordinate],
+    ) {
+        // Arrange
+        let mut service = initialize_server(&[(TEST_URI.clone(), input, Vec::new(), Vec::new())])
+            .await
+            .0;
+
+        // Act
+        let refs = service
+            .ready()
+            .await
+            .unwrap()
+            .call(lsp_request_to_jsonrpc_request::<GotoDefinition>(
+                GotoDefinitionParams {
+                    partial_result_params: PartialResultParams {
+                        partial_result_token: None,
+                    },
+                    work_done_progress_params: WorkDoneProgressParams::default(),
+                    text_document_position_params: TextDocumentPositionParams {
+                        text_document: TextDocumentIdentifier {
+                            uri: TEST_URI.clone(),
+                        },
+                        position,
+                    },
+                },
+            ))
+            .await
+            .unwrap();
+
+        // Assert
+        let actual = if locations.is_empty() {
+            None
+        } else {
+            Some(GotoDefinitionResponse::Array(
+                locations
+                    .iter()
+                    .map(|r| Location {
+                        uri: TEST_URI.clone(),
+                        range: Range {
+                            start: Position {
+                                line: r.0 .0,
+                                character: r.0 .1,
+                            },
+                            end: Position {
+                                line: r.1 .0,
+                                character: r.1 .1,
+                            },
+                        },
+                    })
+                    .collect(),
+            ))
+        };
+        assert_eq!(
+            refs,
+            Some(lsp_response_to_jsonrpc_response::<GotoDefinition>(actual))
+        );
+    }
+}

src/handlers/initialize.rs

@@ -55,7 +55,7 @@ mod test {
         ],
     )])]
     #[tokio::test(flavor = "current_thread")]
-    async fn test_server_initialize(#[case] documents: &[(Url, &str, Vec<SymbolInfo>, Vec<&str>)]) {
+    async fn server_initialize(#[case] documents: &[(Url, &str, Vec<SymbolInfo>, Vec<&str>)]) {
         // Act
         let (service, response) = initialize_server(documents).await;
 

src/handlers/references.rs

@@ -101,7 +101,7 @@ function: (identifier) @function)",
         &[((5, 25), (5, 44)), ((11, 15), (11, 34)), ((17, 16), (17, 35))]
     )]
     #[tokio::test(flavor = "current_thread")]
-    async fn test_capture_references(
+    async fn capture_references(
         #[case] input: &str,
         #[case] position: Position,
         #[case] ranges: &[Coordinate],

src/handlers/rename.rs

@@ -143,7 +143,7 @@ mod test {
         "invariant"
     )]
     #[tokio::test(flavor = "current_thread")]
-    async fn test_server_rename(
+    async fn server_rename(
         #[case] original: &str,
         #[case] cursor_position: Position,
         #[case] edits: &[TestEdit],