1use std::sync::Arc;
 2
 3use crate::lsp_log::LogMenuItem;
 4
 5use super::*;
 6use futures::StreamExt;
 7use gpui::{serde_json::json, TestAppContext};
 8use language::{tree_sitter_rust, FakeLspAdapter, Language, LanguageConfig, LanguageServerName};
 9use project::{FakeFs, Project};
10use settings::SettingsStore;
11
12#[gpui::test]
13async fn test_lsp_logs(cx: &mut TestAppContext) {
14    if std::env::var("RUST_LOG").is_ok() {
15        env_logger::init();
16    }
17
18    init_test(cx);
19
20    let mut rust_language = Language::new(
21        LanguageConfig {
22            name: "Rust".into(),
23            path_suffixes: vec!["rs".to_string()],
24            ..Default::default()
25        },
26        Some(tree_sitter_rust::language()),
27    );
28    let mut fake_rust_servers = rust_language
29        .set_fake_lsp_adapter(Arc::new(FakeLspAdapter {
30            name: "the-rust-language-server",
31            ..Default::default()
32        }))
33        .await;
34
35    let fs = FakeFs::new(cx.background());
36    fs.insert_tree(
37        "/the-root",
38        json!({
39            "test.rs": "",
40            "package.json": "",
41        }),
42    )
43    .await;
44    let project = Project::test(fs.clone(), ["/the-root".as_ref()], cx).await;
45    project.update(cx, |project, _| {
46        project.languages().add(Arc::new(rust_language));
47    });
48
49    let log_store = cx.add_model(|cx| LogStore::new(cx));
50    log_store.update(cx, |store, cx| store.add_project(&project, cx));
51
52    let _rust_buffer = project
53        .update(cx, |project, cx| {
54            project.open_local_buffer("/the-root/test.rs", cx)
55        })
56        .await
57        .unwrap();
58
59    let mut language_server = fake_rust_servers.next().await.unwrap();
60    language_server
61        .receive_notification::<lsp::notification::DidOpenTextDocument>()
62        .await;
63
64    let (_, log_view) = cx.add_window(|cx| LspLogView::new(project.clone(), log_store.clone(), cx));
65
66    language_server.notify::<lsp::notification::LogMessage>(lsp::LogMessageParams {
67        message: "hello from the server".into(),
68        typ: lsp::MessageType::INFO,
69    });
70    cx.foreground().run_until_parked();
71
72    log_view.read_with(cx, |view, cx| {
73        assert_eq!(
74            view.menu_items(cx).unwrap(),
75            &[LogMenuItem {
76                server_id: language_server.server.server_id(),
77                server_name: LanguageServerName("the-rust-language-server".into()),
78                worktree: project.read(cx).worktrees(cx).next().unwrap(),
79                rpc_trace_enabled: false,
80                rpc_trace_selected: false,
81                logs_selected: true,
82            }]
83        );
84        assert_eq!(view.editor.read(cx).text(cx), "hello from the server\n");
85    });
86}
87
88fn init_test(cx: &mut gpui::TestAppContext) {
89    cx.foreground().forbid_parking();
90
91    cx.update(|cx| {
92        cx.set_global(SettingsStore::test(cx));
93        theme::init((), cx);
94        language::init(cx);
95        client::init_settings(cx);
96        Project::init_settings(cx);
97        editor::init_settings(cx);
98    });
99}