add test for publish diagnostics version support

Author: dearfl

pyrefly/lib/test/lsp/lsp_interaction/diagnostic.rs

@@ -7,6 +7,7 @@
 
 use lsp_server::RequestId;
 use lsp_server::Response;
+use lsp_types::Url;
 
 use crate::test::lsp::lsp_interaction::object_model::InitializeSettings;
 use crate::test::lsp::lsp_interaction::object_model::LspInteraction;
@@ -163,3 +164,76 @@ fn test_error_documentation_links() {
 
     interaction.shutdown();
 }
+
+#[test]
+fn test_version_support_publish_diagnostics() {
+    let test_files_root = get_test_files_root();
+    let root = test_files_root.path().to_path_buf();
+    let mut file = root.clone();
+    file.push("text_document.py");
+    let uri = Url::from_file_path(file).unwrap();
+    let mut interaction = LspInteraction::new();
+    interaction.set_root(root);
+    interaction.initialize(InitializeSettings {
+        configuration: Some(None),
+        capabilities: Some(serde_json::json!({
+            "textDocument": {
+                "publishDiagnostics": {
+                    "versionSupport": true,
+                },
+            },
+        })),
+        ..Default::default()
+    });
+
+    interaction.server.did_open("text_document.py");
+    interaction.server.diagnostic("text_document.py");
+
+    interaction
+        .client
+        .expect_message(lsp_server::Message::Notification(
+            lsp_server::Notification {
+                method: "textDocument/publishDiagnostics".to_owned(),
+                params: serde_json::json! {{
+                    "uri": uri,
+                    "diagnostics": [],
+                    "version": 1
+                }},
+            },
+        ));
+
+    interaction.server.did_change("text_document.py", "# test");
+    interaction.server.diagnostic("text_document.py");
+
+    // I don't understand why this version is still 1
+    interaction
+        .client
+        .expect_message(lsp_server::Message::Notification(
+            lsp_server::Notification {
+                method: "textDocument/publishDiagnostics".to_owned(),
+                params: serde_json::json! {{
+                    "uri": uri,
+                    "diagnostics": [],
+                    "version": 2
+                }},
+            },
+        ));
+
+    interaction.server.did_close("text_document.py");
+    interaction.server.diagnostic("text_document.py");
+
+    interaction
+        .client
+        .expect_message(lsp_server::Message::Notification(
+            lsp_server::Notification {
+                method: "textDocument/publishDiagnostics".to_owned(),
+                params: serde_json::json! {{
+                    "uri": uri,
+                    "diagnostics": [],
+                    "version": 3
+                }},
+            },
+        ));
+
+    interaction.shutdown();
+}

pyrefly/lib/test/lsp/lsp_interaction/object_model.rs

@@ -201,6 +201,20 @@ impl TestServer {
         }));
     }
 
+    pub fn did_close(&self, file: &'static str) {
+        let path = self.get_root_or_panic().join(file);
+        self.send_message(Message::Notification(Notification {
+            method: "textDocument/didClose".to_owned(),
+            params: serde_json::json!({
+                "textDocument": {
+                    "uri": Url::from_file_path(&path).unwrap().to_string(),
+                    "languageId": "python",
+                    "version": 3
+                },
+            }),
+        }));
+    }
+
     pub fn did_change_configuration(&self) {
         self.send_message(Message::Notification(Notification {
             method: lsp_types::notification::DidChangeConfiguration::METHOD.to_owned(),